shouldHandle

README.md

@@ -254,8 +254,8 @@ Below is the list of supported features and their compatibility:
 - ✅ server.address()
 - ✅ server.clients
 - ✅ server.close(callback)
+- ✅ server.shouldHandle(request)
 - ❌ server.handleUpgrade(request, socket, head, callback) - see examples above for alternatives
-- ❌ server.shouldHandle(request)
 
 ### Client
 

src/server.js

@@ -93,6 +93,10 @@ module.exports = class WebSocketServer extends EventEmitter {
                     msg.finished = true;
                 });
 
+                if(!this.shouldHandle(msg)) {
+                    return res.writeStatus("400 Bad Request").end();
+                }
+
                 if(this.options.verifyClient) {
                     if(this.options.verifyClient.length === 1) {
                         const result = this.options.verifyClient({
@@ -223,6 +227,10 @@ module.exports = class WebSocketServer extends EventEmitter {
         });
     }
 
+    shouldHandle(req) {
+        return true;
+    }
+
     address() {
         const host = this.options.host ?? '::';
         return { address: host, family: host.includes(':') ? 'IPv6' : 'IPv4', port: this.port };

tests/tests/server/shouldHandle.js

@@ -0,0 +1,32 @@
+// must support server.shouldHandle
+
+const ws = require("ws");
+
+const wss = new ws.WebSocketServer({ port: 8080 }, cb);
+
+wss.shouldHandle = (req) => {
+    return false;
+}
+
+wss.on("connection", (ws, req) => {
+    console.log('connected');
+});
+
+
+function cb() {
+    const c = new ws.WebSocket("ws://localhost:8080");
+    c.on("open", () => {
+        console.log("Connected to server");
+        c.send("Hello from client");
+        c.close();
+    });
+    c.on("close", () => {
+        setTimeout(() => {
+            console.log('closed');
+            process.exit(0);
+        }, 100);
+    });
+    c.on("error", (err) => {
+        console.log(err.message);
+    });
+}