@forklaunch/core 1.6.6 → 2.0.0

package/lib/persistence/index.mjs

@@ -175,15 +175,48 @@ var _tenantContext = new AsyncLocalStorage();
 function registerEncryptor(encryptor) {
   _encryptor = encryptor;
 }
+var EmptyTenantError = class extends Error {
+  constructor(api) {
+    super(
+      `${api}: '' is not a tenant id. It used to mean "global", which made a deliberate global row indistinguishable from a tenant nobody resolved. Bind an explicit constant (for example '_internal') for rows that genuinely belong to no organization.`
+    );
+    this.name = "EmptyTenantError";
+  }
+};
+var UnboundTenantError = class extends Error {
+  constructor(operation) {
+    super(
+      `Cannot ${operation} an encrypted column with no tenant bound. Wrap the work in withEncryptionContext(tenantId, fn), or take the entity manager from a factory that calls wrapEmWithTenantContext. A raw em.fork() (or wrap(entity).__em) escapes the tenant proxy.`
+    );
+    this.name = "UnboundTenantError";
+  }
+};
+function assertBindableTenantId(tenantId, api) {
+  if (tenantId === "") {
+    throw new EmptyTenantError(api);
+  }
+}
 function setEncryptionTenantId(tenantId) {
+  assertBindableTenantId(tenantId, "setEncryptionTenantId");
   _tenantContext.enterWith({ tenantId });
 }
 function withEncryptionContext(tenantId, fn) {
+  assertBindableTenantId(tenantId, "withEncryptionContext");
   return _tenantContext.run({ tenantId }, fn);
 }
 function getCurrentTenantId() {
   return _tenantContext.getStore()?.tenantId ?? "";
 }
+function getBoundTenantId() {
+  return _tenantContext.getStore()?.tenantId;
+}
+function requireBoundTenantId(operation) {
+  const tenantId = _tenantContext.getStore()?.tenantId;
+  if (tenantId === void 0 || tenantId === "") {
+    throw new UnboundTenantError(operation);
+  }
+  return tenantId;
+}
 function resolveTypeInstance(type) {
   if (type instanceof Type) return type;
   if (typeof type === "function") {
@@ -257,7 +290,7 @@ var EncryptedType = class extends Type {
       return value;
     }
     const serialized = this.serialize(value);
-    return _encryptor.encrypt(serialized, getCurrentTenantId()) ?? serialized;
+    return _encryptor.encrypt(serialized, requireBoundTenantId("write")) ?? serialized;
   }
   convertToJSValue(value, _platform) {
     if (value === null || value === void 0) return null;
@@ -271,7 +304,7 @@ var EncryptedType = class extends Type {
       );
     }
     let decrypted;
-    const tenantId = getCurrentTenantId();
+    const tenantId = requireBoundTenantId("read");
     try {
       decrypted = _encryptor.decrypt(value, tenantId);
     } catch (err) {
@@ -1197,9 +1230,8 @@ function wrapEmWithTenantContext(em, tenantId) {
   if (tenantId === void 0) {
     return em;
   }
-  if (tenantId) {
-    em.setFilterParams("tenant", { tenantId });
-  }
+  assertBindableTenantId(tenantId, "wrapEmWithTenantContext");
+  em.setFilterParams("tenant", { tenantId });
   const proxy = new Proxy(em, {
     get(target, prop, receiver) {
       const value = Reflect.get(target, prop, receiver);
@@ -1211,23 +1243,14 @@ function wrapEmWithTenantContext(em, tenantId) {
           tenantId,
           () => value.apply(target, args)
         );
-        return rebind(result, target, proxy, tenantId);
+        return rebind(result, target, proxy);
       };
     }
   });
   return proxy;
 }
-function rebind(result, target, proxy, tenantId) {
-  if (result === target) {
-    return proxy;
-  }
-  if (isEntityManager(result) && result !== proxy) {
-    return wrapEmWithTenantContext(result, tenantId);
-  }
-  return result;
-}
-function isEntityManager(value) {
-  return typeof value === "object" && value !== null && "fork" in value && typeof value.fork === "function" && "flush" in value && typeof value.flush === "function" && "persist" in value && typeof value.persist === "function";
+function rebind(result, target, proxy) {
+  return result === target ? proxy : result;
 }
 export {
   ComplianceLevel,
@@ -1235,6 +1258,7 @@ export {
   ENCRYPTED_PREFIXES2 as ENCRYPTED_PREFIXES,
   ENCRYPTION_FORMAT_ENV,
   ENCRYPTION_KEY_ENV,
+  EmptyTenantError,
   EncryptedType,
   EncryptionRequiredError,
   FieldEncryptor,
@@ -1245,7 +1269,9 @@ export {
   RetentionDuration,
   RlsEventSubscriber,
   TENANT_FILTER_NAME,
+  UnboundTenantError,
   asEncryptionSafe,
+  assertBindableTenantId,
   classifyEncryptedValue,
   collectFallbackTenantIds,
   countValuesByKeyId,
@@ -1258,6 +1284,7 @@ export {
   fp,
   getAllRetentionPolicies,
   getAllUserIdFields,
+  getBoundTenantId,
   getComplianceMetadata,
   getCurrentTenantId,
   getEntityComplianceFields,