import { kebabCase } from 'lodash';
import pluralize from 'pluralize';
import pupa from 'pupa';

export function generateFileName(args: {
  type: string;
  name: string;
  getModelName(name: string): string | undefined;
  template: string;
}) {
  const { template, type, name, getModelName } = args;

  return pupa(template, {
    type,
    get model() {
      const result = getModelName(name) || 'prisma';
      return kebabCase(result);
    },
    get name() {
      let result = kebabCase(name);
      for (const suffix of ['input', 'args', 'enum']) {
        const ending = `-${suffix}`;
        if (type === suffix && result.endsWith(ending)) {
          result = result.slice(0, -ending.length);
        }
      }
      return result;
    },
    plural: {
      get type() {
        return pluralize(type);
      },
    },
  });
}