zenstack 0.6.0-pre.10 → 0.6.0-pre.12

package/plugins/trpc/helpers.js

@@ -0,0 +1,144 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.generateProcedure = generateProcedure;
+exports.generateRouterImport = void 0;
+exports.generateRouterSchemaImports = generateRouterSchemaImports;
+exports.getProcedureTypeByOpName = exports.getInputTypeByOpName = exports.generatetRPCImport = void 0;
+exports.resolveModelsComments = resolveModelsComments;
+var _uncapitalizeFirstLetter = require("./utils/uncapitalizeFirstLetter");
+const generatetRPCImport = sourceFile => {
+  sourceFile.addImportDeclaration({
+    moduleSpecifier: '@trpc/server',
+    namespaceImport: 'trpc'
+  });
+};
+exports.generatetRPCImport = generatetRPCImport;
+const generateRouterImport = (sourceFile, modelNamePlural, modelNameCamelCase) => {
+  sourceFile.addImportDeclaration({
+    moduleSpecifier: `./${modelNameCamelCase}.router`,
+    namedImports: [`${modelNamePlural}Router`]
+  });
+};
+exports.generateRouterImport = generateRouterImport;
+function generateProcedure(writer, opType, typeName, modelName, baseOpType) {
+  const procType = getProcedureTypeByOpName(baseOpType);
+  writer.write(`
+        ${opType}: procedure.input(${typeName}).${procType}(({ctx, input}) => db(ctx).${(0, _uncapitalizeFirstLetter.uncapitalizeFirstLetter)(modelName)}.${opType.replace('One', '')}(input)),
+    `);
+}
+function generateRouterSchemaImports(sourceFile, name) {
+  sourceFile.addStatements(`import { ${name}Schema } from '../schemas/${name}.schema';`);
+}
+const getInputTypeByOpName = (opName, modelName) => {
+  let inputType;
+  switch (opName) {
+    case 'findUnique':
+      inputType = `${modelName}Schema.findUnique`;
+      break;
+    case 'findFirst':
+      inputType = `${modelName}Schema.findFirst`;
+      break;
+    case 'findMany':
+      inputType = `${modelName}Schema.findMany`;
+      break;
+    case 'findRaw':
+      inputType = `${modelName}Schema.findRawObject`;
+      break;
+    case 'createOne':
+      inputType = `${modelName}Schema.create`;
+      break;
+    case 'createMany':
+      inputType = `${modelName}Schema.createMany`;
+      break;
+    case 'deleteOne':
+      inputType = `${modelName}Schema.delete`;
+      break;
+    case 'updateOne':
+      inputType = `${modelName}Schema.update`;
+      break;
+    case 'deleteMany':
+      inputType = `${modelName}Schema.deleteMany`;
+      break;
+    case 'updateMany':
+      inputType = `${modelName}Schema.updateMany`;
+      break;
+    case 'upsertOne':
+      inputType = `${modelName}Schema.upsert`;
+      break;
+    case 'aggregate':
+      inputType = `${modelName}Schema.aggregate`;
+      break;
+    case 'aggregateRaw':
+      inputType = `${modelName}Schema.aggregateRawObject`;
+      break;
+    case 'groupBy':
+      inputType = `${modelName}Schema.groupBy`;
+      break;
+    default:
+      console.log('getInputTypeByOpName: ', {
+        opName,
+        modelName
+      });
+  }
+  return inputType;
+};
+exports.getInputTypeByOpName = getInputTypeByOpName;
+const getProcedureTypeByOpName = opName => {
+  let procType;
+  switch (opName) {
+    case 'findUnique':
+    case 'findFirst':
+    case 'findMany':
+    case 'findRaw':
+    case 'aggregate':
+    case 'aggregateRaw':
+    case 'groupBy':
+      procType = 'query';
+      break;
+    case 'createOne':
+    case 'createMany':
+    case 'deleteOne':
+    case 'updateOne':
+    case 'deleteMany':
+    case 'updateMany':
+    case 'upsertOne':
+      procType = 'mutation';
+      break;
+    default:
+      console.log('getProcedureTypeByOpName: ', {
+        opName
+      });
+  }
+  return procType;
+};
+exports.getProcedureTypeByOpName = getProcedureTypeByOpName;
+function resolveModelsComments(models, hiddenModels) {
+  const modelAttributeRegex = /(@@Gen\.)+([A-z])+(\()+(.+)+(\))+/;
+  const attributeNameRegex = /(?:\.)+([A-Za-z])+(?:\()+/;
+  const attributeArgsRegex = /(?:\()+([A-Za-z])+:+(.+)+(?:\))+/;
+  for (const model of models) {
+    if (model.documentation) {
+      var _model$documentation, _model$documentation$, _attribute$match, _attribute$match$, _attribute$match2, _attribute$match2$;
+      const attribute = (_model$documentation = model.documentation) === null || _model$documentation === void 0 ? void 0 : (_model$documentation$ = _model$documentation.match(modelAttributeRegex)) === null || _model$documentation$ === void 0 ? void 0 : _model$documentation$[0];
+      const attributeName = attribute === null || attribute === void 0 ? void 0 : (_attribute$match = attribute.match(attributeNameRegex)) === null || _attribute$match === void 0 ? void 0 : (_attribute$match$ = _attribute$match[0]) === null || _attribute$match$ === void 0 ? void 0 : _attribute$match$.slice(1, -1);
+      if (attributeName !== 'model') continue;
+      const rawAttributeArgs = attribute === null || attribute === void 0 ? void 0 : (_attribute$match2 = attribute.match(attributeArgsRegex)) === null || _attribute$match2 === void 0 ? void 0 : (_attribute$match2$ = _attribute$match2[0]) === null || _attribute$match2$ === void 0 ? void 0 : _attribute$match2$.slice(1, -1);
+      const parsedAttributeArgs = {};
+      if (rawAttributeArgs) {
+        const rawAttributeArgsParts = rawAttributeArgs.split(':').map(it => it.trim()).map(part => part.startsWith('[') ? part : part.split(',')).flat().map(it => it.trim());
+        for (let i = 0; i < rawAttributeArgsParts.length; i += 2) {
+          const key = rawAttributeArgsParts[i];
+          const value = rawAttributeArgsParts[i + 1];
+          parsedAttributeArgs[key] = JSON.parse(value);
+        }
+      }
+      if (parsedAttributeArgs.hide) {
+        hiddenModels.push(model.name);
+      }
+    }
+  }
+}
+//# sourceMappingURL=helpers.js.map

package/plugins/trpc/helpers.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"helpers.js","names":["generatetRPCImport","sourceFile","addImportDeclaration","moduleSpecifier","namespaceImport","generateRouterImport","modelNamePlural","modelNameCamelCase","namedImports","generateProcedure","writer","opType","typeName","modelName","baseOpType","procType","getProcedureTypeByOpName","write","uncapitalizeFirstLetter","replace","generateRouterSchemaImports","name","addStatements","getInputTypeByOpName","opName","inputType","console","log","resolveModelsComments","models","hiddenModels","modelAttributeRegex","attributeNameRegex","attributeArgsRegex","model","documentation","attribute","match","attributeName","slice","rawAttributeArgs","parsedAttributeArgs","rawAttributeArgsParts","split","map","it","trim","part","startsWith","flat","i","length","key","value","JSON","parse","hide","push"],"sources":["../../../src/plugins/trpc/helpers.ts"],"sourcesContent":["import { DMMF } from '@prisma/generator-helper';\nimport { CodeBlockWriter, SourceFile } from 'ts-morph';\nimport { uncapitalizeFirstLetter } from './utils/uncapitalizeFirstLetter';\n\nexport const generatetRPCImport = (sourceFile: SourceFile) => {\n    sourceFile.addImportDeclaration({\n        moduleSpecifier: '@trpc/server',\n        namespaceImport: 'trpc',\n    });\n};\n\nexport const generateRouterImport = (sourceFile: SourceFile, modelNamePlural: string, modelNameCamelCase: string) => {\n    sourceFile.addImportDeclaration({\n        moduleSpecifier: `./${modelNameCamelCase}.router`,\n        namedImports: [`${modelNamePlural}Router`],\n    });\n};\n\nexport function generateProcedure(\n    writer: CodeBlockWriter,\n    opType: string,\n    typeName: string,\n    modelName: string,\n    baseOpType: string\n) {\n    const procType = getProcedureTypeByOpName(baseOpType);\n    writer.write(`\n        ${opType}: procedure.input(${typeName}).${procType}(({ctx, input}) => db(ctx).${uncapitalizeFirstLetter(\n        modelName\n    )}.${opType.replace('One', '')}(input)),\n    `);\n}\n\nexport function generateRouterSchemaImports(sourceFile: SourceFile, name: string) {\n    sourceFile.addStatements(`import { ${name}Schema } from '../schemas/${name}.schema';`);\n}\n\nexport const getInputTypeByOpName = (opName: string, modelName: string) => {\n    let inputType;\n    switch (opName) {\n        case 'findUnique':\n            inputType = `${modelName}Schema.findUnique`;\n            break;\n        case 'findFirst':\n            inputType = `${modelName}Schema.findFirst`;\n            break;\n        case 'findMany':\n            inputType = `${modelName}Schema.findMany`;\n            break;\n        case 'findRaw':\n            inputType = `${modelName}Schema.findRawObject`;\n            break;\n        case 'createOne':\n            inputType = `${modelName}Schema.create`;\n            break;\n        case 'createMany':\n            inputType = `${modelName}Schema.createMany`;\n            break;\n        case 'deleteOne':\n            inputType = `${modelName}Schema.delete`;\n            break;\n        case 'updateOne':\n            inputType = `${modelName}Schema.update`;\n            break;\n        case 'deleteMany':\n            inputType = `${modelName}Schema.deleteMany`;\n            break;\n        case 'updateMany':\n            inputType = `${modelName}Schema.updateMany`;\n            break;\n        case 'upsertOne':\n            inputType = `${modelName}Schema.upsert`;\n            break;\n        case 'aggregate':\n            inputType = `${modelName}Schema.aggregate`;\n            break;\n        case 'aggregateRaw':\n            inputType = `${modelName}Schema.aggregateRawObject`;\n            break;\n        case 'groupBy':\n            inputType = `${modelName}Schema.groupBy`;\n            break;\n        default:\n            console.log('getInputTypeByOpName: ', { opName, modelName });\n    }\n    return inputType;\n};\n\nexport const getProcedureTypeByOpName = (opName: string) => {\n    let procType;\n    switch (opName) {\n        case 'findUnique':\n        case 'findFirst':\n        case 'findMany':\n        case 'findRaw':\n        case 'aggregate':\n        case 'aggregateRaw':\n        case 'groupBy':\n            procType = 'query';\n            break;\n        case 'createOne':\n        case 'createMany':\n        case 'deleteOne':\n        case 'updateOne':\n        case 'deleteMany':\n        case 'updateMany':\n        case 'upsertOne':\n            procType = 'mutation';\n            break;\n        default:\n            console.log('getProcedureTypeByOpName: ', { opName });\n    }\n    return procType;\n};\n\nexport function resolveModelsComments(models: DMMF.Model[], hiddenModels: string[]) {\n    const modelAttributeRegex = /(@@Gen\\.)+([A-z])+(\\()+(.+)+(\\))+/;\n    const attributeNameRegex = /(?:\\.)+([A-Za-z])+(?:\\()+/;\n    const attributeArgsRegex = /(?:\\()+([A-Za-z])+:+(.+)+(?:\\))+/;\n\n    for (const model of models) {\n        if (model.documentation) {\n            const attribute = model.documentation?.match(modelAttributeRegex)?.[0];\n            const attributeName = attribute?.match(attributeNameRegex)?.[0]?.slice(1, -1);\n            if (attributeName !== 'model') continue;\n            const rawAttributeArgs = attribute?.match(attributeArgsRegex)?.[0]?.slice(1, -1);\n\n            const parsedAttributeArgs: Record<string, unknown> = {};\n            if (rawAttributeArgs) {\n                const rawAttributeArgsParts = rawAttributeArgs\n                    .split(':')\n                    .map((it) => it.trim())\n                    .map((part) => (part.startsWith('[') ? part : part.split(',')))\n                    .flat()\n                    .map((it) => it.trim());\n\n                for (let i = 0; i < rawAttributeArgsParts.length; i += 2) {\n                    const key = rawAttributeArgsParts[i];\n                    const value = rawAttributeArgsParts[i + 1];\n                    parsedAttributeArgs[key] = JSON.parse(value);\n                }\n            }\n            if (parsedAttributeArgs.hide) {\n                hiddenModels.push(model.name);\n            }\n        }\n    }\n}\n"],"mappings":";;;;;;;;;;AAEA;AAEO,MAAMA,kBAAkB,GAAIC,UAAsB,IAAK;EAC1DA,UAAU,CAACC,oBAAoB,CAAC;IAC5BC,eAAe,EAAE,cAAc;IAC/BC,eAAe,EAAE;EACrB,CAAC,CAAC;AACN,CAAC;AAAC;AAEK,MAAMC,oBAAoB,GAAG,CAACJ,UAAsB,EAAEK,eAAuB,EAAEC,kBAA0B,KAAK;EACjHN,UAAU,CAACC,oBAAoB,CAAC;IAC5BC,eAAe,EAAG,KAAII,kBAAmB,SAAQ;IACjDC,YAAY,EAAE,CAAE,GAAEF,eAAgB,QAAO;EAC7C,CAAC,CAAC;AACN,CAAC;AAAC;AAEK,SAASG,iBAAiB,CAC7BC,MAAuB,EACvBC,MAAc,EACdC,QAAgB,EAChBC,SAAiB,EACjBC,UAAkB,EACpB;EACE,MAAMC,QAAQ,GAAGC,wBAAwB,CAACF,UAAU,CAAC;EACrDJ,MAAM,CAACO,KAAK,CAAE;AAClB,UAAUN,MAAO,qBAAoBC,QAAS,KAAIG,QAAS,8BAA6B,IAAAG,gDAAuB,EACvGL,SAAS,CACX,IAAGF,MAAM,CAACQ,OAAO,CAAC,KAAK,EAAE,EAAE,CAAE;AACnC,KAAK,CAAC;AACN;AAEO,SAASC,2BAA2B,CAACnB,UAAsB,EAAEoB,IAAY,EAAE;EAC9EpB,UAAU,CAACqB,aAAa,CAAE,YAAWD,IAAK,6BAA4BA,IAAK,WAAU,CAAC;AAC1F;AAEO,MAAME,oBAAoB,GAAG,CAACC,MAAc,EAAEX,SAAiB,KAAK;EACvE,IAAIY,SAAS;EACb,QAAQD,MAAM;IACV,KAAK,YAAY;MACbC,SAAS,GAAI,GAAEZ,SAAU,mBAAkB;MAC3C;IACJ,KAAK,WAAW;MACZY,SAAS,GAAI,GAAEZ,SAAU,kBAAiB;MAC1C;IACJ,KAAK,UAAU;MACXY,SAAS,GAAI,GAAEZ,SAAU,iBAAgB;MACzC;IACJ,KAAK,SAAS;MACVY,SAAS,GAAI,GAAEZ,SAAU,sBAAqB;MAC9C;IACJ,KAAK,WAAW;MACZY,SAAS,GAAI,GAAEZ,SAAU,eAAc;MACvC;IACJ,KAAK,YAAY;MACbY,SAAS,GAAI,GAAEZ,SAAU,mBAAkB;MAC3C;IACJ,KAAK,WAAW;MACZY,SAAS,GAAI,GAAEZ,SAAU,eAAc;MACvC;IACJ,KAAK,WAAW;MACZY,SAAS,GAAI,GAAEZ,SAAU,eAAc;MACvC;IACJ,KAAK,YAAY;MACbY,SAAS,GAAI,GAAEZ,SAAU,mBAAkB;MAC3C;IACJ,KAAK,YAAY;MACbY,SAAS,GAAI,GAAEZ,SAAU,mBAAkB;MAC3C;IACJ,KAAK,WAAW;MACZY,SAAS,GAAI,GAAEZ,SAAU,eAAc;MACvC;IACJ,KAAK,WAAW;MACZY,SAAS,GAAI,GAAEZ,SAAU,kBAAiB;MAC1C;IACJ,KAAK,cAAc;MACfY,SAAS,GAAI,GAAEZ,SAAU,2BAA0B;MACnD;IACJ,KAAK,SAAS;MACVY,SAAS,GAAI,GAAEZ,SAAU,gBAAe;MACxC;IACJ;MACIa,OAAO,CAACC,GAAG,CAAC,wBAAwB,EAAE;QAAEH,MAAM;QAAEX;MAAU,CAAC,CAAC;EAAC;EAErE,OAAOY,SAAS;AACpB,CAAC;AAAC;AAEK,MAAMT,wBAAwB,GAAIQ,MAAc,IAAK;EACxD,IAAIT,QAAQ;EACZ,QAAQS,MAAM;IACV,KAAK,YAAY;IACjB,KAAK,WAAW;IAChB,KAAK,UAAU;IACf,KAAK,SAAS;IACd,KAAK,WAAW;IAChB,KAAK,cAAc;IACnB,KAAK,SAAS;MACVT,QAAQ,GAAG,OAAO;MAClB;IACJ,KAAK,WAAW;IAChB,KAAK,YAAY;IACjB,KAAK,WAAW;IAChB,KAAK,WAAW;IAChB,KAAK,YAAY;IACjB,KAAK,YAAY;IACjB,KAAK,WAAW;MACZA,QAAQ,GAAG,UAAU;MACrB;IACJ;MACIW,OAAO,CAACC,GAAG,CAAC,4BAA4B,EAAE;QAAEH;MAAO,CAAC,CAAC;EAAC;EAE9D,OAAOT,QAAQ;AACnB,CAAC;AAAC;AAEK,SAASa,qBAAqB,CAACC,MAAoB,EAAEC,YAAsB,EAAE;EAChF,MAAMC,mBAAmB,GAAG,mCAAmC;EAC/D,MAAMC,kBAAkB,GAAG,2BAA2B;EACtD,MAAMC,kBAAkB,GAAG,kCAAkC;EAE7D,KAAK,MAAMC,KAAK,IAAIL,MAAM,EAAE;IACxB,IAAIK,KAAK,CAACC,aAAa,EAAE;MAAA;MACrB,MAAMC,SAAS,2BAAGF,KAAK,CAACC,aAAa,kFAAnB,qBAAqBE,KAAK,CAACN,mBAAmB,CAAC,0DAA/C,sBAAkD,CAAC,CAAC;MACtE,MAAMO,aAAa,GAAGF,SAAS,aAATA,SAAS,2CAATA,SAAS,CAAEC,KAAK,CAACL,kBAAkB,CAAC,0EAApC,iBAAuC,CAAC,CAAC,sDAAzC,kBAA2CO,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;MAC7E,IAAID,aAAa,KAAK,OAAO,EAAE;MAC/B,MAAME,gBAAgB,GAAGJ,SAAS,aAATA,SAAS,4CAATA,SAAS,CAAEC,KAAK,CAACJ,kBAAkB,CAAC,4EAApC,kBAAuC,CAAC,CAAC,uDAAzC,mBAA2CM,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;MAEhF,MAAME,mBAA4C,GAAG,CAAC,CAAC;MACvD,IAAID,gBAAgB,EAAE;QAClB,MAAME,qBAAqB,GAAGF,gBAAgB,CACzCG,KAAK,CAAC,GAAG,CAAC,CACVC,GAAG,CAAEC,EAAE,IAAKA,EAAE,CAACC,IAAI,EAAE,CAAC,CACtBF,GAAG,CAAEG,IAAI,IAAMA,IAAI,CAACC,UAAU,CAAC,GAAG,CAAC,GAAGD,IAAI,GAAGA,IAAI,CAACJ,KAAK,CAAC,GAAG,CAAE,CAAC,CAC9DM,IAAI,EAAE,CACNL,GAAG,CAAEC,EAAE,IAAKA,EAAE,CAACC,IAAI,EAAE,CAAC;QAE3B,KAAK,IAAII,CAAC,GAAG,CAAC,EAAEA,CAAC,GAAGR,qBAAqB,CAACS,MAAM,EAAED,CAAC,IAAI,CAAC,EAAE;UACtD,MAAME,GAAG,GAAGV,qBAAqB,CAACQ,CAAC,CAAC;UACpC,MAAMG,KAAK,GAAGX,qBAAqB,CAACQ,CAAC,GAAG,CAAC,CAAC;UAC1CT,mBAAmB,CAACW,GAAG,CAAC,GAAGE,IAAI,CAACC,KAAK,CAACF,KAAK,CAAC;QAChD;MACJ;MACA,IAAIZ,mBAAmB,CAACe,IAAI,EAAE;QAC1B1B,YAAY,CAAC2B,IAAI,CAACvB,KAAK,CAACb,IAAI,CAAC;MACjC;IACJ;EACJ;AACJ"}

package/plugins/trpc/index.d.ts

@@ -0,0 +1,5 @@
+import { DMMF } from '@prisma/generator-helper';
+import { PluginOptions } from '@zenstackhq/sdk';
+import { Model } from '@zenstackhq/sdk/ast';
+export declare const name = "tRPC";
+export default function run(model: Model, options: PluginOptions, dmmf: DMMF.Document): Promise<void>;