Add WebSocket support (ws)

Makefile

@@ -54,6 +54,11 @@ hono-sqlite3:
 hapi-postgres:
 	cd sample-apps/hapi-postgres && AIKIDO_DEBUG=true AIKIDO_BLOCKING=true node app.js
 
+
+.PHONY: ws-postgres
+ws-postgres:
+	cd sample-apps/ws-postgres && AIKIDO_DEBUG=true AIKIDO_BLOCKING=true node app.js
+
 .PHONY: install
 install:
 	mkdir -p build && cp library/package.json build/package.json

library/agent/Context.ts

@@ -21,6 +21,7 @@ export type Context = {
   graphql?: string[];
   xml?: unknown;
   subdomains?: string[]; // https://expressjs.com/en/5x/api.html#req.subdomains
+  ws?: unknown; // Additional data related to WebSocket connections, like the last message received
 };
 
 /**
@@ -56,6 +57,7 @@ export function runWithContext<T>(context: Context, fn: () => T) {
     current.graphql = context.graphql;
     current.xml = context.xml;
     current.subdomains = context.subdomains;
+    current.ws = context.ws;
 
     return fn();
   }

library/agent/Source.ts

@@ -7,6 +7,7 @@ export const SOURCES = [
   "graphql",
   "xml",
   "subdomains",
+  "ws",
 ] as const;
 
 export type Source = (typeof SOURCES)[number];

library/agent/protect.ts

@@ -37,6 +37,7 @@ import { FastXmlParser } from "../sources/FastXmlParser";
 import { SQLite3 } from "../sinks/SQLite3";
 import { Hapi } from "../sources/Hapi";
 import { Shelljs } from "../sinks/Shelljs";
+import { Ws } from "../sources/Ws";
 
 function isDebugging() {
   return (
@@ -133,6 +134,7 @@ function getWrappers() {
     new SQLite3(),
     new Shelljs(),
     new Hapi(),
+    new Ws(),
   ];
 }
 

library/package.json

@@ -47,6 +47,7 @@
     "@types/shell-quote": "^1.7.5",
     "@types/sinonjs__fake-timers": "^8.1.5",
     "@types/supertest": "^6.0.2",
+    "@types/ws": "^8.5.10",
     "@typescript-eslint/eslint-plugin": "^6.19.0",
     "@typescript-eslint/parser": "^6.19.0",
     "aws-sdk": "^2.1595.0",
@@ -72,6 +73,7 @@
     "tap": "^18.6.1",
     "typescript": "^5.3.3",
     "undici": "^6.12.0",
+    "ws": "^8.18.0",
     "xml2js": "^0.6.2"
   },
   "scripts": {