@dudousxd/nestjs-codegen 0.15.0 → 0.16.0

package/dist/nest/index.cjs

@@ -2272,7 +2272,7 @@ var import_chokidar = __toESM(require("chokidar"), 1);
 // src/discovery/contracts-fast.ts
 var import_node_path14 = require("path");
 var import_fast_glob2 = __toESM(require("fast-glob"), 1);
-var import_ts_morph9 = require("ts-morph");
+var import_ts_morph10 = require("ts-morph");
 
 // src/discovery/dto-type-resolver.ts
 var import_ts_morph7 = require("ts-morph");
@@ -3251,6 +3251,12 @@ function toFilterFieldType(name, r) {
 }
 
 // src/discovery/filter-for.ts
+function resolveMixinEntityClass(mixin, project) {
+  const entityArg = mixin?.classArgs[0];
+  if (!entityArg) return void 0;
+  const file = project.getSourceFile(entityArg.filePath) ?? project.addSourceFileAtPathIfExists(entityArg.filePath);
+  return file?.getClass(entityArg.name);
+}
 function classifyFilterForHint(typeInit) {
   if (import_ts_morph6.Node.isStringLiteral(typeInit)) {
     switch (typeInit.getLiteralValue()) {
@@ -3324,7 +3330,9 @@ function extractFilterForHints(classDecl, project) {
   }
   return hints;
 }
-function extractApplyFilterInfo(method, sourceFile, project) {
+function extractApplyFilterInfo(method, sourceFile, project, mixin) {
+  const declFile = method.getSourceFile();
+  const mixinEntity = resolveMixinEntityClass(mixin, project);
   for (const param of method.getParameters()) {
     const filterDecorator = param.getDecorators().find((d) => d.getName() === "ApplyFilter");
     if (!filterDecorator) continue;
@@ -3344,12 +3352,12 @@ function extractApplyFilterInfo(method, sourceFile, project) {
       }
     }
     const filterClassName = filterClassArg.getText();
-    const resolved = findType(filterClassName, sourceFile, project);
-    if (resolved && resolved.kind === "class") {
-      const classDecl = resolved.decl;
+    const resolved = findType(filterClassName, declFile, project);
+    const classDecl = resolved?.kind === "class" ? resolved.decl : resolveLocalClassDeclaration(filterClassArg);
+    if (classDecl) {
       let fieldTypes = extractClassPropertyTypes(classDecl, project);
       if (fieldTypes.length === 0) {
-        fieldTypes = extractFilterableEntityFields(classDecl, project);
+        fieldTypes = extractFilterableEntityFields(classDecl, project, mixinEntity);
       }
       const filterForHints = extractFilterForHints(classDecl, project);
       if (filterForHints.size > 0) {
@@ -3434,22 +3442,29 @@ function extractClassPropertyTypes(classDecl, project) {
   }
   return fields;
 }
-function extractFilterableEntityFields(filterClass, project) {
+function resolveLocalClassDeclaration(identifier) {
+  if (!import_ts_morph6.Node.isIdentifier(identifier)) return void 0;
+  return identifier.getDefinitions().map((d) => d.getDeclarationNode()).find((n) => n !== void 0 && import_ts_morph6.Node.isClassDeclaration(n));
+}
+function resolveDeclaredEntity(optionsArg, filterClass, project) {
+  if (!import_ts_morph6.Node.isObjectLiteralExpression(optionsArg)) return void 0;
+  const entityProp = optionsArg.getProperty("entity");
+  if (!entityProp || !import_ts_morph6.Node.isPropertyAssignment(entityProp)) return void 0;
+  const entityInit = entityProp.getInitializer();
+  if (!entityInit || !import_ts_morph6.Node.isIdentifier(entityInit)) return void 0;
+  const resolved = findType(entityInit.getText(), filterClass.getSourceFile(), project);
+  if (!resolved || resolved.kind !== "class") return void 0;
+  return resolved.decl;
+}
+function extractFilterableEntityFields(filterClass, project, entityOverride) {
   const filterableDecorator = filterClass.getDecorators().find((d) => d.getName() === "Filterable");
   if (!filterableDecorator) return [];
   const args = filterableDecorator.getArguments();
   if (args.length === 0) return [];
   const optionsArg = args[0];
   if (!import_ts_morph6.Node.isObjectLiteralExpression(optionsArg)) return [];
-  const entityProp = optionsArg.getProperty("entity");
-  if (!entityProp || !import_ts_morph6.Node.isPropertyAssignment(entityProp)) return [];
-  const entityInit = entityProp.getInitializer();
-  if (!entityInit || !import_ts_morph6.Node.isIdentifier(entityInit)) return [];
-  const entityName = entityInit.getText();
-  const filterSourceFile = filterClass.getSourceFile();
-  const resolvedEntity = findType(entityName, filterSourceFile, project);
-  if (!resolvedEntity || resolvedEntity.kind !== "class") return [];
-  const entityDecl = resolvedEntity.decl;
+  const entityDecl = entityOverride ?? resolveDeclaredEntity(optionsArg, filterClass, project);
+  if (!entityDecl) return [];
   const fields = collectEntityFields(
     entityDecl,
     entityDecl.getSourceFile(),
@@ -3874,9 +3889,9 @@ function unwrapNamedContainer(node, names) {
   }
   return node;
 }
-function extractDtoContract(method, sourceFile, project) {
+function extractDtoContract(method, sourceFile, project, mixin) {
   let body = extractBodyType(method, sourceFile, project);
-  const filterInfo = extractApplyFilterInfo(method, sourceFile, project);
+  const filterInfo = extractApplyFilterInfo(method, sourceFile, project, mixin);
   const query = extractQueryType(method, sourceFile, project);
   const uploads = extractUploadedFiles(method);
   const multipartBody = uploads.fields ? `{ ${uploads.fields} }` : null;
@@ -3986,12 +4001,94 @@ function resolveParamClass(method, decoratorName, sourceFile, project) {
   return null;
 }
 
-// src/discovery/zod-ast-to-ts.ts
+// src/discovery/heritage.ts
 var import_ts_morph8 = require("ts-morph");
+function unwrapExpression(node) {
+  let current = node;
+  while (import_ts_morph8.Node.isAsExpression(current) || import_ts_morph8.Node.isSatisfiesExpression(current) || import_ts_morph8.Node.isParenthesizedExpression(current) || import_ts_morph8.Node.isTypeAssertion(current)) {
+    current = current.getExpression();
+  }
+  return current;
+}
+function resolveReturnedClass(factoryDecl) {
+  const body = factoryDecl.getBody();
+  if (!body) return void 0;
+  const returns = body.getDescendants().filter((n) => import_ts_morph8.Node.isReturnStatement(n));
+  for (const ret of returns) {
+    const expr = ret.getExpression();
+    if (!expr) continue;
+    const inner = unwrapExpression(expr);
+    if (import_ts_morph8.Node.isClassExpression(inner)) {
+      return inner;
+    }
+    if (import_ts_morph8.Node.isIdentifier(inner)) {
+      const name = inner.getText();
+      const decl = body.getDescendants().find((n) => import_ts_morph8.Node.isClassDeclaration(n) && n.getName() === name);
+      if (decl) return decl;
+    }
+  }
+  return void 0;
+}
+function resolveInheritedMethods(cls) {
+  const expr = cls.getExtends()?.getExpression();
+  if (!expr || !import_ts_morph8.Node.isCallExpression(expr)) return void 0;
+  const callee = expr.getExpression();
+  if (!import_ts_morph8.Node.isIdentifier(callee)) return void 0;
+  const factoryDecl = callee.getDefinitions().map((d) => d.getDeclarationNode()).find((n) => import_ts_morph8.Node.isFunctionDeclaration(n));
+  if (!factoryDecl) return void 0;
+  const returnedClass = resolveReturnedClass(factoryDecl);
+  if (!returnedClass) return void 0;
+  const classArgs = [];
+  for (const arg of expr.getArguments()) {
+    if (!import_ts_morph8.Node.isIdentifier(arg)) continue;
+    const decl = arg.getDefinitions().map((d) => d.getDeclarationNode()).find((n) => import_ts_morph8.Node.isClassDeclaration(n));
+    if (decl) {
+      classArgs.push({
+        name: decl.getName() ?? arg.getText(),
+        filePath: decl.getSourceFile().getFilePath()
+      });
+    }
+  }
+  return {
+    methods: returnedClass.getMethods(),
+    factoryName: callee.getText(),
+    factoryFilePath: factoryDecl.getSourceFile().getFilePath(),
+    classArgs
+  };
+}
+var mixinTypeProject;
+function getMixinTypeProject() {
+  mixinTypeProject ??= new import_ts_morph8.Project({
+    skipAddingFilesFromTsConfig: true,
+    compilerOptions: { strict: true }
+  });
+  return mixinTypeProject;
+}
+function clearMixinTypeProject() {
+  mixinTypeProject = void 0;
+}
+function resolveInstantiatedReturnType(cls, methodName) {
+  const filePath = cls.getSourceFile().getFilePath();
+  const className = cls.getName();
+  if (!className) return void 0;
+  const project = getMixinTypeProject();
+  const sourceFile = project.getSourceFile(filePath) ?? project.addSourceFileAtPathIfExists(filePath);
+  const typedCls = sourceFile?.getClass(className);
+  if (!typedCls) return void 0;
+  const prop = typedCls.getType().getProperty(methodName);
+  if (!prop) return void 0;
+  const returnType = prop.getTypeAtLocation(typedCls).getCallSignatures()[0]?.getReturnType();
+  if (!returnType) return void 0;
+  const unwrapped = returnType.getSymbol()?.getName() === "Promise" ? returnType.getTypeArguments()[0] ?? returnType : returnType;
+  return unwrapped.getText(typedCls);
+}
+
+// src/discovery/zod-ast-to-ts.ts
+var import_ts_morph9 = require("ts-morph");
 function zodAstToTs(node) {
-  if (!import_ts_morph8.Node.isCallExpression(node)) return "unknown";
+  if (!import_ts_morph9.Node.isCallExpression(node)) return "unknown";
   const expr = node.getExpression();
-  if (import_ts_morph8.Node.isPropertyAccessExpression(expr)) {
+  if (import_ts_morph9.Node.isPropertyAccessExpression(expr)) {
     const methodName = expr.getName();
     const receiver = expr.getExpression();
     if (methodName === "optional") {
@@ -4015,17 +4112,17 @@ function zodAstToTs(node) {
       case "literal": {
         const lit = args[0];
         if (!lit) return "unknown";
-        if (import_ts_morph8.Node.isStringLiteral(lit)) return JSON.stringify(lit.getLiteralValue());
-        if (import_ts_morph8.Node.isNumericLiteral(lit)) return lit.getLiteralValue().toString();
-        if (lit.getKind() === import_ts_morph8.SyntaxKind.TrueKeyword) return "true";
-        if (lit.getKind() === import_ts_morph8.SyntaxKind.FalseKeyword) return "false";
+        if (import_ts_morph9.Node.isStringLiteral(lit)) return JSON.stringify(lit.getLiteralValue());
+        if (import_ts_morph9.Node.isNumericLiteral(lit)) return lit.getLiteralValue().toString();
+        if (lit.getKind() === import_ts_morph9.SyntaxKind.TrueKeyword) return "true";
+        if (lit.getKind() === import_ts_morph9.SyntaxKind.FalseKeyword) return "false";
         return "unknown";
       }
       case "enum": {
         const arrArg = args[0];
-        if (!arrArg || !import_ts_morph8.Node.isArrayLiteralExpression(arrArg)) return "unknown";
+        if (!arrArg || !import_ts_morph9.Node.isArrayLiteralExpression(arrArg)) return "unknown";
         const members = arrArg.getElements().map(
-          (el) => import_ts_morph8.Node.isStringLiteral(el) ? JSON.stringify(el.getLiteralValue()) : "unknown"
+          (el) => import_ts_morph9.Node.isStringLiteral(el) ? JSON.stringify(el.getLiteralValue()) : "unknown"
         );
         return members.join(" | ");
       }
@@ -4036,10 +4133,10 @@ function zodAstToTs(node) {
       }
       case "object": {
         const objArg = args[0];
-        if (!objArg || !import_ts_morph8.Node.isObjectLiteralExpression(objArg)) return "unknown";
+        if (!objArg || !import_ts_morph9.Node.isObjectLiteralExpression(objArg)) return "unknown";
         const lines = [];
         for (const prop of objArg.getProperties()) {
-          if (!import_ts_morph8.Node.isPropertyAssignment(prop)) continue;
+          if (!import_ts_morph9.Node.isPropertyAssignment(prop)) continue;
           const key = prop.getName();
           const valNode = prop.getInitializer();
           if (!valNode) continue;
@@ -4051,7 +4148,7 @@ function zodAstToTs(node) {
       }
       case "union": {
         const arrArg = args[0];
-        if (!arrArg || !import_ts_morph8.Node.isArrayLiteralExpression(arrArg)) return "unknown";
+        if (!arrArg || !import_ts_morph9.Node.isArrayLiteralExpression(arrArg)) return "unknown";
         return arrArg.getElements().map(zodAstToTs).join(" | ");
       }
       case "record": {
@@ -4061,7 +4158,7 @@ function zodAstToTs(node) {
       }
       case "tuple": {
         const arrArg = args[0];
-        if (!arrArg || !import_ts_morph8.Node.isArrayLiteralExpression(arrArg)) return "unknown";
+        if (!arrArg || !import_ts_morph9.Node.isArrayLiteralExpression(arrArg)) return "unknown";
         return `[${arrArg.getElements().map(zodAstToTs).join(", ")}]`;
       }
       default:
@@ -4071,18 +4168,18 @@ function zodAstToTs(node) {
   return "unknown";
 }
 function isOptionalChain(node) {
-  if (!import_ts_morph8.Node.isCallExpression(node)) return false;
+  if (!import_ts_morph9.Node.isCallExpression(node)) return false;
   const expr = node.getExpression();
-  return import_ts_morph8.Node.isPropertyAccessExpression(expr) && expr.getName() === "optional";
+  return import_ts_morph9.Node.isPropertyAccessExpression(expr) && expr.getName() === "optional";
 }
 function parseDefineContractCall(callExpr) {
-  if (!import_ts_morph8.Node.isCallExpression(callExpr)) return null;
+  if (!import_ts_morph9.Node.isCallExpression(callExpr)) return null;
   const callee = callExpr.getExpression();
-  const calleeName = import_ts_morph8.Node.isIdentifier(callee) ? callee.getText() : import_ts_morph8.Node.isPropertyAccessExpression(callee) ? callee.getName() : "";
+  const calleeName = import_ts_morph9.Node.isIdentifier(callee) ? callee.getText() : import_ts_morph9.Node.isPropertyAccessExpression(callee) ? callee.getName() : "";
   if (calleeName !== "defineContract") return null;
   const args = callExpr.getArguments();
   const optsArg = args[0];
-  if (!optsArg || !import_ts_morph8.Node.isObjectLiteralExpression(optsArg)) return null;
+  if (!optsArg || !import_ts_morph9.Node.isObjectLiteralExpression(optsArg)) return null;
   let query = null;
   let body = null;
   let response = "unknown";
@@ -4090,7 +4187,7 @@ function parseDefineContractCall(callExpr) {
   let bodyZodText = null;
   let queryZodText = null;
   for (const prop of optsArg.getProperties()) {
-    if (!import_ts_morph8.Node.isPropertyAssignment(prop)) continue;
+    if (!import_ts_morph9.Node.isPropertyAssignment(prop)) continue;
     const propName = prop.getName();
     const val = prop.getInitializer();
     if (!val) continue;
@@ -4115,14 +4212,14 @@ function resolveTsconfigPath(cwd, tsconfig) {
 }
 function createDiscoveryProject(tsconfigPath) {
   try {
-    return new import_ts_morph9.Project({
+    return new import_ts_morph10.Project({
       tsConfigFilePath: tsconfigPath,
       skipAddingFilesFromTsConfig: true,
       skipLoadingLibFiles: true,
       skipFileDependencyResolution: true
     });
   } catch {
-    return new import_ts_morph9.Project({
+    return new import_ts_morph10.Project({
       skipAddingFilesFromTsConfig: true,
       skipLoadingLibFiles: true,
       skipFileDependencyResolution: true,
@@ -4222,15 +4319,16 @@ var PersistentDiscovery = class _PersistentDiscovery {
   runExtraction() {
     clearTypeResolutionCaches(this.project);
     clearEnumCache(this.project);
+    clearMixinTypeProject();
     return extractRoutesFrom(this.project, this.controllerPaths);
   }
 };
 function decoratorStringArg(decoratorExpr) {
   if (!decoratorExpr) return void 0;
-  if (import_ts_morph9.Node.isStringLiteral(decoratorExpr)) return decoratorExpr.getLiteralValue();
-  if (import_ts_morph9.Node.isArrayLiteralExpression(decoratorExpr)) {
+  if (import_ts_morph10.Node.isStringLiteral(decoratorExpr)) return decoratorExpr.getLiteralValue();
+  if (import_ts_morph10.Node.isArrayLiteralExpression(decoratorExpr)) {
     const first = decoratorExpr.getElements()[0];
-    if (first && import_ts_morph9.Node.isStringLiteral(first)) return first.getLiteralValue();
+    if (first && import_ts_morph10.Node.isStringLiteral(first)) return first.getLiteralValue();
   }
   return void 0;
 }
@@ -4252,7 +4350,8 @@ function joinPaths(prefix, suffix) {
   if (!prefix && !suffix) return "/";
   if (!prefix) return suffix.startsWith("/") ? suffix : `/${suffix}`;
   if (!suffix) return prefix.startsWith("/") ? prefix : `/${prefix}`;
-  const p = prefix.endsWith("/") ? prefix.slice(0, -1) : prefix;
+  const withLeadingSlash = prefix.startsWith("/") ? prefix : `/${prefix}`;
+  const p = withLeadingSlash.endsWith("/") ? withLeadingSlash.slice(0, -1) : withLeadingSlash;
   const s = suffix.startsWith("/") ? suffix : `/${suffix}`;
   const combined = p + s;
   return combined === "" ? "/" : combined;
@@ -4306,7 +4405,8 @@ function buildRoute(args) {
     methodAs,
     sourceFile,
     seenNames,
-    contractSource
+    contractSource,
+    mixin
   } = args;
   const routeName = resolveRouteName(className, methodName, classAs, methodAs);
   const qualifiedRef = `${className}.${methodName}`;
@@ -4322,7 +4422,12 @@ function buildRoute(args) {
     path: combinedPath,
     name: routeName,
     params: extractParams(combinedPath),
-    controllerRef: { className, methodName, filePath: sourceFile.getFilePath() },
+    controllerRef: {
+      className,
+      methodName,
+      filePath: sourceFile.getFilePath(),
+      ...mixin ? { mixin } : {}
+    },
     contract: { contractSource }
   };
 }
@@ -4336,16 +4441,17 @@ function extractContractRoute(args) {
     className,
     sourceFile,
     project,
-    seenNames
+    seenNames,
+    mixin
   } = args;
   const firstDecoratorArg = applyContractDecorator.getArguments()[0];
   if (!firstDecoratorArg) return null;
   let contractDef = null;
   let bodyZodRef = null;
   let queryZodRef = null;
-  if (import_ts_morph9.Node.isCallExpression(firstDecoratorArg)) {
+  if (import_ts_morph10.Node.isCallExpression(firstDecoratorArg)) {
     contractDef = parseDefineContractCall(firstDecoratorArg);
-  } else if (import_ts_morph9.Node.isIdentifier(firstDecoratorArg)) {
+  } else if (import_ts_morph10.Node.isIdentifier(firstDecoratorArg)) {
     const identName = firstDecoratorArg.getText();
     const resolvedVar = resolveImportedVariable(identName, sourceFile, project);
     if (!resolvedVar) {
@@ -4388,6 +4494,7 @@ function extractContractRoute(args) {
     methodAs,
     sourceFile,
     seenNames,
+    mixin,
     contractSource: {
       query: contractDef.query,
       body: contractDef.body,
@@ -4404,13 +4511,14 @@ function extractContractRoute(args) {
   });
 }
 function extractDtoRoute(args) {
-  const { cls, method, verb, prefix, className, sourceFile, project, seenNames } = args;
+  const { cls, method, verb, prefix, className, sourceFile, project, seenNames, mixin } = args;
   if (!verb) return null;
   const combined = joinPaths(prefix, verb.handlerPath);
   const methodName = method.getName();
   const classAs = readAsDecorator(cls, `class ${className}`);
   const methodAs = readAsDecorator(method, `${className}.${methodName}`);
-  const dtoContract = extractDtoContract(method, sourceFile, project);
+  const dtoContract = extractDtoContract(method, sourceFile, project, mixin);
+  const mixinResponse = mixin ? resolveInstantiatedReturnType(cls, methodName) : void 0;
   return buildRoute({
     className,
     methodName,
@@ -4420,10 +4528,11 @@ function extractDtoRoute(args) {
     methodAs,
     sourceFile,
     seenNames,
+    mixin,
     contractSource: {
       query: dtoContract?.query ?? null,
       body: dtoContract?.body ?? null,
-      response: dtoContract?.response ?? "unknown",
+      response: mixinResponse ?? dtoContract?.response ?? "unknown",
       error: dtoContract?.error ?? null,
       queryRef: dtoContract?.queryRef ?? null,
       bodyRef: dtoContract?.bodyRef ?? null,
@@ -4452,7 +4561,17 @@ function extractFromSourceFile(sourceFile, project) {
     const firstArg2 = controllerDecorator.getArguments()[0];
     const prefix = decoratorStringArg(firstArg2) ?? "";
     const className = cls.getName() ?? "Unknown";
-    for (const method of cls.getMethods()) {
+    const inherited = resolveInheritedMethods(cls);
+    const mixinBinding = inherited ? {
+      factoryName: inherited.factoryName,
+      factoryFilePath: inherited.factoryFilePath,
+      classArgs: inherited.classArgs
+    } : void 0;
+    const methods = [
+      ...cls.getMethods().map((method) => ({ method })),
+      ...(inherited?.methods ?? []).map((method) => ({ method, mixin: mixinBinding }))
+    ];
+    for (const { method, mixin } of methods) {
       const verb = resolveVerb(method);
       const applyContractDecorator = method.getDecorator("ApplyContract");
       const route = applyContractDecorator ? extractContractRoute({
@@ -4464,7 +4583,8 @@ function extractFromSourceFile(sourceFile, project) {
         className,
         sourceFile,
         project,
-        seenNames
+        seenNames,
+        mixin
       }) : extractDtoRoute({
         cls,
         method,
@@ -4473,7 +4593,8 @@ function extractFromSourceFile(sourceFile, project) {
         className,
         sourceFile,
         project,
-        seenNames
+        seenNames,
+        mixin
       });
       if (route) routes.push(route);
     }
@@ -4674,7 +4795,7 @@ async function watch(config, onChange, options = {}) {
 }
 
 // src/index.ts
-var VERSION = "0.15.0";
+var VERSION = "0.16.0";
 
 // src/generate-manifest.ts
 var MANIFEST_FILE = ".codegen-manifest.json";