import { Module, DsgContext } from "@amplication/code-gen-types";
import { join } from "path";
import {
  AUTH_ENTITY_ERROR,
  AUTH_ENTITY_LOG_ERROR,
  templatesPath,
} from "../constants";
import { readFile, print } from "@amplication/code-gen-utils";
import {
  addImports,
  importNames,
  interpolate,
  removeTSClassDeclares,
} from "../util/ast";
import { builders, namedTypes } from "ast-types";
import { camelCase } from "lodash";

const authServiceSpecPath = join(
  templatesPath,
  "auth.service.spec.template.ts"
);

export async function createAuthServiceSpec(
  dsgContext: DsgContext
): Promise<Module> {
  const { entities, resourceInfo, serverDirectories } = dsgContext;
  const authEntity = entities?.find(
    (x) => x.name === resourceInfo?.settings.authEntityName
  );
  if (!authEntity) {
    dsgContext.logger.error(AUTH_ENTITY_LOG_ERROR);
    throw new Error(AUTH_ENTITY_ERROR);
  }

  const entityServiceName = `${authEntity?.name}Service`;

  const template = await readFile(authServiceSpecPath);
  const authServiceNameId = builders.identifier(entityServiceName);

  const entityNameToLower = authEntity?.name.toLowerCase();

  const entityServiceImport = importNames(
    [authServiceNameId],
    `../${entityNameToLower}/${entityNameToLower}.service`
  );

  addImports(
    template,
    [entityServiceImport].filter(
      (x) => x //remove nulls and undefined
    ) as namedTypes.ImportDeclaration[]
  );

  const templateMapping = {
    ENTITY_SERVICE: builders.identifier(entityServiceName),
    FIND_ONE_FUNCTION: builders.identifier(`${camelCase(authEntity?.name)}`),
  };

  interpolate(template, templateMapping);

  const filePath = `${serverDirectories.authDirectory}/auth.service.spec.ts`;

  removeTSClassDeclares(template);

  return {
    code: print(template).code,
    path: filePath,
  };
}