astro-tokenkit 1.0.13 → 1.0.15

package/README.md

@@ -116,6 +116,7 @@ const specializedClient = createClient({
 | `interceptors`| `InterceptorsConfig` | Request/Response/Error interceptors. |
 | `context` | `AsyncLocalStorage` | External AsyncLocalStorage instance. |
 | `getContextStore`| `() => TokenKitContext`| Custom method to retrieve the context store. |
+| `setContextStore`| `(ctx) => void`| Custom method to set the context store. |
 | `runWithContext`| `Function`| Custom runner to bind context. |
 
 ### Auth Configuration

package/dist/index.cjs

@@ -826,14 +826,30 @@ function sleep(ms) {
 }
 
 // packages/astro-tokenkit/src/middleware.ts
+const LOGGED_KEY = Symbol.for('astro-tokenkit.middleware.logged');
 /**
  * Create middleware for context binding and automatic token rotation
  */
 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) {
@@ -848,22 +864,19 @@ 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 : runWithContext;
-        return runner(ctx, runLogic);
+        return runWithContext(ctx, runLogic);
     });
 }