WebSocket Subprotocol

README.md

@@ -17,7 +17,7 @@ Usage
 
 ```
 $ wscat -h
-usage: wscat [-h] [-v] [-l PORT] [-b] [-k] [-d] [address]
+usage: wscat [-h] [-v] [-l PORT] [-b] [-k] [-d] [-s SUBP] [address]
 
 Positional arguments:
   address
@@ -30,6 +30,8 @@ Optional arguments:
   -b, --binary          Use binary WebSockets.
   -k, --keep-open       Do not close the socket after EOF.
   -d, --deflate         Use per-message deflate.
+  -s SUBP, --subprotocol SUBP
+                        WebSocket subprotocol
 
 ```
 

src/cli.ts

@@ -10,6 +10,7 @@ interface ICLIOptions {
     deflate: boolean
     keepOpen: boolean
     listen?: number
+    subProto?: string
 }
 
 const parser = new ArgumentParser({version, addHelp: true})
@@ -39,6 +40,13 @@ parser.addArgument(['-d', '--deflate'], {
     help: 'Use per-message deflate.',
 })
 
+parser.addArgument(['-s', '--subprotocol'], {
+    dest: 'subProto',
+    help: 'WebSocket subprotocol',
+    metavar: 'SUBP',
+    type: String,
+})
+
 parser.addArgument(['address'], {
     nargs: '?',
     type: String,
@@ -52,6 +60,7 @@ const options: any = {
     keepOpen: args.keepOpen,
     outputStream: process.stdout,
     perMessageDeflate: args.deflate,
+    protocol: args.subProto,
 }
 
 if (args.address) {

src/wscat.ts

@@ -24,6 +24,7 @@ export interface IOptions {
     keepOpen: boolean
     outputStream: NodeJS.WritableStream
     perMessageDeflate: boolean
+    protocol: string
 }
 
 export interface IConnectOptions extends IOptions {