Class ZModelCodeGenerator

packages/language/src/zmodel-code-generator.ts:72–433

 * Generates ZModel source code from AST.
 */
export class ZModelCodeGenerator {
    private readonly options: ZModelCodeOptions;
    private readonly quote: string;
    constructor(options?: Partial<ZModelCodeOptions>) {
        this.options = {
            binaryExprNumberOfSpaces: options?.binaryExprNumberOfSpaces ?? 1,
            unaryExprNumberOfSpaces: options?.unaryExprNumberOfSpaces ?? 0,
            indent: options?.indent ?? 4,
            quote: options?.quote ?? 'single',
        };
        this.quote = this.options.quote === 'double' ? '"' : "'";
    }

    /**
     * Generates ZModel source code from AST.
     */
    generate(ast: AstNode): string {
        const handler = generationHandlers.get(ast.$type);
        if (!handler) {
            throw new Error(`No generation handler found for ${ast.$type}`);
        }
        return handler.value.call(this, ast);
    }

    private quotedStr(val: string): string {
        const trimmedVal = val.replace(new RegExp(`(?<!\\\\)${this.quote}`, 'g'), `\\${this.quote}`);
        return `${this.quote}${trimmedVal}${this.quote}`;
    }

    @gen(Model)
    private _generateModel(ast: Model) {
        return `${ast.imports.map((d) => this.generate(d)).join('\n')}${ast.imports.length > 0 ? '\n\n' : ''}${ast.declarations
            .map((d) => this.generate(d))
            .join('\n\n')}`;
    }

    @gen(DataSource)
    private _generateDataSource(ast: DataSource) {
        return `datasource ${ast.name} {
${ast.fields.map((x) => this.indent + this.generate(x)).join('\n')}
}`;
    }

    @gen(ModelImport)
    private _generateModelImport(ast: ModelImport) {
        return `import ${this.quotedStr(ast.path)}`;
    }

    @gen(Enum)
    private _generateEnum(ast: Enum) {
        const comments = `${ast.comments.join('\n')}\n`;
        return `${ast.comments.length > 0 ? comments : ''}enum ${ast.name} {
${ast.fields.map((x) => this.indent + this.generate(x)).join('\n')}${
            ast.attributes.length > 0
                ? '\n\n' + ast.attributes.map((x) => this.indent + this.generate(x)).join('\n')
                : ''
        }
}`;