conode 0.1.23 → 0.2.0

package/README.md

@@ -1,13 +1,11 @@
 # CoNode
 
-0-Deps, simple and fast context library for NodeJS
+0-deps async context management for Node.js
 
 [![Build Status][github-image]][github-url]
 [![NPM version][npm-image]][npm-url]
 [![Downloads][downloads-image]][npm-url]
 [![Coverage Status][codecov-image]][codecov-url]
-[![Maintainability][codeclimate-image]][codeclimate-url]
-[![Snyk][snyk-image]][snyk-url]
 
 ## Table of Contents
 
@@ -18,9 +16,10 @@
 
 ## Features
 
-- Provides useful wrappers of [AsyncLocalStorage](https://nodejs.org/api/async_context.html)
-- Supports TypeScript typings
-- Supports CommonJS and ES Module systems
+- Zero dependencies  
+- Lightweight wrappers around [AsyncLocalStorage](https://nodejs.org/api/async_hooks.html)  
+- Full TypeScript support  
+- Works in CommonJS and ES Module environments  
 
 ## Installing
 
@@ -38,6 +37,22 @@ yarn add conode
 
 ## Examples
 
+Quick start
+```typescript
+import { createContext } from 'conode';
+
+interface RequestContext { userId: string }
+
+const reqCtx = createContext<RequestContext>();
+
+async function handleRequest() {
+  console.log(reqCtx.getContext()?.userId);
+}
+
+await reqCtx.contextualize({ userId: 'abc123' }, handleRequest);
+// Logs: 'abc123'
+```
+
 Functional approach
 ```typescript
 import { createContext } from '../index';
@@ -51,16 +66,16 @@ const action = () => {
 auth.contextualize(jwt, action);
 ```
 
-Class approach
+Class approach with decorator
 ```typescript
-import { contextualize, Context } from '../index';
+import { contextualize, Contextable } from '../index';
 
 const action = () => {
   const service = Service.getContext();
   console.log(service.getSomething());
 };
 
-class Service extends Context() {
+class Service extends Contextable() {
   @contextualize
   async run() {
     await action();
@@ -83,12 +98,8 @@ Copyright (c) 2022-present Ivan Zakharchanka
 [npm-url]: https://www.npmjs.com/package/conode
 [downloads-image]: https://img.shields.io/npm/dw/conode.svg?maxAge=43200
 [npm-image]: https://img.shields.io/npm/v/conode.svg?maxAge=43200
-[github-url]: https://github.com/3axap4eHko/conode/actions/workflows/cicd.yml
-[github-image]: https://github.com/3axap4eHko/conode/actions/workflows/cicd.yml/badge.svg
+[github-url]: https://github.com/3axap4eHko/conode/actions/workflows/build.yml
+[github-image]: https://github.com/3axap4eHko/conode/actions/workflows/build.yml/badge.svg
 [codecov-url]: https://codecov.io/gh/3axap4eHko/conode
 [codecov-image]: https://codecov.io/gh/3axap4eHko/conode/branch/master/graph/badge.svg?token=JZ8QCGH6PI
-[codeclimate-url]: https://codeclimate.com/github/3axap4eHko/conode/maintainability
-[codeclimate-image]: https://api.codeclimate.com/v1/badges/0ba20f27f6db2b0fec8c/maintainability
-[snyk-url]: https://snyk.io/test/npm/conode/latest
-[snyk-image]: https://img.shields.io/snyk/vulnerabilities/github/3axap4eHko/conode.svg?maxAge=43200
 

package/build/index.cjs

@@ -1,70 +1,65 @@
 "use strict";
-var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
-    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
-    return new (P || (P = Promise))(function (resolve, reject) {
-        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
-        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
-        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
-        step((generator = generator.apply(thisArg, _arguments || [])).next());
+Object.defineProperty(exports, "__esModule", {
+    value: true
+});
+function _export(target, all) {
+    for(var name in all)Object.defineProperty(target, name, {
+        enumerable: true,
+        get: all[name]
     });
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.contextualize = exports.Context = exports.createContext = void 0;
-const async_hooks_1 = require("async_hooks");
-const createContext = () => {
-    const storage = new async_hooks_1.AsyncLocalStorage();
+}
+_export(exports, {
+    Contextable: function() {
+        return Contextable;
+    },
+    contextualize: function() {
+        return contextualize;
+    },
+    createContext: function() {
+        return createContext;
+    }
+});
+const _nodeasync_hooks = require("node:async_hooks");
+const createContext = ()=>{
+    const storage = new _nodeasync_hooks.AsyncLocalStorage();
     return {
-        contextualize(value, callback) {
-            return new Promise((resolve, reject) => {
-                storage.run(value, () => __awaiter(this, void 0, void 0, function* () {
-                    try {
-                        yield callback();
-                        resolve();
-                    }
-                    catch (e) {
-                        reject(e);
-                    }
-                }));
-            });
+        contextualize (context, callback) {
+            return storage.run({
+                context
+            }, callback);
         },
-        getContext() {
-            return storage.getStore();
+        getContext () {
+            const store = storage.getStore();
+            if (!store) {
+                throw new Error('Context not initialized');
+            }
+            return store.context;
         }
     };
 };
-exports.createContext = createContext;
-function Context(Base) {
-    const context = (0, exports.createContext)();
-    return Base
-        ? class Contextual extends Base {
-            static get context() {
-                return context.getContext();
-            }
-            contextualize(callback) {
-                return __awaiter(this, void 0, void 0, function* () {
-                    return context.contextualize(this, callback);
-                });
-            }
+function Contextable(BaseClass = class {
+}) {
+    const context = createContext();
+    return class Contextual extends BaseClass {
+        static getContext() {
+            return context.getContext();
         }
-        : class Contextual {
-            static get context() {
-                return context.getContext();
-            }
-            contextualize(callback) {
-                return __awaiter(this, void 0, void 0, function* () {
-                    return context.contextualize(this, callback);
-                });
-            }
-        };
+        constructor(...args){
+            super(...args);
+        }
+        contextualize(callback) {
+            return context.contextualize(this, callback);
+        }
+    };
 }
-exports.Context = Context;
-function contextualize(target, _, descriptor) {
-    const method = descriptor.value;
-    function contextualized() {
-        return target.contextualize(() => method.apply(this, arguments));
+function contextualize(...args) {
+    const [, , descriptor] = args;
+    const property = descriptor?.value;
+    if (typeof property === 'function') {
+        descriptor.value = function(...args) {
+            return this.contextualize(()=>property.apply(this, args));
+        };
     }
-    ;
-    descriptor.value = contextualized;
 }
-exports.contextualize = contextualize;
-//# sourceMappingURL=index.cjs.map
+
+//# sourceMappingURL=index.cjs.map

package/build/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,6CAAgD;AAIzC,MAAM,aAAa,GAAG,GAAM,EAAE;IACnC,MAAM,OAAO,GAAG,IAAI,+BAAiB,EAAK,CAAC;IAE3C,OAAO;QACL,aAAa,CAAC,KAAQ,EAAE,QAAkB;YACxC,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC3C,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,GAAS,EAAE;oBAC5B,IAAI;wBACF,MAAM,QAAQ,EAAE,CAAC;wBACjB,OAAO,EAAE,CAAC;qBACX;oBAAC,OAAO,CAAC,EAAE;wBACV,MAAM,CAAC,CAAC,CAAC,CAAC;qBACX;gBACH,CAAC,CAAA,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;QACD,UAAU;YACR,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC;KACF,CAAC;AACJ,CAAC,CAAC;AApBW,QAAA,aAAa,iBAoBxB;AAIF,SAAgB,OAAO,CAAC,IAAkB;IACxC,MAAM,OAAO,GAAG,IAAA,qBAAa,GAAE,CAAC;IAEhC,OAAO,IAAI;QACT,CAAC,CAAC,MAAM,UAAW,SAAQ,IAAI;YAC7B,MAAM,KAAK,OAAO;gBAChB,OAAO,OAAO,CAAC,UAAU,EAAE,CAAC;YAC9B,CAAC;YAEK,aAAa,CAAC,QAAkB;;oBACpC,OAAO,OAAO,CAAC,aAAa,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;gBAC/C,CAAC;aAAA;SACF;QACD,CAAC,CAAC,MAAM,UAAU;YAChB,MAAM,KAAK,OAAO;gBAChB,OAAO,OAAO,CAAC,UAAU,EAAE,CAAC;YAC9B,CAAC;YAEK,aAAa,CAAC,QAAkB;;oBACpC,OAAO,OAAO,CAAC,aAAa,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;gBAC/C,CAAC;aAAA;SACF,CAAC;AACN,CAAC;AAtBD,0BAsBC;AAMD,SAAgB,aAAa,CAA2B,MAAS,EAAE,CAAS,EAAE,UAA6C;IACzH,MAAM,MAAM,GAAG,UAAU,CAAC,KAAM,CAAC;IACjC,SAAS,cAAc;QACrB,OAAO,MAAM,CAAC,aAAa,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC;IACnE,CAAC;IAAA,CAAC;IACF,UAAU,CAAC,KAAK,GAAG,cAAqB,CAAC;AAC3C,CAAC;AAND,sCAMC"}
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import { AsyncLocalStorage } from 'node:async_hooks';\n\nexport type Fn<T extends unknown[], R> = (...args: T) => R;\n\nexport const createContext = <T>() => {\n  const storage = new AsyncLocalStorage<{ context: T }>();\n  return {\n    contextualize<R>(context: T, callback: Fn<[], R>): R {\n      return storage.run({ context }, callback);\n    },\n    getContext() {\n      const store = storage.getStore();\n      if (!store) {\n        throw new Error('Context not initialized');\n      }\n      return store.context;\n    }\n  };\n};\n\ntype Constructor = new (...args: any) => any;\n\nexport interface Contextualized {\n  contextualize<R>(callback: Fn<[], R>): Promise<R>;\n}\n\nexport function Contextable(BaseClass: Constructor = class { }) {\n  const context = createContext();\n  return class Contextual extends BaseClass {\n    static getContext<C extends Constructor>(this: C) {\n      return context.getContext() as InstanceType<C>;\n    }\n    constructor(...args: unknown[]) {\n      super(...args);\n    }\n\n    contextualize<R>(callback: Fn<[], R>): R {\n      return context.contextualize(this, callback);\n    }\n  };\n}\n\nexport function contextualize<T extends Contextualized, K extends keyof T>(...args: unknown[]) {\n  const [, , descriptor] = args as [T, K, TypedPropertyDescriptor<T[K]>];\n  const property = descriptor?.value;\n  if (typeof property === 'function') {\n    descriptor.value = function(this: Contextualized, ...args: unknown[]) {\n      return this.contextualize(() => property.apply(this, args));\n    } as T[K];\n  }\n}\n\n"],"names":["Contextable","contextualize","createContext","storage","AsyncLocalStorage","context","callback","run","getContext","store","getStore","Error","BaseClass","Contextual","constructor","args","descriptor","property","value","apply"],"mappings":";;;;;;;;;;;IA0BgBA,WAAW;eAAXA;;IAgBAC,aAAa;eAAbA;;IAtCHC,aAAa;eAAbA;;;iCAJqB;AAI3B,MAAMA,gBAAgB;IAC3B,MAAMC,UAAU,IAAIC,kCAAiB;IACrC,OAAO;QACLH,eAAiBI,OAAU,EAAEC,QAAmB;YAC9C,OAAOH,QAAQI,GAAG,CAAC;gBAAEF;YAAQ,GAAGC;QAClC;QACAE;YACE,MAAMC,QAAQN,QAAQO,QAAQ;YAC9B,IAAI,CAACD,OAAO;gBACV,MAAM,IAAIE,MAAM;YAClB;YACA,OAAOF,MAAMJ,OAAO;QACtB;IACF;AACF;AAQO,SAASL,YAAYY,YAAyB;AAAQ,CAAC;IAC5D,MAAMP,UAAUH;IAChB,OAAO,MAAMW,mBAAmBD;QAC9B,OAAOJ,aAA2C;YAChD,OAAOH,QAAQG,UAAU;QAC3B;QACAM,YAAY,GAAGC,IAAe,CAAE;YAC9B,KAAK,IAAIA;QACX;QAEAd,cAAiBK,QAAmB,EAAK;YACvC,OAAOD,QAAQJ,aAAa,CAAC,IAAI,EAAEK;QACrC;IACF;AACF;AAEO,SAASL,cAA2D,GAAGc,IAAe;IAC3F,MAAM,KAAKC,WAAW,GAAGD;IACzB,MAAME,WAAWD,YAAYE;IAC7B,IAAI,OAAOD,aAAa,YAAY;QAClCD,WAAWE,KAAK,GAAG,SAA+B,GAAGH,IAAe;YAClE,OAAO,IAAI,CAACd,aAAa,CAAC,IAAMgB,SAASE,KAAK,CAAC,IAAI,EAAEJ;QACvD;IACF;AACF"}

package/build/index.d.ts

@@ -1,17 +1,18 @@
-export type Callback = () => any;
+export type Fn<T extends unknown[], R> = (...args: T) => R;
 export declare const createContext: <T>() => {
-    contextualize(value: T, callback: Callback): Promise<void>;
-    getContext(): T | undefined;
-};
-type Constructor = new (...args: any[]) => {};
-export declare function Context(Base?: Constructor): {
-    new (...args: any[]): {
-        contextualize(callback: Callback): Promise<void>;
-    };
-    readonly context: unknown;
+    contextualize<R>(context: T, callback: Fn<[], R>): R;
+    getContext(): T;
 };
+type Constructor = new (...args: any) => any;
 export interface Contextualized {
-    contextualize(callback: Callback): Promise<any>;
+    contextualize<R>(callback: Fn<[], R>): Promise<R>;
 }
-export declare function contextualize<T extends Contextualized>(target: T, _: string, descriptor: TypedPropertyDescriptor<Callback>): void;
+export declare function Contextable(BaseClass?: Constructor): {
+    new (...args: unknown[]): {
+        [x: string]: any;
+        contextualize<R>(callback: Fn<[], R>): R;
+    };
+    getContext<C extends Constructor>(this: C): InstanceType<C>;
+};
+export declare function contextualize<T extends Contextualized, K extends keyof T>(...args: unknown[]): void;
 export {};

package/build/index.js

@@ -1,51 +1,44 @@
-import { AsyncLocalStorage } from 'async_hooks';
-export const createContext = () => {
+import { AsyncLocalStorage } from 'node:async_hooks';
+export const createContext = ()=>{
     const storage = new AsyncLocalStorage();
     return {
-        contextualize(value, callback) {
-            return new Promise((resolve, reject) => {
-                storage.run(value, async () => {
-                    try {
-                        await callback();
-                        resolve();
-                    }
-                    catch (e) {
-                        reject(e);
-                    }
-                });
-            });
+        contextualize (context, callback) {
+            return storage.run({
+                context
+            }, callback);
         },
-        getContext() {
-            return storage.getStore();
+        getContext () {
+            const store = storage.getStore();
+            if (!store) {
+                throw new Error('Context not initialized');
+            }
+            return store.context;
         }
     };
 };
-export function Context(Base) {
+export function Contextable(BaseClass = class {
+}) {
     const context = createContext();
-    return Base
-        ? class Contextual extends Base {
-            static get context() {
-                return context.getContext();
-            }
-            async contextualize(callback) {
-                return context.contextualize(this, callback);
-            }
+    return class Contextual extends BaseClass {
+        static getContext() {
+            return context.getContext();
         }
-        : class Contextual {
-            static get context() {
-                return context.getContext();
-            }
-            async contextualize(callback) {
-                return context.contextualize(this, callback);
-            }
-        };
+        constructor(...args){
+            super(...args);
+        }
+        contextualize(callback) {
+            return context.contextualize(this, callback);
+        }
+    };
 }
-export function contextualize(target, _, descriptor) {
-    const method = descriptor.value;
-    function contextualized() {
-        return target.contextualize(() => method.apply(this, arguments));
+export function contextualize(...args) {
+    const [, , descriptor] = args;
+    const property = descriptor?.value;
+    if (typeof property === 'function') {
+        descriptor.value = function(...args) {
+            return this.contextualize(()=>property.apply(this, args));
+        };
     }
-    ;
-    descriptor.value = contextualized;
 }
-//# sourceMappingURL=index.js.map
+
+//# sourceMappingURL=index.js.map

package/build/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAIhD,MAAM,CAAC,MAAM,aAAa,GAAG,GAAM,EAAE;IACnC,MAAM,OAAO,GAAG,IAAI,iBAAiB,EAAK,CAAC;IAE3C,OAAO;QACL,aAAa,CAAC,KAAQ,EAAE,QAAkB;YACxC,OAAO,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBAC3C,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,IAAI,EAAE;oBAC5B,IAAI;wBACF,MAAM,QAAQ,EAAE,CAAC;wBACjB,OAAO,EAAE,CAAC;qBACX;oBAAC,OAAO,CAAC,EAAE;wBACV,MAAM,CAAC,CAAC,CAAC,CAAC;qBACX;gBACH,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;QACD,UAAU;YACR,OAAO,OAAO,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC;KACF,CAAC;AACJ,CAAC,CAAC;AAIF,MAAM,UAAU,OAAO,CAAC,IAAkB;IACxC,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;IAEhC,OAAO,IAAI;QACT,CAAC,CAAC,MAAM,UAAW,SAAQ,IAAI;YAC7B,MAAM,KAAK,OAAO;gBAChB,OAAO,OAAO,CAAC,UAAU,EAAE,CAAC;YAC9B,CAAC;YAED,KAAK,CAAC,aAAa,CAAC,QAAkB;gBACpC,OAAO,OAAO,CAAC,aAAa,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YAC/C,CAAC;SACF;QACD,CAAC,CAAC,MAAM,UAAU;YAChB,MAAM,KAAK,OAAO;gBAChB,OAAO,OAAO,CAAC,UAAU,EAAE,CAAC;YAC9B,CAAC;YAED,KAAK,CAAC,aAAa,CAAC,QAAkB;gBACpC,OAAO,OAAO,CAAC,aAAa,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;YAC/C,CAAC;SACF,CAAC;AACN,CAAC;AAMD,MAAM,UAAU,aAAa,CAA2B,MAAS,EAAE,CAAS,EAAE,UAA6C;IACzH,MAAM,MAAM,GAAG,UAAU,CAAC,KAAM,CAAC;IACjC,SAAS,cAAc;QACrB,OAAO,MAAM,CAAC,aAAa,CAAC,GAAG,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC;IACnE,CAAC;IAAA,CAAC;IACF,UAAU,CAAC,KAAK,GAAG,cAAqB,CAAC;AAC3C,CAAC"}
+{"version":3,"sources":["../src/index.ts"],"sourcesContent":["import { AsyncLocalStorage } from 'node:async_hooks';\n\nexport type Fn<T extends unknown[], R> = (...args: T) => R;\n\nexport const createContext = <T>() => {\n  const storage = new AsyncLocalStorage<{ context: T }>();\n  return {\n    contextualize<R>(context: T, callback: Fn<[], R>): R {\n      return storage.run({ context }, callback);\n    },\n    getContext() {\n      const store = storage.getStore();\n      if (!store) {\n        throw new Error('Context not initialized');\n      }\n      return store.context;\n    }\n  };\n};\n\ntype Constructor = new (...args: any) => any;\n\nexport interface Contextualized {\n  contextualize<R>(callback: Fn<[], R>): Promise<R>;\n}\n\nexport function Contextable(BaseClass: Constructor = class { }) {\n  const context = createContext();\n  return class Contextual extends BaseClass {\n    static getContext<C extends Constructor>(this: C) {\n      return context.getContext() as InstanceType<C>;\n    }\n    constructor(...args: unknown[]) {\n      super(...args);\n    }\n\n    contextualize<R>(callback: Fn<[], R>): R {\n      return context.contextualize(this, callback);\n    }\n  };\n}\n\nexport function contextualize<T extends Contextualized, K extends keyof T>(...args: unknown[]) {\n  const [, , descriptor] = args as [T, K, TypedPropertyDescriptor<T[K]>];\n  const property = descriptor?.value;\n  if (typeof property === 'function') {\n    descriptor.value = function(this: Contextualized, ...args: unknown[]) {\n      return this.contextualize(() => property.apply(this, args));\n    } as T[K];\n  }\n}\n\n"],"names":["AsyncLocalStorage","createContext","storage","contextualize","context","callback","run","getContext","store","getStore","Error","Contextable","BaseClass","Contextual","constructor","args","descriptor","property","value","apply"],"mappings":"AAAA,SAASA,iBAAiB,QAAQ,mBAAmB;AAIrD,OAAO,MAAMC,gBAAgB;IAC3B,MAAMC,UAAU,IAAIF;IACpB,OAAO;QACLG,eAAiBC,OAAU,EAAEC,QAAmB;YAC9C,OAAOH,QAAQI,GAAG,CAAC;gBAAEF;YAAQ,GAAGC;QAClC;QACAE;YACE,MAAMC,QAAQN,QAAQO,QAAQ;YAC9B,IAAI,CAACD,OAAO;gBACV,MAAM,IAAIE,MAAM;YAClB;YACA,OAAOF,MAAMJ,OAAO;QACtB;IACF;AACF,EAAE;AAQF,OAAO,SAASO,YAAYC,YAAyB;AAAQ,CAAC;IAC5D,MAAMR,UAAUH;IAChB,OAAO,MAAMY,mBAAmBD;QAC9B,OAAOL,aAA2C;YAChD,OAAOH,QAAQG,UAAU;QAC3B;QACAO,YAAY,GAAGC,IAAe,CAAE;YAC9B,KAAK,IAAIA;QACX;QAEAZ,cAAiBE,QAAmB,EAAK;YACvC,OAAOD,QAAQD,aAAa,CAAC,IAAI,EAAEE;QACrC;IACF;AACF;AAEA,OAAO,SAASF,cAA2D,GAAGY,IAAe;IAC3F,MAAM,KAAKC,WAAW,GAAGD;IACzB,MAAME,WAAWD,YAAYE;IAC7B,IAAI,OAAOD,aAAa,YAAY;QAClCD,WAAWE,KAAK,GAAG,SAA+B,GAAGH,IAAe;YAClE,OAAO,IAAI,CAACZ,aAAa,CAAC,IAAMc,SAASE,KAAK,CAAC,IAAI,EAAEJ;QACvD;IACF;AACF"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "conode",
-  "version": "0.1.23",
+  "version": "0.2.0",
   "description": "0-Deps, simple and fast context library for NodeJS",
   "type": "module",
   "main": "build/index.cjs",
@@ -11,13 +11,6 @@
   ],
   "module": "build/index.js",
   "sideEffects": false,
-  "scripts": {
-    "convert-cjs": "node scripts/convert-cjs.js",
-    "build": "rm -rf build && yarn build:cjs && yarn build:mjs",
-    "build:cjs": "tsc -p tsconfig.cjs.json && yarn convert-cjs",
-    "build:mjs": "tsc -p tsconfig.mjs.json --declaration",
-    "test": "jest"
-  },
   "repository": {
     "type": "git",
     "url": "git+https://github.com/3axap4eHko/conode.git"
@@ -39,10 +32,16 @@
   },
   "homepage": "https://github.com/3axap4eHko/conode#readme",
   "devDependencies": {
-    "@types/jest": "^27.5.1",
-    "@types/node": "^17.0.35",
-    "jest": "^28.1.0",
-    "ts-jest": "^28.0.3",
-    "typescript": "^4.7.2"
+    "@swc/jest": "^0.2.38",
+    "@types/jest": "^29.5.14",
+    "@types/node": "^22.15.18",
+    "inop": "^0.7.8",
+    "jest": "^29.7.0",
+    "typescript": "^5.8.3"
+  },
+  "scripts": {
+    "build": "rm -rf build && inop src build -i __tests__ -i *.tmp.ts && tsc --declaration --emitDeclarationOnly",
+    "test": "jest",
+    "test:build": "node src/__tests__/index.cjs && node src/__tests__/index.mjs"
   }
-}
+}

package/src/index.ts

@@ -1,63 +1,52 @@
-import { AsyncLocalStorage } from 'async_hooks';
+import { AsyncLocalStorage } from 'node:async_hooks';
 
-export type Callback = () => any;
+export type Fn<T extends unknown[], R> = (...args: T) => R;
 
 export const createContext = <T>() => {
-  const storage = new AsyncLocalStorage<T>();
-
+  const storage = new AsyncLocalStorage<{ context: T }>();
   return {
-    contextualize(value: T, callback: Callback) {
-      return new Promise<void>((resolve, reject) => {
-        storage.run(value, async () => {
-          try {
-            await callback();
-            resolve();
-          } catch (e) {
-            reject(e);
-          }
-        });
-      });
+    contextualize<R>(context: T, callback: Fn<[], R>): R {
+      return storage.run({ context }, callback);
     },
     getContext() {
-      return storage.getStore();
+      const store = storage.getStore();
+      if (!store) {
+        throw new Error('Context not initialized');
+      }
+      return store.context;
     }
   };
 };
 
-type Constructor = new (...args: any[]) => {};
+type Constructor = new (...args: any) => any;
 
-export function Context(Base?: Constructor) {
-  const context = createContext();
-
-  return Base
-    ? class Contextual extends Base {
-      static get context() {
-        return context.getContext();
-      }
+export interface Contextualized {
+  contextualize<R>(callback: Fn<[], R>): Promise<R>;
+}
 
-      async contextualize(callback: Callback) {
-        return context.contextualize(this, callback);
-      }
+export function Contextable(BaseClass: Constructor = class { }) {
+  const context = createContext();
+  return class Contextual extends BaseClass {
+    static getContext<C extends Constructor>(this: C) {
+      return context.getContext() as InstanceType<C>;
+    }
+    constructor(...args: unknown[]) {
+      super(...args);
     }
-    : class Contextual {
-      static get context() {
-        return context.getContext();
-      }
 
-      async contextualize(callback: Callback) {
-        return context.contextualize(this, callback);
-      }
-    };
+    contextualize<R>(callback: Fn<[], R>): R {
+      return context.contextualize(this, callback);
+    }
+  };
 }
 
-export interface Contextualized {
-  contextualize(callback: Callback): Promise<any>;
+export function contextualize<T extends Contextualized, K extends keyof T>(...args: unknown[]) {
+  const [, , descriptor] = args as [T, K, TypedPropertyDescriptor<T[K]>];
+  const property = descriptor?.value;
+  if (typeof property === 'function') {
+    descriptor.value = function(this: Contextualized, ...args: unknown[]) {
+      return this.contextualize(() => property.apply(this, args));
+    } as T[K];
+  }
 }
 
-export function contextualize<T extends Contextualized>(target: T, _: string, descriptor: TypedPropertyDescriptor<Callback>) {
-  const method = descriptor.value!;
-  function contextualized() {
-    return target.contextualize(() => method.apply(this, arguments));
-  };
-  descriptor.value = contextualized as any;
-}