@hiliosai/sdk 0.1.18 → 0.1.20

package/dist/index.d.ts

@@ -151,6 +151,9 @@ declare abstract class AbstractDatasource<TContext = AppContext> implements Base
      * Override to implement clearing logic
      */
     clear(): Promise<void>;
+    memoize<P = unknown, R = unknown>(name: string, params: P, fn: () => Promise<R>, options?: {
+        ttl?: number;
+    }): Promise<R>;
 }
 
 interface PrismaClientLike {
@@ -223,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;
     /**
@@ -337,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: {
@@ -865,6 +876,7 @@ declare const isDev: boolean;
 declare const isTest: boolean;
 declare const isProd: boolean;
 declare const REDIS_URL: string | undefined;
+declare const DEFAULT_DATASOURCE_CACHE_TTL: number;
 
 type Env = typeof env;
 
@@ -897,4 +909,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, 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, 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

@@ -249,6 +249,11 @@ var isDev = nodeEnv === "development";
 var isTest = nodeEnv === "test";
 var isProd = nodeEnv === "production";
 var REDIS_URL = env4.string("REDIS_URL");
+var DEFAULT_DATASOURCE_CACHE_TTL = env4.number(
+  "DEFAULT_DATASOURCE_CACHE_TTL",
+  60
+  // 60 seconds
+);
 var env_default = env4;
 
 // src/errors/permission.error.ts
@@ -1389,8 +1394,176 @@ var AbstractDatasource = class {
    */
   async clear() {
   }
+  // memoize
+  async memoize(name, params, fn, options) {
+    let res;
+    if (!this.broker.cacher) return fn();
+    const key = this.broker.cacher.defaultKeygen(
+      `${this.name}:ds-${name}`,
+      params ?? {},
+      {},
+      []
+    );
+    res = await this.broker.cacher.get(key);
+    if (res) return res;
+    res = await fn();
+    this.broker.cacher.set(
+      key,
+      res,
+      options?.ttl ?? DEFAULT_DATASOURCE_CACHE_TTL
+    );
+    return res;
+  }
 };
 
+// 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) {
@@ -1436,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
@@ -1497,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) {
@@ -1678,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",
@@ -1812,4 +1917,4 @@ var retryExtension = {
   }
 };
 
-export { AbstractDatasource, CHANNELS, ContextHelpersMiddleware, CreateHealthCheckMiddleware, DatasourceMixin, HEALTH_CHECK_DEFAULTS, INTEGRATION_CHANNELS, IntegrationCapability, IntegrationPlatform, IntegrationStatus, MemoizeMixin, MessageContentType, NAMESPACE, PERMISSIONS, PermissionsMiddleware, PrismaDatasource, REDIS_URL, ROLE_PERMISSIONS, UserRole, createDatasourceMiddleware, createTenantExtension, defineIntegration, defineService, env_default as env, isDev, isProd, isTest, moleculer_default as moleculer, nodeEnv, omit, retryExtension, softDeleteExtension };
+export { AbstractDatasource, CHANNELS, ContextHelpersMiddleware, CreateHealthCheckMiddleware, DEFAULT_DATASOURCE_CACHE_TTL, DatasourceMixin, HEALTH_CHECK_DEFAULTS, INTEGRATION_CHANNELS, IntegrationCapability, IntegrationPlatform, IntegrationStatus, MemoizeMixin, MessageContentType, NAMESPACE, PERMISSIONS, PermissionsMiddleware, PrismaDatasource, REDIS_URL, ROLE_PERMISSIONS, UserRole, createDatasourceMiddleware, createTenantExtension, defineIntegration, defineService, env_default as env, isDev, isProd, isTest, moleculer_default as moleculer, nodeEnv, omit, retryExtension, softDeleteExtension };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hiliosai/sdk",
-  "version": "0.1.18",
+  "version": "0.1.20",
   "type": "module",
   "license": "UNLICENSED",
   "main": "./dist/index.js",
@@ -31,4 +31,4 @@
     "bun-types": "latest"
   },
   "prettier": "@hiliosai/prettier"
-}
+}