@haustle/notion-orm 0.0.44 → 0.0.46

package/build/src/ast/agents/agent-file-writer.js

@@ -0,0 +1,99 @@
+import fs from "fs";
+import path from "path";
+import * as ts from "typescript";
+import { camelize, toPascalCase } from "../../helpers.js";
+import { createNameImport } from "../shared/ast-builders.js";
+import { AGENTS_DIR, AST_IMPORT_PATHS, AST_RUNTIME_CONSTANTS } from "../shared/constants.js";
+import { codegenArtifactFileName, } from "../shared/codegen-environment.js";
+import { emitValueAsExpression } from "../shared/emit/emit-value-as-expression.js";
+import { createEmitContext, finalizeGeneratedSourceWithTrailingNewline, insertBlankLineAfterDoubleSlashBanner, printTsNodes, transpileTsToJs, writeTextArtifact, } from "../shared/emit/ts-emit-core.js";
+import { TS_EMIT_OPTIONS_GENERATED } from "../shared/emit/ts-emit-options.js";
+function addLeadingSectionComment(node, text) {
+    ts.addSyntheticLeadingComment(node, ts.SyntaxKind.SingleLineCommentTrivia, ` ${text}`, true);
+}
+function addGeneratedAgentModuleBanner(agentClientImport, cliGenerateCommand) {
+    ts.addSyntheticLeadingComment(agentClientImport, ts.SyntaxKind.SingleLineCommentTrivia, " Generated by @haustle/notion-orm — do not edit manually.", true);
+    ts.addSyntheticLeadingComment(agentClientImport, ts.SyntaxKind.SingleLineCommentTrivia, ` Regenerate with \`${cliGenerateCommand}\` (or your package script).`, true);
+}
+function printAgentModule(args) {
+    const context = createEmitContext({
+        fileName: `${args.agentFileBasename}.ts`,
+    });
+    const printed = args.statementSegments
+        .map((segment) => printTsNodes({ nodes: segment, context }).trimEnd())
+        .join("\n\n");
+    let tsCode = insertBlankLineAfterDoubleSlashBanner(printed);
+    tsCode = finalizeGeneratedSourceWithTrailingNewline(tsCode);
+    return { tsCode };
+}
+function buildAgentModuleNodes(args) {
+    const agentModuleName = args.agentModuleName ?? camelize(args.agentName);
+    const agentClientImport = createNameImport({
+        namedImport: "AgentClient",
+        path: AST_IMPORT_PATHS.AGENT_CLIENT,
+    });
+    addGeneratedAgentModuleBanner(agentClientImport, AST_RUNTIME_CONSTANTS.CLI_GENERATE_COMMAND);
+    const idVariable = ts.factory.createVariableStatement(undefined, ts.factory.createVariableDeclarationList([
+        ts.factory.createVariableDeclaration(ts.factory.createIdentifier("id"), undefined, undefined, ts.factory.createStringLiteral(args.agentId)),
+    ], ts.NodeFlags.Const));
+    addLeadingSectionComment(idVariable, "Identity (Notion agent id)");
+    const nameVariable = ts.factory.createVariableStatement(undefined, ts.factory.createVariableDeclarationList([
+        ts.factory.createVariableDeclaration(ts.factory.createIdentifier("name"), undefined, undefined, ts.factory.createStringLiteral(args.agentName)),
+    ], ts.NodeFlags.Const));
+    addLeadingSectionComment(nameVariable, "Display name (from Notion)");
+    const iconValue = emitValueAsExpression(args.agentIcon);
+    const agentClientFunction = ts.factory.createVariableStatement([ts.factory.createModifier(ts.SyntaxKind.ExportKeyword)], ts.factory.createVariableDeclarationList([
+        ts.factory.createVariableDeclaration(ts.factory.createIdentifier(toPascalCase(agentModuleName)), undefined, undefined, ts.factory.createArrowFunction(undefined, undefined, [
+            ts.factory.createParameterDeclaration(undefined, undefined, ts.factory.createIdentifier("auth"), undefined, ts.factory.createKeywordTypeNode(ts.SyntaxKind.StringKeyword), undefined),
+        ], undefined, undefined, ts.factory.createNewExpression(ts.factory.createIdentifier("AgentClient"), undefined, [
+            ts.factory.createObjectLiteralExpression([
+                ts.factory.createPropertyAssignment(ts.factory.createIdentifier("auth"), ts.factory.createIdentifier("auth")),
+                ts.factory.createPropertyAssignment(ts.factory.createIdentifier("id"), ts.factory.createIdentifier("id")),
+                ts.factory.createPropertyAssignment(ts.factory.createIdentifier("name"), ts.factory.createIdentifier("name")),
+                ts.factory.createPropertyAssignment(ts.factory.createIdentifier("icon"), iconValue),
+            ], true),
+        ]))),
+    ], ts.NodeFlags.Const));
+    addLeadingSectionComment(agentClientFunction, "Agent client factory");
+    const statementSegments = [
+        [agentClientImport],
+        [idVariable],
+        [nameVariable],
+        [agentClientFunction],
+    ];
+    return {
+        statementSegments,
+        agentId: args.agentId,
+        agentName: args.agentName,
+        agentModuleName,
+    };
+}
+export function renderAgentModule(args) {
+    const { statementSegments, agentId, agentName, agentModuleName } = buildAgentModuleNodes(args);
+    const { tsCode } = printAgentModule({
+        statementSegments,
+        agentFileBasename: toPascalCase(agentModuleName),
+    });
+    return { tsCode, agentId, agentName, agentModuleName };
+}
+export async function createTypescriptFileForAgent(args) {
+    const { statementSegments, agentModuleName } = buildAgentModuleNodes(args);
+    if (!fs.existsSync(AGENTS_DIR)) {
+        fs.mkdirSync(AGENTS_DIR, { recursive: true });
+    }
+    const agentFileBasename = toPascalCase(agentModuleName);
+    const { tsCode } = printAgentModule({
+        statementSegments,
+        agentFileBasename,
+    });
+    const outputPath = path.resolve(AGENTS_DIR, codegenArtifactFileName(agentFileBasename, args.environment));
+    const content = args.environment === "javascript"
+        ? transpileTsToJs({
+            typescriptCode: tsCode,
+            module: TS_EMIT_OPTIONS_GENERATED.module,
+            target: TS_EMIT_OPTIONS_GENERATED.target,
+        })
+        : tsCode;
+    writeTextArtifact({ filePath: outputPath, content });
+}
+//# sourceMappingURL=agent-file-writer.js.map

package/build/src/ast/agents/agent-file-writer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent-file-writer.js","sourceRoot":"","sources":["../../../../src/ast/agents/agent-file-writer.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,KAAK,EAAE,MAAM,YAAY,CAAC;AAEjC,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,qBAAqB,EAAE,MAAM,qBAAqB,CAAC;AAC1F,OAAO,EAEN,uBAAuB,GACvB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,qBAAqB,EAAE,MAAM,yCAAyC,CAAC;AAChF,OAAO,EACN,iBAAiB,EACjB,0CAA0C,EAC1C,qCAAqC,EACrC,YAAY,EACZ,eAAe,EACf,iBAAiB,GACjB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAS3E,SAAS,wBAAwB,CAAC,IAAkB,EAAE,IAAY;IACjE,EAAE,CAAC,0BAA0B,CAC5B,IAAI,EACJ,EAAE,CAAC,UAAU,CAAC,uBAAuB,EACrC,IAAI,IAAI,EAAE,EACV,IAAI,CACJ,CAAC;AACH,CAAC;AAED,SAAS,6BAA6B,CACrC,iBAAuC,EACvC,kBAA0B;IAE1B,EAAE,CAAC,0BAA0B,CAC5B,iBAAiB,EACjB,EAAE,CAAC,UAAU,CAAC,uBAAuB,EACrC,2DAA2D,EAC3D,IAAI,CACJ,CAAC;IACF,EAAE,CAAC,0BAA0B,CAC5B,iBAAiB,EACjB,EAAE,CAAC,UAAU,CAAC,uBAAuB,EACrC,sBAAsB,kBAAkB,8BAA8B,EACtE,IAAI,CACJ,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB,CAAC,IAGzB;IACA,MAAM,OAAO,GAAG,iBAAiB,CAAC;QACjC,QAAQ,EAAE,GAAG,IAAI,CAAC,iBAAiB,KAAK;KACxC,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB;SACpC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;SACrE,IAAI,CAAC,MAAM,CAAC,CAAC;IACf,IAAI,MAAM,GAAG,qCAAqC,CAAC,OAAO,CAAC,CAAC;IAC5D,MAAM,GAAG,0CAA0C,CAAC,MAAM,CAAC,CAAC;IAC5D,OAAO,EAAE,MAAM,EAAE,CAAC;AACnB,CAAC;AAED,SAAS,qBAAqB,CAAC,IAK9B;IACA,MAAM,eAAe,GAAG,IAAI,CAAC,eAAe,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IACzE,MAAM,iBAAiB,GAAG,gBAAgB,CAAC;QAC1C,WAAW,EAAE,aAAa;QAC1B,IAAI,EAAE,gBAAgB,CAAC,YAAY;KACnC,CAAC,CAAC;IACH,6BAA6B,CAC5B,iBAAiB,EACjB,qBAAqB,CAAC,oBAAoB,CAC1C,CAAC;IAEF,MAAM,UAAU,GAAG,EAAE,CAAC,OAAO,CAAC,uBAAuB,CACpD,SAAS,EACT,EAAE,CAAC,OAAO,CAAC,6BAA6B,CACvC;QACC,EAAE,CAAC,OAAO,CAAC,yBAAyB,CACnC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,EACjC,SAAS,EACT,SAAS,EACT,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,OAAO,CAAC,CAC5C;KACD,EACD,EAAE,CAAC,SAAS,CAAC,KAAK,CAClB,CACD,CAAC;IACF,wBAAwB,CAAC,UAAU,EAAE,4BAA4B,CAAC,CAAC;IAEnE,MAAM,YAAY,GAAG,EAAE,CAAC,OAAO,CAAC,uBAAuB,CACtD,SAAS,EACT,EAAE,CAAC,OAAO,CAAC,6BAA6B,CACvC;QACC,EAAE,CAAC,OAAO,CAAC,yBAAyB,CACnC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,EACnC,SAAS,EACT,SAAS,EACT,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,SAAS,CAAC,CAC9C;KACD,EACD,EAAE,CAAC,SAAS,CAAC,KAAK,CAClB,CACD,CAAC;IACF,wBAAwB,CAAC,YAAY,EAAE,4BAA4B,CAAC,CAAC;IAErE,MAAM,SAAS,GAAG,qBAAqB,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAExD,MAAM,mBAAmB,GAAG,EAAE,CAAC,OAAO,CAAC,uBAAuB,CAC7D,CAAC,EAAE,CAAC,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,CAAC,EACxD,EAAE,CAAC,OAAO,CAAC,6BAA6B,CACvC;QACC,EAAE,CAAC,OAAO,CAAC,yBAAyB,CACnC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC,EAC1D,SAAS,EACT,SAAS,EACT,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAC7B,SAAS,EACT,SAAS,EACT;YACC,EAAE,CAAC,OAAO,CAAC,0BAA0B,CACpC,SAAS,EACT,SAAS,EACT,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,EACnC,SAAS,EACT,EAAE,CAAC,OAAO,CAAC,qBAAqB,CAAC,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAC7D,SAAS,CACT;SACD,EACD,SAAS,EACT,SAAS,EACT,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAC7B,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,aAAa,CAAC,EAC1C,SAAS,EACT;YACC,EAAE,CAAC,OAAO,CAAC,6BAA6B,CACvC;gBACC,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAClC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,EACnC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,CACnC;gBACD,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAClC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,EACjC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,CACjC;gBACD,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAClC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,EACnC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,CACnC;gBACD,EAAE,CAAC,OAAO,CAAC,wBAAwB,CAClC,EAAE,CAAC,OAAO,CAAC,gBAAgB,CAAC,MAAM,CAAC,EACnC,SAAS,CACT;aACD,EACD,IAAI,CACJ;SACD,CACD,CACD,CACD;KACD,EACD,EAAE,CAAC,SAAS,CAAC,KAAK,CAClB,CACD,CAAC;IACF,wBAAwB,CAAC,mBAAmB,EAAE,sBAAsB,CAAC,CAAC;IAEtE,MAAM,iBAAiB,GAAyC;QAC/D,CAAC,iBAAiB,CAAC;QACnB,CAAC,UAAU,CAAC;QACZ,CAAC,YAAY,CAAC;QACd,CAAC,mBAAmB,CAAC;KACrB,CAAC;IAEF,OAAO;QACN,iBAAiB;QACjB,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,eAAe;KACf,CAAC;AACH,CAAC;AAED,MAAM,UAAU,iBAAiB,CAAC,IAKjC;IAMA,MAAM,EAAE,iBAAiB,EAAE,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,GAC/D,qBAAqB,CAAC,IAAI,CAAC,CAAC;IAC7B,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,CAAC;QACnC,iBAAiB;QACjB,iBAAiB,EAAE,YAAY,CAAC,eAAe,CAAC;KAChD,CAAC,CAAC;IACH,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,CAAC;AACxD,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,4BAA4B,CAAC,IAMlD;IACA,MAAM,EAAE,iBAAiB,EAAE,eAAe,EAAE,GAAG,qBAAqB,CAAC,IAAI,CAAC,CAAC;IAE3E,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAChC,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/C,CAAC;IAED,MAAM,iBAAiB,GAAG,YAAY,CAAC,eAAe,CAAC,CAAC;IACxD,MAAM,EAAE,MAAM,EAAE,GAAG,gBAAgB,CAAC;QACnC,iBAAiB;QACjB,iBAAiB;KACjB,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAC9B,UAAU,EACV,uBAAuB,CAAC,iBAAiB,EAAE,IAAI,CAAC,WAAW,CAAC,CAC5D,CAAC;IACF,MAAM,OAAO,GACZ,IAAI,CAAC,WAAW,KAAK,YAAY;QAChC,CAAC,CAAC,eAAe,CAAC;YAChB,cAAc,EAAE,MAAM;YACtB,MAAM,EAAE,yBAAyB,CAAC,MAAM;YACxC,MAAM,EAAE,yBAAyB,CAAC,MAAM;SACxC,CAAC;QACH,CAAC,CAAC,MAAM,CAAC;IACX,iBAAiB,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC;AACtD,CAAC"}

package/build/src/ast/agents/generate-agents-cli.d.ts

@@ -0,0 +1,24 @@
+import type { CodegenDiagnosticSink } from "../shared/codegen-diagnostics";
+type GenerationProgress = {
+    completed: number;
+    total: number;
+};
+type CreateAgentTypesOptions = {
+    onProgress?: (progress: GenerationProgress) => void;
+    skipSourceIndexUpdate?: boolean;
+    onDiagnostic?: CodegenDiagnosticSink;
+};
+export type CreateAgentTypesResult = {
+    agentNames: string[];
+    /** Module/registry keys (camelCase), parallel to `agentNames`. */
+    agentKeys: string[];
+    skipped: boolean;
+    /** All agents returned by the API across pages (0 when `skipped`). */
+    totalAgentsListed: number;
+    /** Agents that failed during file generation (0 when `skipped`). */
+    generationFailureCount: number;
+};
+/** Returns `{ skipped: true }` when the agents SDK is not installed. */
+export declare const createAgentTypes: (options?: CreateAgentTypesOptions) => Promise<CreateAgentTypesResult>;
+export {};
+//# sourceMappingURL=generate-agents-cli.d.ts.map

package/build/src/ast/agents/generate-agents-cli.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-agents-cli.d.ts","sourceRoot":"","sources":["../../../../src/ast/agents/generate-agents-cli.ts"],"names":[],"mappings":"AAaA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,+BAA+B,CAAC;AA0B3E,KAAK,kBAAkB,GAAG;IAAE,SAAS,EAAE,MAAM,CAAC;IAAC,KAAK,EAAE,MAAM,CAAA;CAAE,CAAC;AAC/D,KAAK,uBAAuB,GAAG;IAC9B,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,kBAAkB,KAAK,IAAI,CAAC;IACpD,qBAAqB,CAAC,EAAE,OAAO,CAAC;IAChC,YAAY,CAAC,EAAE,qBAAqB,CAAC;CACrC,CAAC;AAkCF,MAAM,MAAM,sBAAsB,GAAG;IACpC,UAAU,EAAE,MAAM,EAAE,CAAC;IACrB,kEAAkE;IAClE,SAAS,EAAE,MAAM,EAAE,CAAC;IACpB,OAAO,EAAE,OAAO,CAAC;IACjB,sEAAsE;IACtE,iBAAiB,EAAE,MAAM,CAAC;IAC1B,oEAAoE;IACpE,sBAAsB,EAAE,MAAM,CAAC;CAC/B,CAAC;AAEF,wEAAwE;AACxE,eAAO,MAAM,gBAAgB,GAC5B,UAAU,uBAAuB,KAC/B,OAAO,CAAC,sBAAsB,CA6FhC,CAAC"}

package/build/src/ast/agents/generate-agents-cli.js

@@ -0,0 +1,193 @@
+/**
+ * CLI orchestration for agent type generation.
+ * Syncs agent metadata into config, emits agent modules, and refreshes the
+ * generated source index when requested.
+ */
+import fs from "fs";
+import { z } from "zod";
+import { isAgentsSdkAvailable, loadAgentsSdk } from "../../agents-sdk-resolver.js";
+import { syncAgentsInConfigWithAST } from "../../cli/helpers.js";
+import { findConfigFile } from "../../config/findConfigFile.js";
+import { getNotionConfig } from "../../config/loadConfig.js";
+import { camelize, toPascalCase, toUndashedNotionId } from "../../helpers.js";
+import { readDatabaseMetadata, } from "../shared/cached-metadata.js";
+import { codegenArtifactFileName, resolveCodegenEnvironment, } from "../shared/codegen-environment.js";
+import { AGENTS_DIR, AST_FS_PATHS, codegenIndexSourcePath } from "../shared/constants.js";
+import { updateSourceIndexFile } from "../shared/emit/orm-index-emitter.js";
+import { emitRegistryModuleArtifacts } from "../shared/emit/registry-emitter.js";
+import { createTypescriptFileForAgent } from "./agent-file-writer.js";
+function writeAgentMetadata(metadata) {
+    if (!fs.existsSync(AGENTS_DIR)) {
+        fs.mkdirSync(AGENTS_DIR, { recursive: true });
+    }
+    fs.writeFileSync(AST_FS_PATHS.agentMetadataFile, JSON.stringify(metadata, null, 2));
+}
+async function listAllAgentsForSync(client) {
+    const out = [];
+    let cursor;
+    for (;;) {
+        const page = await client.agents.list({
+            page_size: 100,
+            ...(cursor !== undefined ? { start_cursor: cursor } : {}),
+        });
+        for (const a of page.results) {
+            out.push({ id: a.id, name: a.name, icon: a.icon });
+        }
+        if (!page.has_more || !page.next_cursor) {
+            break;
+        }
+        cursor = page.next_cursor;
+    }
+    return out;
+}
+/** Returns `{ skipped: true }` when the agents SDK is not installed. */
+export const createAgentTypes = async (options) => {
+    if (!isAgentsSdkAvailable()) {
+        return {
+            agentNames: [],
+            agentKeys: [],
+            skipped: true,
+            totalAgentsListed: 0,
+            generationFailureCount: 0,
+        };
+    }
+    const sdk = await loadAgentsSdk();
+    const config = await getNotionConfig();
+    const client = new sdk.NotionAgentsClient({
+        auth: config.auth,
+    });
+    const configFile = findConfigFile();
+    const environment = resolveCodegenEnvironment({ configRuntime: configFile });
+    const agentsList = await listAllAgentsForSync(client);
+    const total = agentsList.length;
+    options?.onProgress?.({ completed: 0, total });
+    if (configFile) {
+        const agentsToSync = agentsList.map(({ id, name }) => ({
+            id,
+            name,
+        }));
+        await syncAgentsInConfigWithAST(configFile.path, agentsToSync, configFile.isTS, { onFormatWarning: options?.onDiagnostic });
+    }
+    const metadataMap = new Map();
+    const agentNames = [];
+    const agentKeys = [];
+    let completedCount = 0;
+    let generationFailureCount = 0;
+    for (const agent of agentsList) {
+        try {
+            const normalizedIdForStorage = toUndashedNotionId(agent.id);
+            const agentMetadata = await generateAgentTypes(normalizedIdForStorage, agent.name, parseAgentIcon(agent.icon), environment);
+            metadataMap.set(agentMetadata.id, agentMetadata);
+            agentNames.push(agentMetadata.displayName);
+            agentKeys.push(agentMetadata.name);
+            completedCount += 1;
+            options?.onProgress?.({
+                completed: completedCount,
+                total,
+            });
+        }
+        catch (error) {
+            generationFailureCount += 1;
+            const detail = error instanceof Error ? error.message : String(error);
+            const message = `Error generating types for agent ${agent.id}: ${detail}`;
+            if (options?.onDiagnostic) {
+                options.onDiagnostic({ level: "error", message });
+            }
+            else {
+                // biome-ignore lint/suspicious/noConsole: CLI fallback when no sink
+                console.error(`❌ ${message}`);
+            }
+        }
+    }
+    const agentsMetadata = Array.from(metadataMap.values());
+    writeAgentMetadata(agentsMetadata);
+    createAgentBarrelFile({
+        agentInfo: agentsMetadata.map((agent) => ({ name: agent.name })),
+        environment,
+    });
+    if (!options?.skipSourceIndexUpdate) {
+        const databasesMetadata = readDatabaseMetadata();
+        updateSourceIndexFile(databasesMetadata, agentsMetadata, environment);
+    }
+    return {
+        agentNames,
+        agentKeys,
+        skipped: false,
+        totalAgentsListed: total,
+        generationFailureCount,
+    };
+};
+/** Emits the environment-specific `agents/index` registry module. */
+function createAgentBarrelFile(args) {
+    const { agentInfo, environment } = args;
+    emitRegistryModuleArtifacts({
+        registryName: "agents",
+        entries: agentInfo.map(({ name }) => ({
+            importName: name,
+            importPath: `./${codegenArtifactFileName(toPascalCase(name), environment)}`,
+            registryKey: name,
+        })),
+        outputPath: codegenIndexSourcePath({ scope: "agents", environment }),
+        environment,
+    });
+}
+function createMetadata(id, name, displayName) {
+    return {
+        id,
+        name,
+        displayName,
+    };
+}
+/** Trust boundary for icon payloads returned by the Notion Agents API. */
+const agentIconSchema = z.union([
+    z.object({
+        type: z.literal("emoji"),
+        emoji: z.string(),
+    }),
+    z.object({
+        type: z.literal("file"),
+        file: z.object({
+            url: z.string(),
+            expiry_time: z.string(),
+        }),
+    }),
+    z.object({
+        type: z.literal("external"),
+        external: z.object({
+            url: z.string(),
+        }),
+    }),
+    z.object({
+        type: z.literal("custom_emoji"),
+        custom_emoji: z.object({
+            id: z.string(),
+            name: z.string(),
+            url: z.string(),
+        }),
+    }),
+    z.object({
+        type: z.literal("custom_agent_avatar"),
+        custom_agent_avatar: z.object({
+            static_url: z.string(),
+            animated_url: z.string(),
+        }),
+    }),
+    z.null(),
+]);
+function parseAgentIcon(input) {
+    const parseResult = agentIconSchema.safeParse(input);
+    return parseResult.success ? parseResult.data : null;
+}
+async function generateAgentTypes(agentId, agentName, agentIcon, environment) {
+    const agentModuleName = camelize(agentName);
+    await createTypescriptFileForAgent({
+        agentId,
+        agentName,
+        agentModuleName,
+        agentIcon,
+        environment,
+    });
+    const agentMetadata = createMetadata(agentId, agentModuleName, agentName);
+    return agentMetadata;
+}
+//# sourceMappingURL=generate-agents-cli.js.map

package/build/src/ast/agents/generate-agents-cli.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-agents-cli.js","sourceRoot":"","sources":["../../../../src/ast/agents/generate-agents-cli.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,oBAAoB,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAChF,OAAO,EAAE,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAE9D,OAAO,EAAE,cAAc,EAAE,MAAM,6BAA6B,CAAC;AAC7D,OAAO,EAAE,eAAe,EAAE,MAAM,yBAAyB,CAAC;AAC1D,OAAO,EAAE,QAAQ,EAAE,YAAY,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAE3E,OAAO,EAEN,oBAAoB,GACpB,MAAM,2BAA2B,CAAC;AACnC,OAAO,EACN,uBAAuB,EACvB,yBAAyB,GAEzB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,UAAU,EAAE,YAAY,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC;AACvF,OAAO,EAAE,qBAAqB,EAAE,MAAM,kCAAkC,CAAC;AACzE,OAAO,EAAE,2BAA2B,EAAE,MAAM,iCAAiC,CAAC;AAC9E,OAAO,EAAE,4BAA4B,EAAE,MAAM,qBAAqB,CAAC;AAEnE,SAAS,kBAAkB,CAAC,QAA+B;IAC1D,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;QAChC,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/C,CAAC;IACD,EAAE,CAAC,aAAa,CACf,YAAY,CAAC,iBAAiB,EAC9B,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CACjC,CAAC;AACH,CAAC;AAgBD,KAAK,UAAU,oBAAoB,CAAC,MAOnC;IACA,MAAM,GAAG,GAAuD,EAAE,CAAC;IACnE,IAAI,MAA0B,CAAC;IAC/B,SAAS,CAAC;QACT,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC;YACrC,SAAS,EAAE,GAAG;YACd,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SACzD,CAAC,CAAC;QACH,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YAC9B,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QACpD,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACzC,MAAM;QACP,CAAC;QACD,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC;IAC3B,CAAC;IACD,OAAO,GAAG,CAAC;AACZ,CAAC;AAaD,wEAAwE;AACxE,MAAM,CAAC,MAAM,gBAAgB,GAAG,KAAK,EACpC,OAAiC,EACC,EAAE;IACpC,IAAI,CAAC,oBAAoB,EAAE,EAAE,CAAC;QAC7B,OAAO;YACN,UAAU,EAAE,EAAE;YACd,SAAS,EAAE,EAAE;YACb,OAAO,EAAE,IAAI;YACb,iBAAiB,EAAE,CAAC;YACpB,sBAAsB,EAAE,CAAC;SACzB,CAAC;IACH,CAAC;IAED,MAAM,GAAG,GAAG,MAAM,aAAa,EAAE,CAAC;IAClC,MAAM,MAAM,GAAG,MAAM,eAAe,EAAE,CAAC;IAEvC,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,kBAAkB,CAAC;QACzC,IAAI,EAAE,MAAM,CAAC,IAAI;KACjB,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,cAAc,EAAE,CAAC;IACpC,MAAM,WAAW,GAAG,yBAAyB,CAAC,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC,CAAC;IAE7E,MAAM,UAAU,GAAG,MAAM,oBAAoB,CAAC,MAAM,CAAC,CAAC;IACtD,MAAM,KAAK,GAAG,UAAU,CAAC,MAAM,CAAC;IAChC,OAAO,EAAE,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;IAC/C,IAAI,UAAU,EAAE,CAAC;QAChB,MAAM,YAAY,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YACtD,EAAE;YACF,IAAI;SACJ,CAAC,CAAC,CAAC;QACJ,MAAM,yBAAyB,CAC9B,UAAU,CAAC,IAAI,EACf,YAAY,EACZ,UAAU,CAAC,IAAI,EACf,EAAE,eAAe,EAAE,OAAO,EAAE,YAAY,EAAE,CAC1C,CAAC;IACH,CAAC;IAED,MAAM,WAAW,GAAG,IAAI,GAAG,EAA+B,CAAC;IAC3D,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,MAAM,SAAS,GAAa,EAAE,CAAC;IAC/B,IAAI,cAAc,GAAG,CAAC,CAAC;IACvB,IAAI,sBAAsB,GAAG,CAAC,CAAC;IAE/B,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;QAChC,IAAI,CAAC;YACJ,MAAM,sBAAsB,GAAG,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC5D,MAAM,aAAa,GAAG,MAAM,kBAAkB,CAC7C,sBAAsB,EACtB,KAAK,CAAC,IAAI,EACV,cAAc,CAAC,KAAK,CAAC,IAAI,CAAC,EAC1B,WAAW,CACX,CAAC;YACF,WAAW,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,EAAE,aAAa,CAAC,CAAC;YACjD,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;YAC3C,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC;YACnC,cAAc,IAAI,CAAC,CAAC;YACpB,OAAO,EAAE,UAAU,EAAE,CAAC;gBACrB,SAAS,EAAE,cAAc;gBACzB,KAAK;aACL,CAAC,CAAC;QACJ,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACzB,sBAAsB,IAAI,CAAC,CAAC;YAC5B,MAAM,MAAM,GACX,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACxD,MAAM,OAAO,GAAG,oCAAoC,KAAK,CAAC,EAAE,KAAK,MAAM,EAAE,CAAC;YAC1E,IAAI,OAAO,EAAE,YAAY,EAAE,CAAC;gBAC3B,OAAO,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;YACnD,CAAC;iBAAM,CAAC;gBACP,oEAAoE;gBACpE,OAAO,CAAC,KAAK,CAAC,KAAK,OAAO,EAAE,CAAC,CAAC;YAC/B,CAAC;QACF,CAAC;IACF,CAAC;IAED,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;IACxD,kBAAkB,CAAC,cAAc,CAAC,CAAC;IAEnC,qBAAqB,CAAC;QACrB,SAAS,EAAE,cAAc,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;QAChE,WAAW;KACX,CAAC,CAAC;IAEH,IAAI,CAAC,OAAO,EAAE,qBAAqB,EAAE,CAAC;QACrC,MAAM,iBAAiB,GAAG,oBAAoB,EAAE,CAAC;QACjD,qBAAqB,CAAC,iBAAiB,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;IACvE,CAAC;IAED,OAAO;QACN,UAAU;QACV,SAAS;QACT,OAAO,EAAE,KAAK;QACd,iBAAiB,EAAE,KAAK;QACxB,sBAAsB;KACtB,CAAC;AACH,CAAC,CAAC;AAEF,qEAAqE;AACrE,SAAS,qBAAqB,CAAC,IAG9B;IACA,MAAM,EAAE,SAAS,EAAE,WAAW,EAAE,GAAG,IAAI,CAAC;IAExC,2BAA2B,CAAC;QAC3B,YAAY,EAAE,QAAQ;QACtB,OAAO,EAAE,SAAS,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YACrC,UAAU,EAAE,IAAI;YAChB,UAAU,EAAE,KAAK,uBAAuB,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,WAAW,CAAC,EAAE;YAC3E,WAAW,EAAE,IAAI;SACjB,CAAC,CAAC;QACH,UAAU,EAAE,sBAAsB,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;QACpE,WAAW;KACX,CAAC,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CACtB,EAAU,EACV,IAAY,EACZ,WAAmB;IAEnB,OAAO;QACN,EAAE;QACF,IAAI;QACJ,WAAW;KACX,CAAC;AACH,CAAC;AAED,0EAA0E;AAC1E,MAAM,eAAe,GAAyB,CAAC,CAAC,KAAK,CAAC;IACrD,CAAC,CAAC,MAAM,CAAC;QACR,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC;QACxB,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE;KACjB,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACR,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC;QACvB,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;YACd,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;YACf,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE;SACvB,CAAC;KACF,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACR,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC;QAC3B,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC;YAClB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;SACf,CAAC;KACF,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACR,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC;QAC/B,YAAY,EAAE,CAAC,CAAC,MAAM,CAAC;YACtB,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE;YACd,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;YAChB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;SACf,CAAC;KACF,CAAC;IACF,CAAC,CAAC,MAAM,CAAC;QACR,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,qBAAqB,CAAC;QACtC,mBAAmB,EAAE,CAAC,CAAC,MAAM,CAAC;YAC7B,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE;YACtB,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE;SACxB,CAAC;KACF,CAAC;IACF,CAAC,CAAC,IAAI,EAAE;CACR,CAAC,CAAC;AAEH,SAAS,cAAc,CAAC,KAAc;IACrC,MAAM,WAAW,GAAG,eAAe,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;IACrD,OAAO,WAAW,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;AACtD,CAAC;AAED,KAAK,UAAU,kBAAkB,CAChC,OAAe,EACf,SAAiB,EACjB,SAAoB,EACpB,WAA+B;IAE/B,MAAM,eAAe,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC;IAE5C,MAAM,4BAA4B,CAAC;QAClC,OAAO;QACP,SAAS;QACT,eAAe;QACf,SAAS;QACT,WAAW;KACX,CAAC,CAAC;IAEH,MAAM,aAAa,GAAG,cAAc,CAAC,OAAO,EAAE,eAAe,EAAE,SAAS,CAAC,CAAC;IAC1E,OAAO,aAAa,CAAC;AACtB,CAAC"}

package/build/src/ast/database/database-file-writer.d.ts

@@ -0,0 +1,37 @@
+/**
+ * Database file writer for generated Notion data sources.
+ * Converts a Notion schema response into emitted TS/JS modules and column
+ * metadata consumed by the generated client layer.
+ */
+import type { GetDataSourceResponse } from "@notionhq/client/build/src/api-endpoints.js";
+import { type CodegenEnvironment } from "../shared/codegen-environment";
+import { type CodegenDiagnosticSink } from "../shared/codegen-diagnostics";
+type BuildDatabaseModuleOptions = {
+    onDiagnostic?: CodegenDiagnosticSink;
+};
+/**
+ * Renders the generated database module to TypeScript source text without
+ * writing to disk. Used by golden tests and any caller that needs the
+ * emitted code as strings.
+ */
+export declare function renderDatabaseModule(dataSourceResponse: GetDataSourceResponse, options?: BuildDatabaseModuleOptions): {
+    tsCode: string;
+    databaseName: string;
+    databaseModuleName: string;
+    databaseId: string;
+};
+/**
+ * Creates the generated module for a single database and writes it to disk.
+ * Thin wrapper over the pure builders that adds filesystem I/O.
+ */
+export declare function createCodegenFileForDatabase(args: {
+    dataSourceResponse: GetDataSourceResponse;
+    environment: CodegenEnvironment;
+    onDiagnostic?: CodegenDiagnosticSink;
+}): Promise<{
+    databaseName: string;
+    databaseModuleName: string;
+    databaseId: string;
+}>;
+export {};
+//# sourceMappingURL=database-file-writer.d.ts.map

package/build/src/ast/database/database-file-writer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"database-file-writer.d.ts","sourceRoot":"","sources":["../../../../src/ast/database/database-file-writer.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,KAAK,EAEX,qBAAqB,EACrB,MAAM,6CAA6C,CAAC;AAkCrD,OAAO,EACN,KAAK,kBAAkB,EAEvB,MAAM,+BAA+B,CAAC;AAEvC,OAAO,EACN,KAAK,qBAAqB,EAG1B,MAAM,+BAA+B,CAAC;AA0FvC,KAAK,0BAA0B,GAAG;IACjC,YAAY,CAAC,EAAE,qBAAqB,CAAC;CACrC,CAAC;AAqMF;;;;GAIG;AACH,wBAAgB,oBAAoB,CACnC,kBAAkB,EAAE,qBAAqB,EACzC,OAAO,CAAC,EAAE,0BAA0B,GAClC;IACF,MAAM,EAAE,MAAM,CAAC;IACf,YAAY,EAAE,MAAM,CAAC;IACrB,kBAAkB,EAAE,MAAM,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC;CACnB,CAQA;AAED;;;GAGG;AACH,wBAAsB,4BAA4B,CACjD,IAAI,EAAE;IACL,kBAAkB,EAAE,qBAAqB,CAAC;IAC1C,WAAW,EAAE,kBAAkB,CAAC;IAChC,YAAY,CAAC,EAAE,qBAAqB,CAAC;CACrC;;;;GAmCD"}

package/build/src/ast/database/database-file-writer.js

@@ -0,0 +1,240 @@
+import fs from "fs";
+import path from "path";
+import * as ts from "typescript";
+import { isColumnTypesWithOptions, isSupportedPropertyType, } from "../../client/database/types/index.js";
+import { camelize, toUndashedNotionId } from "../../helpers.js";
+import { toNotionDatabaseId } from "../../client/database/types/notion-database-id.js";
+import { createColumnNameToColumnProperties, createDatabaseClassExport, createCreateSchemaTypeExport, createPageSchemaTypeExport, createNameImport, createQuerySchemaTypeExport, createTypeOnlyNamedImports, toPascalCase, } from "../shared/ast-builders.js";
+import { AST_FS_PATHS, AST_IMPORT_PATHS, AST_RUNTIME_CONSTANTS, } from "../shared/constants.js";
+import { createEmitContext, finalizeGeneratedSourceWithTrailingNewline, printTsNodes, transpileTsToJs, writeTextArtifact, } from "../shared/emit/ts-emit-core.js";
+import { codegenArtifactFileName, } from "../shared/codegen-environment.js";
+import { TS_EMIT_OPTIONS_GENERATED } from "../shared/emit/ts-emit-options.js";
+import { formatCodegenMissingOptionsIdentifierLine, formatCodegenSkippedPropertyLine, } from "../shared/codegen-diagnostics.js";
+import { propertyASTGenerators, } from "./notion-column-generators.js";
+/** Narrows Notion properties to the subset supported by our emitters. */
+function isSupportedNotionProperty(property) {
+    return isSupportedPropertyType(property.type);
+}
+function addLeadingSectionComment(node, text) {
+    ts.addSyntheticLeadingComment(node, ts.SyntaxKind.SingleLineCommentTrivia, ` ${text}`, true);
+}
+function addGeneratedDatabaseModuleBanner(notionOrmImport, cliGenerateCommand) {
+    ts.addSyntheticLeadingComment(notionOrmImport, ts.SyntaxKind.SingleLineCommentTrivia, " Generated by @haustle/notion-orm — do not edit manually.", true);
+    ts.addSyntheticLeadingComment(notionOrmImport, ts.SyntaxKind.SingleLineCommentTrivia, ` Regenerate with \`${cliGenerateCommand}\` (or your package script).`, true);
+}
+/**
+ * Prefer the live Notion title when present, but fall back to a stable label so
+ * generation can still succeed for sparse payloads.
+ */
+function getDataSourceDisplayName(args) {
+    if ("title" in args.dataSourceResponse) {
+        const title = args.dataSourceResponse.title?.[0]?.plain_text?.trim();
+        if (title) {
+            return title;
+        }
+    }
+    return `Database ${args.normalizedDataSourceId.slice(0, 8)}`;
+}
+/** Inserts a blank line between the two-line generated banner and the first import. */
+function insertBlankLineAfterGeneratedBanner(code) {
+    return code.replace(/^(\/\/[^\n]+\r?\n\/\/[^\n]+)\r?\n(?=import\b)/m, "$1\n\n");
+}
+function printDatabaseModule(args) {
+    const context = createEmitContext({
+        fileName: `${args.databaseFileBasename}.ts`,
+    });
+    const printed = args.statementSegments
+        .map((segment) => printTsNodes({ nodes: segment, context }).trimEnd())
+        .join("\n\n");
+    let tsCode = insertBlankLineAfterGeneratedBanner(printed);
+    tsCode = finalizeGeneratedSourceWithTrailingNewline(tsCode);
+    return { tsCode };
+}
+/**
+ * Pure AST builder for a single database module.
+ * Returns statement nodes and metadata without any filesystem side effects,
+ * making it the primary seam for golden/snapshot tests.
+ */
+function buildDatabaseModuleNodes(dataSourceResponse, options) {
+    const onDiagnostic = options?.onDiagnostic;
+    const { id: dataSourceId, properties } = dataSourceResponse;
+    const normalizedDataSourceId = toUndashedNotionId(dataSourceId);
+    const columns = {};
+    const enumConstStatements = [];
+    const databaseName = getDataSourceDisplayName({
+        dataSourceResponse,
+        normalizedDataSourceId,
+    });
+    const databaseModuleName = camelize(databaseName);
+    Object.entries(properties).forEach(([propertyName, value]) => {
+        const unsupportedPropertyType = value.type;
+        if (!isSupportedNotionProperty(value)) {
+            onDiagnostic?.({
+                level: "warn",
+                message: formatCodegenSkippedPropertyLine({
+                    databaseDisplayName: databaseName,
+                    propertyName,
+                    notionType: unsupportedPropertyType,
+                }),
+            });
+            return;
+        }
+        const propertyType = value.type;
+        const camelizedColumnName = camelize(propertyName);
+        const handler = propertyASTGenerators[propertyType];
+        if (!handler) {
+            onDiagnostic?.({
+                level: "warn",
+                message: formatCodegenSkippedPropertyLine({
+                    databaseDisplayName: databaseName,
+                    propertyName,
+                    notionType: propertyType,
+                }),
+            });
+            return;
+        }
+        const result = handler({
+            columnName: propertyName,
+            camelizedName: camelizedColumnName,
+            columnValue: value,
+        });
+        if (!result) {
+            return;
+        }
+        const { zodMeta, enumConstStatement } = result;
+        const optionsIdentifier = zodMeta.type === "optionLiterals"
+            ? zodMeta.propertyValuesIdentifier
+            : undefined;
+        if (enumConstStatement) {
+            enumConstStatements.push(enumConstStatement);
+        }
+        if (isColumnTypesWithOptions(propertyType)) {
+            if (optionsIdentifier === undefined) {
+                onDiagnostic?.({
+                    level: "warn",
+                    message: formatCodegenMissingOptionsIdentifierLine({
+                        databaseModuleName,
+                        propertyName,
+                        propertyType,
+                    }),
+                });
+                return;
+            }
+            columns[camelizedColumnName] = {
+                columnName: propertyName,
+                type: propertyType,
+                optionsIdentifier,
+            };
+        }
+        else if (value.type === "relation") {
+            columns[camelizedColumnName] = {
+                columnName: propertyName,
+                type: "relation",
+                relatedDatabaseId: toNotionDatabaseId(value.relation.database_id),
+            };
+        }
+        else {
+            if (propertyType === "relation") {
+                throw new Error(`[${databaseModuleName}] Internal error: relation column '${propertyName}' reached plain-metadata branch.`);
+            }
+            columns[camelizedColumnName] = {
+                columnName: propertyName,
+                type: propertyType,
+            };
+        }
+    });
+    const queryAndPropertyTypesImport = createTypeOnlyNamedImports({
+        names: [
+            "DatabaseColumns",
+            "DatabaseDefinition",
+            "InferCreateSchema",
+            "InferDatabaseSchema",
+            "Query",
+        ],
+        path: AST_IMPORT_PATHS.DATABASE_CLIENT,
+    });
+    const databaseClientImport = createNameImport({
+        namedImport: "DatabaseClient",
+        path: AST_IMPORT_PATHS.DATABASE_CLIENT,
+    });
+    addGeneratedDatabaseModuleBanner(queryAndPropertyTypesImport, AST_RUNTIME_CONSTANTS.CLI_GENERATE_COMMAND);
+    if (enumConstStatements.length > 0) {
+        addLeadingSectionComment(enumConstStatements[0], "Option literals for select/status/multi_select columns");
+    }
+    const columnNameToColumnPropertiesStatement = ts.factory.createVariableStatement(undefined, createColumnNameToColumnProperties(columns));
+    addLeadingSectionComment(columnNameToColumnPropertiesStatement, "Column metadata");
+    const pageSchemaTypeAlias = createPageSchemaTypeExport();
+    addLeadingSectionComment(pageSchemaTypeAlias, "Row types (TypeScript)");
+    const createSchemaTypeAlias = createCreateSchemaTypeExport();
+    addLeadingSectionComment(createSchemaTypeAlias, "Create/update payload types (writable columns only)");
+    const queryTypeExportStatement = createQuerySchemaTypeExport();
+    addLeadingSectionComment(queryTypeExportStatement, "Typed database query arguments");
+    const databaseFactoryStatement = createDatabaseClassExport({
+        databaseName: toPascalCase(databaseModuleName),
+        schemaTitle: databaseName,
+        databaseId: normalizedDataSourceId,
+    });
+    addLeadingSectionComment(databaseFactoryStatement, "Database client factory");
+    const statementSegments = [
+        [queryAndPropertyTypesImport, databaseClientImport],
+        ...enumConstStatements.map((stmt) => [stmt]),
+        [columnNameToColumnPropertiesStatement],
+        [
+            pageSchemaTypeAlias,
+            createSchemaTypeAlias,
+            queryTypeExportStatement,
+            databaseFactoryStatement,
+        ],
+    ];
+    return {
+        statementSegments,
+        databaseName,
+        databaseModuleName,
+        databaseId: normalizedDataSourceId,
+    };
+}
+/**
+ * Renders the generated database module to TypeScript source text without
+ * writing to disk. Used by golden tests and any caller that needs the
+ * emitted code as strings.
+ */
+export function renderDatabaseModule(dataSourceResponse, options) {
+    const { statementSegments, databaseName, databaseModuleName, databaseId } = buildDatabaseModuleNodes(dataSourceResponse, options);
+    const { tsCode } = printDatabaseModule({
+        statementSegments,
+        databaseFileBasename: toPascalCase(databaseModuleName),
+    });
+    return { tsCode, databaseName, databaseModuleName, databaseId };
+}
+/**
+ * Creates the generated module for a single database and writes it to disk.
+ * Thin wrapper over the pure builders that adds filesystem I/O.
+ */
+export async function createCodegenFileForDatabase(args) {
+    const { dataSourceResponse, environment, onDiagnostic } = args;
+    const { statementSegments, databaseName, databaseModuleName, databaseId } = buildDatabaseModuleNodes(dataSourceResponse, { onDiagnostic });
+    const databasesDir = AST_FS_PATHS.DATABASES_DIR;
+    if (!fs.existsSync(databasesDir)) {
+        fs.mkdirSync(databasesDir, { recursive: true });
+    }
+    const databaseFileBasename = toPascalCase(databaseModuleName);
+    const { tsCode } = printDatabaseModule({
+        statementSegments,
+        databaseFileBasename,
+    });
+    const outputPath = path.resolve(databasesDir, codegenArtifactFileName(databaseFileBasename, environment));
+    const content = environment === "javascript"
+        ? transpileTsToJs({
+            typescriptCode: tsCode,
+            module: TS_EMIT_OPTIONS_GENERATED.module,
+            target: TS_EMIT_OPTIONS_GENERATED.target,
+        })
+        : tsCode;
+    writeTextArtifact({ filePath: outputPath, content });
+    return {
+        databaseName,
+        databaseModuleName,
+        databaseId,
+    };
+}
+//# sourceMappingURL=database-file-writer.js.map

package/build/src/ast/database/database-file-writer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"database-file-writer.js","sourceRoot":"","sources":["../../../../src/ast/database/database-file-writer.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,KAAK,EAAE,MAAM,YAAY,CAAC;AACjC,OAAO,EACN,wBAAwB,EACxB,uBAAuB,GAEvB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAAE,QAAQ,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AAC7D,OAAO,EAAE,kBAAkB,EAAE,MAAM,gDAAgD,CAAC;AACpF,OAAO,EACN,kCAAkC,EAClC,yBAAyB,EACzB,4BAA4B,EAC5B,0BAA0B,EAC1B,gBAAgB,EAChB,2BAA2B,EAC3B,0BAA0B,EAE1B,YAAY,GACZ,MAAM,wBAAwB,CAAC;AAChC,OAAO,EACN,YAAY,EACZ,gBAAgB,EAChB,qBAAqB,GACrB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EACN,iBAAiB,EACjB,0CAA0C,EAC1C,YAAY,EACZ,eAAe,EACf,iBAAiB,GACjB,MAAM,6BAA6B,CAAC;AACrC,OAAO,EAEN,uBAAuB,GACvB,MAAM,+BAA+B,CAAC;AACvC,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAC3E,OAAO,EAEN,yCAAyC,EACzC,gCAAgC,GAChC,MAAM,+BAA+B,CAAC;AACvC,OAAO,EACN,qBAAqB,GAErB,MAAM,4BAA4B,CAAC;AAIpC,yEAAyE;AACzE,SAAS,yBAAyB,CACjC,QAAkC;IAElC,OAAO,uBAAuB,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,wBAAwB,CAAC,IAAkB,EAAE,IAAY;IACjE,EAAE,CAAC,0BAA0B,CAC5B,IAAI,EACJ,EAAE,CAAC,UAAU,CAAC,uBAAuB,EACrC,IAAI,IAAI,EAAE,EACV,IAAI,CACJ,CAAC;AACH,CAAC;AAED,SAAS,gCAAgC,CACxC,eAAqC,EACrC,kBAA0B;IAE1B,EAAE,CAAC,0BAA0B,CAC5B,eAAe,EACf,EAAE,CAAC,UAAU,CAAC,uBAAuB,EACrC,2DAA2D,EAC3D,IAAI,CACJ,CAAC;IACF,EAAE,CAAC,0BAA0B,CAC5B,eAAe,EACf,EAAE,CAAC,UAAU,CAAC,uBAAuB,EACrC,sBAAsB,kBAAkB,8BAA8B,EACtE,IAAI,CACJ,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,SAAS,wBAAwB,CAAC,IAGjC;IACA,IAAI,OAAO,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;QACxC,MAAM,KAAK,GAAG,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;QACrE,IAAI,KAAK,EAAE,CAAC;YACX,OAAO,KAAK,CAAC;QACd,CAAC;IACF,CAAC;IACD,OAAO,YAAY,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;AAC9D,CAAC;AAUD,uFAAuF;AACvF,SAAS,mCAAmC,CAAC,IAAY;IACxD,OAAO,IAAI,CAAC,OAAO,CAClB,gDAAgD,EAChD,QAAQ,CACR,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,IAG5B;IACA,MAAM,OAAO,GAAG,iBAAiB,CAAC;QACjC,QAAQ,EAAE,GAAG,IAAI,CAAC,oBAAoB,KAAK;KAC3C,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,IAAI,CAAC,iBAAiB;SACpC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC;SACrE,IAAI,CAAC,MAAM,CAAC,CAAC;IACf,IAAI,MAAM,GAAG,mCAAmC,CAAC,OAAO,CAAC,CAAC;IAC1D,MAAM,GAAG,0CAA0C,CAAC,MAAM,CAAC,CAAC;IAC5D,OAAO,EAAE,MAAM,EAAE,CAAC;AACnB,CAAC;AAMD;;;;GAIG;AACH,SAAS,wBAAwB,CAChC,kBAAyC,EACzC,OAAoC;IAEpC,MAAM,YAAY,GAAG,OAAO,EAAE,YAAY,CAAC;IAC1C,MAAM,EAAE,EAAE,EAAE,YAAY,EAAE,UAAU,EAAE,GAAG,kBAAkB,CAAC;IAC5D,MAAM,sBAAsB,GAAG,kBAAkB,CAAC,YAAY,CAAC,CAAC;IAEhE,MAAM,OAAO,GACZ,EAAE,CAAC;IACJ,MAAM,mBAAmB,GAAmB,EAAE,CAAC;IAE/C,MAAM,YAAY,GAAG,wBAAwB,CAAC;QAC7C,kBAAkB;QAClB,sBAAsB;KACtB,CAAC,CAAC;IAEH,MAAM,kBAAkB,GAAG,QAAQ,CAAC,YAAY,CAAC,CAAC;IAElD,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,YAAY,EAAE,KAAK,CAAC,EAAE,EAAE;QAC5D,MAAM,uBAAuB,GAAG,KAAK,CAAC,IAAI,CAAC;QAC3C,IAAI,CAAC,yBAAyB,CAAC,KAAK,CAAC,EAAE,CAAC;YACvC,YAAY,EAAE,CAAC;gBACd,KAAK,EAAE,MAAM;gBACb,OAAO,EAAE,gCAAgC,CAAC;oBACzC,mBAAmB,EAAE,YAAY;oBACjC,YAAY;oBACZ,UAAU,EAAE,uBAAuB;iBACnC,CAAC;aACF,CAAC,CAAC;YACH,OAAO;QACR,CAAC;QAED,MAAM,YAAY,GAA8B,KAAK,CAAC,IAAI,CAAC;QAE3D,MAAM,mBAAmB,GAAG,QAAQ,CAAC,YAAY,CAAC,CAAC;QAEnD,MAAM,OAAO,GAAG,qBAAqB,CAAC,YAAY,CAAC,CAAC;QACpD,IAAI,CAAC,OAAO,EAAE,CAAC;YACd,YAAY,EAAE,CAAC;gBACd,KAAK,EAAE,MAAM;gBACb,OAAO,EAAE,gCAAgC,CAAC;oBACzC,mBAAmB,EAAE,YAAY;oBACjC,YAAY;oBACZ,UAAU,EAAE,YAAY;iBACxB,CAAC;aACF,CAAC,CAAC;YACH,OAAO;QACR,CAAC;QAED,MAAM,MAAM,GAAG,OAAO,CAAC;YACtB,UAAU,EAAE,YAAY;YACxB,aAAa,EAAE,mBAAmB;YAClC,WAAW,EAAE,KAAK;SAClB,CAAC,CAAC;QACH,IAAI,CAAC,MAAM,EAAE,CAAC;YACb,OAAO;QACR,CAAC;QAED,MAAM,EAAE,OAAO,EAAE,kBAAkB,EAAE,GAAG,MAAM,CAAC;QAC/C,MAAM,iBAAiB,GACtB,OAAO,CAAC,IAAI,KAAK,gBAAgB;YAChC,CAAC,CAAC,OAAO,CAAC,wBAAwB;YAClC,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,kBAAkB,EAAE,CAAC;YACxB,mBAAmB,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC9C,CAAC;QAED,IAAI,wBAAwB,CAAC,YAAY,CAAC,EAAE,CAAC;YAC5C,IAAI,iBAAiB,KAAK,SAAS,EAAE,CAAC;gBACrC,YAAY,EAAE,CAAC;oBACd,KAAK,EAAE,MAAM;oBACb,OAAO,EAAE,yCAAyC,CAAC;wBAClD,kBAAkB;wBAClB,YAAY;wBACZ,YAAY;qBACZ,CAAC;iBACF,CAAC,CAAC;gBACH,OAAO;YACR,CAAC;YACD,OAAO,CAAC,mBAAmB,CAAC,GAAG;gBAC9B,UAAU,EAAE,YAAY;gBACxB,IAAI,EAAE,YAAY;gBAClB,iBAAiB;aACjB,CAAC;QACH,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YACtC,OAAO,CAAC,mBAAmB,CAAC,GAAG;gBAC9B,UAAU,EAAE,YAAY;gBACxB,IAAI,EAAE,UAAU;gBAChB,iBAAiB,EAAE,kBAAkB,CACpC,KAAK,CAAC,QAAQ,CAAC,WAAW,CAC1B;aACD,CAAC;QACH,CAAC;aAAM,CAAC;YACP,IAAI,YAAY,KAAK,UAAU,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CACd,IAAI,kBAAkB,sCAAsC,YAAY,kCAAkC,CAC1G,CAAC;YACH,CAAC;YACD,OAAO,CAAC,mBAAmB,CAAC,GAAG;gBAC9B,UAAU,EAAE,YAAY;gBACxB,IAAI,EAAE,YAAY;aAClB,CAAC;QACH,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,MAAM,2BAA2B,GAAG,0BAA0B,CAAC;QAC9D,KAAK,EAAE;YACN,iBAAiB;YACjB,oBAAoB;YACpB,mBAAmB;YACnB,qBAAqB;YACrB,OAAO;SACP;QACD,IAAI,EAAE,gBAAgB,CAAC,eAAe;KACtC,CAAC,CAAC;IACH,MAAM,oBAAoB,GAAG,gBAAgB,CAAC;QAC7C,WAAW,EAAE,gBAAgB;QAC7B,IAAI,EAAE,gBAAgB,CAAC,eAAe;KACtC,CAAC,CAAC;IACH,gCAAgC,CAC/B,2BAA2B,EAC3B,qBAAqB,CAAC,oBAAoB,CAC1C,CAAC;IAEF,IAAI,mBAAmB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpC,wBAAwB,CACvB,mBAAmB,CAAC,CAAC,CAAC,EACtB,wDAAwD,CACxD,CAAC;IACH,CAAC;IAED,MAAM,qCAAqC,GAAG,EAAE,CAAC,OAAO,CAAC,uBAAuB,CAC/E,SAAS,EACT,kCAAkC,CAAC,OAAO,CAAC,CAC3C,CAAC;IACF,wBAAwB,CACvB,qCAAqC,EACrC,iBAAiB,CACjB,CAAC;IAEF,MAAM,mBAAmB,GAAG,0BAA0B,EAAE,CAAC;IACzD,wBAAwB,CACvB,mBAAmB,EACnB,wBAAwB,CACxB,CAAC;IAEF,MAAM,qBAAqB,GAAG,4BAA4B,EAAE,CAAC;IAC7D,wBAAwB,CACvB,qBAAqB,EACrB,qDAAqD,CACrD,CAAC;IAEF,MAAM,wBAAwB,GAAG,2BAA2B,EAAE,CAAC;IAC/D,wBAAwB,CACvB,wBAAwB,EACxB,gCAAgC,CAChC,CAAC;IAEF,MAAM,wBAAwB,GAAG,yBAAyB,CAAC;QAC1D,YAAY,EAAE,YAAY,CAAC,kBAAkB,CAAC;QAC9C,WAAW,EAAE,YAAY;QACzB,UAAU,EAAE,sBAAsB;KAClC,CAAC,CAAC;IACH,wBAAwB,CACvB,wBAAwB,EACxB,yBAAyB,CACzB,CAAC;IAEF,MAAM,iBAAiB,GAAqB;QAC3C,CAAC,2BAA2B,EAAE,oBAAoB,CAAC;QACnD,GAAG,mBAAmB,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;QAC5C,CAAC,qCAAqC,CAAC;QACvC;YACC,mBAAmB;YACnB,qBAAqB;YACrB,wBAAwB;YACxB,wBAAwB;SACxB;KACD,CAAC;IAEF,OAAO;QACN,iBAAiB;QACjB,YAAY;QACZ,kBAAkB;QAClB,UAAU,EAAE,sBAAsB;KAClC,CAAC;AACH,CAAC;AAEF;;;;GAIG;AACH,MAAM,UAAU,oBAAoB,CACnC,kBAAyC,EACzC,OAAoC;IAOpC,MAAM,EAAE,iBAAiB,EAAE,YAAY,EAAE,kBAAkB,EAAE,UAAU,EAAE,GACxE,wBAAwB,CAAC,kBAAkB,EAAE,OAAO,CAAC,CAAC;IACvD,MAAM,EAAE,MAAM,EAAE,GAAG,mBAAmB,CAAC;QACtC,iBAAiB;QACjB,oBAAoB,EAAE,YAAY,CAAC,kBAAkB,CAAC;KACtD,CAAC,CAAC;IACH,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,kBAAkB,EAAE,UAAU,EAAE,CAAC;AACjE,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,4BAA4B,CACjD,IAIC;IAED,MAAM,EAAE,kBAAkB,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC;IAC/D,MAAM,EAAE,iBAAiB,EAAE,YAAY,EAAE,kBAAkB,EAAE,UAAU,EAAE,GACxE,wBAAwB,CAAC,kBAAkB,EAAE,EAAE,YAAY,EAAE,CAAC,CAAC;IAEhE,MAAM,YAAY,GAAG,YAAY,CAAC,aAAa,CAAC;IAChD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;QAClC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,MAAM,oBAAoB,GAAG,YAAY,CAAC,kBAAkB,CAAC,CAAC;IAC9D,MAAM,EAAE,MAAM,EAAE,GAAG,mBAAmB,CAAC;QACtC,iBAAiB;QACjB,oBAAoB;KACpB,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,IAAI,CAAC,OAAO,CAC9B,YAAY,EACZ,uBAAuB,CAAC,oBAAoB,EAAE,WAAW,CAAC,CAC1D,CAAC;IACF,MAAM,OAAO,GACZ,WAAW,KAAK,YAAY;QAC3B,CAAC,CAAC,eAAe,CAAC;YAChB,cAAc,EAAE,MAAM;YACtB,MAAM,EAAE,yBAAyB,CAAC,MAAM;YACxC,MAAM,EAAE,yBAAyB,CAAC,MAAM;SACxC,CAAC;QACH,CAAC,CAAC,MAAM,CAAC;IACX,iBAAiB,CAAC,EAAE,QAAQ,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,CAAC;IAErD,OAAO;QACN,YAAY;QACZ,kBAAkB;QAClB,UAAU;KACV,CAAC;AACH,CAAC"}