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

package/dist/index.d.mts

@@ -534,14 +534,16 @@ declare class Server {
     get handlers(): NileHandlers;
     get paths(): ConfigurablePaths;
     set paths(paths: ConfigurablePaths);
-    /** Allows setting of context outside of the request lifecycle
-     * Basically means you want to disregard cookies and do everything manually
-     * If we elect to DDL, we don't want to use tenant id or user id, so remove those.
+    /**
+     * Sets the context for a particular set of requests or db calls to be sure the context is fully managed for the entire lifecycle
      */
-    withContext(context?: PartialContext): Promise<this>;
-    withContext<T>(context: PartialContext, fn: (sdk: this) => Promise<T>): Promise<T>;
+    withContext(): Promise<this>;
+    withContext<T>(context: PartialContext, fn: AsyncCallback<this, T>): Promise<T>;
+    withContext(context: PartialContext): Promise<this>;
+    withContext<T>(fn: AsyncCallback<this, T>): Promise<T>;
     /**
      * Creates a context without a user id and a tenant id, but keeps the headers around for auth at least.
+     * This is useful for DDL/DML, since most extensions will set the context by default
      */
     noContext(): Promise<this>;
     noContext<T>(fn: (sdk: this) => Promise<T>): Promise<T>;
@@ -552,6 +554,7 @@ declare class Server {
     getContext(): Context;
 }
 declare function create<T = Server>(config?: NileConfig): T;
+type AsyncCallback<TInstance, TResult> = (sdk: TInstance) => Promise<TResult>;
 
 type Opts = {
     basePath?: string;
@@ -561,13 +564,12 @@ type Context = {
     headers: Headers;
     tenantId: string | undefined | null;
     userId: string | undefined | null;
-    preserveHeaders: boolean;
 };
 type PartialContext = {
     headers?: null | Headers;
     tenantId?: string | undefined | null;
     userId?: string | undefined | null;
-    preserveHeaders?: boolean;
+    useLastContext?: boolean;
 };
 type CTX = {
     run: <T>(ctx: Partial<Context>, fn: () => T) => T;
@@ -583,6 +585,8 @@ type ExtensionResult<TParams> = {
     onResponse?: (params: TParams, ctx: CTX) => void | Promise<void>;
     onHandleRequest?: (params?: TParams) => RouteReturn | Promise<RouteReturn>;
     onConfigure?: (params?: TParams) => void;
+    withUserId?: () => string;
+    withTenantId?: () => string;
     replace?: {
         handlers: (handlers: NileHandlers) => Any;
     };
@@ -595,7 +599,9 @@ declare enum ExtensionState {
     onHandleRequest = "onHandleRequest",
     onRequest = "onRequest",
     onResponse = "onResponse",
-    withContext = "withContext"
+    withContext = "withContext",
+    withTenantId = "withTenantId",
+    withUserId = "withUserId"
 }
 type NilePoolConfig = PoolConfig & {
     afterCreate?: AfterCreate;
@@ -692,9 +698,9 @@ type NileConfig = {
     /** Hooks executed before and after each request. */
     extensions?: Extension[];
     /**
-     * Preserve incoming request headers when running extensions.
+     * Re-use the last set context
      */
-    preserveHeaders?: boolean;
+    useLastContext?: boolean;
 };
 type NileDb = NilePoolConfig & {
     tenantId?: string;

package/dist/index.d.ts

@@ -534,14 +534,16 @@ declare class Server {
     get handlers(): NileHandlers;
     get paths(): ConfigurablePaths;
     set paths(paths: ConfigurablePaths);
-    /** Allows setting of context outside of the request lifecycle
-     * Basically means you want to disregard cookies and do everything manually
-     * If we elect to DDL, we don't want to use tenant id or user id, so remove those.
+    /**
+     * Sets the context for a particular set of requests or db calls to be sure the context is fully managed for the entire lifecycle
      */
-    withContext(context?: PartialContext): Promise<this>;
-    withContext<T>(context: PartialContext, fn: (sdk: this) => Promise<T>): Promise<T>;
+    withContext(): Promise<this>;
+    withContext<T>(context: PartialContext, fn: AsyncCallback<this, T>): Promise<T>;
+    withContext(context: PartialContext): Promise<this>;
+    withContext<T>(fn: AsyncCallback<this, T>): Promise<T>;
     /**
      * Creates a context without a user id and a tenant id, but keeps the headers around for auth at least.
+     * This is useful for DDL/DML, since most extensions will set the context by default
      */
     noContext(): Promise<this>;
     noContext<T>(fn: (sdk: this) => Promise<T>): Promise<T>;
@@ -552,6 +554,7 @@ declare class Server {
     getContext(): Context;
 }
 declare function create<T = Server>(config?: NileConfig): T;
+type AsyncCallback<TInstance, TResult> = (sdk: TInstance) => Promise<TResult>;
 
 type Opts = {
     basePath?: string;
@@ -561,13 +564,12 @@ type Context = {
     headers: Headers;
     tenantId: string | undefined | null;
     userId: string | undefined | null;
-    preserveHeaders: boolean;
 };
 type PartialContext = {
     headers?: null | Headers;
     tenantId?: string | undefined | null;
     userId?: string | undefined | null;
-    preserveHeaders?: boolean;
+    useLastContext?: boolean;
 };
 type CTX = {
     run: <T>(ctx: Partial<Context>, fn: () => T) => T;
@@ -583,6 +585,8 @@ type ExtensionResult<TParams> = {
     onResponse?: (params: TParams, ctx: CTX) => void | Promise<void>;
     onHandleRequest?: (params?: TParams) => RouteReturn | Promise<RouteReturn>;
     onConfigure?: (params?: TParams) => void;
+    withUserId?: () => string;
+    withTenantId?: () => string;
     replace?: {
         handlers: (handlers: NileHandlers) => Any;
     };
@@ -595,7 +599,9 @@ declare enum ExtensionState {
     onHandleRequest = "onHandleRequest",
     onRequest = "onRequest",
     onResponse = "onResponse",
-    withContext = "withContext"
+    withContext = "withContext",
+    withTenantId = "withTenantId",
+    withUserId = "withUserId"
 }
 type NilePoolConfig = PoolConfig & {
     afterCreate?: AfterCreate;
@@ -692,9 +698,9 @@ type NileConfig = {
     /** Hooks executed before and after each request. */
     extensions?: Extension[];
     /**
-     * Preserve incoming request headers when running extensions.
+     * Re-use the last set context
      */
-    preserveHeaders?: boolean;
+    useLastContext?: boolean;
 };
 type NileDb = NilePoolConfig & {
     tenantId?: string;

package/dist/index.js

@@ -14,6 +14,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 = {
@@ -226,6 +228,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);
         }
@@ -241,10 +251,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;
           }
@@ -255,7 +261,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);
@@ -300,6 +306,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
@@ -308,8 +319,7 @@ var storage = new async_hooks.AsyncLocalStorage();
 var defaultContext = {
   headers: new Headers(),
   tenantId: void 0,
-  userId: void 0,
-  preserveHeaders: false
+  userId: void 0
 };
 var lastUsedContext = defaultContext;
 var ctx = {
@@ -359,8 +369,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 };
   },
@@ -378,8 +386,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);
@@ -392,12 +399,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 () => {
@@ -414,8 +419,7 @@ function serializeContext(context) {
   return JSON.stringify({
     headers,
     tenantId: context.tenantId,
-    userId: context.userId,
-    preserveHeaders: context.preserveHeaders
+    userId: context.userId
   });
 }
 function parseCookieHeader(header) {
@@ -2020,8 +2024,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),
@@ -2733,7 +2736,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
@@ -2855,7 +2858,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) {
@@ -2874,7 +2877,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) {
@@ -3164,12 +3167,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);
       }
@@ -3179,7 +3182,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);
       }
@@ -3197,7 +3200,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);
@@ -3320,10 +3323,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 });
       }
     }
   }
@@ -3410,7 +3413,7 @@ function updateConfig(response, config) {
     ...config,
     origin,
     headers: headers ?? void 0,
-    preserveHeaders: true
+    useLastContext: true
   };
 }
 
@@ -3442,7 +3445,6 @@ var Server = class {
     watchHeaders((headers) => {
       if (headers) {
         this.#config.context.headers = new Headers(headers);
-        this.#config.context.preserveHeaders = true;
         this.#reset();
       }
     });
@@ -3450,7 +3452,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);
@@ -3516,18 +3517,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) {