Supports supplying custom sec-websocket-protocol in EdenWS subscribe

src/treaty/index.ts

@@ -60,8 +60,8 @@ export class EdenWS<Schema extends InputSchema<any> = InputSchema> {
     ws: WebSocket
     url: string
 
-    constructor(url: string) {
-        this.ws = new WebSocket(url)
+    constructor(url: string, protocols?: string[]) {
+        this.ws = new WebSocket(url, protocols)
         this.url = url
     }
 
@@ -172,7 +172,7 @@ const createProxy = (
                     transform?: EdenTreaty.Transform
                     headers?: Record<string, string>
                     query?: Record<string, string | number>
-                }
+                },
             ] = [{}, {}]
         ) {
             let bodyObj: any =
@@ -223,7 +223,8 @@ const createProxy = (
                     url.replace(
                         /^([^]+):\/\//,
                         url.startsWith('https://') ? 'wss://' : 'ws://'
-                    )
+                    ),
+                    options as unknown as string[],
                 )
 
             const execute = async <T extends EdenTreaty.ExecuteOptions>(

src/treaty/types.ts

@@ -40,8 +40,8 @@ export namespace EdenTreaty {
                   undefined extends Route['query']
                     ? (params?: {
                           $query?: Record<string, string>
-                      }) => EdenWS<Route>
-                    : (params: { $query: Route['query'] }) => EdenWS<Route['subscribe']>
+                      }, protocols?: string[]) => EdenWS<Route>
+                    : (params: { $query: Route['query'] }, protocols?: string[]) => EdenWS<Route['subscribe']>
                 : // ? HTTP route
                 ((
                       params: Prettify<

src/treaty2/index.ts

@@ -223,7 +223,7 @@ const createProxy = (
 						path +
 						q
 
-					return new EdenWS(url)
+					return new EdenWS(url, options)
 				}
 
 				return (async () => {

src/treaty2/types.ts

@@ -76,8 +76,8 @@ export namespace Treaty {
                             query: Route['subscribe']['query']
                         }) extends infer Param
                 ? {} extends Param
-                    ? (options?: Param) => EdenWS<Route['subscribe']>
-                    : (options?: Param) => EdenWS<Route['subscribe']>
+                    ? (options?: Param, protocols?: string[]) => EdenWS<Route['subscribe']>
+                    : (options?: Param, protocols?: string[]) => EdenWS<Route['subscribe']>
                 : never
             : Route[K] extends {
                     body: infer Body

src/treaty2/ws.ts

@@ -5,8 +5,8 @@ import { parseMessageEvent } from '../utils/parsingUtils'
 export class EdenWS<in out Schema extends InputSchema<any> = {}> {
     ws: WebSocket
 
-    constructor(public url: string) {
-        this.ws = new WebSocket(url)
+    constructor(public url: string, protocols?: string[]) {
+        this.ws = new WebSocket(url, protocols)
     }
 
     send(data: Schema['body'] | Schema['body'][]) {

test/treaty.test.ts

@@ -51,6 +51,15 @@ const app = new Elysia()
     .post('/string', ({ body }) => body, {
         body: t.String()
     })
+	.ws('/custom-protocol', {
+        headers: t.Object({
+			'sec-websocket-protocol': t.Literal('customprotocol'),
+		}),
+        open: async (ws) => { 
+			ws.send('success');
+			ws.close();
+		},
+	})
     .listen(8082)
 
 const client = edenTreaty<typeof app>('http://localhost:8082')
@@ -270,4 +279,15 @@ describe('Eden Treaty', () => {
 
         expect(data).toEqual({ body: null })
     })
+
+
+	it('sends the correct custom protocols', async (done) => {
+		await new Promise<void>(res => {
+			const socket = client['custom-protocol'].subscribe(undefined, ['customprotocol']);
+
+			socket.subscribe(() => res());
+		}) 
+
+		done()
+	});
 })

test/treaty2.test.ts

@@ -152,6 +152,15 @@ const app = new Elysia()
 			ws.close()
 		}
 	})
+	.ws('/custom-protocol', {
+		headers: t.Object({
+			'sec-websocket-protocol': t.Literal('customprotocol'),
+		}),
+		open: async (ws) => { 
+			ws.send('success');
+			ws.close();
+		},
+	})
 	.get('/stream', function* stream() {
 		yield 'a'
 		yield 'b'
@@ -640,4 +649,16 @@ describe('Treaty2 - Using endpoint URL', () => {
 			done()
 		})
 	})
+
+	it('sends the correct custom protocols', async (done) => {
+		const client = treaty<typeof app>('http://localhost:8080')
+
+		await new Promise<void>(res => {
+			const socket = client['custom-protocol'].subscribe(undefined, ['customprotocol']);
+
+			socket.subscribe(() => res());
+		}) 
+
+		done()
+	});
 })