iframe-pubsub 1.0.11 → 1.0.13

package/dist/index.d.mts

@@ -42,7 +42,7 @@ declare class Client {
      *
      * Note: aichat itself does not know the iframe is removed then we have to clean up from parent
      */
-    cleanAIChat(): void;
+    cleanAIChat(): boolean;
     /**
      * Listen for messages from the parent page with a callback.
      *
@@ -197,7 +197,7 @@ declare class PubSub {
      *
      * @param pageId The ID of the page or component to unregister from.
      */
-    unregister(pageId: string): void;
+    unregister(pageId: string): boolean;
     sendMessage(message: IMessage): void;
     /**
      * Try to send a message to a client with retry logic.

package/dist/index.d.ts

@@ -42,7 +42,7 @@ declare class Client {
      *
      * Note: aichat itself does not know the iframe is removed then we have to clean up from parent
      */
-    cleanAIChat(): void;
+    cleanAIChat(): boolean;
     /**
      * Listen for messages from the parent page with a callback.
      *
@@ -197,7 +197,7 @@ declare class PubSub {
      *
      * @param pageId The ID of the page or component to unregister from.
      */
-    unregister(pageId: string): void;
+    unregister(pageId: string): boolean;
     sendMessage(message: IMessage): void;
     /**
      * Try to send a message to a client with retry logic.

package/dist/index.js

@@ -63,7 +63,7 @@ var _PubSub = class _PubSub {
    * @param pageId The ID of the page or component to unregister from.
    */
   unregister(pageId) {
-    this.subscribers.delete(pageId);
+    return this.subscribers.delete(pageId);
   }
   sendMessage(message) {
     if (this.mainCallback) {
@@ -96,7 +96,8 @@ var _PubSub = class _PubSub {
       console.warn(`Failed to send message to client ${message.to} after 10 retries`);
     }
   }
-  async handleMessage(event) {
+  handleMessage(event) {
+    console.log("Pubsub Received message:", event);
     const data = event.data;
     const source = event.source;
     if (data?.type === "REGISTER") {
@@ -125,14 +126,14 @@ var _PubSub = class _PubSub {
     if (!data || !data.from || !data.to) return;
     const message = data;
     if (this.mainCallback) {
-      await this.mainCallback(message);
+      this.mainCallback(message);
     }
     const subscriber = this.subscribers.get(message.to);
     if (!subscriber) return;
     if (subscriber.source) {
       subscriber.source.postMessage(message, "*");
     } else {
-      await subscriber.callback(message);
+      subscriber.callback(message);
     }
   }
   /**
@@ -195,7 +196,8 @@ var Client = class {
     if (this.isIframe) {
       throw new Error("You are not allowed to clean up aichat from iframe.");
     }
-    this.pubsub.unregister("aichat");
+    window.removeEventListener("message", this.handleMessage.bind(this));
+    return this.pubsub.unregister("aichat" /* AI_CHAT_CLIENT_ID */);
   }
   /**
    * Listen for messages from the parent page with a callback.
@@ -203,6 +205,7 @@ var Client = class {
    * @param callback The callback function to register.
    */
   onMessage(callback) {
+    console.info("Register new onMessage callback for", this.pageId, callback);
     this.callback = callback;
   }
   /**
@@ -288,7 +291,8 @@ var Client = class {
       }
     });
   }
-  async handleMessage(event) {
+  handleMessage(event) {
+    console.log("Client Received message:", event);
     let message;
     if (event.data) {
       const evt = event;
@@ -299,7 +303,7 @@ var Client = class {
     if (!message || !message.from || !message.to || message.to !== this.pageId) return;
     if (this.callback) {
       try {
-        await this.callback(message);
+        this.callback(message);
       } catch (error) {
         console.error(`Client ${this.pageId} failed to process message:`, error);
       }

package/dist/index.mjs

@@ -36,7 +36,7 @@ var _PubSub = class _PubSub {
    * @param pageId The ID of the page or component to unregister from.
    */
   unregister(pageId) {
-    this.subscribers.delete(pageId);
+    return this.subscribers.delete(pageId);
   }
   sendMessage(message) {
     if (this.mainCallback) {
@@ -69,7 +69,8 @@ var _PubSub = class _PubSub {
       console.warn(`Failed to send message to client ${message.to} after 10 retries`);
     }
   }
-  async handleMessage(event) {
+  handleMessage(event) {
+    console.log("Pubsub Received message:", event);
     const data = event.data;
     const source = event.source;
     if (data?.type === "REGISTER") {
@@ -98,14 +99,14 @@ var _PubSub = class _PubSub {
     if (!data || !data.from || !data.to) return;
     const message = data;
     if (this.mainCallback) {
-      await this.mainCallback(message);
+      this.mainCallback(message);
     }
     const subscriber = this.subscribers.get(message.to);
     if (!subscriber) return;
     if (subscriber.source) {
       subscriber.source.postMessage(message, "*");
     } else {
-      await subscriber.callback(message);
+      subscriber.callback(message);
     }
   }
   /**
@@ -168,7 +169,8 @@ var Client = class {
     if (this.isIframe) {
       throw new Error("You are not allowed to clean up aichat from iframe.");
     }
-    this.pubsub.unregister("aichat");
+    window.removeEventListener("message", this.handleMessage.bind(this));
+    return this.pubsub.unregister("aichat" /* AI_CHAT_CLIENT_ID */);
   }
   /**
    * Listen for messages from the parent page with a callback.
@@ -176,6 +178,7 @@ var Client = class {
    * @param callback The callback function to register.
    */
   onMessage(callback) {
+    console.info("Register new onMessage callback for", this.pageId, callback);
     this.callback = callback;
   }
   /**
@@ -261,7 +264,8 @@ var Client = class {
       }
     });
   }
-  async handleMessage(event) {
+  handleMessage(event) {
+    console.log("Client Received message:", event);
     let message;
     if (event.data) {
       const evt = event;
@@ -272,7 +276,7 @@ var Client = class {
     if (!message || !message.from || !message.to || message.to !== this.pageId) return;
     if (this.callback) {
       try {
-        await this.callback(message);
+        this.callback(message);
       } catch (error) {
         console.error(`Client ${this.pageId} failed to process message:`, error);
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "iframe-pubsub",
-  "version": "1.0.11",
+  "version": "1.0.13",
   "description": "A PubSub library for iframe communication.",
   "author": "Lap Tran",
   "license": "ISC",