feat(core): add middleware data in request object

lokesh-coder · Nov 22, 2020 · cf62dba · cf62dba
1 parent f5b7563
commit cf62dba
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 9 deletions.
diff --git a/packages/core/__tests__/__snapshots__/core.spec.ts.snap b/packages/core/__tests__/__snapshots__/core.spec.ts.snap
@@ -89,6 +89,7 @@ Object {
     "getCommandById": [Function],
     "getCommandByName": [Function],
     "getCommands": [Function],
+    "getMiddlewares": [Function],
     "runCommand": [Function],
   },
   "root": StringContaining "/packages/core/__tests__",

diff --git a/packages/core/__tests__/core.spec.ts b/packages/core/__tests__/core.spec.ts
@@ -103,17 +103,29 @@ describe("Core", () => {
     it("should get all middlewares", () => {
       expect(core["mwCtrl"].get()).toEqual({
         END: [
-          { on: "END", run: expect.any(Function) },
-          { on: "END", run: expect.any(Function) },
+          { on: "END", run: expect.any(Function), source: "__OBJ__" },
+          {
+            on: "END",
+            run: expect.any(Function),
+            source: `${__dirname}/plugin/middleware.js`,
+          },
+        ],
+        INIT: [{ on: "INIT", run: expect.any(Function), source: "__OBJ__" }],
+        POST_PARSE: [
+          { on: "POST_PARSE", run: expect.any(Function), source: "__OBJ__" },
         ],
-        INIT: [{ on: "INIT", run: expect.any(Function) }],
-        POST_PARSE: [{ on: "POST_PARSE", run: expect.any(Function) }],
         POST_RUN: [],
         POST_VALIDATE: [],
-        PRE_PARSE: [{ on: "PRE_PARSE", run: expect.any(Function) }],
-        PRE_RUN: [{ on: "PRE_RUN", run: expect.any(Function) }],
-        PRE_VALIDATE: [{ on: "PRE_VALIDATE", run: expect.any(Function) }],
-        START: [{ on: "START", run: expect.any(Function) }],
+        PRE_PARSE: [
+          { on: "PRE_PARSE", run: expect.any(Function), source: "__OBJ__" },
+        ],
+        PRE_RUN: [
+          { on: "PRE_RUN", run: expect.any(Function), source: "__OBJ__" },
+        ],
+        PRE_VALIDATE: [
+          { on: "PRE_VALIDATE", run: expect.any(Function), source: "__OBJ__" },
+        ],
+        START: [{ on: "START", run: expect.any(Function), source: "__OBJ__" }],
       });
     });
     it("should get config", () => {

diff --git a/packages/core/src/core.ts b/packages/core/src/core.ts
@@ -107,6 +107,7 @@ class LesyCoreClass {
       getCommandById: this.cmdCtrl.getCommandById.bind(this.cmdCtrl),
       getCommandByName: this.cmdCtrl.getCommandByName.bind(this.cmdCtrl),
       getCommands: this.cmdCtrl.getCommands.bind(this.cmdCtrl),
+      getMiddlewares: this.mwCtrl ? this.mwCtrl.get.bind(this.mwCtrl) : {},
     };
   }
 

diff --git a/packages/core/src/middleware.ts b/packages/core/src/middleware.ts
@@ -19,7 +19,8 @@ class LesyMiddleware {
     END: [],
   };
 
-  add(mw: Middleware) {
+  add(mw: Middleware, source: string) {
+    mw.source = source;
     this.middlewares[mw.on].push(mw);
   }
 

diff --git a/packages/core/src/model.ts b/packages/core/src/model.ts
@@ -34,6 +34,7 @@ export type MiddlewareFn = (
 export interface Middleware {
   on: MiddlewarePlacement;
   run: MiddlewareFn;
+  source?: string;
 }
 
 export type MiddlewareContext = Record<string, any>;