import { AutoBeOpenApi } from "@autobe/interface";
import {
  IHttpMigrateApplication,
  OpenApi,
  OpenApiV3_2,
} from "@typia/interface";
import { HttpMigration, OpenApiConverter } from "@typia/utils";
import { HashMap } from "tstl";

import { StringUtil } from "../StringUtil";
import { AutoBeOpenApiEndpointComparator } from "./AutoBeOpenApiEndpointComparator";

export function transformOpenApiDocument(
  input: AutoBeOpenApi.IDocument,
): OpenApi.IDocument {
  const dict: HashMap<AutoBeOpenApi.IEndpoint, string> = new HashMap(
    AutoBeOpenApiEndpointComparator.hashCode,
    AutoBeOpenApiEndpointComparator.equals,
  );
  const paths: Record<string, OpenApi.IPath> = {};

  for (const op of input.operations) {
    dict.set(op, op.name);
    paths[op.path] ??= {};
    paths[op.path][op.method] = {
      summary: StringUtil.summary(op.description),
      description:
        op.description +
        (op.authorizationType !== null &&
        op.responseBody?.typeName.endsWith(".IAuthorized") === true
          ? "\n\n@setHeader token.access Authorization"
          : ""),
      parameters: op.parameters.map((p) => ({
        name: p.name,
        in: "path",
        schema: p.schema,
        required: true,
        description: p.description,
      })),
      requestBody: op.requestBody
        ? {
            content: {
              "application/json": {
                schema: {
                  $ref: `#/components/schemas/${op.requestBody.typeName}`,
                },
              },
            },
            description: op.requestBody.description,
            required: true,
          }
        : undefined,
      responses: op.responseBody
        ? {
            [op.method === "post" ? 201 : 200]: {
              content: {
                "application/json": {
                  schema: {
                    $ref: `#/components/schemas/${op.responseBody.typeName}`,
                  },
                },
              },
              description: op.responseBody.description,
            },
          }
        : undefined,
      ...{
        "x-autobe-authorization-type": op.authorizationType,
        "x-autobe-authorization-actor": op.authorizationActor,
        "x-autobe-prerequisites": op.prerequisites,
        "x-samchon-accessor": op.accessor,
        "x-autobe-specification": op.specification,
      },
    };
  }

  const document: OpenApi.IDocument = OpenApiConverter.upgradeDocument({
    openapi: "3.2.0",
    paths,
    components: input.components,
  } as OpenApiV3_2.IDocument);
  const migrate: IHttpMigrateApplication = HttpMigration.application(document);
  migrate.routes.forEach((r) => {
    if (r.method === "head") return;
    const name: string = dict.get({
      method: r.method as "post",
      path: r.path,
    });
    if (r.accessor.length >= 2 && r.accessor.at(-2) === name) r.accessor.pop();
    r.accessor[r.accessor.length - 1] = name;
    r.operation()["x-samchon-accessor"] = r.accessor;
  });
  return document;
}