fix(pencil): Added cls context implementation

Author: alexivsn

package-lock.json

@@ -23,6 +23,7 @@
     "semantic-release": "semantic-release"
   },
   "dependencies": {
+    "cls-hooked": "^4.2.2",
     "cookie": "^0.3.1",
     "futoin-ipset": "^1.3.4",
     "ip": "^1.1.5",
@@ -37,6 +38,7 @@
     "@types/assert": "^1.4.6",
     "@types/chai": "^4.2.10",
     "@types/chai-as-promised": "^7.1.2",
+    "@types/cls-hooked": "^4.3.0",
     "@types/cookie": "^0.3.3",
     "@types/express": "^4.17.3",
     "@types/fetch-mock": "^7.3.2",

package.json

@@ -32,13 +32,13 @@ export default class SDKEvent implements IEvent {
 
   constructor(event: EventOptions, options: SecureNativeOptions) {
     Logger.debug('Building SDK event');
+
     // extract info from session
-    const { req } = SessionManager.getLastSession();
+    const req  = SessionManager.getNs().get('req');
     if (!req) {
       Logger.debug('request is empty', req);
     }
     const reqCtx = mergeRequestContexts(event.context || {}, contextFromRequest(req));
-
     const decryptedToken = decrypt(reqCtx?.clientToken, options.apiKey);
     Logger.debug('Decrypted client token', decryptedToken);
     const parsedToken = JSON.parse(decryptedToken) || {};

src/events/sdk-event.ts

@@ -1,13 +1,15 @@
 import { IncomingMessage, ServerResponse } from 'http';
 import { Logger } from './logger';
+import { createNamespace, Namespace } from 'cls-hooked';
+
 export interface Session {
   req: IncomingMessage | any;
   res: ServerResponse | any;
 }
 
 export default class SessionManager {
   private static stack: Array<Session> = [];
-
+  private static ns = createNamespace('sn_session');
   static getLastSession(): Session {
     const [session = { req: null, res: null }] = SessionManager.stack;
     return session;
@@ -38,4 +40,8 @@ export default class SessionManager {
     Logger.debug(`[SessionManager] Cleaning all sessions`);
     SessionManager.stack = [];
   }
+
+  static getNs(): Namespace {
+    return SessionManager.ns;
+  }
 }

src/session-manager.ts

@@ -1,4 +1,5 @@
 import { wrap } from 'shimmer';
+import SessionManager from '../session-manager';
 
 export const wrapListener = (module, method, listner, callback) => {
   wrap(module, method, function (original) {
@@ -8,8 +9,20 @@ export const wrapListener = (module, method, listner, callback) => {
         if (!status) {
           return;
         }
+        // @ts-ignore
+        const [ , req, res ] = arguments;
+
+        const ns = SessionManager.getNs();
+        ns.bindEmitter(req);
+        ns.bindEmitter(res);
+
+        return ns.run(() => {
+          ns.set('req', req);
+          ns.set('res', res);
+          return original.apply(this, arguments);
+        });
       }
       return original.apply(this, arguments);
     };
   });
-}
+};