fix: patch local copy of open api document

nestjs · Aug 10, 2023 · 8af173c · 8af173c
1 parent 3ddc224
commit 8af173c
Showing 1 changed file with 36 additions and 43 deletions.
diff --git a/lib/swagger-module.ts b/lib/swagger-module.ts
@@ -113,11 +113,13 @@ export class SwaggerModule {
         }
 
         if (options.swaggerOptions.patchDocumentOnRequest) {
-          document = options.swaggerOptions.patchDocumentOnRequest(
-            req,
-            res,
-            document
+          const documentToSerialize =
+            options.swaggerOptions.patchDocumentOnRequest(req, res, document);
+          const swaggerInitJsPerRequest = buildSwaggerInitJS(
+            documentToSerialize,
+            options.swaggerOptions
           );
+          return res.send(swaggerInitJsPerRequest);
         }
 
         if (!swaggerInitJS) {
@@ -145,11 +147,13 @@ export class SwaggerModule {
           }
 
           if (options.swaggerOptions.patchDocumentOnRequest) {
-            document = options.swaggerOptions.patchDocumentOnRequest(
-              req,
-              res,
-              document
+            const documentToSerialize =
+              options.swaggerOptions.patchDocumentOnRequest(req, res, document);
+            const swaggerInitJsPerRequest = buildSwaggerInitJS(
+              documentToSerialize,
+              options.swaggerOptions
             );
+            return res.send(swaggerInitJsPerRequest);
           }
 
           if (!swaggerInitJS) {
@@ -177,11 +181,14 @@ export class SwaggerModule {
       }
 
       if (options.swaggerOptions.patchDocumentOnRequest) {
-        document = options.swaggerOptions.patchDocumentOnRequest(
-          req,
-          res,
-          document
+        const documentToSerialize =
+          options.swaggerOptions.patchDocumentOnRequest(req, res, document);
+        const htmlPerRequest = buildSwaggerHTML(
+          baseUrlForSwaggerUI,
+          documentToSerialize,
+          options.swaggerOptions
         );
+        return res.send(htmlPerRequest);
       }
 
       if (!html) {
@@ -205,11 +212,14 @@ export class SwaggerModule {
         }
 
         if (options.swaggerOptions.patchDocumentOnRequest) {
-          document = options.swaggerOptions.patchDocumentOnRequest(
-            req,
-            res,
-            document
+          const documentToSerialize =
+            options.swaggerOptions.patchDocumentOnRequest(req, res, document);
+          const htmlPerRequest = buildSwaggerHTML(
+            baseUrlForSwaggerUI,
+            documentToSerialize,
+            options.swaggerOptions
           );
+          return res.send(htmlPerRequest);
         }
 
         if (!html) {
@@ -219,7 +229,6 @@ export class SwaggerModule {
             options.swaggerOptions
           );
         }
-
         res.send(html);
       });
     } catch (err) {
@@ -231,29 +240,18 @@ export class SwaggerModule {
        */
     }
 
-    let yamlDocument: string;
-    let jsonDocument: string;
-
     httpAdapter.get(normalizeRelPath(options.jsonDocumentUrl), (req, res) => {
       res.type('application/json');
 
       if (!document) {
         document = lazyBuildDocument();
       }
 
-      if (options.swaggerOptions.patchDocumentOnRequest) {
-        document = options.swaggerOptions.patchDocumentOnRequest(
-          req,
-          res,
-          document
-        );
-      }
-
-      if (!jsonDocument) {
-        jsonDocument = JSON.stringify(document);
-      }
+      const documentToSerialize = options.swaggerOptions.patchDocumentOnRequest
+        ? options.swaggerOptions.patchDocumentOnRequest(req, res, document)
+        : document;
 
-      res.send(jsonDocument);
+      res.send(JSON.stringify(documentToSerialize));
     });
 
     httpAdapter.get(normalizeRelPath(options.yamlDocumentUrl), (req, res) => {
@@ -263,18 +261,13 @@ export class SwaggerModule {
         document = lazyBuildDocument();
       }
 
-      if (options.swaggerOptions.patchDocumentOnRequest) {
-        document = options.swaggerOptions.patchDocumentOnRequest(
-          req,
-          res,
-          document
-        );
-      }
-
-      if (!yamlDocument) {
-        yamlDocument = jsyaml.dump(document, { skipInvalid: true });
-      }
+      const documentToSerialize = options.swaggerOptions.patchDocumentOnRequest
+        ? options.swaggerOptions.patchDocumentOnRequest(req, res, document)
+        : document;
 
+      const yamlDocument = jsyaml.dump(documentToSerialize, {
+        skipInvalid: true
+      });
       res.send(yamlDocument);
     });
   }