feat(bun): export getBunServer (#4626)

artemtam · web-flow · commit b6e5a97cd44b · 2026-01-26T23:44:09.000+09:00
* feat(bun): export getBunServer

* fix: add generic to jsdoc
diff --git a/src/adapter/bun/conninfo.ts b/src/adapter/bun/conninfo.ts
@@ -8,7 +8,13 @@ import { getBunServer } from './server'
  * @returns ConnInfo
  */
 export const getConnInfo: GetConnInfo = (c: Context) => {
-  const server = getBunServer(c)
+  const server = getBunServer<{
+    requestIP?: (req: Request) => {
+      address: string
+      family: string
+      port: number
+    } | null
+  }>(c)
 
   if (!server) {
     throw new TypeError('env has to include the 2nd argument of fetch.')
diff --git a/src/adapter/bun/index.ts b/src/adapter/bun/index.ts
@@ -8,3 +8,4 @@ export { bunFileSystemModule, toSSG } from './ssg'
 export { createBunWebSocket, upgradeWebSocket, websocket } from './websocket'
 export type { BunWebSocketData, BunWebSocketHandler } from './websocket'
 export { getConnInfo } from './conninfo'
+export { getBunServer } from './server'
diff --git a/src/adapter/bun/server.test.ts b/src/adapter/bun/server.test.ts
@@ -1,10 +1,9 @@
 import { Context } from '../../context'
 import { getBunServer } from './server'
-import type { BunServer } from './server'
 
 describe('getBunServer', () => {
   it('Should success to pick Server', () => {
-    const server = {} as BunServer
+    const server = {}
 
     expect(getBunServer(new Context(new Request('http://localhost/'), { env: server }))).toBe(
       server
diff --git a/src/adapter/bun/server.ts b/src/adapter/bun/server.ts
@@ -4,27 +4,11 @@
  */
 import type { Context } from '../../context'
 
-/**
- * Bun Server Object
- */
-export interface BunServer {
-  requestIP?: (req: Request) => {
-    address: string
-    family: string
-    port: number
-  } | null
-  upgrade<T>(
-    req: Request,
-    options?: {
-      data: T
-    }
-  ): boolean
-}
-
 /**
  * Get Bun Server Object from Context
+ * @template T - The type of Bun Server
  * @param c Context
  * @returns Bun Server
  */
-export const getBunServer = (c: Context): BunServer | undefined =>
-  ('server' in c.env ? c.env.server : c.env) as BunServer | undefined
+export const getBunServer = <T>(c: Context): T | undefined =>
+  ('server' in c.env ? c.env.server : c.env) as T | undefined
diff --git a/src/adapter/bun/websocket.ts b/src/adapter/bun/websocket.ts
@@ -46,7 +46,15 @@ export const createWSContext = (ws: BunServerWebSocket<BunWebSocketData>): WSCon
 }
 
 export const upgradeWebSocket: UpgradeWebSocket<any> = defineWebSocketHelper((c, events) => {
-  const server = getBunServer(c)
+  const server = getBunServer<{
+    upgrade<T>(
+      req: Request,
+      options?: {
+        data: T
+      }
+    ): boolean
+  }>(c)
+
   if (!server) {
     throw new TypeError('env has to include the 2nd argument of fetch.')
   }
