import { mkdirSync, writeFileSync } from "node:fs";
import path from "node:path";
import type { Common } from "@/common/common.js";
import { type GraphQLSchema, printSchema } from "graphql";

export const ponderEnv = `// This file enables type checking and editor autocomplete for this Ponder project.
// After upgrading, you may find that changes have been made to this file.
// If this happens, please commit the changes. Do not manually edit this file.
// See https://ponder.sh/docs/getting-started/installation#typescript for more information.

declare module "@/generated" {
  import type { Virtual } from "@ponder/core";

  type config = typeof import("./ponder.config.ts").default;
  type schema = typeof import("./ponder.schema.ts").default;

  export const ponder: Virtual.Registry<config, schema>;

  export type EventNames = Virtual.EventNames<config>;
  export type Event<name extends EventNames = EventNames> = Virtual.Event<
    config,
    name
  >;
  export type Context<name extends EventNames = EventNames> = Virtual.Context<
    config,
    schema,
    name
  >;
  export type ApiContext = Virtual.Drizzle<schema>;
  export type IndexingFunctionArgs<name extends EventNames = EventNames> =
    Virtual.IndexingFunctionArgs<config, schema, name>;
  export type Schema = Virtual.Schema<schema>;
}
`;

export function runCodegen({
  common,
  graphqlSchema,
}: {
  common: Common;
  graphqlSchema: GraphQLSchema;
}) {
  writeFileSync(
    path.join(common.options.rootDir, "ponder-env.d.ts"),
    ponderEnv,
    "utf8",
  );

  common.logger.debug({
    service: "codegen",
    msg: "Wrote new file at ponder-env.d.ts",
  });

  mkdirSync(common.options.generatedDir, { recursive: true });
  writeFileSync(
    path.join(common.options.generatedDir, "schema.graphql"),
    printSchema(graphqlSchema),
    "utf-8",
  );

  common.logger.debug({
    service: "codegen",
    msg: "Wrote new file at generated/schema.graphql",
  });
}