@hiliosai/sdk 0.1.19 → 0.1.21

package/dist/index.d.ts

@@ -156,14 +156,6 @@ declare abstract class AbstractDatasource<TContext = AppContext> implements Base
     }): Promise<R>;
 }
 
-interface PrismaClientLike {
-    $connect(): Promise<void>;
-    $disconnect(): Promise<void>;
-    $queryRaw: any;
-    $transaction: any;
-    $extends: any;
-    [key: string]: any;
-}
 interface SoftDeleteExtension {
     softDelete: {
         [model: string]: {
@@ -180,10 +172,25 @@ interface AuditTrailExtension {
     };
 }
 interface TenantExtension {
-    $tenant: {
-        setContext: (tenantId: string) => void;
-        getCurrentTenant: () => string | null;
-    };
+    $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>;
+}
+type PrismaClientWithTenant<T = any> = T & TenantExtension;
+interface PrismaClientLike extends Partial<TenantExtension> {
+    $connect(): Promise<void>;
+    $disconnect(): Promise<void>;
+    $queryRaw: any;
+    $transaction: any;
+    $extends: any;
+    [key: string]: any;
 }
 declare global {
     var __prisma: PrismaClientLike | undefined;
@@ -226,7 +233,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 +348,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: {
@@ -901,4 +916,4 @@ declare function DatasourceMixin(datasourceConstructors?: DatasourceConstructorR
 
 declare function omit<T extends Record<string, unknown>, K extends keyof T>(obj: T, keys: K[]): Omit<T, K>;
 
-export { AbstractDatasource, type ActionHandler, type ActionSchemaWithContext, type ActionWithPermissions, type AppContext, type AppMeta, type AuditTrailExtension, type BaseDatasource, type BaseSpec, CHANNELS, type CarouselItem, type ChannelSendOptions, ContextHelpersMiddleware, CreateHealthCheckMiddleware, DEFAULT_DATASOURCE_CACHE_TTL, type DatasourceConstructorRegistry, type DatasourceContext, type DatasourceInstanceRegistry, type DatasourceInstanceTypes, DatasourceMixin, type Env, HEALTH_CHECK_DEFAULTS, INTEGRATION_CHANNELS, IntegrationCapability, type IntegrationChannelName, type IntegrationConfig, type IntegrationMessageFailedPayload, type IntegrationMessageReceivedPayload, type IntegrationMessageSentPayload, IntegrationPlatform, type IntegrationRegisteredPayload, type IntegrationServiceConfig, type IntegrationServiceSchema, IntegrationStatus, type IntegrationUnregisteredPayload, MemoizeMixin, type MemoizeMixinOptions, type Message, type MessageAttachment, type MessageButton, type MessageContent, MessageContentType, type MessageDirection, type MessageParticipant, type MessageStatus, type MessageType, NAMESPACE, PERMISSIONS, type Permission, type PermissionHelpers, PermissionsMiddleware, type PlatformMessage, PrismaDatasource, REDIS_URL, ROLE_PERMISSIONS, type SendResult, type SendToChannelMethod, type ServiceActionsSchema, type ServiceConfig, type ServiceSchema, type ServiceWithDatasources, type SoftDeleteExtension, type Tenant, type TenantExtension, type User, UserRole, type WebhookEvent, createDatasourceMiddleware, createTenantExtension, defineIntegration, defineService, isDev, isProd, isTest, configs as moleculer, nodeEnv, omit, retryExtension, softDeleteExtension };
+export { AbstractDatasource, type ActionHandler, type ActionSchemaWithContext, type ActionWithPermissions, type AppContext, type AppMeta, type AuditTrailExtension, type BaseDatasource, type BaseSpec, CHANNELS, type CarouselItem, type ChannelSendOptions, ContextHelpersMiddleware, CreateHealthCheckMiddleware, DEFAULT_DATASOURCE_CACHE_TTL, type DatasourceConstructorRegistry, type DatasourceContext, type DatasourceInstanceRegistry, type DatasourceInstanceTypes, DatasourceMixin, type Env, HEALTH_CHECK_DEFAULTS, INTEGRATION_CHANNELS, IntegrationCapability, type IntegrationChannelName, type IntegrationConfig, type IntegrationMessageFailedPayload, type IntegrationMessageReceivedPayload, type IntegrationMessageSentPayload, IntegrationPlatform, type IntegrationRegisteredPayload, type IntegrationServiceConfig, type IntegrationServiceSchema, IntegrationStatus, type IntegrationUnregisteredPayload, MemoizeMixin, type MemoizeMixinOptions, type Message, type MessageAttachment, type MessageButton, type MessageContent, MessageContentType, type MessageDirection, type MessageParticipant, type MessageStatus, type MessageType, NAMESPACE, PERMISSIONS, type Permission, type PermissionHelpers, PermissionsMiddleware, type PlatformMessage, type PrismaClientWithTenant, PrismaDatasource, REDIS_URL, ROLE_PERMISSIONS, type SendResult, type SendToChannelMethod, type ServiceActionsSchema, type ServiceConfig, type ServiceSchema, type ServiceWithDatasources, type SoftDeleteExtension, type Tenant, type TenantExtension, type User, UserRole, type WebhookEvent, createDatasourceMiddleware, createTenantExtension, defineIntegration, defineService, isDev, isProd, isTest, configs as moleculer, nodeEnv, omit, retryExtension, softDeleteExtension };

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,13 @@ var PrismaDatasource = class extends AbstractDatasource {
   async healthCheck() {
     try {
       this.broker.logger.info("Running Prisma health check");
-      await this.client.$queryRaw`SELECT 1`;
+      if (this.client.$withSystemMode) {
+        await this.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) {
@@ -1577,9 +1736,8 @@ var PrismaDatasource = class extends AbstractDatasource {
    */
   setTenantContext(tenantId) {
     this._client ?? (this._client = this.initializePrismaClient());
-    const tenantClient = this._client;
-    if (tenantClient.$setTenant) {
-      tenantClient.$setTenant(tenantId);
+    if (this._client.$setTenant) {
+      this._client.$setTenant(tenantId);
       this.broker.logger.debug("Tenant context set:", { tenantId });
     } else {
       this.broker.logger.warn(
@@ -1594,9 +1752,8 @@ var PrismaDatasource = class extends AbstractDatasource {
     if (!this._client) {
       return null;
     }
-    const tenantClient = this._client;
-    if (tenantClient.$getCurrentTenant) {
-      return tenantClient.$getCurrentTenant();
+    if (this._client.$getCurrentTenant) {
+      return this._client.$getCurrentTenant();
     }
     return null;
   }
@@ -1612,9 +1769,8 @@ var PrismaDatasource = class extends AbstractDatasource {
       if (previousTenant) {
         this.setTenantContext(previousTenant);
       } else {
-        const tenantClient = this.client;
-        if (tenantClient.$clearTenant) {
-          tenantClient.$clearTenant();
+        if (this.client.$clearTenant) {
+          this.client.$clearTenant();
         }
       }
     }
@@ -1703,86 +1859,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.21",
   "type": "module",
   "license": "UNLICENSED",
   "main": "./dist/index.js",