astro-tokenkit 1.0.13 → 1.0.14

package/dist/middleware.js

@@ -15,7 +15,6 @@ import { getConfig, getTokenManager } from './config';
  */
 export function createMiddleware() {
     return (ctx, next) => __awaiter(this, void 0, void 0, function* () {
-        var _a;
         const tokenManager = getTokenManager();
         const config = getConfig();
         const runLogic = () => __awaiter(this, void 0, void 0, function* () {
@@ -32,21 +31,18 @@ export function createMiddleware() {
             }
             return next();
         });
-        // If getContextStore is defined, it means the context is managed externally (e.g., by a superior ALS)
-        // We skip runWithContext to avoid nesting ALS.run() unnecessarily,
-        // UNLESS a custom runWithContext is provided.
-        if (config.getContextStore && !config.runWithContext) {
-            let storage = config.getContextStore();
-            if (storage)
-                // Update existing reference
-                storage.cookies = ctx.cookies;
-            else if (config.setContextStore)
-                config.setContextStore({ cookies: ctx.cookies });
-            else
-                console.error("[TokenKit] getContextStore returned null or undefined and no setter was found");
+        const setupAndRun = () => __awaiter(this, void 0, void 0, function* () {
+            if (config.setContextStore) {
+                config.setContextStore(ctx);
+            }
             return runLogic();
+        });
+        if (config.runWithContext) {
+            return config.runWithContext(ctx, setupAndRun);
+        }
+        if (config.setContextStore) {
+            return setupAndRun();
         }
-        const runner = (_a = config.runWithContext) !== null && _a !== void 0 ? _a : defaultRunWithContext;
-        return runner(ctx, runLogic);
+        return defaultRunWithContext(ctx, runLogic);
     });
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "astro-tokenkit",
-  "version": "1.0.13",
+  "version": "1.0.14",
   "description": "A powerful API client for Astro with automatic token rotation, session management, and seamless context integration.",
   "main": "dist/index.js",
   "module": "dist/index.js",