astro-tokenkit 1.0.13 → 1.0.15

package/dist/middleware.js

@@ -10,14 +10,30 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 };
 import { runWithContext as defaultRunWithContext } from './client/context';
 import { getConfig, getTokenManager } from './config';
+const LOGGED_KEY = Symbol.for('astro-tokenkit.middleware.logged');
 /**
  * Create middleware for context binding and automatic token rotation
  */
 export function createMiddleware() {
     return (ctx, next) => __awaiter(this, void 0, void 0, function* () {
-        var _a;
         const tokenManager = getTokenManager();
         const config = getConfig();
+        const globalStorage = globalThis;
+        if (!globalStorage[LOGGED_KEY]) {
+            const authStatus = tokenManager ? 'enabled' : 'disabled';
+            let contextStrategy = 'default';
+            if (config.runWithContext) {
+                contextStrategy = 'custom (runWithContext)';
+            }
+            else if (config.setContextStore) {
+                contextStrategy = 'custom (getter/setter)';
+            }
+            else if (config.context) {
+                contextStrategy = 'custom (external AsyncLocalStorage)';
+            }
+            console.log(`[TokenKit] Middleware initialized (auth: ${authStatus}, context: ${contextStrategy})`);
+            globalStorage[LOGGED_KEY] = true;
+        }
         const runLogic = () => __awaiter(this, void 0, void 0, function* () {
             // Proactively ensure a valid session if auth is configured
             if (tokenManager) {
@@ -32,21 +48,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.15",
   "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",