@dudousxd/nestjs-codegen 0.12.0 → 0.13.1

package/dist/nest/index.cjs

@@ -1589,6 +1589,55 @@ function extractParamsType(method, sourceFile, project) {
   }
   return entries.length > 0 ? `{ ${entries.join("; ")} }` : null;
 }
+function extractUploadedFiles(method) {
+  const FILE = "File | Blob";
+  const entries = [];
+  let multipart = false;
+  const hasUploadedFileParam = method.getParameters().some(
+    (p) => p.getDecorators().some((d) => {
+      const name = d.getName();
+      return name === "UploadedFile" || name === "UploadedFiles";
+    })
+  );
+  for (const decorator of method.getDecorators()) {
+    if (decorator.getName() !== "UseInterceptors") continue;
+    for (const arg of decorator.getArguments()) {
+      if (!import_ts_morph5.Node.isCallExpression(arg)) continue;
+      const interceptor = arg.getExpression().getText();
+      const callArgs = arg.getArguments();
+      const firstArg2 = callArgs[0];
+      if (interceptor === "FileInterceptor") {
+        if (firstArg2 && import_ts_morph5.Node.isStringLiteral(firstArg2)) {
+          entries.push(`${firstArg2.getLiteralValue()}: ${FILE}`);
+          multipart = true;
+        }
+      } else if (interceptor === "FilesInterceptor") {
+        if (firstArg2 && import_ts_morph5.Node.isStringLiteral(firstArg2)) {
+          entries.push(`${firstArg2.getLiteralValue()}: Array<${FILE}>`);
+          multipart = true;
+        }
+      } else if (interceptor === "FileFieldsInterceptor") {
+        if (firstArg2 && import_ts_morph5.Node.isArrayLiteralExpression(firstArg2)) {
+          for (const el of firstArg2.getElements()) {
+            if (!import_ts_morph5.Node.isObjectLiteralExpression(el)) continue;
+            const nameProp = el.getProperty("name");
+            if (nameProp && import_ts_morph5.Node.isPropertyAssignment(nameProp)) {
+              const init = nameProp.getInitializer();
+              if (init && import_ts_morph5.Node.isStringLiteral(init)) {
+                entries.push(`${init.getLiteralValue()}: Array<${FILE}>`);
+              }
+            }
+          }
+          multipart = true;
+        }
+      } else if (interceptor === "AnyFilesInterceptor") {
+        multipart = true;
+      }
+    }
+  }
+  if (hasUploadedFileParam) multipart = true;
+  return { fields: entries.length > 0 ? entries.join("; ") : null, multipart };
+}
 function extractResponseType(method, sourceFile, project) {
   const apiResponseDecorator = method.getDecorators().find((d) => d.getName() === "ApiResponse" && (apiResponseStatus(d) ?? 0) < 400);
   if (apiResponseDecorator) {
@@ -1723,6 +1772,8 @@ function extractDtoContract(method, sourceFile, project) {
   let body = extractBodyType(method, sourceFile, project);
   const filterInfo = extractApplyFilterInfo(method, sourceFile, project);
   const query = extractQueryType(method, sourceFile, project);
+  const uploads = extractUploadedFiles(method);
+  const multipartBody = uploads.fields ? `{ ${uploads.fields} }` : null;
   const streamElement = detectStreamElement(method);
   const isStream = streamElement !== null;
   if (filterInfo && filterInfo.source === "body") {
@@ -1732,7 +1783,7 @@ function extractDtoContract(method, sourceFile, project) {
   const paramsType = extractParamsType(method, sourceFile, project);
   const response = isStream ? resolveTypeNodeToString(streamElement, sourceFile, project, 3) : extractResponseType(method, sourceFile, project);
   const errorInfo = extractErrorType(method, sourceFile, project);
-  if (body === null && query === null && paramsType === null && response === "unknown" && errorInfo === null && filterInfo === null && !isStream) {
+  if (body === null && query === null && paramsType === null && response === "unknown" && errorInfo === null && filterInfo === null && !isStream && !uploads.multipart) {
     return null;
   }
   let bodyRef = null;
@@ -1805,7 +1856,9 @@ function extractDtoContract(method, sourceFile, project) {
     formWarnings,
     bodySchema,
     querySchema,
-    stream: isStream
+    stream: isStream,
+    multipart: uploads.multipart,
+    multipartBody
   };
 }
 function resolveParamClass(method, decoratorName, sourceFile, project) {
@@ -2272,7 +2325,9 @@ function extractDtoRoute(args) {
       formWarnings: dtoContract?.formWarnings ?? [],
       bodySchema: dtoContract?.bodySchema ?? null,
       querySchema: dtoContract?.querySchema ?? null,
-      stream: dtoContract?.stream ?? false
+      stream: dtoContract?.stream ?? false,
+      multipart: dtoContract?.multipart ?? false,
+      multipartBody: dtoContract?.multipartBody ?? null
     }
   });
 }
@@ -2883,7 +2938,15 @@ function emitRouterTypeBlock(tree, indent, outDir, serialization) {
       const isFilterQuery = c.contractSource.filterSource === "query" && !!c.contractSource.filterFields?.length;
       const query = queryRef ? queryRef.isArray ? `Array<${queryRef.name}>` : queryRef.name : isFilterQuery ? emitFilterQueryType(c) : c.contractSource.query ?? "never";
       const bodyRef = c.contractSource.bodyRef;
-      const body = method === "GET" ? "never" : bodyRef ? bodyRef.isArray ? `Array<${bodyRef.name}>` : bodyRef.name : c.contractSource.body ?? "never";
+      let body = method === "GET" ? "never" : bodyRef ? bodyRef.isArray ? `Array<${bodyRef.name}>` : bodyRef.name : c.contractSource.body ?? "never";
+      const multipartBody = c.contractSource.multipartBody;
+      if (c.contractSource.multipart && multipartBody) {
+        if (body === "never") {
+          body = multipartBody;
+        } else if (!bodyAcceptsAnything(body)) {
+          body = `(${body}) & ${multipartBody}`;
+        }
+      }
       const response = buildResponseType(c, outDir, serialization);
       const error = buildErrorType(c);
       const params = buildParamsType(c.params);
@@ -2902,6 +2965,25 @@ function emitRouterTypeBlock(tree, indent, outDir, serialization) {
   }
   return lines;
 }
+function topLevelUnionArms(type) {
+  const arms = [];
+  let depth = 0;
+  let start = 0;
+  for (let i = 0; i < type.length; i++) {
+    const ch = type[i];
+    if (ch === "{" || ch === "[" || ch === "<" || ch === "(") depth++;
+    else if (ch === "}" || ch === "]" || ch === ">" || ch === ")") depth--;
+    else if (ch === "|" && depth === 0) {
+      arms.push(type.slice(start, i).trim());
+      start = i + 1;
+    }
+  }
+  arms.push(type.slice(start).trim());
+  return arms;
+}
+function bodyAcceptsAnything(body) {
+  return topLevelUnionArms(body).some((arm) => arm === "unknown" || arm === "any");
+}
 function buildRequestModel(c) {
   const m = c.method.toLowerCase();
   const flat = JSON.stringify(c.name);
@@ -2918,6 +3000,7 @@ function buildRequestModel(c) {
   const optsParts = [];
   if (hasQuery) optsParts.push("query: input?.query as Record<string, unknown> | undefined");
   if (hasBody) optsParts.push("body: input?.body");
+  if (hasBody && c.contractSource.multipart) optsParts.push("multipart: true");
   const optsExpr = optsParts.length ? `{ ${optsParts.join(", ")} }` : "{}";
   return {
     routeName: c.name,
@@ -4264,7 +4347,7 @@ async function acquireLock(outDir) {
 }
 
 // src/index.ts
-var VERSION = "0.12.0";
+var VERSION = "0.13.1";
 
 // src/generate-manifest.ts
 var MANIFEST_FILE = ".codegen-manifest.json";