@niledatabase/server 5.0.0-alpha.16 → 5.0.0-alpha.17

package/dist/index.d.mts

@@ -557,12 +557,7 @@ declare class Server {
         tenantId?: string;
         userId?: string;
     }, ...remaining: unknown[]) => void;
-    getContext(): {
-        headers: Headers | undefined;
-        userId: string | null | undefined;
-        tenantId: string | null | undefined;
-        preserveHeaders: boolean;
-    };
+    getContext(): any;
 }
 declare function create(config?: NileConfig): Server;
 
@@ -578,6 +573,7 @@ type ExtensionResult = {
     onHandleRequest?: (...params: Any) => RouteReturn | Promise<RouteReturn>;
     onConfigure?: (...params: Any) => void;
     onSetContext?: (...params: Any) => void;
+    onGetContext?: () => Any;
 };
 type Extension = (instance: Server) => ExtensionResult;
 declare enum ExtensionState {

package/dist/index.d.ts

@@ -557,12 +557,7 @@ declare class Server {
         tenantId?: string;
         userId?: string;
     }, ...remaining: unknown[]) => void;
-    getContext(): {
-        headers: Headers | undefined;
-        userId: string | null | undefined;
-        tenantId: string | null | undefined;
-        preserveHeaders: boolean;
-    };
+    getContext(): any;
 }
 declare function create(config?: NileConfig): Server;
 
@@ -578,6 +573,7 @@ type ExtensionResult = {
     onHandleRequest?: (...params: Any) => RouteReturn | Promise<RouteReturn>;
     onConfigure?: (...params: Any) => void;
     onSetContext?: (...params: Any) => void;
+    onGetContext?: () => Any;
 };
 type Extension = (instance: Server) => ExtensionResult;
 declare enum ExtensionState {

package/dist/index.js

@@ -3229,6 +3229,19 @@ var Server = class {
    * @returns undefined
    */
   setContext = (req, ...remaining) => {
+    let ok = false;
+    if (req && typeof req === "object" && "tenantId" in req) {
+      ok = true;
+      this.#config.tenantId = req.tenantId;
+    }
+    if (req && typeof req === "object" && "userId" in req) {
+      ok = true;
+      this.#config.userId = req.userId;
+    }
+    if (req && typeof req === "object" && "preserveHeaders" in req) {
+      ok = true;
+      this.#preserveHeaders = Boolean(req.preserveHeaders);
+    }
     let atLeastOne = false;
     if (this.#config?.extensions) {
       for (const create2 of this.#config.extensions) {
@@ -3261,19 +3274,6 @@ var Server = class {
       }
     } catch {
     }
-    let ok = false;
-    if (req && typeof req === "object" && "tenantId" in req) {
-      ok = true;
-      this.#config.tenantId = req.tenantId;
-    }
-    if (req && typeof req === "object" && "userId" in req) {
-      ok = true;
-      this.#config.userId = req.userId;
-    }
-    if (req && typeof req === "object" && "preserveHeaders" in req) {
-      ok = true;
-      this.#preserveHeaders = Boolean(req.preserveHeaders);
-    }
     if (ok) {
       return;
     }
@@ -3296,6 +3296,17 @@ var Server = class {
     }
   };
   getContext() {
+    if (this.#config?.extensions) {
+      for (const create2 of this.#config.extensions) {
+        if (typeof create2 !== "function") {
+          continue;
+        }
+        const ext = create2(this);
+        if (typeof ext.onGetContext === "function") {
+          return ext.onGetContext();
+        }
+      }
+    }
     return {
       headers: this.#headers,
       userId: this.#config?.userId,