@dudousxd/nestjs-codegen 0.7.1 → 0.8.0

package/dist/nest/index.cjs

@@ -1535,14 +1535,33 @@ function extractQueryType(method, sourceFile, project) {
   for (const param of method.getParameters()) {
     const queryDecorator = param.getDecorators().find((d) => d.getName() === "Query");
     if (!queryDecorator) continue;
-    const queryArgs = queryDecorator.getArguments();
-    if (queryArgs.length > 0) continue;
+    if (queryDecorator.getArguments().length > 0) continue;
     const typeNode = param.getTypeNode();
     if (typeNode) {
       return resolveTypeNodeToString(typeNode, sourceFile, project, 3);
     }
   }
-  return null;
+  const entries = [];
+  for (const param of method.getParameters()) {
+    const queryDecorator = param.getDecorators().find((d) => d.getName() === "Query");
+    if (!queryDecorator) continue;
+    const queryArgs = queryDecorator.getArguments();
+    const nameArg = queryArgs[0];
+    if (!nameArg || !import_ts_morph5.Node.isStringLiteral(nameArg)) continue;
+    const queryName = nameArg.getLiteralValue();
+    const typeNode = param.getTypeNode();
+    const queryType = typeNode ? resolveTypeNodeToString(typeNode, sourceFile, project, 3) : "string";
+    entries.push(`${queryName}${isParamOptional(param) ? "?" : ""}: ${queryType}`);
+  }
+  return entries.length > 0 ? `{ ${entries.join("; ")} }` : null;
+}
+function isParamOptional(param) {
+  if (param.hasQuestionToken() || param.hasInitializer()) return true;
+  const typeNode = param.getTypeNode();
+  if (typeNode && import_ts_morph5.Node.isUnionTypeNode(typeNode)) {
+    return typeNode.getTypeNodes().some((t) => t.getKind() === import_ts_morph5.SyntaxKind.UndefinedKeyword);
+  }
+  return false;
 }
 function extractParamsType(method, sourceFile, project) {
   const entries = [];