diff --git a/package.json b/package.json index 394690c..4dbd1c6 100644 --- a/package.json +++ b/package.json @@ -1,6 +1,6 @@ { "name": "defer-to-connect", - "version": "2.0.0", + "version": "2.0.1", "description": "The safe way to handle the `connect` socket event", "main": "dist/source", "files": [ diff --git a/source/index.ts b/source/index.ts index 0d94384..d075a33 100644 --- a/source/index.ts +++ b/source/index.ts @@ -7,6 +7,10 @@ interface Listeners { close?: (hadError: boolean) => void; } +function isTLSSocket(socket: any): socket is TLSSocket { + return socket.encrypted; +} + const deferToConnect = (socket: Socket | TLSSocket, fn: Listeners | (() => void)): void => { let listeners: Listeners; @@ -26,7 +30,7 @@ const deferToConnect = (socket: Socket | TLSSocket, fn: Listeners | (() => void) listeners.connect!(); } - if (socket.encrypted && hasSecureConnectListener) { + if (isTLSSocket(socket) && hasSecureConnectListener) { if (socket.authorized) { listeners.secureConnect!(); } else if (!socket.authorizationError) {