feat(node-ws): Environment variable support

.changeset/modern-clubs-dig.md

@@ -0,0 +1,5 @@
+---
+'@hono/node-ws': major
+---
+
+Environment variable support

packages/node-ws/src/index.test.ts

@@ -2,7 +2,7 @@
 // eslint-disable-next-line @typescript-eslint/ban-ts-comment
 // @ts-ignore
 import type { ServerType } from '@hono/node-server/dist/types'
-import { Hono } from 'hono'
+import { Env, Hono } from 'hono'
 import { cors } from 'hono/cors'
 import { HTTPException } from 'hono/http-exception'
 import type { WSMessageReceive } from 'hono/ws'
@@ -15,10 +15,12 @@
   let injectWebSocket: ReturnType<typeof createNodeWebSocket>['injectWebSocket']
   let upgradeWebSocket: ReturnType<typeof createNodeWebSocket>['upgradeWebSocket']
   let wss: ReturnType<typeof createNodeWebSocket>['wss']
+  let env: Env["Bindings"]
 
   beforeEach(async () => {
     app = new Hono()
-    ;({ injectWebSocket, upgradeWebSocket, wss } = createNodeWebSocket({ app }))
+    env = { customVar: '1' } as Env["Bindings"]
+    ({ injectWebSocket, upgradeWebSocket, wss } = createNodeWebSocket({ app, env }))
 
     server = await new Promise<ServerType>((resolve) => {
       const server = serve({ fetch: app.fetch, port: 3030 }, () => {
@@ -26,7 +28,7 @@
       })
     })
     injectWebSocket(server)
   })
 
   afterEach(() => {
     server.close()
@@ -329,4 +331,20 @@
       })
     )
   })
+
+  it('Should server can obtain environment variables', async () => {
+    const mainPromise = new Promise<string>((resolve) =>
+      app.get(
+        '/',
+        (c, next) => {
+          let ws = upgradeWebSocket((c) => ({}))
+          resolve(c.env.customVar)
+          return ws(c, next)
+        }
+      )
+    )
+
+    new WebSocket('ws://localhost:3030/')
+    expect(await mainPromise).toBe('1')
+  })
 })

packages/node-ws/src/index.ts

@@ -1,4 +1,4 @@
-import type { Hono } from 'hono'
+import type { Hono, Env } from 'hono'
 import { defineWebSocketHelper } from 'hono/ws'
 import type { UpgradeWebSocket, WSContext } from 'hono/ws'
 import type { WebSocket } from 'ws'
@@ -23,6 +23,7 @@ export interface NodeWebSocketInit {
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
   app: Hono<any, any, any>
   baseUrl?: string | URL
+  env?: Env["Bindings"]
 }
 
 const generateConnectionSymbol = () => Symbol('connection')
@@ -77,6 +78,7 @@ export const createNodeWebSocket = (init: NodeWebSocketInit): NodeWebSocket => {
         } = {
           incoming: request,
           outgoing: undefined,
+          ...init.env
         }
         const response = await init.app.request(url, { headers: headers }, env)
         const waiter = waiterMap.get(request)