okai/dist/ts-ast.js

AI-powered code generation tool for ServiceStack Apps. Generate TypeScript data models, C# APIs, migrations, and UI components from natural language prompts using LLMs.

import { toMetadataTypes } from "./cs-ast.js";
import { createTdAstFromAIAst, transformUserRefs } from "./ts-once.js";
import { TypeScriptParser } from "./ts-parser.js";
import { toTsd } from "./tsd-gen.js";
import { getGroupName } from "./utils.js";
// Extract TypeScript source code from a AI Response
export function extractTypeScriptSrc(msg) {
    msg = msg.trim();
    const startPos = msg.indexOf("```typescript");
    if (startPos >= 0) {
        const ts = msg.substring(msg.indexOf('\n', startPos) + 1);
        const src = ts.substring(0, ts.indexOf("```"));
        return src;
    }
    if (msg.startsWith("//") || msg.startsWith("/*") || msg.startsWith("export ")
        || msg.startsWith("interface ") || msg.startsWith("class ") || msg.startsWith("type ")) {
        return msg;
    }
}
// Parse TypeScript source code into AST
export function toAst(src) {
    const parser = new TypeScriptParser();
    return parser.parse(src);
}
// Transforms that are only applied once on AI TypeScript AST
export function createAstFromAITypeScript(ts) {
    const parser = new TypeScriptParser();
    const tdAst = parser.parse(ts);
    return createTdAstFromAIAst(tdAst);
}
// Apply Project Info to the AST once after it's downloaded
export function astForProject(tsAst, info) {
    transformUserRefs(tsAst, info);
    return tsAst;
}
// Convert User TypeScript into CS AST and Updated TSD
export function generateCsAstFromTsd(userTs, opt) {
    const userTsAst = toAst(userTs); // user modified tsd
    if (opt?.references) {
        userTsAst.references = userTsAst.references || [];
        opt.references
            .filter(x => !userTsAst.references.some(y => y.path === x))
            .forEach(path => userTsAst.references.push({ path }));
    }
    const tsd = toTsd(userTsAst);
    const tsdAst = toAst(tsd);
    const csAst = toMetadataTypes(tsdAst);
    const result = {
        tsdAst,
        tsd,
        csAst,
        groupName: getGroupName(csAst)
    };
    return result;
}
//# sourceMappingURL=ts-ast.js.map