@autobe/agent/src/orchestrate/realize/internal/compileRealizeFiles.ts

AI backend server code generator

import {
  AutoBeRealizeAuthorization,
  AutoBeRealizeFunction,
  AutoBeRealizeValidateEvent,
  IAutoBeCompiler,
  IAutoBeTypeScriptCompileResult,
} from "@autobe/interface";
import { ILlmSchema } from "@samchon/openapi";
import { v7 } from "uuid";

import { AutoBeContext } from "../../../context/AutoBeContext";

export async function compileRealizeFiles<Model extends ILlmSchema.Model>(
  ctx: AutoBeContext<Model>,
  props: {
    authorizations: AutoBeRealizeAuthorization[];
    functions: AutoBeRealizeFunction[];
  },
): Promise<AutoBeRealizeValidateEvent> {
  const prisma = ctx.state().prisma?.compiled;
  const payloads: Record<string, string> = Object.fromEntries(
    props.authorizations.map((el) => [el.payload.location, el.payload.content]),
  );
  const compiler: IAutoBeCompiler = await ctx.compiler();
  const templateFiles: Record<string, string> = await compiler.getTemplate({
    phase: "realize",
    dbms: "sqlite",
  });
  const nodeModules: Record<string, string> =
    prisma?.type === "success" ? prisma.nodeModules : {};

  const filterTsFiles = (location: string) =>
    location.startsWith("src/") && location.endsWith(".ts");

  const files: Record<string, string> = {
    ...nodeModules,
    ...payloads,
    ...Object.fromEntries(
      Object.entries(await ctx.files({ dbms: "sqlite" })).filter(([key]) =>
        filterTsFiles(key),
      ),
    ),
    ...Object.fromEntries(
      Object.entries(templateFiles).filter(([key]) => filterTsFiles(key)),
    ),
    ...Object.fromEntries(
      props.functions.map((el) => [el.location, el.content]),
    ),
  };
  const compiled: IAutoBeTypeScriptCompileResult =
    await compiler.typescript.compile({
      files: files,
    });
  return {
    type: "realizeValidate",
    id: v7(),
    files: files,
    result: compiled,
    step: ctx.state().analyze?.step ?? 0,
    created_at: new Date().toISOString(),
  };
}