@heritageai/messaging 1.0.24 → 1.0.26

package/dist/base-listener.js

@@ -20,12 +20,26 @@ class Listener {
             this.onMessage(data, msg);
         });
     }
-    parseMessageData(data) {
-        if (data instanceof String) {
-            return JSON.parse(data.valueOf());
+    parseMessageData(msgData) {
+        if (typeof msgData === "string") {
+            return JSON.parse(msgData);
         }
-        else if (Buffer.isBuffer(data)) {
-            return JSON.parse(data.toString("utf8"));
+        // Buffer in Node
+        if (Buffer.isBuffer(msgData)) {
+            return JSON.parse(msgData.toString("utf8"));
+        }
+        // Uint8Array support
+        if (msgData instanceof Uint8Array) {
+            return JSON.parse(Buffer.from(msgData).toString("utf8"));
+        }
+        // Fallback: sometimes data can be an object already
+        if (typeof msgData === "object") {
+            try {
+                return JSON.parse(JSON.stringify(msgData));
+            }
+            catch {
+                return msgData; // Just return as-is if parsing fails
+            }
         }
         throw new Error("Invalid data type");
     }

package/dist/services/ws-notifier.js

@@ -12,9 +12,11 @@ const pushNotificationToWS = async (payload) => {
         const WS_GATEWAY_URL = process.env.WS_GATEWAY_URL;
         if (!WS_GATEWAY_URL)
             throw new Error("WS_GATEWAY_URL not defined");
-        await axios_1.default.post(`${WS_GATEWAY_URL}/internal/notify`, payload, {
-            headers: { "Content-Type": "application/json" },
-        });
+        // await axios.post(`${WS_GATEWAY_URL}/internal/notify`, payload, {
+        //   headers: { "Content-Type": "application/json" },
+        // });
+        const { userId, notification } = payload;
+        await axios_1.default.post(`${process.env.WS_GATEWAY_URL}/internal/notify`, { userId, notification }, { headers: { "x-internal-key": process.env.INTERNAL_API_KEY } });
     }
     catch (err) {
         logger_1.logger.error("Failed to push notification to WS-Gateway", err);

package/package.json

@@ -2,7 +2,7 @@
   "author": "future_platform",
   "name": "@heritageai/messaging",
   "license": "ISC",
-  "version": "1.0.24",
+  "version": "1.0.26",
   "description": "",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/base-listener.ts

@@ -23,7 +23,7 @@ export abstract class Listener<T> {
     const subscription = this.client.subscribe(
       this.subject,
       this.queueGroupName,
-      this.subscriptionOptions()
+      this.subscriptionOptions(),
     );
 
     // @ts-ignore (node-nats-streaming typing bug)
@@ -33,12 +33,30 @@ export abstract class Listener<T> {
     });
   }
 
-  private parseMessageData(data: String | Buffer<ArrayBufferLike>): T {
-    if (data instanceof String) {
-      return JSON.parse(data.valueOf());
-    } else if (Buffer.isBuffer(data)) {
-      return JSON.parse(data.toString("utf8"));
+  private parseMessageData(msgData: any) {
+    if (typeof msgData === "string") {
+      return JSON.parse(msgData);
     }
+
+    // Buffer in Node
+    if (Buffer.isBuffer(msgData)) {
+      return JSON.parse(msgData.toString("utf8"));
+    }
+
+    // Uint8Array support
+    if (msgData instanceof Uint8Array) {
+      return JSON.parse(Buffer.from(msgData).toString("utf8"));
+    }
+
+    // Fallback: sometimes data can be an object already
+    if (typeof msgData === "object") {
+      try {
+        return JSON.parse(JSON.stringify(msgData));
+      } catch {
+        return msgData; // Just return as-is if parsing fails
+      }
+    }
+
     throw new Error("Invalid data type");
   }
 }

package/src/services/ws-notifier.ts

@@ -17,9 +17,16 @@ export const pushNotificationToWS = async (payload: WSNotificationPayload) => {
     const WS_GATEWAY_URL = process.env.WS_GATEWAY_URL;
     if (!WS_GATEWAY_URL) throw new Error("WS_GATEWAY_URL not defined");
 
-    await axios.post(`${WS_GATEWAY_URL}/internal/notify`, payload, {
-      headers: { "Content-Type": "application/json" },
-    });
+    // await axios.post(`${WS_GATEWAY_URL}/internal/notify`, payload, {
+    //   headers: { "Content-Type": "application/json" },
+    // });
+
+    const { userId, notification } = payload;
+    await axios.post(
+      `${process.env.WS_GATEWAY_URL}/internal/notify`,
+      { userId, notification },
+      { headers: { "x-internal-key": process.env.INTERNAL_API_KEY! } },
+    );
   } catch (err) {
     logger.error("Failed to push notification to WS-Gateway", err);
     throw err;