@zenstackhq/trpc 1.0.0-beta.1 → 1.0.0-beta.10

package/generator.d.ts

@@ -1,4 +1,4 @@
-import { DMMF } from '@prisma/generator-helper';
+import type { DMMF } from '@prisma/generator-helper';
 import { PluginOptions } from '@zenstackhq/sdk';
 import { Model } from '@zenstackhq/sdk/ast';
 export declare function generate(model: Model, options: PluginOptions, dmmf: DMMF.Document): Promise<void>;

package/generator.js

@@ -25,56 +25,51 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.generate = void 0;
 const sdk_1 = require("@zenstackhq/sdk");
-const fs_1 = require("fs");
+const fs_1 = __importDefault(require("fs"));
 const lower_case_first_1 = require("lower-case-first");
 const path_1 = __importDefault(require("path"));
+const ts_morph_1 = require("ts-morph");
+const upper_case_first_1 = require("upper-case-first");
 const _1 = require(".");
 const helpers_1 = require("./helpers");
 const project_1 = require("./project");
 const removeDir_1 = __importDefault(require("./utils/removeDir"));
-const generator_1 = require("./zod/generator");
 function generate(model, options, dmmf) {
+    var _a;
     return __awaiter(this, void 0, void 0, function* () {
         let outDir = (0, sdk_1.requireOption)(options, 'output');
         outDir = (0, sdk_1.resolvePath)(outDir, options);
         // resolve "generateModelActions" option
-        let generateModelActions = undefined;
-        if (options.generateModelActions) {
-            if (typeof options.generateModelActions === 'string') {
-                // comma separated string
-                generateModelActions = options.generateModelActions
-                    .split(',')
-                    .filter((i) => !!i)
-                    .map((i) => i.trim());
-            }
-            else if (Array.isArray(options.generateModelActions) &&
-                options.generateModelActions.every((i) => typeof i === 'string')) {
-                // string array
-                generateModelActions = options.generateModelActions;
-            }
-            else {
-                throw new sdk_1.PluginError(_1.name, `Invalid "generateModelActions" option: must be a comma-separated string or an array of strings`);
-            }
+        const generateModelActions = parseOptionAsStrings(options, 'generateModelActions');
+        // resolve "generateClientHelpers" option
+        const generateClientHelpers = parseOptionAsStrings(options, 'generateClientHelpers');
+        if (generateClientHelpers && !generateClientHelpers.every((v) => ['react', 'next'].includes(v))) {
+            throw new sdk_1.PluginError(_1.name, `Option "generateClientHelpers" only support values "react" and "next"`);
+        }
+        if (options.zodSchemasImport && typeof options.zodSchemasImport !== 'string') {
+            throw new sdk_1.PluginError(_1.name, `Option "zodSchemasImport" must be a string`);
         }
-        yield fs_1.promises.mkdir(outDir, { recursive: true });
+        yield fs_1.default.promises.mkdir(outDir, { recursive: true });
         yield (0, removeDir_1.default)(outDir, true);
-        yield (0, generator_1.generate)(model, options, dmmf);
         const prismaClientDmmf = dmmf;
         const modelOperations = prismaClientDmmf.mappings.modelOperations;
         const models = prismaClientDmmf.datamodel.models;
         const hiddenModels = [];
         (0, helpers_1.resolveModelsComments)(models, hiddenModels);
-        createAppRouter(outDir, modelOperations, hiddenModels, generateModelActions);
+        const zodSchemasImport = (_a = options.zodSchemasImport) !== null && _a !== void 0 ? _a : '@zenstackhq/runtime/zod';
+        createAppRouter(outDir, modelOperations, hiddenModels, generateModelActions, generateClientHelpers, model, zodSchemasImport);
         createHelper(outDir);
         yield (0, sdk_1.saveProject)(project_1.project);
     });
 }
 exports.generate = generate;
-function createAppRouter(outDir, modelOperations, hiddenModels, generateModelActions) {
-    const appRouter = project_1.project.createSourceFile(path_1.default.resolve(outDir, 'routers', `index.ts`), undefined, {
+function createAppRouter(outDir, modelOperations, hiddenModels, generateModelActions, generateClientHelpers, zmodel, zodSchemasImport) {
+    const indexFile = path_1.default.resolve(outDir, 'routers', `index.ts`);
+    const appRouter = project_1.project.createSourceFile(indexFile, undefined, {
         overwrite: true,
     });
     appRouter.addStatements('/* eslint-disable */');
+    const prismaImport = (0, sdk_1.getPrismaClientImportSpec)(zmodel, path_1.default.dirname(indexFile));
     appRouter.addImportDeclarations([
         {
             namedImports: ['AnyRootConfig'],
@@ -82,10 +77,10 @@ function createAppRouter(outDir, modelOperations, hiddenModels, generateModelAct
         },
         {
             namedImports: ['PrismaClient'],
-            moduleSpecifier: '@prisma/client',
+            moduleSpecifier: prismaImport,
         },
         {
-            namedImports: ['createRouterFactory'],
+            namedImports: ['createRouterFactory', 'AnyRouter'],
             moduleSpecifier: '@trpc/server/dist/core/router',
         },
         {
@@ -112,23 +107,22 @@ function createAppRouter(outDir, modelOperations, hiddenModels, generateModelAct
         }
         
     `);
-    const createFunction = appRouter.addFunction({
+    const filteredModelOperations = modelOperations.filter((mo) => !hiddenModels.includes(mo.model));
+    appRouter
+        .addFunction({
         name: 'createRouter<Config extends BaseConfig>',
         parameters: [
             { name: 'router', type: 'RouterFactory<Config>' },
             { name: 'procedure', type: 'ProcBuilder<Config>' },
         ],
         isExported: true,
-    });
-    createFunction.setBodyText((writer) => {
+    })
+        .setBodyText((writer) => {
         writer.write('return router(');
         writer.block(() => {
-            for (const modelOperation of modelOperations) {
+            for (const modelOperation of filteredModelOperations) {
                 const { model } = modelOperation, operations = __rest(modelOperation, ["model"]);
-                if (hiddenModels.includes(model)) {
-                    continue;
-                }
-                generateModelCreateRouter(project_1.project, model, operations, outDir, generateModelActions);
+                generateModelCreateRouter(project_1.project, model, operations, outDir, generateModelActions, generateClientHelpers, zmodel, zodSchemasImport);
                 appRouter.addImportDeclaration({
                     defaultImport: `create${model}Router`,
                     moduleSpecifier: `./${model}.router`,
@@ -138,9 +132,49 @@ function createAppRouter(outDir, modelOperations, hiddenModels, generateModelAct
         });
         writer.write(');');
     });
+    if (generateClientHelpers) {
+        appRouter.addInterface({
+            name: 'ClientType',
+            typeParameters: ['AppRouter extends AnyRouter'],
+            isExported: true,
+            properties: filteredModelOperations.map(({ model }) => {
+                appRouter.addImportDeclaration({
+                    namedImports: [{ name: 'ClientType', alias: `${(0, upper_case_first_1.upperCaseFirst)(model)}ClientType` }],
+                    moduleSpecifier: `./${model}.router`,
+                });
+                return {
+                    name: (0, lower_case_first_1.lowerCaseFirst)(model),
+                    type: `${(0, upper_case_first_1.upperCaseFirst)(model)}ClientType<AppRouter>`,
+                };
+            }),
+        });
+        createClientHelpers(outDir, generateClientHelpers);
+    }
     appRouter.formatText();
 }
-function generateModelCreateRouter(project, model, operations, outputDir, generateModelActions) {
+function createClientHelpers(outputDir, generateClientHelpers) {
+    const utils = project_1.project.createSourceFile(path_1.default.resolve(outputDir, 'client', `utils.ts`), undefined, {
+        overwrite: true,
+    });
+    utils.replaceWithText(fs_1.default.readFileSync(path_1.default.join(__dirname, './res/client/utils.ts'), 'utf-8'));
+    for (const client of generateClientHelpers) {
+        switch (client) {
+            case 'react': {
+                const content = fs_1.default.readFileSync(path_1.default.join(__dirname, './res/client/react.ts'), 'utf-8');
+                project_1.project.createSourceFile(path_1.default.resolve(outputDir, 'client', 'react.ts'), content, {
+                    overwrite: true,
+                });
+                break;
+            }
+            case 'next': {
+                const content = fs_1.default.readFileSync(path_1.default.join(__dirname, './res/client/next.ts'), 'utf-8');
+                project_1.project.createSourceFile(path_1.default.resolve(outputDir, 'client', 'next.ts'), content, { overwrite: true });
+                break;
+            }
+        }
+    }
+}
+function generateModelCreateRouter(project, model, operations, outputDir, generateModelActions, generateClientHelpers, zmodel, zodSchemasImport) {
     const modelRouter = project.createSourceFile(path_1.default.resolve(outputDir, 'routers', `${model}.router.ts`), undefined, {
         overwrite: true,
     });
@@ -151,10 +185,12 @@ function generateModelCreateRouter(project, model, operations, outputDir, genera
             moduleSpecifier: '.',
         },
     ]);
-    (0, helpers_1.generateRouterSchemaImports)(modelRouter, model);
+    (0, helpers_1.generateRouterSchemaImports)(modelRouter, model, zodSchemasImport);
     (0, helpers_1.generateHelperImport)(modelRouter);
-    modelRouter
-        .addFunction({
+    if (generateClientHelpers) {
+        (0, helpers_1.generateRouterTypingImports)(modelRouter, zmodel);
+    }
+    const createRouterFunc = modelRouter.addFunction({
         name: 'createRouter<Config extends BaseConfig>',
         parameters: [
             { name: 'router', type: 'RouterFactory<Config>' },
@@ -162,23 +198,48 @@ function generateModelCreateRouter(project, model, operations, outputDir, genera
         ],
         isExported: true,
         isDefaultExport: true,
-    })
-        .setBodyText((writer) => {
-        writer.write('return router(');
-        writer.block(() => {
+    });
+    let routerTypingStructure = undefined;
+    if (generateClientHelpers) {
+        // generate an interface for precise Prisma-like typing for the router procedures
+        // which will be used to correct tRPC's typing on the client side
+        routerTypingStructure = {
+            kind: ts_morph_1.StructureKind.Interface,
+            name: 'ClientType',
+            isExported: true,
+            typeParameters: ['AppRouter extends AnyRouter', `Context = AppRouter['_def']['_config']['$types']['ctx']`],
+            properties: [],
+        };
+    }
+    createRouterFunc.setBodyText((funcWriter) => {
+        funcWriter.write('return router(');
+        funcWriter.block(() => {
+            var _a;
             for (const [opType, opNameWithModel] of Object.entries(operations)) {
                 const baseOpType = opType.replace('OrThrow', '');
-                const inputType = (0, helpers_1.getInputTypeByOpName)(baseOpType, model);
+                const inputType = (0, helpers_1.getInputSchemaByOpName)(baseOpType, model);
                 const generateOpName = opType.replace(/One$/, '');
                 if (opNameWithModel &&
                     inputType &&
                     (!generateModelActions || generateModelActions.includes(generateOpName))) {
-                    (0, helpers_1.generateProcedure)(writer, generateOpName, inputType, model, baseOpType);
+                    (0, helpers_1.generateProcedure)(funcWriter, generateOpName, inputType, model, baseOpType);
+                    if (routerTypingStructure) {
+                        (_a = routerTypingStructure.properties) === null || _a === void 0 ? void 0 : _a.push({
+                            kind: ts_morph_1.StructureKind.PropertySignature,
+                            name: generateOpName,
+                            type: (writer) => {
+                                (0, helpers_1.generateRouterTyping)(writer, generateOpName, model, baseOpType);
+                            },
+                        });
+                    }
                 }
             }
         });
-        writer.write(');');
+        funcWriter.write(');');
     });
+    if (routerTypingStructure) {
+        modelRouter.addInterface(routerTypingStructure);
+    }
     modelRouter.formatText();
 }
 function createHelper(outDir) {
@@ -276,4 +337,24 @@ function createHelper(outDir) {
     `);
     checkRead.formatText();
 }
+function parseOptionAsStrings(options, optionaName) {
+    const value = options[optionaName];
+    if (value === undefined) {
+        return undefined;
+    }
+    else if (typeof value === 'string') {
+        // comma separated string
+        return value
+            .split(',')
+            .filter((i) => !!i)
+            .map((i) => i.trim());
+    }
+    else if (Array.isArray(value) && value.every((i) => typeof i === 'string')) {
+        // string array
+        return value;
+    }
+    else {
+        throw new sdk_1.PluginError(_1.name, `Invalid "${optionaName}" option: must be a comma-separated string or an array of strings`);
+    }
+}
 //# sourceMappingURL=generator.js.map

package/generator.js.map

@@ -1 +1 @@
-{"version":3,"file":"generator.js","sourceRoot":"","sources":["../src/generator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AACA,yCAQyB;AAEzB,2BAAoC;AACpC,uDAAkD;AAClD,gDAAwB;AAExB,wBAAyB;AACzB,uCAMmB;AACnB,uCAAoC;AACpC,kEAA0C;AAC1C,+CAAiE;AAEjE,SAAsB,QAAQ,CAAC,KAAY,EAAE,OAAsB,EAAE,IAAmB;;QACpF,IAAI,MAAM,GAAG,IAAA,mBAAa,EAAS,OAAO,EAAE,QAAQ,CAAC,CAAC;QACtD,MAAM,GAAG,IAAA,iBAAW,EAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAEtC,wCAAwC;QACxC,IAAI,oBAAoB,GAAyB,SAAS,CAAC;QAC3D,IAAI,OAAO,CAAC,oBAAoB,EAAE;YAC9B,IAAI,OAAO,OAAO,CAAC,oBAAoB,KAAK,QAAQ,EAAE;gBAClD,yBAAyB;gBACzB,oBAAoB,GAAG,OAAO,CAAC,oBAAoB;qBAC9C,KAAK,CAAC,GAAG,CAAC;qBACV,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;qBAClB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;aAC7B;iBAAM,IACH,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,oBAAoB,CAAC;gBAC3C,OAAO,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,EAClE;gBACE,eAAe;gBACf,oBAAoB,GAAG,OAAO,CAAC,oBAAgC,CAAC;aACnE;iBAAM;gBACH,MAAM,IAAI,iBAAW,CACjB,OAAI,EACJ,gGAAgG,CACnG,CAAC;aACL;SACJ;QAED,MAAM,aAAE,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC5C,MAAM,IAAA,mBAAS,EAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAE9B,MAAM,IAAA,oBAAkB,EAAC,KAAK,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;QAE/C,MAAM,gBAAgB,GAAG,IAAI,CAAC;QAE9B,MAAM,eAAe,GAAG,gBAAgB,CAAC,QAAQ,CAAC,eAAe,CAAC;QAClE,MAAM,MAAM,GAAG,gBAAgB,CAAC,SAAS,CAAC,MAAM,CAAC;QACjD,MAAM,YAAY,GAAa,EAAE,CAAC;QAClC,IAAA,+BAAqB,EAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAE5C,eAAe,CAAC,MAAM,EAAE,eAAe,EAAE,YAAY,EAAE,oBAAoB,CAAC,CAAC;QAC7E,YAAY,CAAC,MAAM,CAAC,CAAC;QAErB,MAAM,IAAA,iBAAW,EAAC,iBAAO,CAAC,CAAC;IAC/B,CAAC;CAAA;AA3CD,4BA2CC;AAED,SAAS,eAAe,CACpB,MAAc,EACd,eAAoC,EACpC,YAAsB,EACtB,oBAA0C;IAE1C,MAAM,SAAS,GAAG,iBAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC,EAAE,SAAS,EAAE;QAC/F,SAAS,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,SAAS,CAAC,aAAa,CAAC,sBAAsB,CAAC,CAAC;IAEhD,SAAS,CAAC,qBAAqB,CAAC;QAC5B;YACI,YAAY,EAAE,CAAC,eAAe,CAAC;YAC/B,eAAe,EAAE,cAAc;SAClC;QACD;YACI,YAAY,EAAE,CAAC,cAAc,CAAC;YAC9B,eAAe,EAAE,gBAAgB;SACpC;QACD;YACI,YAAY,EAAE,CAAC,qBAAqB,CAAC;YACrC,eAAe,EAAE,+BAA+B;SACnD;QACD;YACI,YAAY,EAAE,CAAC,eAAe,CAAC;YAC/B,eAAe,EAAE,mDAAmD;SACvE;KACJ,CAAC,CAAC;IAEH,SAAS,CAAC,aAAa,CAAC;;;;;;;;;;;;;;;;;;KAkBvB,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,SAAS,CAAC,WAAW,CAAC;QACzC,IAAI,EAAE,yCAAyC;QAC/C,UAAU,EAAE;YACR,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,uBAAuB,EAAE;YACjD,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,qBAAqB,EAAE;SACrD;QACD,UAAU,EAAE,IAAI;KACnB,CAAC,CAAC;IAEH,cAAc,CAAC,WAAW,CAAC,CAAC,MAAM,EAAE,EAAE;QAClC,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;YACd,KAAK,MAAM,cAAc,IAAI,eAAe,EAAE;gBAC1C,MAAM,EAAE,KAAK,KAAoB,cAAc,EAA7B,UAAU,UAAK,cAAc,EAAzC,SAAwB,CAAiB,CAAC;gBAChD,IAAI,YAAY,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE;oBAC9B,SAAS;iBACZ;gBAED,yBAAyB,CAAC,iBAAO,EAAE,KAAK,EAAE,UAAU,EAAE,MAAM,EAAE,oBAAoB,CAAC,CAAC;gBAEpF,SAAS,CAAC,oBAAoB,CAAC;oBAC3B,aAAa,EAAE,SAAS,KAAK,QAAQ;oBACrC,eAAe,EAAE,KAAK,KAAK,SAAS;iBACvC,CAAC,CAAC;gBAEH,MAAM,CAAC,SAAS,CAAC,GAAG,IAAA,iCAAc,EAAC,KAAK,CAAC,WAAW,KAAK,oCAAoC,CAAC,CAAC;aAClG;QACL,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,UAAU,EAAE,CAAC;AAC3B,CAAC;AAED,SAAS,yBAAyB,CAC9B,OAAgB,EAChB,KAAa,EACb,UAAqD,EACrD,SAAiB,EACjB,oBAA0C;IAE1C,MAAM,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,GAAG,KAAK,YAAY,CAAC,EAAE,SAAS,EAAE;QAC9G,SAAS,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,WAAW,CAAC,aAAa,CAAC,sBAAsB,CAAC,CAAC;IAElD,WAAW,CAAC,qBAAqB,CAAC;QAC9B;YACI,YAAY,EAAE,CAAC,oBAAoB,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,IAAI,CAAC;YACjF,eAAe,EAAE,GAAG;SACvB;KACJ,CAAC,CAAC;IAEH,IAAA,qCAA2B,EAAC,WAAW,EAAE,KAAK,CAAC,CAAC;IAChD,IAAA,8BAAoB,EAAC,WAAW,CAAC,CAAC;IAElC,WAAW;SACN,WAAW,CAAC;QACT,IAAI,EAAE,yCAAyC;QAC/C,UAAU,EAAE;YACR,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,uBAAuB,EAAE;YACjD,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,qBAAqB,EAAE;SACrD;QACD,UAAU,EAAE,IAAI;QAChB,eAAe,EAAE,IAAI;KACxB,CAAC;SACD,WAAW,CAAC,CAAC,MAAM,EAAE,EAAE;QACpB,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;YACd,KAAK,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE;gBAChE,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;gBACjD,MAAM,SAAS,GAAG,IAAA,8BAAoB,EAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAC1D,MAAM,cAAc,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBAElD,IACI,eAAe;oBACf,SAAS;oBACT,CAAC,CAAC,oBAAoB,IAAI,oBAAoB,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,EAC1E;oBACE,IAAA,2BAAiB,EAAC,MAAM,EAAE,cAAc,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;iBAC3E;aACJ;QACL,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC,CAAC,CAAC;IAEP,WAAW,CAAC,UAAU,EAAE,CAAC;AAC7B,CAAC;AAED,SAAS,YAAY,CAAC,MAAc;IAChC,MAAM,EAAE,GAAG,iBAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,OAAO,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE,SAAS,EAAE;QAC9E,SAAS,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,EAAE,CAAC,aAAa,CAAC,2CAA2C,CAAC,CAAC;IAC9D,EAAE,CAAC,aAAa,CAAC,oDAAoD,qBAAe,IAAI,CAAC,CAAC;IAE1F,MAAM,WAAW,GAAG,EAAE,CAAC,WAAW,CAAC;QAC/B,IAAI,EAAE,aAAa;QACnB,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;QAC/B,UAAU,EAAE;YACR;gBACI,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,YAAY;aACrB;SACJ;QACD,OAAO,EAAE,IAAI;QACb,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,wBAAwB;KACvC,CAAC,CAAC;IAEH,WAAW,CAAC,WAAW,CACnB;;;;;gDAKwC,uBAAiB,CAAC,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;KAuBhF,CACA,CAAC;IACF,WAAW,CAAC,UAAU,EAAE,CAAC;IAEzB,MAAM,SAAS,GAAG,EAAE,CAAC,WAAW,CAAC;QAC7B,IAAI,EAAE,WAAW;QACjB,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;QAC/B,UAAU,EAAE;YACR;gBACI,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,YAAY;aACrB;SACJ;QACD,OAAO,EAAE,IAAI;QACb,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,YAAY;KAC3B,CAAC,CAAC;IAEH,SAAS,CAAC,WAAW,CACjB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA8BH,CACA,CAAC;IACF,SAAS,CAAC,UAAU,EAAE,CAAC;AAC3B,CAAC"}
+{"version":3,"file":"generator.js","sourceRoot":"","sources":["../src/generator.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;AACA,yCASyB;AAEzB,4CAAoB;AACpB,uDAAkD;AAClD,gDAAwB;AACxB,uCAA6G;AAC7G,uDAAkD;AAClD,wBAAyB;AACzB,uCAQmB;AACnB,uCAAoC;AACpC,kEAA0C;AAE1C,SAAsB,QAAQ,CAAC,KAAY,EAAE,OAAsB,EAAE,IAAmB;;;QACpF,IAAI,MAAM,GAAG,IAAA,mBAAa,EAAS,OAAO,EAAE,QAAQ,CAAC,CAAC;QACtD,MAAM,GAAG,IAAA,iBAAW,EAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QAEtC,wCAAwC;QACxC,MAAM,oBAAoB,GAAG,oBAAoB,CAAC,OAAO,EAAE,sBAAsB,CAAC,CAAC;QAEnF,yCAAyC;QACzC,MAAM,qBAAqB,GAAG,oBAAoB,CAAC,OAAO,EAAE,uBAAuB,CAAC,CAAC;QACrF,IAAI,qBAAqB,IAAI,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE;YAC7F,MAAM,IAAI,iBAAW,CAAC,OAAI,EAAE,uEAAuE,CAAC,CAAC;SACxG;QAED,IAAI,OAAO,CAAC,gBAAgB,IAAI,OAAO,OAAO,CAAC,gBAAgB,KAAK,QAAQ,EAAE;YAC1E,MAAM,IAAI,iBAAW,CAAC,OAAI,EAAE,4CAA4C,CAAC,CAAC;SAC7E;QAED,MAAM,YAAE,CAAC,QAAQ,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QACrD,MAAM,IAAA,mBAAS,EAAC,MAAM,EAAE,IAAI,CAAC,CAAC;QAE9B,MAAM,gBAAgB,GAAG,IAAI,CAAC;QAE9B,MAAM,eAAe,GAAG,gBAAgB,CAAC,QAAQ,CAAC,eAAe,CAAC;QAClE,MAAM,MAAM,GAAG,gBAAgB,CAAC,SAAS,CAAC,MAAM,CAAC;QACjD,MAAM,YAAY,GAAa,EAAE,CAAC;QAClC,IAAA,+BAAqB,EAAC,MAAM,EAAE,YAAY,CAAC,CAAC;QAE5C,MAAM,gBAAgB,GAAG,MAAC,OAAO,CAAC,gBAA2B,mCAAI,yBAAyB,CAAC;QAC3F,eAAe,CACX,MAAM,EACN,eAAe,EACf,YAAY,EACZ,oBAAoB,EACpB,qBAAqB,EACrB,KAAK,EACL,gBAAgB,CACnB,CAAC;QACF,YAAY,CAAC,MAAM,CAAC,CAAC;QAErB,MAAM,IAAA,iBAAW,EAAC,iBAAO,CAAC,CAAC;;CAC9B;AAxCD,4BAwCC;AAED,SAAS,eAAe,CACpB,MAAc,EACd,eAAoC,EACpC,YAAsB,EACtB,oBAA0C,EAC1C,qBAA2C,EAC3C,MAAa,EACb,gBAAwB;IAExB,MAAM,SAAS,GAAG,cAAI,CAAC,OAAO,CAAC,MAAM,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;IAC9D,MAAM,SAAS,GAAG,iBAAO,CAAC,gBAAgB,CAAC,SAAS,EAAE,SAAS,EAAE;QAC7D,SAAS,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,SAAS,CAAC,aAAa,CAAC,sBAAsB,CAAC,CAAC;IAEhD,MAAM,YAAY,GAAG,IAAA,+BAAyB,EAAC,MAAM,EAAE,cAAI,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC;IAChF,SAAS,CAAC,qBAAqB,CAAC;QAC5B;YACI,YAAY,EAAE,CAAC,eAAe,CAAC;YAC/B,eAAe,EAAE,cAAc;SAClC;QACD;YACI,YAAY,EAAE,CAAC,cAAc,CAAC;YAC9B,eAAe,EAAE,YAAY;SAChC;QACD;YACI,YAAY,EAAE,CAAC,qBAAqB,EAAE,WAAW,CAAC;YAClD,eAAe,EAAE,+BAA+B;SACnD;QACD;YACI,YAAY,EAAE,CAAC,eAAe,CAAC;YAC/B,eAAe,EAAE,mDAAmD;SACvE;KACJ,CAAC,CAAC;IAEH,SAAS,CAAC,aAAa,CAAC;;;;;;;;;;;;;;;;;;KAkBvB,CAAC,CAAC;IAEH,MAAM,uBAAuB,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;IAEjG,SAAS;SACJ,WAAW,CAAC;QACT,IAAI,EAAE,yCAAyC;QAC/C,UAAU,EAAE;YACR,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,uBAAuB,EAAE;YACjD,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,qBAAqB,EAAE;SACrD;QACD,UAAU,EAAE,IAAI;KACnB,CAAC;SACD,WAAW,CAAC,CAAC,MAAM,EAAE,EAAE;QACpB,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QAC/B,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE;YACd,KAAK,MAAM,cAAc,IAAI,uBAAuB,EAAE;gBAClD,MAAM,EAAE,KAAK,KAAoB,cAAc,EAA7B,UAAU,UAAK,cAAc,EAAzC,SAAwB,CAAiB,CAAC;gBAChD,yBAAyB,CACrB,iBAAO,EACP,KAAK,EACL,UAAU,EACV,MAAM,EACN,oBAAoB,EACpB,qBAAqB,EACrB,MAAM,EACN,gBAAgB,CACnB,CAAC;gBAEF,SAAS,CAAC,oBAAoB,CAAC;oBAC3B,aAAa,EAAE,SAAS,KAAK,QAAQ;oBACrC,eAAe,EAAE,KAAK,KAAK,SAAS;iBACvC,CAAC,CAAC;gBAEH,MAAM,CAAC,SAAS,CAAC,GAAG,IAAA,iCAAc,EAAC,KAAK,CAAC,WAAW,KAAK,oCAAoC,CAAC,CAAC;aAClG;QACL,CAAC,CAAC,CAAC;QACH,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC,CAAC,CAAC;IAEP,IAAI,qBAAqB,EAAE;QACvB,SAAS,CAAC,YAAY,CAAC;YACnB,IAAI,EAAE,YAAY;YAClB,cAAc,EAAE,CAAC,6BAA6B,CAAC;YAC/C,UAAU,EAAE,IAAI;YAChB,UAAU,EAAE,uBAAuB,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE;gBAClD,SAAS,CAAC,oBAAoB,CAAC;oBAC3B,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,GAAG,IAAA,iCAAc,EAAC,KAAK,CAAC,YAAY,EAAE,CAAC;oBACnF,eAAe,EAAE,KAAK,KAAK,SAAS;iBACvC,CAAC,CAAC;gBACH,OAAO;oBACH,IAAI,EAAE,IAAA,iCAAc,EAAC,KAAK,CAAC;oBAC3B,IAAI,EAAE,GAAG,IAAA,iCAAc,EAAC,KAAK,CAAC,uBAAuB;iBAC1B,CAAC;YACpC,CAAC,CAAC;SACL,CAAC,CAAC;QAEH,mBAAmB,CAAC,MAAM,EAAE,qBAAqB,CAAC,CAAC;KACtD;IAED,SAAS,CAAC,UAAU,EAAE,CAAC;AAC3B,CAAC;AAED,SAAS,mBAAmB,CAAC,SAAiB,EAAE,qBAA+B;IAC3E,MAAM,KAAK,GAAG,iBAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,EAAE,UAAU,CAAC,EAAE,SAAS,EAAE;QAC7F,SAAS,EAAE,IAAI;KAClB,CAAC,CAAC;IACH,KAAK,CAAC,eAAe,CAAC,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,uBAAuB,CAAC,EAAE,OAAO,CAAC,CAAC,CAAC;IAE/F,KAAK,MAAM,MAAM,IAAI,qBAAqB,EAAE;QACxC,QAAQ,MAAM,EAAE;YACZ,KAAK,OAAO,CAAC,CAAC;gBACV,MAAM,OAAO,GAAG,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,uBAAuB,CAAC,EAAE,OAAO,CAAC,CAAC;gBACxF,iBAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,EAAE,UAAU,CAAC,EAAE,OAAO,EAAE;oBAC7E,SAAS,EAAE,IAAI;iBAClB,CAAC,CAAC;gBACH,MAAM;aACT;YAED,KAAK,MAAM,CAAC,CAAC;gBACT,MAAM,OAAO,GAAG,YAAE,CAAC,YAAY,CAAC,cAAI,CAAC,IAAI,CAAC,SAAS,EAAE,sBAAsB,CAAC,EAAE,OAAO,CAAC,CAAC;gBACvF,iBAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,EAAE,SAAS,CAAC,EAAE,OAAO,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBACrG,MAAM;aACT;SACJ;KACJ;AACL,CAAC;AAED,SAAS,yBAAyB,CAC9B,OAAgB,EAChB,KAAa,EACb,UAAqD,EACrD,SAAiB,EACjB,oBAA0C,EAC1C,qBAA2C,EAC3C,MAAa,EACb,gBAAwB;IAExB,MAAM,WAAW,GAAG,OAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,OAAO,CAAC,SAAS,EAAE,SAAS,EAAE,GAAG,KAAK,YAAY,CAAC,EAAE,SAAS,EAAE;QAC9G,SAAS,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,WAAW,CAAC,aAAa,CAAC,sBAAsB,CAAC,CAAC;IAElD,WAAW,CAAC,qBAAqB,CAAC;QAC9B;YACI,YAAY,EAAE,CAAC,oBAAoB,EAAE,kBAAkB,EAAE,iBAAiB,EAAE,IAAI,CAAC;YACjF,eAAe,EAAE,GAAG;SACvB;KACJ,CAAC,CAAC;IAEH,IAAA,qCAA2B,EAAC,WAAW,EAAE,KAAK,EAAE,gBAAgB,CAAC,CAAC;IAClE,IAAA,8BAAoB,EAAC,WAAW,CAAC,CAAC;IAClC,IAAI,qBAAqB,EAAE;QACvB,IAAA,qCAA2B,EAAC,WAAW,EAAE,MAAM,CAAC,CAAC;KACpD;IAED,MAAM,gBAAgB,GAAG,WAAW,CAAC,WAAW,CAAC;QAC7C,IAAI,EAAE,yCAAyC;QAC/C,UAAU,EAAE;YACR,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,uBAAuB,EAAE;YACjD,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,qBAAqB,EAAE;SACrD;QACD,UAAU,EAAE,IAAI;QAChB,eAAe,EAAE,IAAI;KACxB,CAAC,CAAC;IAEH,IAAI,qBAAqB,GAA8C,SAAS,CAAC;IACjF,IAAI,qBAAqB,EAAE;QACvB,iFAAiF;QACjF,iEAAiE;QACjE,qBAAqB,GAAG;YACpB,IAAI,EAAE,wBAAa,CAAC,SAAS;YAC7B,IAAI,EAAE,YAAY;YAClB,UAAU,EAAE,IAAI;YAChB,cAAc,EAAE,CAAC,6BAA6B,EAAE,yDAAyD,CAAC;YAC1G,UAAU,EAAE,EAAkC;SACjD,CAAC;KACL;IAED,gBAAgB,CAAC,WAAW,CAAC,CAAC,UAAU,EAAE,EAAE;QACxC,UAAU,CAAC,KAAK,CAAC,gBAAgB,CAAC,CAAC;QACnC,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE;;YAClB,KAAK,MAAM,CAAC,MAAM,EAAE,eAAe,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE;gBAChE,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;gBACjD,MAAM,SAAS,GAAG,IAAA,gCAAsB,EAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAC5D,MAAM,cAAc,GAAG,MAAM,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBAElD,IACI,eAAe;oBACf,SAAS;oBACT,CAAC,CAAC,oBAAoB,IAAI,oBAAoB,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,EAC1E;oBACE,IAAA,2BAAiB,EAAC,UAAU,EAAE,cAAc,EAAE,SAAS,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;oBAE5E,IAAI,qBAAqB,EAAE;wBACvB,MAAA,qBAAqB,CAAC,UAAU,0CAAE,IAAI,CAAC;4BACnC,IAAI,EAAE,wBAAa,CAAC,iBAAiB;4BACrC,IAAI,EAAE,cAAc;4BACpB,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE;gCACb,IAAA,8BAAoB,EAAC,MAAM,EAAE,cAAc,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC;4BACpE,CAAC;yBACJ,CAAC,CAAC;qBACN;iBACJ;aACJ;QACL,CAAC,CAAC,CAAC;QACH,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,IAAI,qBAAqB,EAAE;QACvB,WAAW,CAAC,YAAY,CAAC,qBAAqB,CAAC,CAAC;KACnD;IAED,WAAW,CAAC,UAAU,EAAE,CAAC;AAC7B,CAAC;AAED,SAAS,YAAY,CAAC,MAAc;IAChC,MAAM,EAAE,GAAG,iBAAO,CAAC,gBAAgB,CAAC,cAAI,CAAC,OAAO,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE,SAAS,EAAE;QAC9E,SAAS,EAAE,IAAI;KAClB,CAAC,CAAC;IAEH,EAAE,CAAC,aAAa,CAAC,2CAA2C,CAAC,CAAC;IAC9D,EAAE,CAAC,aAAa,CAAC,oDAAoD,qBAAe,IAAI,CAAC,CAAC;IAE1F,MAAM,WAAW,GAAG,EAAE,CAAC,WAAW,CAAC;QAC/B,IAAI,EAAE,aAAa;QACnB,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;QAC/B,UAAU,EAAE;YACR;gBACI,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,YAAY;aACrB;SACJ;QACD,OAAO,EAAE,IAAI;QACb,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,wBAAwB;KACvC,CAAC,CAAC;IAEH,WAAW,CAAC,WAAW,CACnB;;;;;gDAKwC,uBAAiB,CAAC,mBAAmB;;;;;;;;;;;;;;;;;;;;;;;KAuBhF,CACA,CAAC;IACF,WAAW,CAAC,UAAU,EAAE,CAAC;IAEzB,MAAM,SAAS,GAAG,EAAE,CAAC,WAAW,CAAC;QAC7B,IAAI,EAAE,WAAW;QACjB,cAAc,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC;QAC/B,UAAU,EAAE;YACR;gBACI,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,YAAY;aACrB;SACJ;QACD,OAAO,EAAE,IAAI;QACb,UAAU,EAAE,IAAI;QAChB,UAAU,EAAE,YAAY;KAC3B,CAAC,CAAC;IAEH,SAAS,CAAC,WAAW,CACjB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;KA8BH,CACA,CAAC;IACF,SAAS,CAAC,UAAU,EAAE,CAAC;AAC3B,CAAC;AAED,SAAS,oBAAoB,CAAC,OAAsB,EAAE,WAAmB;IACrE,MAAM,KAAK,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IACnC,IAAI,KAAK,KAAK,SAAS,EAAE;QACrB,OAAO,SAAS,CAAC;KACpB;SAAM,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE;QAClC,yBAAyB;QACzB,OAAO,KAAK;aACP,KAAK,CAAC,GAAG,CAAC;aACV,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;aAClB,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;KAC7B;SAAM,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,OAAO,CAAC,KAAK,QAAQ,CAAC,EAAE;QAC1E,eAAe;QACf,OAAO,KAAiB,CAAC;KAC5B;SAAM;QACH,MAAM,IAAI,iBAAW,CACjB,OAAI,EACJ,YAAY,WAAW,mEAAmE,CAC7F,CAAC;KACL;AACL,CAAC"}

package/helpers.d.ts

@@ -1,8 +1,14 @@
-import { DMMF } from '@prisma/generator-helper';
+import type { DMMF } from '@prisma/generator-helper';
+import { Model } from '@zenstackhq/sdk/ast';
 import { CodeBlockWriter, SourceFile } from 'ts-morph';
 export declare function generateProcedure(writer: CodeBlockWriter, opType: string, typeName: string, modelName: string, baseOpType: string): void;
-export declare function generateRouterSchemaImports(sourceFile: SourceFile, name: string): void;
+/**
+ * Generate precise Prisma-like typing for router procedures.
+ */
+export declare function generateRouterTyping(writer: CodeBlockWriter, opType: string, modelName: string, baseOpType: string): void;
+export declare function generateRouterTypingImports(sourceFile: SourceFile, model: Model): void;
+export declare function generateRouterSchemaImports(sourceFile: SourceFile, name: string, zodSchemasImport: string): void;
 export declare function generateHelperImport(sourceFile: SourceFile): void;
-export declare const getInputTypeByOpName: (opName: string, modelName: string) => string | undefined;
+export declare const getInputSchemaByOpName: (opName: string, modelName: string) => string | undefined;
 export declare const getProcedureTypeByOpName: (opName: string) => string | undefined;
 export declare function resolveModelsComments(models: DMMF.Model[], hiddenModels: string[]): void;

package/helpers.js

@@ -1,81 +1,272 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.resolveModelsComments = exports.getProcedureTypeByOpName = exports.getInputTypeByOpName = exports.generateHelperImport = exports.generateRouterSchemaImports = exports.generateProcedure = void 0;
-const uncapitalizeFirstLetter_1 = require("./utils/uncapitalizeFirstLetter");
+exports.resolveModelsComments = exports.getProcedureTypeByOpName = exports.getInputSchemaByOpName = exports.generateHelperImport = exports.generateRouterSchemaImports = exports.generateRouterTypingImports = exports.generateRouterTyping = exports.generateProcedure = void 0;
+const sdk_1 = require("@zenstackhq/sdk");
+const lower_case_first_1 = require("lower-case-first");
+const upper_case_first_1 = require("upper-case-first");
+const _1 = require(".");
 function generateProcedure(writer, opType, typeName, modelName, baseOpType) {
     const procType = (0, exports.getProcedureTypeByOpName)(baseOpType);
     const prismaMethod = opType.replace('One', '');
     if (procType === 'query') {
+        // the cast "as any" is to circumvent a TS compiler misfired error in certain cases
         writer.write(`
-        ${opType}: procedure.input(${typeName}).query(({ctx, input}) => checkRead(db(ctx).${(0, uncapitalizeFirstLetter_1.uncapitalizeFirstLetter)(modelName)}.${prismaMethod}(input))),
+        ${opType}: procedure.input(${typeName}).query(({ctx, input}) => checkRead(db(ctx).${(0, lower_case_first_1.lowerCaseFirst)(modelName)}.${prismaMethod}(input as any))),
     `);
     }
     else if (procType === 'mutation') {
+        // the cast "as any" is to circumvent a TS compiler misfired error in certain cases
         writer.write(`
-        ${opType}: procedure.input(${typeName}).mutation(async ({ctx, input}) => checkMutate(db(ctx).${(0, uncapitalizeFirstLetter_1.uncapitalizeFirstLetter)(modelName)}.${prismaMethod}(input))),
+        ${opType}: procedure.input(${typeName}).mutation(async ({ctx, input}) => checkMutate(db(ctx).${(0, lower_case_first_1.lowerCaseFirst)(modelName)}.${prismaMethod}(input as any))),
     `);
     }
 }
 exports.generateProcedure = generateProcedure;
-function generateRouterSchemaImports(sourceFile, name) {
-    sourceFile.addStatements(`import { ${name}Schema } from '../schemas/${name}.schema';`);
+/**
+ * Given a model and Prisma operation, returns related TS types.
+ */
+function getPrismaOperationTypes(model, operation) {
+    // TODO: find a way to derive from Prisma Client API's generic types
+    // instead of duplicating them
+    const capModel = (0, upper_case_first_1.upperCaseFirst)(model);
+    const capOperation = (0, upper_case_first_1.upperCaseFirst)(operation);
+    let genericBase = `Prisma.${capModel}${capOperation}Args`;
+    const getPayload = `Prisma.${capModel}GetPayload<T>`;
+    const selectSubset = `Prisma.SelectSubset<T, ${genericBase}>`;
+    let argsType;
+    let resultType;
+    switch (operation) {
+        case 'findUnique':
+        case 'findUniqueOrThrow':
+        case 'findFirst':
+        case 'findFirstOrThrow':
+            argsType = selectSubset;
+            resultType = getPayload;
+            break;
+        case 'findMany':
+            argsType = selectSubset;
+            resultType = `Array<${getPayload}>`;
+            break;
+        case 'create':
+            argsType = selectSubset;
+            resultType = getPayload;
+            break;
+        case 'createMany':
+            argsType = selectSubset;
+            resultType = `Prisma.BatchPayload`;
+            break;
+        case 'update':
+            argsType = selectSubset;
+            resultType = getPayload;
+            break;
+        case 'updateMany':
+            argsType = selectSubset;
+            resultType = `Prisma.BatchPayload`;
+            break;
+        case 'upsert':
+            argsType = selectSubset;
+            resultType = getPayload;
+            break;
+        case 'delete':
+            argsType = selectSubset;
+            resultType = getPayload;
+            break;
+        case 'deleteMany':
+            argsType = selectSubset;
+            resultType = `Prisma.BatchPayload`;
+            break;
+        case 'count':
+            argsType = `Prisma.Subset<T, ${genericBase}>`;
+            resultType = `'select' extends keyof T'
+            ? T['select'] extends true
+              ? number
+              : GetScalarType<T['select'], ${capModel}CountAggregateOutputType>
+            : number`;
+            break;
+        case 'aggregate':
+            argsType = `Prisma.Subset<T, ${genericBase}>`;
+            resultType = `Prisma.Get${capModel}AggregateType<T>`;
+            break;
+        case 'groupBy':
+            genericBase = `Prisma.${capModel}GroupByArgs,
+            HasSelectOrTake extends Prisma.Or<
+              Prisma.Extends<'skip', Prisma.Keys<T>>,
+              Prisma.Extends<'take', Prisma.Keys<T>>
+            >,
+            OrderByArg extends Prisma.True extends HasSelectOrTake
+              ? { orderBy: Prisma.${capModel}GroupByArgs['orderBy'] }
+              : { orderBy?: Prisma.${capModel}GroupByArgs['orderBy'] },
+            OrderFields extends Prisma.ExcludeUnderscoreKeys<Prisma.Keys<Prisma.MaybeTupleToUnion<T['orderBy']>>>,
+            ByFields extends Prisma.MaybeTupleToUnion<T['by']>,
+            ByValid extends Prisma.Has<ByFields, OrderFields>,
+            HavingFields extends Prisma.GetHavingFields<T['having']>,
+            HavingValid extends Prisma.Has<ByFields, HavingFields>,
+            ByEmpty extends T['by'] extends never[] ? Prisma.True : Prisma.False,
+            InputErrors extends ByEmpty extends Prisma.True
+            ? \`Error: "by" must not be empty.\`
+            : HavingValid extends Prisma.False
+            ? {
+                [P in HavingFields]: P extends ByFields
+                  ? never
+                  : P extends string
+                  ? \`Error: Field "\${P}" used in "having" needs to be provided in "by".\`
+                  : [
+                      Error,
+                      'Field ',
+                      P,
+                      \` in "having" needs to be provided in "by"\`,
+                    ]
+              }[HavingFields]
+            : 'take' extends Prisma.Keys<T>
+            ? 'orderBy' extends Prisma.Keys<T>
+              ? ByValid extends Prisma.True
+                ? {}
+                : {
+                    [P in OrderFields]: P extends ByFields
+                      ? never
+                      : \`Error: Field "\${P}" in "orderBy" needs to be provided in "by"\`
+                  }[OrderFields]
+              : 'Error: If you provide "take", you also need to provide "orderBy"'
+            : 'skip' extends Prisma.Keys<T>
+            ? 'orderBy' extends Prisma.Keys<T>
+              ? ByValid extends Prisma.True
+                ? {}
+                : {
+                    [P in OrderFields]: P extends ByFields
+                      ? never
+                      : \`Error: Field "\${P}" in "orderBy" needs to be provided in "by"\`
+                  }[OrderFields]
+              : 'Error: If you provide "skip", you also need to provide "orderBy"'
+            : ByValid extends Prisma.True
+            ? {}
+            : {
+                [P in OrderFields]: P extends ByFields
+                  ? never
+                  : \`Error: Field "\${P}" in "orderBy" needs to be provided in "by"\`
+              }[OrderFields]
+          `;
+            argsType = `Prisma.SubsetIntersection<T, Prisma.${capModel}GroupByArgs, OrderByArg> & InputErrors`;
+            resultType = `{} extends InputErrors ? Prisma.Get${capModel}GroupByPayload<T> : InputErrors`;
+            break;
+        default:
+            throw new sdk_1.PluginError(_1.name, `Unsupported operation: "${operation}"`);
+    }
+    return { genericBase, argsType, resultType };
+}
+/**
+ * Generate precise Prisma-like typing for router procedures.
+ */
+function generateRouterTyping(writer, opType, modelName, baseOpType) {
+    const procType = (0, exports.getProcedureTypeByOpName)(baseOpType);
+    const { genericBase, argsType, resultType } = getPrismaOperationTypes(modelName, opType);
+    const errorType = `TRPCClientErrorLike<AppRouter>`;
+    writer.block(() => {
+        if (procType === 'query') {
+            writer.writeLine(`
+                useQuery: <T extends ${genericBase}>(
+                    input: ${argsType},
+                    opts?: UseTRPCQueryOptions<string, T, ${resultType}, ${resultType}, Error>
+                    ) => UseTRPCQueryResult<
+                    ${resultType},
+                        ${errorType}
+                    >;
+                useInfiniteQuery: <T extends ${genericBase}>(
+                    input: Omit<${argsType}, 'cursor'>,
+                    opts?: UseTRPCInfiniteQueryOptions<string, T, ${resultType}, Error>
+                    ) => UseTRPCInfiniteQueryResult<
+                    ${resultType},
+                        ${errorType}
+                    >;
+                    `);
+        }
+        else if (procType === 'mutation') {
+            writer.writeLine(`
+                useMutation: <T extends ${genericBase}>(opts?: UseTRPCMutationOptions<
+                    ${genericBase},
+                    ${errorType},
+                    Prisma.${(0, upper_case_first_1.upperCaseFirst)(modelName)}GetPayload<null>,
+                    Context
+                >,) =>
+                Omit<UseTRPCMutationResult<${resultType}, ${errorType}, ${argsType}, Context>, 'mutateAsync'> & {
+                    mutateAsync:
+                        <T extends ${genericBase}>(variables: T, opts?: UseTRPCMutationOptions<T, ${errorType}, ${resultType}, Context>) => Promise<${resultType}>
+                };
+                `);
+        }
+    });
+}
+exports.generateRouterTyping = generateRouterTyping;
+function generateRouterTypingImports(sourceFile, model) {
+    const importingDir = sourceFile.getDirectoryPath();
+    const prismaImport = (0, sdk_1.getPrismaClientImportSpec)(model, importingDir);
+    sourceFile.addStatements([
+        `import type { Prisma } from '${prismaImport}';`,
+        `import type { UseTRPCMutationOptions, UseTRPCMutationResult, UseTRPCQueryOptions, UseTRPCQueryResult, UseTRPCInfiniteQueryOptions, UseTRPCInfiniteQueryResult } from '@trpc/react-query/shared';`,
+        `import type { TRPCClientErrorLike } from '@trpc/client';`,
+        `import type { AnyRouter } from '@trpc/server';`,
+    ]);
+}
+exports.generateRouterTypingImports = generateRouterTypingImports;
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
+function generateRouterSchemaImports(sourceFile, name, zodSchemasImport) {
+    sourceFile.addStatements(`import { ${name}InputSchema } from '${zodSchemasImport}/input';`);
 }
 exports.generateRouterSchemaImports = generateRouterSchemaImports;
 function generateHelperImport(sourceFile) {
     sourceFile.addStatements(`import { checkRead, checkMutate } from '../helper';`);
 }
 exports.generateHelperImport = generateHelperImport;
-const getInputTypeByOpName = (opName, modelName) => {
+const getInputSchemaByOpName = (opName, modelName) => {
     let inputType;
     switch (opName) {
         case 'findUnique':
-            inputType = `${modelName}Schema.findUnique`;
+            inputType = `${modelName}InputSchema.findUnique`;
             break;
         case 'findFirst':
-            inputType = `${modelName}Schema.findFirst`;
+            inputType = `${modelName}InputSchema.findFirst`;
             break;
         case 'findMany':
-            inputType = `${modelName}Schema.findMany`;
+            inputType = `${modelName}InputSchema.findMany`;
             break;
         case 'findRaw':
-            inputType = `${modelName}Schema.findRawObject`;
+            inputType = `${modelName}InputSchema.findRawObject`;
             break;
         case 'createOne':
-            inputType = `${modelName}Schema.create`;
+            inputType = `${modelName}InputSchema.create`;
             break;
         case 'createMany':
-            inputType = `${modelName}Schema.createMany`;
+            inputType = `${modelName}InputSchema.createMany`;
             break;
         case 'deleteOne':
-            inputType = `${modelName}Schema.delete`;
+            inputType = `${modelName}InputSchema.delete`;
             break;
         case 'updateOne':
-            inputType = `${modelName}Schema.update`;
+            inputType = `${modelName}InputSchema.update`;
             break;
         case 'deleteMany':
-            inputType = `${modelName}Schema.deleteMany`;
+            inputType = `${modelName}InputSchema.deleteMany`;
             break;
         case 'updateMany':
-            inputType = `${modelName}Schema.updateMany`;
+            inputType = `${modelName}InputSchema.updateMany`;
             break;
         case 'upsertOne':
-            inputType = `${modelName}Schema.upsert`;
+            inputType = `${modelName}InputSchema.upsert`;
             break;
         case 'aggregate':
-            inputType = `${modelName}Schema.aggregate`;
+            inputType = `${modelName}InputSchema.aggregate`;
             break;
         case 'aggregateRaw':
-            inputType = `${modelName}Schema.aggregateRawObject`;
+            inputType = `${modelName}InputSchema.aggregateRawObject`;
             break;
         case 'groupBy':
-            inputType = `${modelName}Schema.groupBy`;
+            inputType = `${modelName}InputSchema.groupBy`;
             break;
         default:
             console.log('getInputTypeByOpName: ', { opName, modelName });
     }
     return inputType;
 };
-exports.getInputTypeByOpName = getInputTypeByOpName;
+exports.getInputSchemaByOpName = getInputSchemaByOpName;
 const getProcedureTypeByOpName = (opName) => {
     let procType;
     switch (opName) {