@aichatwar/shared 1.0.168 → 1.0.170

package/build/events/kafka/baseListener.d.ts

@@ -5,6 +5,20 @@ export declare abstract class Listener<T extends BaseEvent> {
     abstract topic: T['subject'];
     abstract groupId: string;
     abstract onMessage(data: T['data'], payload: EachMessagePayload): Promise<void>;
+    private static registry;
+    /**
+     * Returns true only if every registered listener's consumer is connected.
+     * Wire this into your /ready endpoint alongside the MongoDB check.
+     */
+    static allHealthy(): boolean;
+    /**
+     * Returns per-listener health details for diagnostics.
+     */
+    static healthDetails(): Array<{
+        topic: string;
+        groupId: string;
+        healthy: boolean;
+    }>;
     protected consumer: Consumer;
     protected ackDeadline: number;
     protected fromBeginning: boolean;
@@ -16,7 +30,11 @@ export declare abstract class Listener<T extends BaseEvent> {
     private isListening;
     private crashHandlerSetup;
     private crashRestartCount;
+    private readonly maxCrashRestarts;
     private readonly maxCrashRestartDelay;
+    private lastSuccessfulMessageAt;
+    /** Expose consumer health for readiness probes */
+    get healthy(): boolean;
     constructor(consumer: Consumer);
     private setupCrashHandler;
     ack(payload?: EachMessagePayload): Promise<void>;

package/build/events/kafka/baseListener.js

@@ -11,6 +11,29 @@ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, ge
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.Listener = void 0;
 class Listener {
+    /**
+     * Returns true only if every registered listener's consumer is connected.
+     * Wire this into your /ready endpoint alongside the MongoDB check.
+     */
+    static allHealthy() {
+        if (Listener.registry.length === 0)
+            return true;
+        return Listener.registry.every(l => l.healthy);
+    }
+    /**
+     * Returns per-listener health details for diagnostics.
+     */
+    static healthDetails() {
+        return Listener.registry.map(l => ({
+            topic: l.topic,
+            groupId: l.groupId,
+            healthy: l.healthy,
+        }));
+    }
+    /** Expose consumer health for readiness probes */
+    get healthy() {
+        return this.isListening;
+    }
     constructor(consumer) {
         this.ackDeadline = 5 * 1000; // 5 seconds
         this.fromBeginning = false; // Override in subclasses to read from beginning
@@ -21,8 +44,11 @@ class Listener {
         this.isListening = false; // Track if listener is active
         this.crashHandlerSetup = false; // Track if crash handler has been set up
         this.crashRestartCount = 0;
+        this.maxCrashRestarts = 5; // Exit process after this many consecutive crashes
         this.maxCrashRestartDelay = 120000; // 2 minutes cap
+        this.lastSuccessfulMessageAt = 0; // Timestamp of last successfully processed message
         this.consumer = consumer;
+        Listener.registry.push(this);
         this.setupCrashHandler();
     }
     setupCrashHandler() {
@@ -37,21 +63,34 @@ class Listener {
                 return;
             }
             this.crashRestartCount++;
+            this.isListening = false;
+            if (this.crashRestartCount >= this.maxCrashRestarts) {
+                console.error(`💀 [${this.topic}] Consumer crashed ${this.crashRestartCount} times without recovery. ` +
+                    `Exiting process to let Kubernetes restart the pod with a fresh state.`);
+                process.exit(1);
+            }
             const delay = Math.min(5000 * Math.pow(2, Math.min(this.crashRestartCount - 1, 5)), this.maxCrashRestartDelay);
-            console.error(`[${this.topic}] Consumer crashed (restart #${this.crashRestartCount}, ` +
+            console.error(`[${this.topic}] Consumer crashed (restart #${this.crashRestartCount}/${this.maxCrashRestarts}, ` +
                 `retrying in ${delay}ms):`, error);
-            this.isListening = false;
             try {
                 yield this.consumer.disconnect();
             }
             catch (_) { /* best effort */ }
             setTimeout(() => {
-                console.log(`[${this.topic}] Auto-restarting consumer after crash (attempt #${this.crashRestartCount})...`);
+                console.log(`[${this.topic}] Auto-restarting consumer after crash (attempt #${this.crashRestartCount}/${this.maxCrashRestarts})...`);
                 this.listen().catch((err) => {
                     console.error(`[${this.topic}] Auto-restart failed:`, err);
+                    console.error(`💀 [${this.topic}] Auto-restart listen() threw. Exiting process to let Kubernetes restart the pod.`);
+                    process.exit(1);
                 });
             }, delay);
         }));
+        this.consumer.on('consumer.disconnect', () => {
+            if (this.isListening) {
+                console.warn(`⚠️ [${this.topic}] Consumer disconnected unexpectedly`);
+                this.isListening = false;
+            }
+        });
         this.crashHandlerSetup = true;
     }
     // Manual acknowledgment method
@@ -67,6 +106,12 @@ class Listener {
                     partition: targetPayload.partition,
                     offset: (BigInt(targetPayload.message.offset) + BigInt(1)).toString()
                 }]);
+            // Reset crash counter on successful processing — proves the consumer is healthy
+            if (this.crashRestartCount > 0) {
+                console.log(`[${this.topic}] Crash counter reset (was ${this.crashRestartCount}) after successful message processing`);
+                this.crashRestartCount = 0;
+            }
+            this.lastSuccessfulMessageAt = Date.now();
             console.log(`Message manually acknowledged for topic: ${this.topic}`);
         });
     }
@@ -198,3 +243,4 @@ class Listener {
     }
 }
 exports.Listener = Listener;
+Listener.registry = [];

package/build/events/types/subscriptionTier.d.ts

@@ -0,0 +1,7 @@
+export declare enum SubscriptionTier {
+    Free = "free",
+    Basic = "basic",
+    Pro = "pro",
+    Enterprise = "enterprise"
+}
+export declare const AGENT_LIMITS_BY_TIER: Record<SubscriptionTier, number>;

package/build/events/types/subscriptionTier.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AGENT_LIMITS_BY_TIER = exports.SubscriptionTier = void 0;
+var SubscriptionTier;
+(function (SubscriptionTier) {
+    SubscriptionTier["Free"] = "free";
+    SubscriptionTier["Basic"] = "basic";
+    SubscriptionTier["Pro"] = "pro";
+    SubscriptionTier["Enterprise"] = "enterprise";
+})(SubscriptionTier || (exports.SubscriptionTier = SubscriptionTier = {}));
+exports.AGENT_LIMITS_BY_TIER = {
+    [SubscriptionTier.Free]: 2,
+    [SubscriptionTier.Basic]: 3,
+    [SubscriptionTier.Pro]: 5,
+    [SubscriptionTier.Enterprise]: 50,
+};

package/build/events/userEvents.d.ts

@@ -1,6 +1,7 @@
 import { Subjects } from "./subjects";
 import { BaseEvent } from "./baseEvent";
 import { UserStatus } from "./types/userStatus";
+import { SubscriptionTier } from "./types/subscriptionTier";
 interface UserCreatedEvent extends BaseEvent {
     subject: Subjects.UserCreated;
     data: {
@@ -11,6 +12,7 @@ interface UserCreatedEvent extends BaseEvent {
         isAgent?: boolean;
         ownerUserId?: string;
         role?: 'user' | 'admin';
+        subscriptionTier?: SubscriptionTier;
     };
 }
 interface UserUpdatedEvent extends BaseEvent {
@@ -21,6 +23,7 @@ interface UserUpdatedEvent extends BaseEvent {
         status: UserStatus;
         version: number;
         role?: 'user' | 'admin';
+        subscriptionTier?: SubscriptionTier;
     };
 }
 interface UserSingedInEvent extends BaseEvent {

package/build/index.d.ts

@@ -36,6 +36,7 @@ export * from "./events/types/friendshipStatus";
 export * from "./events/types/userStatus";
 export * from "./events/types/visibility";
 export * from "./events/types/postStatus";
+export * from "./events/types/subscriptionTier";
 export * from "./events/userEvents";
 export * from "./events/mediaEvents";
 export * from "./events/agentManagerEvents";

package/build/index.js

@@ -52,6 +52,7 @@ __exportStar(require("./events/types/friendshipStatus"), exports);
 __exportStar(require("./events/types/userStatus"), exports);
 __exportStar(require("./events/types/visibility"), exports);
 __exportStar(require("./events/types/postStatus"), exports);
+__exportStar(require("./events/types/subscriptionTier"), exports);
 __exportStar(require("./events/userEvents"), exports);
 __exportStar(require("./events/mediaEvents"), exports);
 __exportStar(require("./events/agentManagerEvents"), exports);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aichatwar/shared",
-  "version": "1.0.168",
+  "version": "1.0.170",
   "main": "./build/index.js",
   "typs": "./build/index.d.ts",
   "files": [