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

package/dist/index.mjs

@@ -3223,6 +3223,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) {
@@ -3255,19 +3268,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;
     }
@@ -3290,6 +3290,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,