Added typescript for websocket req to client obj (#325)

moscajs · Sep 14, 2019 · 24b73f5 · 24b73f5
1 parent 18aeb22
commit 24b73f5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/test/typescript/typings.ts b/test/typescript/typings.ts
@@ -9,6 +9,9 @@ const broker = Server({
   heartbeatInterval: 60000,
   connectTimeout: 30000,
   preConnect: (client: Client, callback) => {
+    if (client.req) {
+      callback(new Error('not websocket stream'), false)
+    }
     if (client.conn.remoteAddress === '::1') {
       callback(null, true)
     } else {

diff --git a/types/index.d.ts b/types/index.d.ts
@@ -7,6 +7,7 @@
 import { IPublishPacket, ISubscribePacket, ISubscription, IUnsubscribePacket } from 'mqtt-packet'
 import { Duplex } from 'stream'
 import { Socket } from 'net'
+import { IncomingMessage } from 'http'
 import EventEmitter = NodeJS.EventEmitter
 
 declare function aedes (options?: aedes.AedesOptions): aedes.Aedes
@@ -24,6 +25,7 @@ declare namespace aedes {
     id: string
     clean: boolean
     conn: Socket
+    req?: IncomingMessage
 
     on (event: 'error', cb: (err: Error) => void): this