@dudousxd/nestjs-codegen 0.12.0 → 0.13.0

package/dist/cli/main.js

@@ -786,6 +786,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,
@@ -3681,6 +3682,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 (!Node6.isCallExpression(arg)) continue;
+      const interceptor = arg.getExpression().getText();
+      const callArgs = arg.getArguments();
+      const firstArg2 = callArgs[0];
+      if (interceptor === "FileInterceptor") {
+        if (firstArg2 && Node6.isStringLiteral(firstArg2)) {
+          entries.push(`${firstArg2.getLiteralValue()}: ${FILE}`);
+          multipart = true;
+        }
+      } else if (interceptor === "FilesInterceptor") {
+        if (firstArg2 && Node6.isStringLiteral(firstArg2)) {
+          entries.push(`${firstArg2.getLiteralValue()}: Array<${FILE}>`);
+          multipart = true;
+        }
+      } else if (interceptor === "FileFieldsInterceptor") {
+        if (firstArg2 && Node6.isArrayLiteralExpression(firstArg2)) {
+          for (const el of firstArg2.getElements()) {
+            if (!Node6.isObjectLiteralExpression(el)) continue;
+            const nameProp = el.getProperty("name");
+            if (nameProp && Node6.isPropertyAssignment(nameProp)) {
+              const init = nameProp.getInitializer();
+              if (init && Node6.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) {
@@ -3815,6 +3865,11 @@ 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);
+  if (uploads.fields) {
+    const fileObject = `{ ${uploads.fields} }`;
+    body = body ? `(${body}) & ${fileObject}` : fileObject;
+  }
   const streamElement = detectStreamElement(method);
   const isStream = streamElement !== null;
   if (filterInfo && filterInfo.source === "body") {
@@ -3824,7 +3879,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;
@@ -3897,7 +3952,8 @@ function extractDtoContract(method, sourceFile, project) {
     formWarnings,
     bodySchema,
     querySchema,
-    stream: isStream
+    stream: isStream,
+    multipart: uploads.multipart
   };
 }
 function resolveParamClass(method, decoratorName, sourceFile, project) {
@@ -4382,7 +4438,8 @@ 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
     }
   });
 }
@@ -4612,7 +4669,7 @@ async function watch(config, onChange, options = {}) {
 }
 
 // src/index.ts
-var VERSION = "0.12.0";
+var VERSION = "0.13.0";
 
 // src/cli/codegen.ts
 async function runCodegen(opts = {}) {