@hiliosai/sdk 0.1.19 → 0.1.20

package/dist/index.d.ts

@@ -226,7 +226,8 @@ declare class PrismaDatasource<TPrismaClient extends PrismaClientLike = PrismaCl
     protected createClient(): TPrismaClient;
     /**
      * Apply extensions to the Prisma client
-     * Override this method to add production extensions like soft delete, audit trails, etc.
+     * By default, applies tenant extension for multi-tenant support
+     * Override this method to add additional extensions like soft delete, audit trails, etc.
      */
     protected applyExtensions(client: TPrismaClient): TPrismaClient;
     /**
@@ -340,6 +341,13 @@ declare function createTenantExtension(): {
         $setTenant(tenantId: string): void;
         $getCurrentTenant(): string | null;
         $clearTenant(): void;
+        $enableBypassMode(): void;
+        $disableBypassMode(): void;
+        $enableStrictMode(): void;
+        $disableStrictMode(): void;
+        $withBypassMode<T>(fn: () => Promise<T>): Promise<T>;
+        $withTenant<T>(tenantId: string, fn: () => Promise<T>): Promise<T>;
+        $withSystemMode<T>(fn: () => Promise<T>): Promise<T>;
     };
     query: {
         $allModels: {

package/dist/index.js

@@ -1416,6 +1416,154 @@ var AbstractDatasource = class {
   }
 };
 
+// src/datasources/extensions/tenant.extension.ts
+function createTenantExtension() {
+  const tenantContext = {
+    currentTenantId: null,
+    bypassMode: false,
+    strictMode: false
+  };
+  return {
+    name: "Tenant",
+    client: {
+      // Set tenant context
+      $setTenant(tenantId) {
+        tenantContext.currentTenantId = tenantId;
+        tenantContext.bypassMode = false;
+      },
+      // Get current tenant
+      $getCurrentTenant() {
+        return tenantContext.currentTenantId;
+      },
+      // Clear tenant context
+      $clearTenant() {
+        tenantContext.currentTenantId = null;
+        tenantContext.bypassMode = false;
+      },
+      // Enable bypass mode (admin operations)
+      $enableBypassMode() {
+        tenantContext.bypassMode = true;
+      },
+      // Disable bypass mode
+      $disableBypassMode() {
+        tenantContext.bypassMode = false;
+      },
+      // Enable strict mode (requires tenantId for operations)
+      $enableStrictMode() {
+        tenantContext.strictMode = true;
+      },
+      // Disable strict mode
+      $disableStrictMode() {
+        tenantContext.strictMode = false;
+      },
+      // Execute with bypass mode temporarily
+      async $withBypassMode(fn) {
+        const previousMode = tenantContext.bypassMode;
+        tenantContext.bypassMode = true;
+        try {
+          return await fn();
+        } finally {
+          tenantContext.bypassMode = previousMode;
+        }
+      },
+      // Execute with specific tenant temporarily
+      async $withTenant(tenantId, fn) {
+        const previousTenant = tenantContext.currentTenantId;
+        const previousBypass = tenantContext.bypassMode;
+        tenantContext.currentTenantId = tenantId;
+        tenantContext.bypassMode = false;
+        try {
+          return await fn();
+        } finally {
+          tenantContext.currentTenantId = previousTenant;
+          tenantContext.bypassMode = previousBypass;
+        }
+      },
+      // Execute without tenant filtering (system operations)
+      async $withSystemMode(fn) {
+        const previousTenant = tenantContext.currentTenantId;
+        const previousBypass = tenantContext.bypassMode;
+        tenantContext.currentTenantId = null;
+        tenantContext.bypassMode = true;
+        try {
+          return await fn();
+        } finally {
+          tenantContext.currentTenantId = previousTenant;
+          tenantContext.bypassMode = previousBypass;
+        }
+      }
+    },
+    query: {
+      $allModels: {
+        // Automatically add tenantId filter to all read operations
+        async findMany({ args, query }) {
+          var _a;
+          if (!tenantContext.bypassMode && tenantContext.currentTenantId) {
+            args.where ?? (args.where = {});
+            (_a = args.where).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
+          } else if (tenantContext.strictMode && !tenantContext.bypassMode && !tenantContext.currentTenantId) {
+            throw new Error("Tenant context required for this operation. Use $withSystemMode() for system-level operations.");
+          }
+          return query(args);
+        },
+        async findUnique({ args, query }) {
+          var _a;
+          if (!tenantContext.bypassMode && tenantContext.currentTenantId) {
+            args.where ?? (args.where = {});
+            (_a = args.where).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
+          } else if (tenantContext.strictMode && !tenantContext.bypassMode && !tenantContext.currentTenantId) {
+            throw new Error("Tenant context required for this operation. Use $withSystemMode() for system-level operations.");
+          }
+          return query(args);
+        },
+        async findFirst({ args, query }) {
+          var _a;
+          if (!tenantContext.bypassMode && tenantContext.currentTenantId) {
+            args.where ?? (args.where = {});
+            (_a = args.where).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
+          } else if (tenantContext.strictMode && !tenantContext.bypassMode && !tenantContext.currentTenantId) {
+            throw new Error("Tenant context required for this operation. Use $withSystemMode() for system-level operations.");
+          }
+          return query(args);
+        },
+        // Automatically add tenantId to create operations
+        async create({ args, query }) {
+          var _a;
+          if (!tenantContext.bypassMode && tenantContext.currentTenantId) {
+            args.data ?? (args.data = {});
+            (_a = args.data).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
+          } else if (tenantContext.strictMode && !tenantContext.bypassMode && !tenantContext.currentTenantId) {
+            throw new Error("Tenant context required for this operation. Use $withSystemMode() for system-level operations.");
+          }
+          return query(args);
+        },
+        // Add tenantId filter to update operations
+        async update({ args, query }) {
+          var _a;
+          if (!tenantContext.bypassMode && tenantContext.currentTenantId) {
+            args.where ?? (args.where = {});
+            (_a = args.where).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
+          } else if (tenantContext.strictMode && !tenantContext.bypassMode && !tenantContext.currentTenantId) {
+            throw new Error("Tenant context required for this operation. Use $withSystemMode() for system-level operations.");
+          }
+          return query(args);
+        },
+        // Add tenantId filter to delete operations
+        async delete({ args, query }) {
+          var _a;
+          if (!tenantContext.bypassMode && tenantContext.currentTenantId) {
+            args.where ?? (args.where = {});
+            (_a = args.where).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
+          } else if (tenantContext.strictMode && !tenantContext.bypassMode && !tenantContext.currentTenantId) {
+            throw new Error("Tenant context required for this operation. Use $withSystemMode() for system-level operations.");
+          }
+          return query(args);
+        }
+      }
+    }
+  };
+}
+
 // src/datasources/prisma.datasource.ts
 var PrismaDatasource = class extends AbstractDatasource {
   constructor(prismaClient) {
@@ -1461,10 +1609,15 @@ var PrismaDatasource = class extends AbstractDatasource {
   }
   /**
    * Apply extensions to the Prisma client
-   * Override this method to add production extensions like soft delete, audit trails, etc.
+   * By default, applies tenant extension for multi-tenant support
+   * Override this method to add additional extensions like soft delete, audit trails, etc.
    */
   applyExtensions(client) {
-    return client;
+    let extended = client;
+    if (typeof client.$extends === "function") {
+      extended = extended.$extends(createTenantExtension());
+    }
+    return extended;
   }
   /**
    * Get extended client with all applied extensions
@@ -1522,7 +1675,14 @@ var PrismaDatasource = class extends AbstractDatasource {
   async healthCheck() {
     try {
       this.broker.logger.info("Running Prisma health check");
-      await this.client.$queryRaw`SELECT 1`;
+      const client = this.client;
+      if (client.$withSystemMode) {
+        await client.$withSystemMode(async () => {
+          await this.client.$queryRaw`SELECT 1`;
+        });
+      } else {
+        await this.client.$queryRaw`SELECT 1`;
+      }
       this.broker.logger.info("Prisma health check passed");
       return true;
     } catch (error) {
@@ -1703,86 +1863,6 @@ var softDeleteExtension = {
   }
 };
 
-// src/datasources/extensions/tenant.extension.ts
-function createTenantExtension() {
-  const tenantContext = {
-    currentTenantId: null
-  };
-  return {
-    name: "Tenant",
-    client: {
-      // Set tenant context
-      $setTenant(tenantId) {
-        tenantContext.currentTenantId = tenantId;
-      },
-      // Get current tenant
-      $getCurrentTenant() {
-        return tenantContext.currentTenantId;
-      },
-      // Clear tenant context
-      $clearTenant() {
-        tenantContext.currentTenantId = null;
-      }
-    },
-    query: {
-      $allModels: {
-        // Automatically add tenantId filter to all read operations
-        async findMany({ args, query }) {
-          var _a;
-          if (tenantContext.currentTenantId) {
-            args.where ?? (args.where = {});
-            (_a = args.where).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
-          }
-          return query(args);
-        },
-        async findUnique({ args, query }) {
-          var _a;
-          if (tenantContext.currentTenantId) {
-            args.where ?? (args.where = {});
-            (_a = args.where).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
-          }
-          return query(args);
-        },
-        async findFirst({ args, query }) {
-          var _a;
-          if (tenantContext.currentTenantId) {
-            args.where ?? (args.where = {});
-            (_a = args.where).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
-          }
-          return query(args);
-        },
-        // Automatically add tenantId to create operations
-        async create({ args, query }) {
-          var _a;
-          if (tenantContext.currentTenantId) {
-            args.data ?? (args.data = {});
-            (_a = args.data).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
-          }
-          return query(args);
-        },
-        // Add tenantId filter to update operations
-        async update({ args, query }) {
-          var _a;
-          if (tenantContext.currentTenantId) {
-            args.where ?? (args.where = {});
-            (_a = args.where).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
-          }
-          return query(args);
-        },
-        // Add tenantId filter to delete operations
-        async delete({ args, query }) {
-          var _a;
-          if (tenantContext.currentTenantId) {
-            args.where ?? (args.where = {});
-            (_a = args.where).tenantId ?? (_a.tenantId = tenantContext.currentTenantId);
-          }
-          return query(args);
-        }
-      }
-    }
-  };
-}
-
 // src/datasources/extensions/retry.extension.ts
 var retryExtension = {
   name: "Retry",

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hiliosai/sdk",
-  "version": "0.1.19",
+  "version": "0.1.20",
   "type": "module",
   "license": "UNLICENSED",
   "main": "./dist/index.js",