From ec32130c14789c901d3e2d3e0d2a73a1c65e1be4 Mon Sep 17 00:00:00 2001
From: Stephen Scott <stephen.scott@digital.homeoffice.gov.uk>
Date: Fri, 13 May 2022 10:24:47 +0100
Subject: [PATCH] fix dynamic loading for serveFiles and added customJsStr

---
 README.md           |  4 +++-
 index.js            |  9 ++++++++-
 test/testapp/app.js | 20 ++++++++++++++++++++
 3 files changed, 31 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 3101364..85cb48f 100644
--- a/README.md
+++ b/README.md
@@ -223,11 +223,13 @@ const app = express();
 const swaggerUi = require('swagger-ui-express');
 const swaggerDocument = require('./swagger.json');
 
+var options = {}
+
 app.use('/api-docs', function(req, res, next){
     swaggerDocument.host = req.get('host');
     req.swaggerDoc = swaggerDocument;
     next();
-}, swaggerUi.serve, swaggerUi.setup());
+}, swaggerUi.serveFiles(swaggerDocument, options), swaggerUi.setup());
 ```
 
 ### Two swagger documents
diff --git a/index.js b/index.js
index 9b74230..a8cd5ac 100644
--- a/index.js
+++ b/index.js
@@ -82,6 +82,7 @@ var htmlTplString = `
 <script src="./swagger-ui-standalone-preset.js"> </script>
 <script src="./swagger-ui-init.js"> </script>
 <% customJs %>
+<% customJsStr %>
 <% customCssUrl %>
 <style>
   <% customCss %>
@@ -140,12 +141,14 @@ window.onload = function() {
 var generateHTML = function (swaggerDoc, opts, options, customCss, customfavIcon, swaggerUrl, customSiteTitle, _htmlTplString, _jsTplString) {
   var isExplorer
   var customJs
+  var customJsStr
   var swaggerUrls
   var customCssUrl
   if (opts && typeof opts === 'object') {
     options = opts.swaggerOptions
     customCss = opts.customCss
     customJs = opts.customJs
+    customJsStr = opts.customJsStr
     customfavIcon = opts.customfavIcon
     swaggerUrl = opts.swaggerUrl
     swaggerUrls = opts.swaggerUrls
@@ -167,7 +170,8 @@ var generateHTML = function (swaggerDoc, opts, options, customCss, customfavIcon
   var favIconString = customfavIcon ? '<link rel="icon" href="' + customfavIcon + '" />' : favIconHtml
   var htmlWithCustomCss = _htmlTplString.toString().replace('<% customCss %>', customCss)
   var htmlWithFavIcon = htmlWithCustomCss.replace('<% favIconString %>', favIconString)
-  var htmlWithCustomJs = htmlWithFavIcon.replace('<% customJs %>', customJs ? `<script src="${customJs}"></script>` : '')
+  var htmlWithCustomJsUrl = htmlWithFavIcon.replace('<% customJs %>', customJs ? `<script src="${customJs}"></script>` : '')
+  var htmlWithCustomJs = htmlWithCustomJsUrl.replace('<% customJsStr %>', customJsStr ? `<script>${customJsStr}</script>` : '')
   var htmlWithCustomCssUrl = htmlWithCustomJs.replace('<% customCssUrl %>', customCssUrl ? `<link href="${customCssUrl}" rel="stylesheet">` : '')
 
   var initOptions = {
@@ -210,6 +214,9 @@ var swaggerInitFunction = function (swaggerDoc, opts) {
     if (trimQuery(req.url) === '/package.json') {
       res.sendStatus(404)
     } else if (trimQuery(req.url) === '/swagger-ui-init.js') {
+      if (req.swaggerDoc) {
+        swaggerInitFile = jsTplString.toString().replace('<% swaggerOptions %>', stringify(opts))
+      }
       res.set('Content-Type', 'application/javascript')
       res.send(swaggerInitFile)
     } else {
diff --git a/test/testapp/app.js b/test/testapp/app.js
index 05ede2e..76a0586 100644
--- a/test/testapp/app.js
+++ b/test/testapp/app.js
@@ -91,6 +91,26 @@ var swaggerHtml = swaggerUi.generateHTML(swaggerDocument, swaggerUiOpts)
 app.use('/api-docs-html1', swaggerUi.serveFiles(swaggerDocument, swaggerUiOpts))
 app.get('/api-docs-html1', (req, res) => { res.send(swaggerHtml) });
 
+var swaggerUiOpts3 = {
+	explorer: false,
+	swaggerOptions: options,
+	customCss: '.swagger-ui .topbar { background-color: pink }',
+	swaggerUrl: '/swagger.json',
+	customJsStr: 'window.alert("123")',
+	operationsSorter: 'alpha',
+}
+
+app.use('/api-docs-jsstr', swaggerUi.serve)
+app.get('/api-docs-jsstr', swaggerUi.setup(null, swaggerUiOpts3));
+
+// let count = 0
+// app.use('/api-docs-dynamic', function(req, res, next){
+// 	count = count + 1
+//     swaggerDocument.info.description = `Hello ${count}!`;
+//     req.swaggerDoc = swaggerDocument;
+//     next();
+// }, swaggerUi.serveFiles(swaggerDocument, options), swaggerUi.setup());
+
 app.use(function(req, res) {
     res.send(404, 'Page not found');
 });