@autobe/compiler/src/AutoBeCompiler.ts

AI backend server code generator

import {
  AutoBePhase,
  IAutoBeCompiler,
  IAutoBeCompilerListener,
  IAutoBeGetFilesOptions,
  IAutoBeInterfaceCompiler,
  IAutoBePrismaCompiler,
  IAutoBeRealizeCompiler,
  IAutoBeTestCompiler,
  IAutoBeTypeScriptCompiler,
} from "@autobe/interface";

import { AutoBeTypeScriptCompiler } from "./AutoBeTypeScriptCompiler";
import { AutoBeInterfaceCompiler } from "./interface/AutoBeInterfaceCompiler";
import { AutoBePrismaCompiler } from "./prisma/AutoBePrismaCompiler";
import { AutoBeCompilerCommonTemplate } from "./raw/AutoBeCompilerCommonTemplate";
import { AutoBeCompilerInterfaceTemplate } from "./raw/AutoBeCompilerInterfaceTemplate";
import { AutoBeCompilerRealizeTemplate } from "./raw/AutoBeCompilerRealizeTemplate";
import { AutoBeCompilerRealizeTemplateOfPostgres } from "./raw/AutoBeCompilerRealizeTemplateOfPostgres";
import { AutoBeCompilerRealizeTemplateOfSQLite } from "./raw/AutoBeCompilerRealizeTemplateOfSQLite";
import { AutoBeCompilerTestTemplate } from "./raw/AutoBeCompilerTestTemplate";
import { AutoBeRealizeCompiler } from "./realize/AutoBeRealizeCompiler";
import { AutoBeTestCompiler } from "./test/AutoBeTestCompiler";

/**
 * Comprehensive compilation infrastructure for the vibe coding pipeline.
 *
 * This class provides the three-tier compiler system that transforms AST
 * structures into production-ready code across all development phases. The
 * AutoBeCompiler integrates the custom Prisma compiler, Interface compiler, and
 * official TypeScript compiler into a unified compilation infrastructure.
 *
 * The compilation system ensures 100% syntactic correctness and semantic
 * integrity throughout the automated development workflow by operating on
 * validated AST data and providing continuous validation feedback loops. This
 * enables the revolutionary "structure first, validate continuously, generate
 * deterministically" approach that guarantees generated applications work
 * correctly on the first attempt.
 *
 * For high-performance scenarios with multiple concurrent users, individual
 * compiler components can be separated into dedicated worker processes to
 * prevent blocking during computationally intensive compilation operations
 * while maintaining the same interface compatibility.
 *
 * @author Samchon
 */
export class AutoBeCompiler implements IAutoBeCompiler {
  public prisma: IAutoBePrismaCompiler;
  public interface: IAutoBeInterfaceCompiler;
  public typescript: IAutoBeTypeScriptCompiler;
  public test: IAutoBeTestCompiler;
  public realize: IAutoBeRealizeCompiler;

  public constructor(private readonly listener: IAutoBeCompilerListener) {
    this.prisma = new AutoBePrismaCompiler();
    this.interface = new AutoBeInterfaceCompiler();
    this.typescript = new AutoBeTypeScriptCompiler();
    this.test = new AutoBeTestCompiler();
    this.realize = new AutoBeRealizeCompiler(this.listener.realize);
  }

  public async getTemplate(
    options: Required<IAutoBeGetFilesOptions>,
  ): Promise<Record<string, string>> {
    const result: Record<string, string> = {
      ...AutoBeCompilerCommonTemplate,
    };
    const index: number = PHASES.indexOf(options.phase);
    if (index >= OF_INTERFACE)
      Object.assign(result, AutoBeCompilerInterfaceTemplate);
    if (index >= OF_TEST) Object.assign(result, AutoBeCompilerTestTemplate);
    if (index >= OF_REALIZE)
      Object.assign(
        result,
        AutoBeCompilerRealizeTemplate,
        options.dbms === "postgres"
          ? AutoBeCompilerRealizeTemplateOfPostgres
          : AutoBeCompilerRealizeTemplateOfSQLite,
      );
    return result;
  }
}

const PHASES: AutoBePhase[] = [
  "analyze",
  "prisma",
  "interface",
  "test",
  "realize",
];
const OF_INTERFACE: number = PHASES.indexOf("interface");
const OF_TEST: number = PHASES.indexOf("test");
const OF_REALIZE: number = PHASES.indexOf("realize");