@autofleet/zehut 1.3.0-alpha → 1.3.0-alpha2

package/lib/index.d.ts

@@ -1,2 +1,15 @@
-import User, { middleware, eagerLoadPermissionsMiddleware, middlewareWithDecode, getDecodedBearer } from './user';
-export { User, middleware, middlewareWithDecode, eagerLoadPermissionsMiddleware, getDecodedBearer };
+import User from './user';
+export declare const exportConst: {
+    User: typeof User;
+    middleware: (options?: {
+        eagerLoadUserPermissions?: boolean;
+        eagerLoadUserPermissionsLegacy?: boolean;
+    }) => (req: any, res: any, next: any) => Promise<void>;
+    middlewareWithDecode: (options?: {
+        eagerLoadUserPermissions?: boolean;
+        eagerLoadUserPermissionsLegacy?: boolean;
+    }) => (req: any, res: any, next: any) => Promise<void>;
+    eagerLoadPermissionsMiddleware: (req: any, res: any, next: any) => Promise<void>;
+    getCurrentPayload: () => any;
+    getDecodedBearer: (req: any) => any;
+};

package/lib/index.js

@@ -19,10 +19,15 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getDecodedBearer = exports.eagerLoadPermissionsMiddleware = exports.middlewareWithDecode = exports.middleware = exports.User = void 0;
+exports.exportConst = void 0;
 const user_1 = __importStar(require("./user"));
-exports.User = user_1.default;
-Object.defineProperty(exports, "middleware", { enumerable: true, get: function () { return user_1.middleware; } });
-Object.defineProperty(exports, "eagerLoadPermissionsMiddleware", { enumerable: true, get: function () { return user_1.eagerLoadPermissionsMiddleware; } });
-Object.defineProperty(exports, "middlewareWithDecode", { enumerable: true, get: function () { return user_1.middlewareWithDecode; } });
-Object.defineProperty(exports, "getDecodedBearer", { enumerable: true, get: function () { return user_1.getDecodedBearer; } });
+const tracer_1 = require("./tracer");
+tracer_1.enable();
+exports.exportConst = {
+    User: user_1.default,
+    middleware: user_1.middleware(tracer_1.newTrace),
+    middlewareWithDecode: user_1.middlewareWithDecode(tracer_1.newTrace),
+    eagerLoadPermissionsMiddleware: user_1.eagerLoadPermissionsMiddleware,
+    getCurrentPayload: tracer_1.getCurrentTrace,
+    getDecodedBearer: user_1.getDecodedBearer
+};

package/lib/tracer.d.ts

@@ -0,0 +1,9 @@
+import asyncHooks from 'async_hooks';
+declare const tracer: {
+    currentTrace: any;
+    traces: {};
+};
+export declare const newTrace: (type: any) => void;
+export declare const enable: () => () => asyncHooks.AsyncHook;
+export declare const getCurrentTrace: () => any;
+export default tracer;

package/lib/tracer.js

@@ -0,0 +1,55 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getCurrentTrace = exports.enable = exports.newTrace = void 0;
+const async_hooks_1 = __importDefault(require("async_hooks"));
+const uuid_1 = __importDefault(require("uuid"));
+const prevStates = {};
+const tracer = {
+    currentTrace: null,
+    traces: {},
+};
+function init(asyncId, type, triggerAsyncId) {
+    if (tracer.traces[triggerAsyncId]) {
+        tracer.traces[asyncId] = tracer.traces[triggerAsyncId];
+    }
+}
+function before(asyncId) {
+    if (!tracer.traces[asyncId]) {
+        return;
+    }
+    prevStates[asyncId] = tracer.currentTrace;
+    tracer.currentTrace = tracer.traces[asyncId];
+}
+function after(asyncId) {
+    if (!tracer.traces[asyncId]) {
+        return;
+    }
+    tracer.currentTrace = prevStates[asyncId];
+}
+function destroy(asyncId) {
+    if (tracer.traces[asyncId]) {
+        delete tracer.traces[asyncId];
+        delete prevStates[asyncId];
+    }
+}
+const hook = async_hooks_1.default.createHook({
+    init, before, after, destroy,
+});
+class Trace {
+    constructor(type) {
+        this.id = uuid_1.default.v1();
+        this.type = type;
+        this.context = new Map();
+    }
+}
+exports.newTrace = (type) => {
+    tracer.currentTrace = new Trace(type);
+    tracer.traces[async_hooks_1.default.executionAsyncId()] = tracer.currentTrace;
+    return tracer.currentTrace;
+};
+exports.enable = () => () => hook.enable();
+exports.getCurrentTrace = () => tracer.currentTrace || {};
+exports.default = tracer;

package/lib/user/index.d.ts

@@ -1,9 +1,9 @@
 import ApiUser from './ApiUser';
-export declare const middleware: (options?: {
+export declare const middleware: (newTrace: any) => (options?: {
     eagerLoadUserPermissions?: boolean;
     eagerLoadUserPermissionsLegacy?: boolean;
 }) => (req: any, res: any, next: any) => Promise<void>;
-export declare const middlewareWithDecode: (options?: {
+export declare const middlewareWithDecode: (newTrace: any) => (options?: {
     eagerLoadUserPermissions?: boolean;
     eagerLoadUserPermissionsLegacy?: boolean;
 }) => (req: any, res: any, next: any) => Promise<void>;

package/lib/user/index.js

@@ -15,7 +15,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.getDecodedBearer = exports.eagerLoadPermissionsMiddleware = exports.middlewareWithDecode = exports.middleware = void 0;
 const ApiUser_1 = __importDefault(require("./ApiUser"));
 const utils_1 = require("../utils");
-exports.middleware = (options = {}) => (req, res, next) => __awaiter(void 0, void 0, void 0, function* () {
+exports.middleware = (newTrace) => (options = {}) => (req, res, next) => __awaiter(void 0, void 0, void 0, function* () {
     const userId = req.headers['x-af-user-id'];
     if (!userId) {
         next();
@@ -29,11 +29,13 @@ exports.middleware = (options = {}) => (req, res, next) => __awaiter(void 0, voi
         yield userObject.getUserPermissionsLegacy();
     }
     req.user = userObject;
+    const trace = newTrace('userPayload');
+    trace.context.set('userObject', userObject);
     // Added in order to support outbreak.
     req.headers['x-af-user-permissions'] = userObject;
     next();
 });
-exports.middlewareWithDecode = (options = {}) => (req, res, next) => __awaiter(void 0, void 0, void 0, function* () {
+exports.middlewareWithDecode = (newTrace) => (options = {}) => (req, res, next) => __awaiter(void 0, void 0, void 0, function* () {
     var _a;
     if (req.headers.authorization) {
         const decoded = utils_1.decodeBearer(req.headers.authorization);
@@ -49,6 +51,8 @@ exports.middlewareWithDecode = (options = {}) => (req, res, next) => __awaiter(v
             yield userObject.getUserPermissionsLegacy();
         }
         req.user = userObject;
+        const trace = newTrace('userPayload');
+        trace.context.set('userObject', userObject);
         // Added in order to support outbreak.
         req.headers['x-af-user-permissions'] = userObject;
         const entities = req.user.permissions;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@autofleet/zehut",
-  "version": "1.3.0-alpha",
+  "version": "1.3.0-alpha2",
   "description": "manage user's identity",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",