From 0cb5887e91dde792b807cc3d7f0862042f5287a0 Mon Sep 17 00:00:00 2001
From: Ian Sinnott <ian@iansinnott.com>
Date: Sat, 29 Jul 2023 21:21:20 +0800
Subject: [PATCH] feat: allow configuring the path of the openapi json spec

---
 src/index.ts       | 10 ++++++++--
 src/types.ts       |  8 ++++++++
 test/index.test.ts | 13 +++++++++++++
 3 files changed, 29 insertions(+), 2 deletions(-)

diff --git a/src/index.ts b/src/index.ts
index eefbfbd..cfe7f8e 100644
--- a/src/index.ts
+++ b/src/index.ts
@@ -17,12 +17,14 @@ export const swagger =
             version = '4.18.2',
             excludeStaticFile = true,
             path = '/swagger' as Path,
+            specPathname = 'json',
             exclude = []
         }: ElysiaSwaggerConfig<Path> = {
             documentation: {},
             version: '4.18.2',
             excludeStaticFile: true,
             path: '/swagger' as Path,
+            specPathname: 'json',
             exclude: []
         }
     ) =>
@@ -33,6 +35,10 @@ export const swagger =
             version: '0.0.0',
             ...documentation.info
         }
+        
+        if (specPathname.startsWith('/')) {
+            specPathname = specPathname.slice(1)
+        }
 
         app.get(path, (context) => {
             return new Response(
@@ -58,7 +64,7 @@ export const swagger =
     <script>
         window.onload = () => {
             window.ui = SwaggerUIBundle({
-                url: '${path}/json',
+                url: '${path}/${specPathname}',
                 dom_id: '#swagger-ui',
             });
         };
@@ -73,7 +79,7 @@ export const swagger =
             )
         }).route(
             'GET',
-            `${path}/json`,
+            `${path}/${specPathname}`,
             (context) =>
                 ({
                     openapi: '3.0.3',
diff --git a/src/types.ts b/src/types.ts
index 860af2b..ae11f7c 100644
--- a/src/types.ts
+++ b/src/types.ts
@@ -31,6 +31,14 @@ export interface ElysiaSwaggerConfig<Path extends string = '/swagger'> {
      * @default '/swagger'
      */
     path?: Path
+    /**
+     * The pathname for the OpenAPI spec, which will be appended to the path option.
+     * 
+     * @example 'openapi.json' => '/swagger/openapi.json'
+     *
+     * @default 'json'
+     */
+    specPathname?: string
     /**
      * Paths to exclude from Swagger endpoint
      *
diff --git a/test/index.test.ts b/test/index.test.ts
index e696f48..f277d7d 100644
--- a/test/index.test.ts
+++ b/test/index.test.ts
@@ -65,4 +65,17 @@ describe('Swagger', () => {
         const res = await app.handle(req('/v2/swagger'))
         expect(res.status).toBe(200)
     })
+
+    it('use custom spec path', async () => {
+        const app = new Elysia().use(
+            swagger({
+                path: '/v2/swagger',
+                specPathname: 'openapi.json'
+            })
+        )
+
+        const res = await app.handle(req('/v2/swagger/openapi.json'))
+        expect(res.status).toBe(200)
+        expect(res.headers.get('content-type')).toContain('application/json')
+    })
 })