@niledatabase/server 5.0.0-alpha.30 → 5.0.0-alpha.32

package/dist/index.mjs

@@ -8,6 +8,8 @@ var ExtensionState = /* @__PURE__ */ ((ExtensionState2) => {
   ExtensionState2["onRequest"] = "onRequest";
   ExtensionState2["onResponse"] = "onResponse";
   ExtensionState2["withContext"] = "withContext";
+  ExtensionState2["withTenantId"] = "withTenantId";
+  ExtensionState2["withUserId"] = "withUserId";
   return ExtensionState2;
 })(ExtensionState || {});
 var APIErrorErrorCodeEnum = {
@@ -220,6 +222,14 @@ function bindRunExtensions(instance) {
         if (extensionConfig.disableExtensions?.includes(ext.id)) {
           continue;
         }
+        if (ext.withTenantId && toRun === "withTenantId" /* withTenantId */) {
+          ctx.set({
+            tenantId: await ext.withTenantId()
+          });
+        }
+        if (ext.withUserId && toRun === "withUserId" /* withUserId */) {
+          ctx.set({ userId: await ext.withUserId() });
+        }
         if (ext.withContext && toRun === "withContext" /* withContext */) {
           await ext.withContext(ctx);
         }
@@ -235,10 +245,6 @@ function bindRunExtensions(instance) {
         const [param] = Array.isArray(params) ? params : [params];
         if (ext.onRequest && toRun === "onRequest" /* onRequest */) {
           const { ...previousContext } = ctx.get();
-          const preserveHeaders = previousContext.preserveHeaders;
-          if (preserveHeaders) {
-            ctx.set({ preserveHeaders: false });
-          }
           if (!_init) {
             continue;
           }
@@ -249,7 +255,7 @@ function bindRunExtensions(instance) {
             const cookie = updatedContext.headers.get("cookie");
             if (cookie && param.headers) {
               const updatedCookies = mergeCookies(
-                preserveHeaders ? previousHeaders?.get("cookie") : null,
+                previousHeaders?.get("cookie"),
                 updatedContext.headers.get("cookie")
               );
               param.headers.set("cookie", updatedCookies);
@@ -294,6 +300,11 @@ function mergeCookies(...cookieStrings) {
 }
 async function runExtensionContext(config) {
   await config?.extensionCtx?.runExtensions("withContext" /* withContext */, config);
+  await config?.extensionCtx?.runExtensions(
+    "withTenantId" /* withTenantId */,
+    config
+  );
+  await config?.extensionCtx?.runExtensions("withUserId" /* withUserId */, config);
 }
 
 // src/api/utils/request-context.ts
@@ -302,8 +313,7 @@ var storage = new AsyncLocalStorage();
 var defaultContext = {
   headers: new Headers(),
   tenantId: void 0,
-  userId: void 0,
-  preserveHeaders: false
+  userId: void 0
 };
 var lastUsedContext = defaultContext;
 var ctx = {
@@ -353,8 +363,6 @@ var ctx = {
     }
     if ("tenantId" in partial) store.tenantId = partial.tenantId;
     if ("userId" in partial) store.userId = partial.userId;
-    if ("preserveHeaders" in partial)
-      store.preserveHeaders = Boolean(partial.preserveHeaders);
     silly(`[SET] ${serializeContext(store)}`);
     lastUsedContext = { ...store };
   },
@@ -372,8 +380,7 @@ function withNileContext(config, fn, name = "unknown") {
     const context2 = {
       headers: mergedHeaders,
       tenantId: existing.tenantId,
-      userId: existing.userId,
-      preserveHeaders: existing.preserveHeaders ?? false
+      userId: existing.userId
     };
     silly(`${name} [INITIAL - Request] ${serializeContext(context2)}`);
     return ctx.run(context2, fn);
@@ -386,12 +393,10 @@ function withNileContext(config, fn, name = "unknown") {
   }
   const hasTenantId = "tenantId" in initialContext;
   const hasUserId = "userId" in initialContext;
-  const hasPreserveHeaders = "preserveHeaders" in initialContext;
   const context = {
     headers: mergedHeaders,
     tenantId: hasTenantId ? initialContext.tenantId : existing.tenantId,
-    userId: hasUserId ? initialContext.userId : existing.userId,
-    preserveHeaders: hasPreserveHeaders ? Boolean(initialContext.preserveHeaders) : existing.preserveHeaders ?? false
+    userId: hasUserId ? initialContext.userId : existing.userId
   };
   silly(`${name} [INITIAL - Partial<Context>] ${serializeContext(context)}`);
   return ctx.run(context, async () => {
@@ -408,8 +413,7 @@ function serializeContext(context) {
   return JSON.stringify({
     headers,
     tenantId: context.tenantId,
-    userId: context.userId,
-    preserveHeaders: context.preserveHeaders
+    userId: context.userId
   });
 }
 function parseCookieHeader(header) {
@@ -2014,8 +2018,7 @@ var Config = class {
     this.context = {
       tenantId: config?.tenantId,
       userId: config?.userId,
-      headers: config?.headers ? new Headers(config.headers) : new Headers(),
-      preserveHeaders: false
+      headers: config?.headers ? new Headers(config.headers) : new Headers()
     };
     this.routes = {
       ...appRoutes(config?.routePrefix),
@@ -2727,7 +2730,7 @@ var Auth = class {
         ].filter(Boolean).join("; ");
         const uHeaders = new Headers({ cookie });
         updateHeaders(uHeaders);
-        ctx.set({ headers: uHeaders, preserveHeaders: true });
+        ctx.set({ headers: uHeaders });
       } else {
         error("Unable to set context after sign in", {
           headers: signInRes.headers
@@ -2849,7 +2852,7 @@ async function obtainCsrf(config, rawResponse = false) {
         parseToken(res.headers)
       ].filter(Boolean).join("; ");
       headers.set("cookie", cookie);
-      ctx.set({ headers, preserveHeaders: true });
+      ctx.set({ headers });
       updateHeaders(headers);
     }
     if (!rawResponse) {
@@ -2868,7 +2871,7 @@ async function obtainCsrf(config, rawResponse = false) {
     }
     const cookie = cookieParts.filter(Boolean).join("; ");
     headers.set("cookie", cookie);
-    ctx.set({ headers, preserveHeaders: true });
+    ctx.set({ headers });
     updateHeaders(new Headers({ cookie }));
   }
   if (rawResponse) {
@@ -3158,12 +3161,12 @@ var Tenants = class {
       try {
         const json = await me.json();
         if ("id" in json) {
-          ctx.set({ userId: json.id, preserveHeaders: true });
+          ctx.set({ userId: json.id });
         }
       } catch {
       }
       if (typeof req === "string") {
-        ctx.set({ tenantId: req, preserveHeaders: true });
+        ctx.set({ tenantId: req });
       } else {
         this.#handleContext(req);
       }
@@ -3173,7 +3176,7 @@ var Tenants = class {
   async addMember(req, rawResponse) {
     return withNileContext(this.#config, async () => {
       if (typeof req === "string") {
-        ctx.set({ userId: req, preserveHeaders: true });
+        ctx.set({ userId: req });
       } else {
         this.#handleContext(req);
       }
@@ -3191,7 +3194,7 @@ var Tenants = class {
     return withNileContext(this.#config, async () => {
       this.#handleContext(req);
       if (typeof req === "string") {
-        ctx.set({ userId: req, preserveHeaders: true });
+        ctx.set({ userId: req });
       }
       const res = await fetchTenantUser(this.#config, "DELETE");
       return responseHandler(res, rawResponse);
@@ -3314,10 +3317,10 @@ var Tenants = class {
   #handleContext(req) {
     if (typeof req === "object") {
       if ("tenantId" in req) {
-        ctx.set({ tenantId: req.tenantId, preserveHeaders: true });
+        ctx.set({ tenantId: req.tenantId });
       }
       if ("userId" in req) {
-        ctx.set({ userId: req.userId, preserveHeaders: true });
+        ctx.set({ userId: req.userId });
       }
     }
   }
@@ -3404,7 +3407,7 @@ function updateConfig(response, config) {
     ...config,
     origin,
     headers: headers ?? void 0,
-    preserveHeaders: true
+    useLastContext: true
   };
 }
 
@@ -3436,7 +3439,6 @@ var Server = class {
     watchHeaders((headers) => {
       if (headers) {
         this.#config.context.headers = new Headers(headers);
-        this.#config.context.preserveHeaders = true;
         this.#reset();
       }
     });
@@ -3444,7 +3446,6 @@ var Server = class {
       ...this.#config.handlers,
       withContext: handlersWithContext(this.#config)
     };
-    this.#config.context.preserveHeaders = config?.preserveHeaders ?? false;
     this.#config.context.tenantId = getTenantId({ config: this.#config });
     this.#manager = new DBManager(this.#config);
     this.#handleHeaders(config);
@@ -3510,18 +3511,18 @@ var Server = class {
   set paths(paths) {
     this.#config.paths = paths;
   }
-  async withContext(context, fn) {
-    const { ...initialContext } = context ?? defaultContext;
-    this.#config.context = { ...initialContext };
-    const preserve = (context && "preserveHeaders" in context && context.preserveHeaders) ?? true;
+  async withContext(contextOrFn, maybeFn) {
+    const isFn = typeof contextOrFn === "function";
+    const context = isFn ? {} : contextOrFn ?? {};
+    const fn = isFn ? contextOrFn : maybeFn;
+    const preserve = "useLastContext" in context ? context.useLastContext : true;
     if (preserve) {
       this.#config.context = { ...this.getContext(), ...context };
+    } else {
+      this.#config.context = { ...defaultContext, ...context };
     }
     return withNileContext(this.#config, async () => {
-      if (fn) {
-        return fn(this);
-      }
-      return this;
+      return fn ? fn(this) : this;
     });
   }
   async noContext(fn) {