From ba83a268e010a7b18172e7de01e3901b58a7ec62 Mon Sep 17 00:00:00 2001
From: Karibash <40270352+Karibash@users.noreply.github.com>
Date: Sat, 18 Nov 2023 06:04:56 +0900
Subject: [PATCH] =?UTF-8?q?feat(zod-openapi):=20Make=20it=20possible=20to?=
 =?UTF-8?q?=20do=20method=20chaining=20even=20for=20do=E2=80=A6=20(#260)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

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

* fix: Fix incorrect generics type specification
---
 .changeset/tame-dodos-grab.md     |  5 +++++
 packages/zod-openapi/src/index.ts | 18 ++++++++++++------
 2 files changed, 17 insertions(+), 6 deletions(-)
 create mode 100644 .changeset/tame-dodos-grab.md

diff --git a/.changeset/tame-dodos-grab.md b/.changeset/tame-dodos-grab.md
new file mode 100644
index 00000000..ccb1b237
--- /dev/null
+++ 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
index 08f51a59..6b8cf865 100644
--- a/packages/zod-openapi/src/index.ts
+++ b/packages/zod-openapi/src/index.ts
@@ -277,18 +277,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, {}, {}>, 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, {}, {}>, BasePath> => {
+    return this.get(path, (c) => {
       const document = this.getOpenAPI31Document(config)
       return c.json(document)
-    })
+    }) as any
   }
 
   route<