zenstack 0.5.0 → 0.6.0-pre.2

package/src/generator/index.ts

@@ -1,86 +0,0 @@
-/* eslint-disable @typescript-eslint/no-var-requires */
-import { Context, Generator } from './types';
-import * as fs from 'fs';
-import colors from 'colors';
-import PrismaGenerator from './prisma';
-import ServiceGenerator from './service';
-import ReactHooksGenerator from './react-hooks';
-import { TypescriptCompilation } from './tsc';
-import FieldConstraintGenerator from './field-constraint';
-import telemetry from '../telemetry';
-import ora from 'ora';
-
-/**
- * ZenStack code generator
- */
-export class ZenStackGenerator {
-    /**
-     * Runs a series of nested generators
-     */
-    async generate(
-        context: Context,
-        includeGenerators?: string[],
-        clearOutput = true
-    ): Promise<void> {
-        // ensure folder that stores generated prisma schema and migrations
-        if (!fs.existsSync(context.outDir)) {
-            fs.mkdirSync(context.outDir);
-        }
-
-        if (clearOutput) {
-            // recreate folder that stores generated zenstack code
-            if (fs.existsSync(context.generatedCodeDir)) {
-                fs.rmSync(context.generatedCodeDir, {
-                    force: true,
-                    recursive: true,
-                });
-            }
-            fs.mkdirSync(context.generatedCodeDir);
-        }
-
-        // TODO: plugin mechanism
-        const generators: Generator[] = [
-            new PrismaGenerator(),
-            new ServiceGenerator(),
-            new ReactHooksGenerator(),
-            new FieldConstraintGenerator(),
-            new TypescriptCompilation(),
-        ];
-
-        const version = require('../../package.json').version;
-        console.log(colors.bold(`⌛️ Running ZenStack generator v${version}`));
-
-        const warnings: string[] = [];
-        for (const generator of generators) {
-            if (
-                includeGenerators &&
-                !includeGenerators.includes(generator.name)
-            ) {
-                continue;
-            }
-
-            const spinner = ora(generator.startMessage).start();
-            await telemetry.trackSpan(
-                'cli:generator:start',
-                'cli:generator:complete',
-                'cli:generator:error',
-                {
-                    generator: generator.name,
-                },
-                async () => {
-                    const genWarnings = await generator.generate(context);
-                    warnings.push(...genWarnings);
-                }
-            );
-            spinner.succeed(`${colors.cyan(generator.successMessage)}`);
-        }
-
-        console.log(
-            colors.green(
-                colors.bold('👻 All generators completed successfully!')
-            )
-        );
-
-        warnings.forEach((w) => console.warn(colors.yellow(w)));
-    }
-}

package/src/generator/prisma/expression-writer.ts

@@ -1,360 +0,0 @@
-import {
-    BinaryExpr,
-    Expression,
-    isDataModel,
-    isDataModelField,
-    isEnumField,
-    isMemberAccessExpr,
-    isReferenceExpr,
-    isThisExpr,
-    LiteralExpr,
-    MemberAccessExpr,
-    ReferenceExpr,
-    UnaryExpr,
-} from '@lang/generated/ast';
-import { CodeBlockWriter } from 'ts-morph';
-import { GUARD_FIELD_NAME } from '../constants';
-import { GeneratorError } from '../types';
-import TypeScriptExpressionTransformer from './typescript-expression-transformer';
-
-type ComparisonOperator = '==' | '!=' | '>' | '>=' | '<' | '<=';
-
-/**
- * Utility for writing ZModel expression as Prisma query argument objects into a ts-morph writer
- */
-export default class ExpressionWriter {
-    private readonly plainExprBuilder = new TypeScriptExpressionTransformer();
-
-    constructor(private readonly writer: CodeBlockWriter) {}
-
-    /**
-     * Writes the given ZModel expression.
-     */
-    write(expr: Expression): void {
-        const _write = () => {
-            switch (expr.$type) {
-                case LiteralExpr:
-                    this.writeLiteral(expr as LiteralExpr);
-                    break;
-
-                case UnaryExpr:
-                    this.writeUnary(expr as UnaryExpr);
-                    break;
-
-                case BinaryExpr:
-                    this.writeBinary(expr as BinaryExpr);
-                    break;
-
-                case ReferenceExpr:
-                    this.writeReference(expr as ReferenceExpr);
-                    break;
-
-                case MemberAccessExpr:
-                    this.writeMemberAccess(expr as MemberAccessExpr);
-                    break;
-
-                default:
-                    throw new Error(`Not implemented: ${expr.$type}`);
-            }
-        };
-
-        this.block(_write);
-    }
-
-    private writeReference(expr: ReferenceExpr) {
-        if (isEnumField(expr.target.ref)) {
-            throw new Error('We should never get here');
-        } else {
-            this.writer.write(`${expr.target.ref?.name}: true`);
-        }
-    }
-
-    private writeMemberAccess(expr: MemberAccessExpr) {
-        this.writeFieldCondition(
-            expr.operand,
-            () => {
-                this.block(() => {
-                    this.writer.write(`${expr.member.ref?.name}: true`);
-                });
-            },
-            'is'
-        );
-    }
-
-    private writeExprList(exprs: Expression[]) {
-        this.writer.writeLine('[');
-        for (let i = 0; i < exprs.length; i++) {
-            this.write(exprs[i]);
-            if (i !== exprs.length - 1) {
-                this.writer.writeLine(',');
-            }
-        }
-        this.writer.writeLine(']');
-    }
-
-    private writeBinary(expr: BinaryExpr) {
-        switch (expr.operator) {
-            case '&&':
-            case '||':
-                this.writeLogical(expr, expr.operator);
-                break;
-
-            case '==':
-            case '!=':
-            case '>':
-            case '>=':
-            case '<':
-            case '<=':
-                this.writeComparison(expr, expr.operator);
-                break;
-
-            case '?':
-            case '!':
-            case '^':
-                this.writeCollectionPredicate(expr, expr.operator);
-                break;
-        }
-    }
-
-    private writeCollectionPredicate(expr: BinaryExpr, operator: string) {
-        this.writeFieldCondition(
-            expr.left,
-            () => {
-                this.write(expr.right);
-            },
-            operator === '?' ? 'some' : operator === '!' ? 'every' : 'none'
-        );
-    }
-
-    private isFieldAccess(expr: Expression): boolean {
-        if (isThisExpr(expr)) {
-            return true;
-        }
-        if (isMemberAccessExpr(expr)) {
-            return this.isFieldAccess(expr.operand);
-        }
-        if (isReferenceExpr(expr) && isDataModelField(expr.target.ref)) {
-            return true;
-        }
-        return false;
-    }
-
-    private guard(write: () => void) {
-        this.writer.write(`${GUARD_FIELD_NAME}: `);
-        write();
-    }
-
-    private plain(expr: Expression) {
-        this.writer.write(this.plainExprBuilder.transform(expr));
-    }
-
-    private writeComparison(expr: BinaryExpr, operator: ComparisonOperator) {
-        const leftIsFieldAccess = this.isFieldAccess(expr.left);
-        const rightIsFieldAccess = this.isFieldAccess(expr.right);
-
-        if (leftIsFieldAccess && rightIsFieldAccess) {
-            throw new GeneratorError(
-                `Comparison between fields are not supported yet`
-            );
-        }
-
-        if (!leftIsFieldAccess && !rightIsFieldAccess) {
-            // compile down to a plain expression
-            this.guard(() => {
-                this.plain(expr.left);
-                this.writer.write(' ' + operator + ' ');
-                this.plain(expr.right);
-            });
-
-            return;
-        }
-
-        let fieldAccess: Expression;
-        let operand: Expression;
-        if (leftIsFieldAccess) {
-            fieldAccess = expr.left;
-            operand = expr.right;
-        } else {
-            fieldAccess = expr.right;
-            operand = expr.left;
-            operator = this.negateOperator(operator);
-        }
-
-        this.writeFieldCondition(
-            fieldAccess,
-            () => {
-                this.block(
-                    () => {
-                        if (this.isModelTyped(fieldAccess)) {
-                            // comparing with an object, conver to "id" comparison instead
-                            this.writer.write('id: ');
-                            this.block(() => {
-                                this.writeOperator(operator, () => {
-                                    this.plain(operand);
-                                    this.writer.write('?.id');
-                                });
-                            });
-                        } else {
-                            this.writeOperator(operator, () => {
-                                this.plain(operand);
-                            });
-                        }
-                    },
-                    // "this" expression is compiled away (to .id access), so we should
-                    // avoid generating a new layer
-                    !isThisExpr(fieldAccess)
-                );
-            },
-            'is'
-        );
-    }
-
-    private writeOperator(
-        operator: ComparisonOperator,
-        writeOperand: () => void
-    ) {
-        if (operator === '!=') {
-            // wrap a 'not'
-            this.writer.write('not: ');
-            this.block(() => {
-                this.writeOperator('==', writeOperand);
-            });
-        } else {
-            this.writer.write(`${this.mapOperator(operator)}: `);
-            writeOperand();
-        }
-    }
-
-    private writeFieldCondition(
-        fieldAccess: Expression,
-        writeCondition: () => void,
-        relationOp: 'is' | 'some' | 'every' | 'none'
-    ) {
-        let selector: string | undefined;
-        let operand: Expression | undefined;
-
-        if (isThisExpr(fieldAccess)) {
-            // pass on
-            writeCondition();
-            return;
-        } else if (isReferenceExpr(fieldAccess)) {
-            selector = fieldAccess.target.ref?.name;
-        } else if (isMemberAccessExpr(fieldAccess)) {
-            selector = fieldAccess.member.ref?.name;
-            operand = fieldAccess.operand;
-        } else {
-            throw new GeneratorError(
-                `Unsupported expression type: ${fieldAccess.$type}`
-            );
-        }
-
-        if (!selector) {
-            throw new GeneratorError(`Failed to write FieldAccess expression`);
-        }
-
-        if (operand) {
-            // member access expression
-            this.writeFieldCondition(
-                operand,
-                () => {
-                    this.block(
-                        () => {
-                            this.writer.write(selector + ': ');
-                            if (this.isModelTyped(fieldAccess)) {
-                                // expression is resolved to a model, generate relation query
-                                this.block(() => {
-                                    this.writer.write(`${relationOp}: `);
-                                    writeCondition();
-                                });
-                            } else {
-                                // generate plain query
-                                writeCondition();
-                            }
-                        },
-                        // if operand is "this", it doesn't really generate a new layer of query,
-                        // so we should avoid generating a new block
-                        !isThisExpr(operand)
-                    );
-                },
-                'is'
-            );
-        } else if (this.isModelTyped(fieldAccess)) {
-            // reference resolved to a model, generate relation query
-            this.writer.write(selector + ': ');
-            this.block(() => {
-                this.writer.write(`${relationOp}: `);
-                writeCondition();
-            });
-        } else {
-            // generate a plain query
-            this.writer.write(selector + ': ');
-            writeCondition();
-        }
-    }
-
-    private block(write: () => void, condition = true) {
-        if (condition) {
-            this.writer.block(write);
-        } else {
-            write();
-        }
-    }
-
-    private isModelTyped(expr: Expression) {
-        return isDataModel(expr.$resolvedType?.decl);
-    }
-
-    private mapOperator(operator: '==' | '!=' | '>' | '>=' | '<' | '<=') {
-        switch (operator) {
-            case '==':
-                return 'equals';
-            case '!=':
-                throw new Error('Operation != should have been compiled away');
-            case '>':
-                return 'gt';
-            case '>=':
-                return 'ge';
-            case '<':
-                return 'lt';
-            case '<=':
-                return 'le';
-        }
-    }
-
-    private negateOperator(operator: '==' | '!=' | '>' | '>=' | '<' | '<=') {
-        switch (operator) {
-            case '>':
-                return '<=';
-            case '<':
-                return '>=';
-            case '>=':
-                return '<';
-            case '<=':
-                return '>';
-            default:
-                return operator;
-        }
-    }
-
-    private writeLogical(expr: BinaryExpr, operator: '&&' | '||') {
-        this.writer.writeLine(`${operator === '&&' ? 'AND' : 'OR'}: `);
-        this.writeExprList([expr.left, expr.right]);
-    }
-
-    private writeUnary(expr: UnaryExpr) {
-        if (expr.operator !== '!') {
-            throw new GeneratorError(
-                `Unary operator "${expr.operator}" is not supported`
-            );
-        }
-
-        this.writer.writeLine('NOT: ');
-        this.write(expr.operand);
-    }
-
-    private writeLiteral(expr: LiteralExpr) {
-        this.guard(() => {
-            this.plain(expr);
-        });
-    }
-}

package/src/generator/prisma/index.ts

@@ -1,44 +0,0 @@
-import { execSync } from '../../utils/exec-utils';
-import { Context, Generator, GeneratorError } from '../types';
-import QueryGuardGenerator from './query-guard-generator';
-import PrismaSchemaGenerator from './schema-generator';
-
-/**
- * Generates Prisma schema and db client
- */
-export default class PrismaGenerator implements Generator {
-    get name() {
-        return 'prisma';
-    }
-
-    get startMessage() {
-        return 'Generating Prisma client...';
-    }
-
-    get successMessage() {
-        return 'Successfully generated Prisma client';
-    }
-
-    async generate(context: Context) {
-        // generate prisma schema
-        const schemaFile = await new PrismaSchemaGenerator(context).generate();
-
-        // run prisma generate and install @prisma/client
-        await this.generatePrismaClient(schemaFile);
-
-        // generate prisma query guard
-        await new QueryGuardGenerator(context).generate();
-
-        return [];
-    }
-
-    private async generatePrismaClient(schemaFile: string) {
-        try {
-            execSync(`npx prisma generate --schema "${schemaFile}"`);
-        } catch {
-            throw new GeneratorError(
-                `Failed to generate client code with Prisma. Check errors above for clues.\nThis normally shouldn't happen. Please file an issue at: http://go.zenstack.dev/bug.`
-            );
-        }
-    }
-}