import ts from "typescript";

import { Singleton } from "../utils/Singleton";

import { IProject } from "./IProject";
import { NodeTransformer } from "./NodeTransformer";
import { TransformerError } from "./TransformerError";

export namespace FileTransformer {
  export const transform =
    (environments: Omit<IProject, "context">) =>
    (context: ts.TransformationContext) =>
    (file: ts.SourceFile): ts.SourceFile => {
      if (file.isDeclarationFile) return file;

      const project: IProject = {
        ...environments,
        context,
      };
      checkJsDocParsingMode.get(project, file);

      return ts.visitEachChild(
        file,
        (node) => iterate_node(project)(node),
        context,
      );
    };

  const iterate_node =
    (project: IProject) =>
    (node: ts.Node): ts.Node =>
      ts.visitEachChild(
        try_transform_node(project)(node) ?? node,
        (child) => iterate_node(project)(child),
        project.context,
      );

  const try_transform_node =
    (project: IProject) =>
    (node: ts.Node): ts.Node | null => {
      try {
        return NodeTransformer.transform(project)(node);
      } catch (exp) {
        // ONLY ACCEPT TRANSFORMER-ERROR
        if (!isTransformerError(exp)) throw exp;

        // REPORT DIAGNOSTIC
        const diagnostic = ts.createDiagnosticForNode(node, {
          key: exp.code,
          category: ts.DiagnosticCategory.Error,
          message: exp.message,
          code: `(${exp.code})` as any,
        });
        project.extras.addDiagnostic(diagnostic);
        return null;
      }
    };
}

const isTransformerError = (error: any): error is TransformerError =>
  typeof error === "object" &&
  error !== null &&
  error.constructor.name === "TransformerError" &&
  typeof error.code === "string" &&
  typeof error.message === "string";

const checkJsDocParsingMode = new Singleton(
  (project: IProject, file: ts.SourceFile) => {
    if (
      typeof file.jsDocParsingMode === "number" &&
      file.jsDocParsingMode !== 0
    ) {
      project.extras.addDiagnostic(
        ts.createDiagnosticForNode(file, {
          code: `(typia setup)` as any,
          key: "jsDocParsingMode",
          category: ts.DiagnosticCategory.Warning,
          message: [
            `Run "npx typia setup" or "npx typia patch" command again.`,
            ``,
            `Since TypeScript v5.3 update, "tsc" no more parses JSDoc comments. Therefore, "typia" also cannot utilize those JSDoc comments too, and it damages on some features of "typia" like "comment tags" or "JSON schema" generator.`,
            ``,
            `To solve this problem, run "npx typia setup" or "npx typia patch" command to hack the TypeScript compiler to revive the JSDoc parsing feature.`,
            ``,
            `  - reference: https://github.com/microsoft/TypeScript/pull/55739`,
          ].join("\n"),
        }),
      );
    }
  },
);