@hotmeshio/hotmesh 0.5.4 → 0.5.6

package/build/services/store/providers/postgres/kvtypes/hash/jsonb.js

@@ -18,7 +18,7 @@ function createJsonbOperations(context) {
     };
     function handleContextSet(key, fields, options) {
         const tableName = context.tableForKey(key, 'hash');
-        const replayId = Object.keys(fields).find(k => k.includes('-') && k !== '@context');
+        const replayId = Object.keys(fields).find((k) => k.includes('-') && k !== '@context');
         const params = [];
         let sql = '';
         if (options?.nx) {
@@ -93,7 +93,7 @@ function createJsonbOperations(context) {
     }
     function handleContextMerge(key, fields, options) {
         const tableName = context.tableForKey(key, 'hash');
-        const replayId = Object.keys(fields).find(k => k.includes('-') && k !== '@context:merge');
+        const replayId = Object.keys(fields).find((k) => k.includes('-') && k !== '@context:merge');
         const params = [];
         let sql = '';
         if (options?.nx) {
@@ -227,7 +227,7 @@ function createJsonbOperations(context) {
         const tableName = context.tableForKey(key, 'hash');
         const path = fields['@context:delete'];
         const pathParts = path.split('.');
-        const replayId = Object.keys(fields).find(k => k.includes('-') && k !== '@context:delete');
+        const replayId = Object.keys(fields).find((k) => k.includes('-') && k !== '@context:delete');
         const params = [];
         let sql = '';
         if (pathParts.length === 1) {
@@ -300,7 +300,7 @@ function createJsonbOperations(context) {
         const tableName = context.tableForKey(key, 'hash');
         const { path, value } = JSON.parse(fields['@context:append']);
         const pathParts = path.split('.');
-        const replayId = Object.keys(fields).find(k => k.includes('-') && k !== '@context:append');
+        const replayId = Object.keys(fields).find((k) => k.includes('-') && k !== '@context:append');
         const params = [];
         let sql = '';
         if (replayId) {
@@ -348,7 +348,7 @@ function createJsonbOperations(context) {
         const tableName = context.tableForKey(key, 'hash');
         const { path, value } = JSON.parse(fields['@context:prepend']);
         const pathParts = path.split('.');
-        const replayId = Object.keys(fields).find(k => k.includes('-') && k !== '@context:prepend');
+        const replayId = Object.keys(fields).find((k) => k.includes('-') && k !== '@context:prepend');
         const params = [];
         let sql = '';
         if (replayId) {
@@ -396,7 +396,7 @@ function createJsonbOperations(context) {
         const tableName = context.tableForKey(key, 'hash');
         const { path, index } = JSON.parse(fields['@context:remove']);
         const pathParts = path.split('.');
-        const replayId = Object.keys(fields).find(k => k.includes('-') && k !== '@context:remove');
+        const replayId = Object.keys(fields).find((k) => k.includes('-') && k !== '@context:remove');
         const params = [];
         let sql = '';
         if (replayId) {
@@ -458,7 +458,7 @@ function createJsonbOperations(context) {
         const tableName = context.tableForKey(key, 'hash');
         const { path, value } = JSON.parse(fields['@context:increment']);
         const pathParts = path.split('.');
-        const replayId = Object.keys(fields).find(k => k.includes('-') && k !== '@context:increment');
+        const replayId = Object.keys(fields).find((k) => k.includes('-') && k !== '@context:increment');
         const params = [];
         let sql = '';
         if (replayId) {
@@ -506,7 +506,7 @@ function createJsonbOperations(context) {
         const tableName = context.tableForKey(key, 'hash');
         const path = fields['@context:toggle'];
         const pathParts = path.split('.');
-        const replayId = Object.keys(fields).find(k => k.includes('-') && k !== '@context:toggle');
+        const replayId = Object.keys(fields).find((k) => k.includes('-') && k !== '@context:toggle');
         const params = [];
         let sql = '';
         if (replayId) {
@@ -554,7 +554,7 @@ function createJsonbOperations(context) {
         const tableName = context.tableForKey(key, 'hash');
         const { path, value } = JSON.parse(fields['@context:setIfNotExists']);
         const pathParts = path.split('.');
-        const replayId = Object.keys(fields).find(k => k.includes('-') && k !== '@context:setIfNotExists');
+        const replayId = Object.keys(fields).find((k) => k.includes('-') && k !== '@context:setIfNotExists');
         const params = [];
         let sql = '';
         if (replayId) {
@@ -598,7 +598,7 @@ function createJsonbOperations(context) {
     }
     function handleContextGetPath(key, fields, options) {
         const tableName = context.tableForKey(key, 'hash');
-        const getField = Object.keys(fields).find(k => k.startsWith('@context:get:'));
+        const getField = Object.keys(fields).find((k) => k.startsWith('@context:get:'));
         const pathKey = getField.replace('@context:get:', '');
         const pathParts = JSON.parse(fields[getField]);
         const params = [];
@@ -627,7 +627,7 @@ function createJsonbOperations(context) {
     function handleContextGet(key, fields, options) {
         const tableName = context.tableForKey(key, 'hash');
         const path = fields['@context:get'];
-        const replayId = Object.keys(fields).find(k => k.includes('-') && k !== '@context:get');
+        const replayId = Object.keys(fields).find((k) => k.includes('-') && k !== '@context:get');
         const params = [];
         let sql = '';
         if (path === '') {

package/build/services/store/providers/postgres/postgres.js

@@ -1053,14 +1053,14 @@ class PostgresStoreService extends __1.StoreService {
             this.logger.info('postgres-time-notifications-deployed', {
                 appId,
                 schemaName,
-                message: 'Time-aware notifications ENABLED - using LISTEN/NOTIFY instead of polling'
+                message: 'Time-aware notifications ENABLED - using LISTEN/NOTIFY instead of polling',
             });
         }
         catch (error) {
             this.logger.error('postgres-time-notifications-deploy-error', {
                 appId,
                 schemaName,
-                error: error.message
+                error: error.message,
             });
             // Don't throw - fall back to polling mode
         }
@@ -1084,13 +1084,13 @@ class PostgresStoreService extends __1.StoreService {
                     // This is likely a notification from sub or stream provider, ignore it
                     this.logger.debug('postgres-store-ignoring-non-time-notification', {
                         channel: notification.channel,
-                        payloadPreview: notification.payload.substring(0, 100)
+                        payloadPreview: notification.payload.substring(0, 100),
                     });
                 }
             });
             this.logger.debug('postgres-time-scout-notifications-started', {
                 appId: this.appId,
-                channelName
+                channelName,
             });
             // Start the enhanced time scout loop
             await this.processTimeHooksWithNotifications(timeEventCallback);
@@ -1098,7 +1098,7 @@ class PostgresStoreService extends __1.StoreService {
         catch (error) {
             this.logger.error('postgres-time-scout-notifications-error', {
                 appId: this.appId,
-                error
+                error,
             });
             // Fall back to regular polling mode
             throw error;
@@ -1118,7 +1118,7 @@ class PostgresStoreService extends __1.StoreService {
                 type,
                 appId: app_id,
                 nextAwakening: next_awakening,
-                readyAt: ready_at
+                readyAt: ready_at,
             });
             if (type === 'time_hooks_ready') {
                 // Process any ready time hooks immediately
@@ -1132,7 +1132,7 @@ class PostgresStoreService extends __1.StoreService {
         catch (error) {
             this.logger.error('postgres-time-notification-handle-error', {
                 notification,
-                error
+                error,
             });
         }
     }
@@ -1233,7 +1233,7 @@ class PostgresStoreService extends __1.StoreService {
         // For now, just log the schedule update
         this.logger.debug('postgres-time-schedule-updated', {
             nextAwakening,
-            currentTime: Date.now()
+            currentTime: Date.now(),
         });
     }
     /**

package/build/services/stream/providers/postgres/postgres.js

@@ -62,16 +62,17 @@ class PostgresStreamService extends index_1.StreamService {
         if (!clientNotificationConsumers) {
             return;
         }
-        for (const [consumerKey, instanceMap] of clientNotificationConsumers.entries()) {
+        for (const [consumerKey, instanceMap,] of clientNotificationConsumers.entries()) {
             for (const [instance, consumer] of instanceMap.entries()) {
-                if (consumer.isListening && now - consumer.lastFallbackCheck > this.getFallbackInterval()) {
+                if (consumer.isListening &&
+                    now - consumer.lastFallbackCheck > this.getFallbackInterval()) {
                     try {
                         const messages = await instance.fetchMessages(consumer.streamName, consumer.groupName, consumer.consumerName, { batchSize: 10, enableBackoff: false, maxRetries: 1 });
                         if (messages.length > 0) {
                             instance.logger.debug('postgres-stream-fallback-messages', {
                                 streamName: consumer.streamName,
                                 groupName: consumer.groupName,
-                                messageCount: messages.length
+                                messageCount: messages.length,
                             });
                             consumer.callback(messages);
                         }
@@ -81,7 +82,7 @@ class PostgresStreamService extends index_1.StreamService {
                         instance.logger.error('postgres-stream-fallback-error', {
                             streamName: consumer.streamName,
                             groupName: consumer.groupName,
-                            error
+                            error,
                         });
                     }
                 }
@@ -96,17 +97,19 @@ class PostgresStreamService extends index_1.StreamService {
                 // This is likely a pub/sub notification from the sub provider, ignore it
                 this.logger.debug('postgres-stream-ignoring-sub-notification', {
                     channel: notification.channel,
-                    payloadPreview: notification.payload.substring(0, 100)
+                    payloadPreview: notification.payload.substring(0, 100),
                 });
                 return;
             }
             this.logger.debug('postgres-stream-processing-notification', {
-                channel: notification.channel
+                channel: notification.channel,
             });
             const payload = JSON.parse(notification.payload);
             const { stream_name, group_name } = payload;
             if (!stream_name || !group_name) {
-                this.logger.warn('postgres-stream-invalid-notification', { notification });
+                this.logger.warn('postgres-stream-invalid-notification', {
+                    notification,
+                });
                 return;
             }
             const consumerKey = this.getConsumerKey(stream_name, group_name);
@@ -128,7 +131,7 @@ class PostgresStreamService extends index_1.StreamService {
         catch (error) {
             this.logger.error('postgres-stream-notification-parse-error', {
                 notification,
-                error
+                error,
             });
         }
     }
@@ -143,7 +146,7 @@ class PostgresStreamService extends index_1.StreamService {
             this.logger.error('postgres-stream-fetch-deliver-error', {
                 streamName: consumer.streamName,
                 groupName: consumer.groupName,
-                error
+                error,
             });
         }
     }
@@ -298,7 +301,7 @@ class PostgresStreamService extends index_1.StreamService {
                     streamName,
                     groupName,
                     channelName,
-                    listenDuration: Date.now() - listenStart
+                    listenDuration: Date.now() - listenStart,
                 });
             }
             catch (error) {
@@ -306,7 +309,7 @@ class PostgresStreamService extends index_1.StreamService {
                     streamName,
                     groupName,
                     channelName,
-                    error
+                    error,
                 });
                 // Fall back to polling if LISTEN fails
                 return this.fetchMessages(streamName, groupName, consumerName, options);
@@ -319,7 +322,7 @@ class PostgresStreamService extends index_1.StreamService {
             consumerName,
             callback,
             isListening: true,
-            lastFallbackCheck: Date.now()
+            lastFallbackCheck: Date.now(),
         };
         instanceMap.set(this, consumer);
         // Track this consumer for cleanup
@@ -328,7 +331,7 @@ class PostgresStreamService extends index_1.StreamService {
             streamName,
             groupName,
             instanceCount: instanceMap.size,
-            setupDuration: Date.now() - startTime
+            setupDuration: Date.now() - startTime,
         });
         // Do an initial fetch asynchronously to avoid blocking setup
         // This ensures we don't miss any messages that were already in the queue
@@ -338,13 +341,13 @@ class PostgresStreamService extends index_1.StreamService {
                 const initialMessages = await this.fetchMessages(streamName, groupName, consumerName, {
                     ...options,
                     enableBackoff: false,
-                    maxRetries: 1
+                    maxRetries: 1,
                 });
                 this.logger.debug('postgres-stream-initial-fetch-complete', {
                     streamName,
                     groupName,
                     messageCount: initialMessages.length,
-                    fetchDuration: Date.now() - fetchStart
+                    fetchDuration: Date.now() - fetchStart,
                 });
                 // If we got messages, call the callback
                 if (initialMessages.length > 0) {
@@ -355,7 +358,7 @@ class PostgresStreamService extends index_1.StreamService {
                 this.logger.error('postgres-stream-initial-fetch-error', {
                     streamName,
                     groupName,
-                    error
+                    error,
                 });
             }
         });
@@ -387,7 +390,7 @@ class PostgresStreamService extends index_1.StreamService {
                     this.logger.debug('postgres-stream-unlisten', {
                         streamName,
                         groupName,
-                        channelName
+                        channelName,
                     });
                 }
                 catch (error) {
@@ -395,7 +398,7 @@ class PostgresStreamService extends index_1.StreamService {
                         streamName,
                         groupName,
                         channelName,
-                        error
+                        error,
                     });
                 }
             }
@@ -583,7 +586,7 @@ class PostgresStreamService extends index_1.StreamService {
                                 this.logger.debug('postgres-stream-cleanup-unlisten', {
                                     streamName: consumer.streamName,
                                     groupName: consumer.groupName,
-                                    channelName
+                                    channelName,
                                 });
                             }
                             catch (error) {
@@ -591,7 +594,7 @@ class PostgresStreamService extends index_1.StreamService {
                                     streamName: consumer.streamName,
                                     groupName: consumer.groupName,
                                     channelName,
-                                    error
+                                    error,
                                 });
                             }
                         }

package/build/services/sub/providers/postgres/postgres.js

@@ -37,7 +37,7 @@ class PostgresSubService extends index_1.SubService {
                 try {
                     const payload = JSON.parse(msg.payload || '{}');
                     // Call all callbacks registered for this channel across all SubService instances
-                    callbacks.forEach(callback => {
+                    callbacks.forEach((callback) => {
                         try {
                             callback(msg.channel, payload);
                         }
@@ -101,7 +101,11 @@ class PostgresSubService extends index_1.SubService {
         callbacks.set(this, callback);
         // Track this subscription for cleanup
         this.instanceSubscriptions.add(safeKey);
-        this.logger.debug(`postgres-subscribe`, { originalKey, safeKey, totalCallbacks: callbacks.size });
+        this.logger.debug(`postgres-subscribe`, {
+            originalKey,
+            safeKey,
+            totalCallbacks: callbacks.size,
+        });
     }
     async unsubscribe(keyType, appId, topic) {
         const [originalKey, safeKey] = this.mintSafeKey(keyType, {
@@ -125,7 +129,11 @@ class PostgresSubService extends index_1.SubService {
             clientSubscriptions.delete(safeKey);
             await this.eventClient.query(`UNLISTEN "${safeKey}"`);
         }
-        this.logger.debug(`postgres-unsubscribe`, { originalKey, safeKey, remainingCallbacks: callbacks.size });
+        this.logger.debug(`postgres-unsubscribe`, {
+            originalKey,
+            safeKey,
+            remainingCallbacks: callbacks.size,
+        });
     }
     /**
      * Cleanup method to remove all subscriptions for this instance.

package/build/services/task/index.js

@@ -211,7 +211,7 @@ class TaskService {
             try {
                 this.logger.info('task-using-notification-mode', {
                     appId: this.store.appId,
-                    message: 'Time scout using PostgreSQL LISTEN/NOTIFY mode for efficient task processing'
+                    message: 'Time scout using PostgreSQL LISTEN/NOTIFY mode for efficient task processing',
                 });
                 // Use the PostgreSQL store's notification-based approach
                 await this.store.startTimeScoutWithNotifications(timeEventCallback);
@@ -221,7 +221,7 @@ class TaskService {
                     appId: this.store.appId,
                     error: error.message,
                     fallbackTo: 'polling',
-                    message: 'Notification mode failed - falling back to traditional polling'
+                    message: 'Notification mode failed - falling back to traditional polling',
                 });
                 // Fall back to regular polling
                 await this.processTimeHooks(timeEventCallback);
@@ -231,7 +231,7 @@ class TaskService {
             this.logger.info('task-using-polling-mode', {
                 appId: this.store.appId,
                 storeType: this.store.constructor.name,
-                message: 'Time scout using traditional polling mode (notifications not available)'
+                message: 'Time scout using traditional polling mode (notifications not available)',
             });
             // Use regular polling for non-PostgreSQL stores
             await this.processTimeHooks(timeEventCallback);
@@ -247,7 +247,7 @@ class TaskService {
      * Check if the store supports notifications
      */
     supportsNotifications() {
-        return typeof this.store.startTimeScoutWithNotifications === 'function';
+        return (typeof this.store.startTimeScoutWithNotifications === 'function');
     }
 }
 exports.TaskService = TaskService;

package/build/types/memflow.d.ts

@@ -495,4 +495,82 @@ interface ClientWorkflow {
     getHandle(taskQueue: string, workflowName: string, workflowId: string, namespace?: string): Promise<WorkflowHandleService>;
     search(taskQueue: string, workflowName: string, namespace: string | null, index: string, ...query: string[]): Promise<string[]>;
 }
+/**
+ * Workflow interceptor that can wrap workflow execution in an onion-like pattern.
+ * Each interceptor wraps the next one, with the actual workflow execution at the center.
+ *
+ * Interceptors are executed in the order they are registered. Each interceptor can:
+ * - Perform actions before workflow execution
+ * - Modify or enhance the workflow context
+ * - Handle or transform workflow results
+ * - Catch and handle errors
+ * - Add cross-cutting concerns like logging, metrics, or tracing
+ *
+ * @example
+ * ```typescript
+ * // Simple logging interceptor
+ * const loggingInterceptor: WorkflowInterceptor = {
+ *   async execute(ctx, next) {
+ *     console.log('Before workflow');
+ *     try {
+ *       const result = await next();
+ *       console.log('After workflow');
+ *       return result;
+ *     } catch (err) {
+ *       console.error('Workflow error:', err);
+ *       throw err;
+ *     }
+ *   }
+ * };
+ *
+ * // Register the interceptor
+ * MemFlow.registerInterceptor(loggingInterceptor);
+ * ```
+ */
+export interface WorkflowInterceptor {
+    /**
+     * Called before workflow execution to wrap the workflow in custom logic
+     *
+     * @param ctx - The workflow context map containing workflow metadata and state
+     * @param next - Function to call the next interceptor or the workflow itself
+     * @returns The result of the workflow execution
+     *
+     * @example
+     * ```typescript
+     * // Metrics interceptor implementation
+     * {
+     *   async execute(ctx, next) {
+     *     const workflowName = ctx.get('workflowName');
+     *     const metrics = getMetricsClient();
+     *
+     *     metrics.increment(`workflow.start.${workflowName}`);
+     *     const timer = metrics.startTimer();
+     *
+     *     try {
+     *       const result = await next();
+     *       metrics.increment(`workflow.success.${workflowName}`);
+     *       return result;
+     *     } catch (err) {
+     *       metrics.increment(`workflow.error.${workflowName}`);
+     *       throw err;
+     *     } finally {
+     *       timer.end();
+     *     }
+     *   }
+     * }
+     * ```
+     */
+    execute(ctx: Map<string, any>, next: () => Promise<any>): Promise<any>;
+}
+/**
+ * Registry for workflow interceptors that are executed in order
+ * for each workflow execution
+ */
+export interface InterceptorRegistry {
+    /**
+     * Array of registered interceptors that will wrap workflow execution
+     * in the order they were registered (first registered = outermost wrapper)
+     */
+    interceptors: WorkflowInterceptor[];
+}
 export { ActivityConfig, ActivityWorkflowDataType, ChildResponseType, ClientConfig, ClientWorkflow, ContextType, Connection, ProxyResponseType, ProxyType, Registry, SignalOptions, FindJobsOptions, FindOptions, FindWhereOptions, FindWhereQuery, HookOptions, SearchResults, WorkerConfig, WorkflowConfig, WorkerOptions, WorkflowSearchOptions, WorkflowDataType, WorkflowOptions, WorkflowContext, };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hotmeshio/hotmesh",
-  "version": "0.5.4",
+  "version": "0.5.6",
   "description": "Permanent-Memory Workflows & AI Agents",
   "main": "./build/index.js",
   "types": "./build/index.d.ts",
@@ -31,7 +31,8 @@
     "test:memflow:basic": "HMSH_LOGLEVEL=info NODE_ENV=test jest ./tests/memflow/basic/postgres.test.ts --detectOpenHandles --forceExit --verbose",
     "test:memflow:collision": "NODE_ENV=test jest ./tests/memflow/collision/*.test.ts --detectOpenHandles --forceExit --verbose",
     "test:memflow:fatal": "NODE_ENV=test jest ./tests/memflow/fatal/*.test.ts --detectOpenHandles --forceExit --verbose",
-    "test:memflow:goodbye": "NODE_ENV=test jest ./tests/memflow/goodbye/*.test.ts --detectOpenHandles --forceExit --verbose",
+    "test:memflow:goodbye": "NODE_ENV=test HMSH_LOGLEVEL=debug jest ./tests/memflow/goodbye/postgres.test.ts --detectOpenHandles --forceExit --verbose",
+    "test:memflow:interceptor": "NODE_ENV=test HMSH_LOGLEVEL=debug jest ./tests/memflow/interceptor/postgres.test.ts --detectOpenHandles --forceExit --verbose",
     "test:memflow:entity": "NODE_ENV=test HMSH_LOGLEVEL=debug jest ./tests/memflow/entity/postgres.test.ts --detectOpenHandles --forceExit --verbose",
     "test:memflow:agent": "NODE_ENV=test HMSH_LOGLEVEL=debug jest ./tests/memflow/agent/postgres.test.ts --detectOpenHandles --forceExit --verbose",
     "test:memflow:hello": "HMSH_TELEMETRY=debug HMSH_LOGLEVEL=debug HMSH_IS_CLUSTER=true NODE_ENV=test jest ./tests/memflow/helloworld/postgres.test.ts --detectOpenHandles --forceExit --verbose",