fix(core): Swagger stats authentication (#204)

sourcefuse · May 12, 2021 · ed7d59f · ed7d59f
1 parent 5567fe9
commit ed7d59f
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/packages/core/src/component.ts b/packages/core/src/component.ts
@@ -41,6 +41,17 @@ export class CoreComponent implements Component {
       const swStatsMiddleware = swstats.getMiddleware({
         uriPath: this.coreConfig?.obfPath ?? `/obf`,
         swaggerSpec: this.coreConfig?.openapiSpec,
+        authentication: this.coreConfig.authentication
+          ? this.coreConfig.authentication
+          : false,
+        onAuthenticate: this.coreConfig.swaggerAuthenticate
+          ? this.coreConfig.swaggerAuthenticate
+          : (req, username, password) => {
+              return (
+                username === this.coreConfig.swaggerUsername &&
+                password === this.coreConfig.swaggerPassword
+              );
+            },
       });
       middlewares.push(swStatsMiddleware);
     }

diff --git a/packages/core/src/types.ts b/packages/core/src/types.ts
@@ -1,3 +1,5 @@
+import {IncomingMessage} from 'http';
+
 export interface IServiceConfig {
   useCustomSequence: boolean;
 }
@@ -7,4 +9,12 @@ export interface CoreConfig {
   enableObf?: boolean;
   obfPath?: string;
   openapiSpec?: Record<string, unknown>;
+  authentication?: boolean;
+  swaggerUsername?: string;
+  swaggerPassword?: string;
+  swaggerAuthenticate?: (
+    req?: IncomingMessage,
+    username?: string,
+    password?: string,
+  ) => boolean;
 }