@rpcbase/server 0.481.0 → 0.483.0

package/dist/{handler-B_mMDLBO.js → handler-BwK8qxLn.js}

@@ -1,7 +1,7 @@
-import { loadModel } from "@rpcbase/db";
+import { models } from "@rpcbase/db";
 import { buildAbilityFromSession, getAccessibleByQuery } from "@rpcbase/db/acl";
 import { createNotification, sendNotificationsDigestForUser } from "./notifications.js";
-import { o as object, b as boolean, n as number, a as array, s as string, r as record, u as unknown, _ as _enum } from "./schemas-D5T9tDtI.js";
+import { o as object, b as boolean, n as number, a as array, s as string, r as record, _ as _enum, u as unknown } from "./schemas-7qqi9OQy.js";
 const getSessionUser = (ctx) => {
   const rawSessionUser = ctx.req.session?.user;
   const userId = typeof rawSessionUser?.id === "string" ? rawSessionUser.id.trim() : "";
@@ -139,10 +139,10 @@ const listNotifications = async (payload, ctx) => {
   const unreadOnly = parsed.data.unreadOnly === true;
   const limit = parsed.data.limit ?? 50;
   const markSeen = parsed.data.markSeen === true;
-  const SettingsModel = await loadModel("RBNotificationSettings", ctx);
+  const SettingsModel = await models.get("RBNotificationSettings", ctx);
   const settings = await SettingsModel.findOne({ userId }).lean();
   const disabledTopics = buildDisabledTopics(settings, "inApp");
-  const NotificationModel = await loadModel("RBNotification", ctx);
+  const NotificationModel = await models.get("RBNotification", ctx);
   const queryFilters = [
     { userId },
     getAccessibleByQuery(ability, "read", "RBNotification")
@@ -234,7 +234,7 @@ const markRead = async (_payload, ctx) => {
     ctx.res.status(400);
     return { ok: false, error: "missing_notification_id" };
   }
-  const NotificationModel = await loadModel("RBNotification", ctx);
+  const NotificationModel = await models.get("RBNotification", ctx);
   const now = /* @__PURE__ */ new Date();
   try {
     await NotificationModel.updateOne(
@@ -257,10 +257,10 @@ const markAllRead = async (_payload, ctx) => {
     ctx.res.status(403);
     return { ok: false, error: "forbidden" };
   }
-  const SettingsModel = await loadModel("RBNotificationSettings", ctx);
+  const SettingsModel = await models.get("RBNotificationSettings", ctx);
   const settings = await SettingsModel.findOne({ userId: session.userId }).lean();
   const disabledTopics = buildDisabledTopics(settings, "inApp");
-  const NotificationModel = await loadModel("RBNotification", ctx);
+  const NotificationModel = await models.get("RBNotification", ctx);
   const queryFilters = [
     { userId: session.userId },
     { archivedAt: { $exists: false } },
@@ -288,7 +288,7 @@ const archiveNotification = async (_payload, ctx) => {
     ctx.res.status(400);
     return { ok: false, error: "missing_notification_id" };
   }
-  const NotificationModel = await loadModel("RBNotification", ctx);
+  const NotificationModel = await models.get("RBNotification", ctx);
   try {
     await NotificationModel.updateOne(
       { $and: [{ _id: notificationId }, { archivedAt: { $exists: false } }, getAccessibleByQuery(ability, "update", "RBNotification")] },
@@ -310,7 +310,7 @@ const getSettings = async (_payload, ctx) => {
     ctx.res.status(403);
     return { ok: false, error: "forbidden" };
   }
-  const SettingsModel = await loadModel("RBNotificationSettings", ctx);
+  const SettingsModel = await models.get("RBNotificationSettings", ctx);
   const settings = await SettingsModel.findOne(
     { $and: [{ userId: session.userId }, getAccessibleByQuery(ability, "read", "RBNotificationSettings")] }
   ).lean();
@@ -346,7 +346,7 @@ const updateSettings = async (payload, ctx) => {
     ctx.res.status(400);
     return { ok: false, error: "invalid_payload" };
   }
-  const SettingsModel = await loadModel("RBNotificationSettings", ctx);
+  const SettingsModel = await models.get("RBNotificationSettings", ctx);
   const nextValues = {};
   if (parsed.data.digestFrequency) {
     nextValues.digestFrequency = parsed.data.digestFrequency;

package/dist/{handler-Dd20DHyz.js → handler-CedzJJg0.js}

@@ -1,6 +1,6 @@
-import { loadModel, ZRBRtsChangeOp } from "@rpcbase/db";
+import { models, ZRBRtsChangeOp } from "@rpcbase/db";
 import { buildAbilityFromSession } from "@rpcbase/db/acl";
-import { o as object, a as array, s as string, n as number, b as boolean, _ as _enum } from "./schemas-D5T9tDtI.js";
+import { o as object, a as array, s as string, n as number, b as boolean, _ as _enum } from "./schemas-7qqi9OQy.js";
 const Route = "/api/rb/rts/changes";
 const requestSchema = object({
   sinceSeq: number().int().min(0).default(0),
@@ -72,8 +72,8 @@ const changesHandler = async (payload, ctx) => {
   const ability = buildAbilityFromSession({ tenantId, session: ctx.req.session });
   const modelCtx = getModelCtx(ctx, tenantId);
   const [RtsChange, RtsCounter] = await Promise.all([
-    loadModel("RBRtsChange", modelCtx),
-    loadModel("RBRtsCounter", modelCtx)
+    models.get("RBRtsChange", modelCtx),
+    models.get("RBRtsCounter", modelCtx)
   ]);
   const counter = await RtsCounter.findOne({ _id: "rts" }, { seq: 1 }).lean();
   const latestSeq = Number(counter?.seq ?? 0) || 0;

package/dist/{handler-BA2YiqnG.js → handler-qCAUmVgd.js}

@@ -1,10 +1,10 @@
-import { loadModel, getTenantFilesystemDb } from "@rpcbase/db";
+import { models, getTenantFilesystemDb } from "@rpcbase/db";
 import { GridFSBucket, ObjectId } from "mongodb";
 import { JSDOM } from "jsdom";
 import createDOMPurify from "dompurify";
 import { g as getTenantId, a as getModelCtx, b as buildUploadsAbility, c as getUploadSessionAccessQuery, e as ensureUploadIndexes, d as getBucketName, f as getUserId, h as getChunkSizeBytes, i as getSessionTtlMs, j as computeSha256Hex, t as toBufferPayload, n as normalizeSha256Hex, k as getMaxClientUploadBytesPerSecond, l as getRawBodyLimitBytes } from "./shared-BJomDDWK.js";
 import { randomBytes } from "node:crypto";
-import { o as object, n as number, b as boolean, s as string, a as array, _ as _enum } from "./schemas-D5T9tDtI.js";
+import { o as object, n as number, b as boolean, s as string, a as array, _ as _enum } from "./schemas-7qqi9OQy.js";
 const MAX_SVG_BYTES = 128 * 1024;
 const window = new JSDOM("").window;
 const DOMPurify = createDOMPurify(window);
@@ -124,8 +124,8 @@ const completeUpload = async (_payload, ctx) => {
   }
   const modelCtx = getModelCtx(ctx, tenantId);
   const [UploadSession, UploadChunk] = await Promise.all([
-    loadModel("RBUploadSession", modelCtx),
-    loadModel("RBUploadChunk", modelCtx)
+    models.get("RBUploadSession", modelCtx),
+    models.get("RBUploadChunk", modelCtx)
   ]);
   const ability = buildUploadsAbility(ctx, tenantId);
   if (!ability.can("update", "RBUploadSession")) {
@@ -352,8 +352,8 @@ const getStatus = async (_payload, ctx) => {
   }
   const modelCtx = getModelCtx(ctx, tenantId);
   const [UploadSession, UploadChunk] = await Promise.all([
-    loadModel("RBUploadSession", modelCtx),
-    loadModel("RBUploadChunk", modelCtx)
+    models.get("RBUploadSession", modelCtx),
+    models.get("RBUploadChunk", modelCtx)
   ]);
   const ability = buildUploadsAbility(ctx, tenantId);
   if (!ability.can("read", "RBUploadSession")) {
@@ -428,8 +428,8 @@ const initUpload = async (payload, ctx) => {
   const chunksTotal = Math.ceil(totalSize / chunkSize);
   const modelCtx = getModelCtx(ctx, tenantId);
   const [UploadSession, UploadChunk] = await Promise.all([
-    loadModel("RBUploadSession", modelCtx),
-    loadModel("RBUploadChunk", modelCtx)
+    models.get("RBUploadSession", modelCtx),
+    models.get("RBUploadChunk", modelCtx)
   ]);
   await ensureUploadIndexes(UploadSession, UploadChunk);
   const uploadId = new ObjectId().toString();
@@ -474,8 +474,8 @@ const uploadChunk = async (payload, ctx) => {
   }
   const modelCtx = getModelCtx(ctx, tenantId);
   const [UploadSession, UploadChunk] = await Promise.all([
-    loadModel("RBUploadSession", modelCtx),
-    loadModel("RBUploadChunk", modelCtx)
+    models.get("RBUploadSession", modelCtx),
+    models.get("RBUploadChunk", modelCtx)
   ]);
   const ability = buildUploadsAbility(ctx, tenantId);
   if (!ability.can("update", "RBUploadSession")) {

package/dist/index.js

@@ -2096,7 +2096,7 @@ class ErrorTracking {
     this._rateLimiter.stop();
   }
 }
-const version = "5.18.0";
+const version = "5.19.0";
 const FeatureFlagError = {
   ERRORS_WHILE_COMPUTING: "errors_while_computing_flags",
   FLAG_MISSING: "flag_missing",
@@ -2374,6 +2374,7 @@ class FeatureFlagsPoller {
   }
   async loadFeatureFlags(forceReload = false) {
     if (this.loadedSuccessfullyOnce && !forceReload) return;
+    if (!forceReload && this.nextFetchAllowedAt && Date.now() < this.nextFetchAllowedAt) return void this.logMsgIfDebug(() => console.debug("[FEATURE FLAGS] Skipping fetch, in backoff period"));
     if (!this.loadingPromise) this.loadingPromise = this._loadFeatureFlags().catch((err) => this.logMsgIfDebug(() => console.debug(`[FEATURE FLAGS] Failed to load feature flags: ${err}`))).finally(() => {
       this.loadingPromise = void 0;
     });
@@ -2386,6 +2387,16 @@ class FeatureFlagsPoller {
     if (!this.shouldBeginExponentialBackoff) return this.pollingInterval;
     return Math.min(SIXTY_SECONDS, this.pollingInterval * 2 ** this.backOffCount);
   }
+  beginBackoff() {
+    this.shouldBeginExponentialBackoff = true;
+    this.backOffCount += 1;
+    this.nextFetchAllowedAt = Date.now() + this.getPollingInterval();
+  }
+  clearBackoff() {
+    this.shouldBeginExponentialBackoff = false;
+    this.backOffCount = 0;
+    this.nextFetchAllowedAt = void 0;
+  }
   async _loadFeatureFlags() {
     if (this.poller) {
       clearTimeout(this.poller);
@@ -2411,12 +2422,10 @@ class FeatureFlagsPoller {
           this.logMsgIfDebug(() => console.debug("[FEATURE FLAGS] Flags not modified (304), using cached data"));
           this.flagsEtag = res.headers?.get("ETag") ?? this.flagsEtag;
           this.loadedSuccessfullyOnce = true;
-          this.shouldBeginExponentialBackoff = false;
-          this.backOffCount = 0;
+          this.clearBackoff();
           return;
         case 401:
-          this.shouldBeginExponentialBackoff = true;
-          this.backOffCount += 1;
+          this.beginBackoff();
           throw new ClientError(`Your project key or personal API key is invalid. Setting next polling interval to ${this.getPollingInterval()}ms. More information: https://posthog.com/docs/api#rate-limiting`);
         case 402:
           console.warn("[FEATURE FLAGS] Feature flags quota limit exceeded - unsetting all local flags. Learn more about billing limits at https://posthog.com/docs/billing/limits-alerts");
@@ -2426,12 +2435,10 @@ class FeatureFlagsPoller {
           this.cohorts = {};
           return;
         case 403:
-          this.shouldBeginExponentialBackoff = true;
-          this.backOffCount += 1;
+          this.beginBackoff();
           throw new ClientError(`Your personal API key does not have permission to fetch feature flag definitions for local evaluation. Setting next polling interval to ${this.getPollingInterval()}ms. Are you sure you're using the correct personal and Project API key pair? More information: https://posthog.com/docs/api/overview`);
         case 429:
-          this.shouldBeginExponentialBackoff = true;
-          this.backOffCount += 1;
+          this.beginBackoff();
           throw new ClientError(`You are being rate limited. Setting next polling interval to ${this.getPollingInterval()}ms. More information: https://posthog.com/docs/api#rate-limiting`);
         case 200: {
           const responseJson = await res.json() ?? {};
@@ -2443,8 +2450,7 @@ class FeatureFlagsPoller {
             cohorts: responseJson.cohorts || {}
           };
           this.updateFlagState(flagData);
-          this.shouldBeginExponentialBackoff = false;
-          this.backOffCount = 0;
+          this.clearBackoff();
           if (this.cacheProvider && shouldFetch) try {
             await this.cacheProvider.onFlagDefinitionsReceived(flagData);
           } catch (err) {
@@ -2812,6 +2818,7 @@ class PostHogBackendClient extends PostHogCoreStateless {
     });
   }
   async getFeatureFlag(key, distinctId, options) {
+    if (void 0 !== this._flagOverrides && key in this._flagOverrides) return this._flagOverrides[key];
     const { groups, disableGeoip } = options || {};
     let { onlyEvaluateLocally, sendFeatureFlagEvents, personProperties, groupProperties } = options || {};
     const adjustedProperties = this.addLocalPersonAndGroupProperties(distinctId, groups, personProperties, groupProperties);
@@ -2872,6 +2879,7 @@ class PostHogBackendClient extends PostHogCoreStateless {
     return response;
   }
   async getFeatureFlagPayload(key, distinctId, matchValue, options) {
+    if (void 0 !== this._payloadOverrides && key in this._payloadOverrides) return this._payloadOverrides[key];
     const { groups, disableGeoip } = options || {};
     let { onlyEvaluateLocally, personProperties, groupProperties } = options || {};
     const adjustedProperties = this.addLocalPersonAndGroupProperties(distinctId, groups, personProperties, groupProperties);
@@ -2945,6 +2953,14 @@ class PostHogBackendClient extends PostHogCoreStateless {
         ...remoteEvaluationResult.payloads || {}
       };
     }
+    if (void 0 !== this._flagOverrides) featureFlags = {
+      ...featureFlags,
+      ...this._flagOverrides
+    };
+    if (void 0 !== this._payloadOverrides) featureFlagPayloads = {
+      ...featureFlagPayloads,
+      ...this._payloadOverrides
+    };
     return {
       featureFlags,
       featureFlagPayloads
@@ -2958,6 +2974,53 @@ class PostHogBackendClient extends PostHogCoreStateless {
   async reloadFeatureFlags() {
     await this.featureFlagsPoller?.loadFeatureFlags(true);
   }
+  overrideFeatureFlags(overrides) {
+    const flagArrayToRecord = (flags) => Object.fromEntries(flags.map((f) => [
+      f,
+      true
+    ]));
+    if (false === overrides) {
+      this._flagOverrides = void 0;
+      this._payloadOverrides = void 0;
+      return;
+    }
+    if (Array.isArray(overrides)) {
+      this._flagOverrides = flagArrayToRecord(overrides);
+      return;
+    }
+    if (this._isFeatureFlagOverrideOptions(overrides)) {
+      if ("flags" in overrides) {
+        if (false === overrides.flags) this._flagOverrides = void 0;
+        else if (Array.isArray(overrides.flags)) this._flagOverrides = flagArrayToRecord(overrides.flags);
+        else if (void 0 !== overrides.flags) this._flagOverrides = {
+          ...overrides.flags
+        };
+      }
+      if ("payloads" in overrides) {
+        if (false === overrides.payloads) this._payloadOverrides = void 0;
+        else if (void 0 !== overrides.payloads) this._payloadOverrides = {
+          ...overrides.payloads
+        };
+      }
+      return;
+    }
+    this._flagOverrides = {
+      ...overrides
+    };
+  }
+  _isFeatureFlagOverrideOptions(overrides) {
+    if ("object" != typeof overrides || null === overrides || Array.isArray(overrides)) return false;
+    const obj = overrides;
+    if ("flags" in obj) {
+      const flagsValue = obj["flags"];
+      if (false === flagsValue || Array.isArray(flagsValue) || "object" == typeof flagsValue && null !== flagsValue) return true;
+    }
+    if ("payloads" in obj) {
+      const payloadsValue = obj["payloads"];
+      if (false === payloadsValue || "object" == typeof payloadsValue && null !== payloadsValue) return true;
+    }
+    return false;
+  }
   withContext(data, fn, options) {
     if (!this.context) return fn();
     return this.context.run(data, fn, options);

package/dist/notifications.js

@@ -1,7 +1,7 @@
-import { loadModel, loadRbModel } from "@rpcbase/db";
+import { models } from "@rpcbase/db";
 import { s as sendEmail } from "./email-DEw8keax.js";
 const routes = Object.entries({
-  .../* @__PURE__ */ Object.assign({ "./api/notifications/handler.ts": () => import("./handler-B_mMDLBO.js") })
+  .../* @__PURE__ */ Object.assign({ "./api/notifications/handler.ts": () => import("./handler-BwK8qxLn.js") })
 }).reduce((acc, [path, mod]) => {
   acc[path.replace("./api/", "@rpcbase/server/notifications/api/")] = mod;
   return acc;
@@ -18,7 +18,7 @@ const createNotification = async (ctx, input) => {
   const topic = typeof input.topic === "string" ? input.topic.trim() : "";
   const body = typeof input.body === "string" ? input.body.trim() : "";
   const url = typeof input.url === "string" ? input.url.trim() : "";
-  const NotificationModel = await loadModel("RBNotification", ctx);
+  const NotificationModel = await models.get("RBNotification", ctx);
   const doc = await NotificationModel.create({
     userId,
     ...topic ? { topic } : {},
@@ -57,8 +57,8 @@ const sendNotificationsDigestForUser = async (ctx, {
   userId,
   force = false
 }) => {
-  const SettingsModel = await loadModel("RBNotificationSettings", ctx);
-  const NotificationModel = await loadModel("RBNotification", ctx);
+  const SettingsModel = await models.get("RBNotificationSettings", ctx);
+  const NotificationModel = await models.get("RBNotification", ctx);
   const settings = await SettingsModel.findOne({ userId }).lean();
   const digestFrequencyRaw = typeof settings?.digestFrequency === "string" ? settings.digestFrequency : "weekly";
   const digestFrequency = digestFrequencyRaw === "daily" || digestFrequencyRaw === "weekly" || digestFrequencyRaw === "off" ? digestFrequencyRaw : "weekly";
@@ -92,7 +92,7 @@ const sendNotificationsDigestForUser = async (ctx, {
     );
     return { ok: true, sent: false, skippedReason: "empty" };
   }
-  const UserModel = await loadRbModel("RBUser", ctx);
+  const UserModel = await models.getGlobal("RBUser", ctx);
   const user = await UserModel.findById(userId, { email: 1 }).lean();
   const email = typeof user?.email === "string" ? user.email.trim() : "";
   if (!email) {

package/dist/rts/index.js

@@ -1,9 +1,9 @@
 import { randomUUID } from "node:crypto";
-import { loadRbModel, loadModel } from "@rpcbase/db";
+import { models } from "@rpcbase/db";
 import { buildAbilityFromSession, getTenantRolesFromSessionUser, buildAbility, getAccessibleByQuery } from "@rpcbase/db/acl";
 import { WebSocketServer } from "ws";
 const routes = Object.entries({
-  .../* @__PURE__ */ Object.assign({ "./api/changes/handler.ts": () => import("../handler-Dd20DHyz.js") })
+  .../* @__PURE__ */ Object.assign({ "./api/changes/handler.ts": () => import("../handler-CedzJJg0.js") })
 }).reduce((acc, [path, mod]) => {
   acc[path.replace("./api/", "@rpcbase/server/rts/api/")] = mod;
   return acc;
@@ -166,7 +166,7 @@ const parseUpgradeMeta = async ({
     throw new Error("Missing rb-user-id header (reverse-proxy) and no session middleware configured");
   }
   const rbCtx = { req: { session: null } };
-  const User = await loadRbModel("RBUser", rbCtx);
+  const User = await models.getGlobal("RBUser", rbCtx);
   const user = await User.findById(headerUserId, { tenants: 1, tenantRoles: 1 }).lean();
   const tenantsRaw = user?.tenants;
   const tenants = Array.isArray(tenantsRaw) ? tenantsRaw.map((t) => String(t)) : [];
@@ -187,7 +187,7 @@ const getTenantModel = async (tenantId, modelName) => {
       }
     }
   };
-  return loadModel(modelName, ctx);
+  return models.get(modelName, ctx);
 };
 const normalizeLimit = (limit) => {
   if (typeof limit !== "number") return QUERY_MAX_LIMIT;