feat(zod-openapi): Make it possible to do method chaining even for do…

…c methods
honojs · Nov 16, 2023 · 904fb48 · 904fb48
1 parent 4c5b5be
commit 904fb48
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 6 deletions.
diff --git a/.changeset/tame-dodos-grab.md b/.changeset/tame-dodos-grab.md
@@ -0,0 +1,5 @@
+---
+'@hono/zod-openapi': patch
+---
+
+Make it possible to do method chaining even for doc methods
diff --git a/packages/zod-openapi/src/index.ts b/packages/zod-openapi/src/index.ts
@@ -28,6 +28,7 @@ import type {
 import type { MergePath, MergeSchemaPath } from 'hono/types'
 import type { RemoveBlankRecord } from 'hono/utils/types'
 import { mergePath } from 'hono/utils/url'
+import type { OpenAPIObject } from 'openapi3-ts/oas30'
 import type { AnyZodObject, ZodSchema, ZodError } from 'zod'
 import { z, ZodType } from 'zod'
 
@@ -277,18 +278,24 @@ export class OpenAPIHono<
     return document
   }
 
-  doc = (path: string, config: OpenAPIObjectConfig) => {
-    this.get(path, (c) => {
+  doc = <P extends string>(
+    path: P,
+    config: OpenAPIObjectConfig
+  ): OpenAPIHono<E, S & ToSchema<'get', P, {}, OpenAPIObject>, BasePath> => {
+    return this.get(path, (c) => {
       const document = this.getOpenAPIDocument(config)
       return c.json(document)
-    })
+    }) as any
   }
 
-  doc31 = (path: string, config: OpenAPIObjectConfig) => {
-    this.get(path, (c) => {
+  doc31 = <P extends string>(
+    path: P,
+    config: OpenAPIObjectConfig
+  ): OpenAPIHono<E, S & ToSchema<'get', P, {}, OpenAPIObject>, BasePath> => {
+    return this.get(path, (c) => {
       const document = this.getOpenAPI31Document(config)
       return c.json(document)
-    })
+    }) as any
   }
 
   route<