iframe-pubsub 1.0.0 → 1.0.2

package/dist/index.d.mts

@@ -10,7 +10,153 @@ interface IMessage {
     to: string;
     payload: any;
 }
+/**
+ * Message to register a page or component.
+ *
+ * @property {string} type - The type of the message.
+ * @property {string} pageId - The ID of the page or component.
+ */
+interface IRegistrationMessage {
+    type: 'REGISTER' | 'UNREGISTER';
+    pageId: string;
+}
 type MessageCallback = (message: IMessage) => void | Promise<void>;
+
+declare class Client {
+    private pageId;
+    private callback?;
+    private pubsub;
+    private isIframe;
+    /**
+     * Create a new client instance.
+     *
+     * @param pageId The ID of the page or component to register to.
+     */
+    constructor(pageId: string);
+    /**
+     * Unregister the client from the pubsub.
+     */
+    unregister(): void;
+    /**
+     * Listen for messages from the parent page with a callback.
+     *
+     * @param callback The callback function to register.
+     */
+    onMessage(callback: MessageCallback): void;
+    /**
+     * Send a message to the another page or component.
+     *
+     * @param to The ID of the page or component to send the message to.
+     * @param payload The payload of the message.
+     */
+    sendMessage(to: string, payload: any): void;
+    /**
+     * Check if a client with the given ID exists in the PubSub system.
+     *
+     * @param clientId The ID of the client to check.
+     * @param maxRetries Maximum number of retries. Default is 3.
+     * @param retryInterval Interval between retries in milliseconds. Default is 1000ms.
+     * @returns A Promise that resolves to true if the client exists, false otherwise.
+     */
+    checkClientExists(clientId: string, maxRetries?: number, retryInterval?: number): Promise<boolean>;
+    /**
+     * Check if a client with the given ID exists in the PubSub system with retry mechanism.
+     * Will retry up to 3 times with 1 second delay between retries.
+     *
+     * @param clientId The ID of the client to check.
+     * @param retryCount The current retry count.
+     * @returns A Promise that resolves to true if the client exists, false otherwise.
+     * @private
+     */
+    private checkClientExistsWithRetry;
+    private handleMessage;
+}
+
+declare enum AIChatNameEnum {
+    AI_CHAT_CLIENT_ID = "aichat",
+    AI_CHAT_INTERNAL_COMM_TYPE = "pubsub",
+    SET_PARENT_NAME = "setParentName",
+    SET_USER_ID = "setUserId",
+    SET_CUSTOM_HEADERS = "setCustomHeaders",
+    SET_CUSTOM_PAYLOAD = "setCustomPayload",
+    SET_SUGGESTIONS = "setSuggestions",
+    SHOW_SUGGESTIONS = "showSuggestions",
+    SHOW_CHAT_MESSAGE = "showChatMessage",
+    CONFIRM_ACTION = "confirmAction",
+    SHOW_OPTIONS = "showOptions",
+    SEND_CHAT_PROMPT = "sendChatPrompt",
+    DEFAULT = "default"
+}
+declare class AIChatClient extends Client {
+    /**
+     * Create a new client instance.
+     *
+     * @param pageId The ID of the page or component to register to.
+     */
+    constructor(pageId: string);
+    private sendAIChatMethod;
+    /**
+     * Set the parent name then the AI Chat can know who should it communicate with.
+     *
+     * @param parentName The name of the parent page.
+     */
+    setParentName(parentName: string): void;
+    /**
+     * Set the user ID for the AI Chat component.
+     *
+     * @param userId The user ID.
+     */
+    setUserId(userId: string): void;
+    /**
+     * Set custom headers for the AI Chat component.
+     *
+     * @param headers The custom headers.
+     */
+    setCustomHeaders(headers: Record<string, string>): void;
+    /**
+     * Set custom payload for the AI Chat component.
+     *
+     * @param payload The custom payload.
+     */
+    setCustomPayload(payload: Record<string, any>): void;
+    /**
+     * Set suggestions for the AI Chat component.
+     *
+     * @param suggestions The suggestions to set.
+     */
+    setSuggestions(suggestions: any[]): void;
+    /**
+     * Show suggestions in the AI Chat component.
+     *
+     * @param show Whether to show suggestions.
+     */
+    showSuggestions(show: boolean): void;
+    /**
+     * Show a chat message in the AI Chat component.
+     *
+     * @param message The message to show.
+     */
+    showChatMessage(message: string): void;
+    /**
+     * Confirm an action in the AI Chat component.
+     *
+     * @param action The action to confirm.
+     */
+    confirmAction(action: any): void;
+    /**
+     * Show options in the AI Chat component.
+     *
+     * @param options The options to show.
+     */
+    showOptions(options: any[]): void;
+    /**
+     * Send a chat prompt to the AI Chat component.
+     *
+     * @param prompt The prompt to send.
+     */
+    sendChatPrompt(prompt: string): void;
+}
+
 declare class PubSub {
     private static instance;
     private subscribers;
@@ -33,33 +179,22 @@ declare class PubSub {
      */
     unregister(pageId: string): void;
     sendMessage(message: IMessage): void;
-    private handleMessage;
-}
-declare class Client {
-    private pageId;
-    private callback?;
-    private pubsub;
-    private isIframe;
     /**
-     * Create a new client instance.
+     * Try to send a message to a client with retry logic.
+     * Will retry up to 3 times with 1 second delay between retries.
      *
-     * @param pageId The ID of the page or component to register to.
+     * @param message The message to send.
+     * @param retryCount The current retry count.
      */
-    constructor(pageId: string);
-    /**
-     * Listen for messages from the parent page with a callback.
-     *
-     * @param callback The callback function to register.
-     */
-    onMessage(callback: MessageCallback): void;
+    private trySendMessageWithRetry;
+    private handleMessage;
     /**
-     * Send a message to the another page or component.
+     * Check if a client with the given ID exists in the PubSub system.
      *
-     * @param to The ID of the page or component to send the message to.
-     * @param payload The payload of the message.
+     * @param clientId The ID of the client to check.
+     * @returns True if the client exists, false otherwise.
      */
-    sendMessage(to: string, payload: any): void;
-    private handleMessage;
+    isClientExists(clientId: string): boolean;
 }
 
-export { Client, type IMessage, type MessageCallback, PubSub };
+export { AIChatClient, AIChatNameEnum, Client, type IMessage, type IRegistrationMessage, type MessageCallback, PubSub };

package/dist/index.d.ts

@@ -10,7 +10,153 @@ interface IMessage {
     to: string;
     payload: any;
 }
+/**
+ * Message to register a page or component.
+ *
+ * @property {string} type - The type of the message.
+ * @property {string} pageId - The ID of the page or component.
+ */
+interface IRegistrationMessage {
+    type: 'REGISTER' | 'UNREGISTER';
+    pageId: string;
+}
 type MessageCallback = (message: IMessage) => void | Promise<void>;
+
+declare class Client {
+    private pageId;
+    private callback?;
+    private pubsub;
+    private isIframe;
+    /**
+     * Create a new client instance.
+     *
+     * @param pageId The ID of the page or component to register to.
+     */
+    constructor(pageId: string);
+    /**
+     * Unregister the client from the pubsub.
+     */
+    unregister(): void;
+    /**
+     * Listen for messages from the parent page with a callback.
+     *
+     * @param callback The callback function to register.
+     */
+    onMessage(callback: MessageCallback): void;
+    /**
+     * Send a message to the another page or component.
+     *
+     * @param to The ID of the page or component to send the message to.
+     * @param payload The payload of the message.
+     */
+    sendMessage(to: string, payload: any): void;
+    /**
+     * Check if a client with the given ID exists in the PubSub system.
+     *
+     * @param clientId The ID of the client to check.
+     * @param maxRetries Maximum number of retries. Default is 3.
+     * @param retryInterval Interval between retries in milliseconds. Default is 1000ms.
+     * @returns A Promise that resolves to true if the client exists, false otherwise.
+     */
+    checkClientExists(clientId: string, maxRetries?: number, retryInterval?: number): Promise<boolean>;
+    /**
+     * Check if a client with the given ID exists in the PubSub system with retry mechanism.
+     * Will retry up to 3 times with 1 second delay between retries.
+     *
+     * @param clientId The ID of the client to check.
+     * @param retryCount The current retry count.
+     * @returns A Promise that resolves to true if the client exists, false otherwise.
+     * @private
+     */
+    private checkClientExistsWithRetry;
+    private handleMessage;
+}
+
+declare enum AIChatNameEnum {
+    AI_CHAT_CLIENT_ID = "aichat",
+    AI_CHAT_INTERNAL_COMM_TYPE = "pubsub",
+    SET_PARENT_NAME = "setParentName",
+    SET_USER_ID = "setUserId",
+    SET_CUSTOM_HEADERS = "setCustomHeaders",
+    SET_CUSTOM_PAYLOAD = "setCustomPayload",
+    SET_SUGGESTIONS = "setSuggestions",
+    SHOW_SUGGESTIONS = "showSuggestions",
+    SHOW_CHAT_MESSAGE = "showChatMessage",
+    CONFIRM_ACTION = "confirmAction",
+    SHOW_OPTIONS = "showOptions",
+    SEND_CHAT_PROMPT = "sendChatPrompt",
+    DEFAULT = "default"
+}
+declare class AIChatClient extends Client {
+    /**
+     * Create a new client instance.
+     *
+     * @param pageId The ID of the page or component to register to.
+     */
+    constructor(pageId: string);
+    private sendAIChatMethod;
+    /**
+     * Set the parent name then the AI Chat can know who should it communicate with.
+     *
+     * @param parentName The name of the parent page.
+     */
+    setParentName(parentName: string): void;
+    /**
+     * Set the user ID for the AI Chat component.
+     *
+     * @param userId The user ID.
+     */
+    setUserId(userId: string): void;
+    /**
+     * Set custom headers for the AI Chat component.
+     *
+     * @param headers The custom headers.
+     */
+    setCustomHeaders(headers: Record<string, string>): void;
+    /**
+     * Set custom payload for the AI Chat component.
+     *
+     * @param payload The custom payload.
+     */
+    setCustomPayload(payload: Record<string, any>): void;
+    /**
+     * Set suggestions for the AI Chat component.
+     *
+     * @param suggestions The suggestions to set.
+     */
+    setSuggestions(suggestions: any[]): void;
+    /**
+     * Show suggestions in the AI Chat component.
+     *
+     * @param show Whether to show suggestions.
+     */
+    showSuggestions(show: boolean): void;
+    /**
+     * Show a chat message in the AI Chat component.
+     *
+     * @param message The message to show.
+     */
+    showChatMessage(message: string): void;
+    /**
+     * Confirm an action in the AI Chat component.
+     *
+     * @param action The action to confirm.
+     */
+    confirmAction(action: any): void;
+    /**
+     * Show options in the AI Chat component.
+     *
+     * @param options The options to show.
+     */
+    showOptions(options: any[]): void;
+    /**
+     * Send a chat prompt to the AI Chat component.
+     *
+     * @param prompt The prompt to send.
+     */
+    sendChatPrompt(prompt: string): void;
+}
+
 declare class PubSub {
     private static instance;
     private subscribers;
@@ -33,33 +179,22 @@ declare class PubSub {
      */
     unregister(pageId: string): void;
     sendMessage(message: IMessage): void;
-    private handleMessage;
-}
-declare class Client {
-    private pageId;
-    private callback?;
-    private pubsub;
-    private isIframe;
     /**
-     * Create a new client instance.
+     * Try to send a message to a client with retry logic.
+     * Will retry up to 3 times with 1 second delay between retries.
      *
-     * @param pageId The ID of the page or component to register to.
+     * @param message The message to send.
+     * @param retryCount The current retry count.
      */
-    constructor(pageId: string);
-    /**
-     * Listen for messages from the parent page with a callback.
-     *
-     * @param callback The callback function to register.
-     */
-    onMessage(callback: MessageCallback): void;
+    private trySendMessageWithRetry;
+    private handleMessage;
     /**
-     * Send a message to the another page or component.
+     * Check if a client with the given ID exists in the PubSub system.
      *
-     * @param to The ID of the page or component to send the message to.
-     * @param payload The payload of the message.
+     * @param clientId The ID of the client to check.
+     * @returns True if the client exists, false otherwise.
      */
-    sendMessage(to: string, payload: any): void;
-    private handleMessage;
+    isClientExists(clientId: string): boolean;
 }
 
-export { Client, type IMessage, type MessageCallback, PubSub };
+export { AIChatClient, AIChatNameEnum, Client, type IMessage, type IRegistrationMessage, type MessageCallback, PubSub };

package/dist/index.js

@@ -22,10 +22,14 @@ var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "sy
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  AIChatClient: () => AIChatClient,
+  AIChatNameEnum: () => AIChatNameEnum,
   Client: () => Client,
   PubSub: () => PubSub
 });
 module.exports = __toCommonJS(index_exports);
+
+// src/PubSub.ts
 var _PubSub = class _PubSub {
   constructor() {
     __publicField(this, "subscribers");
@@ -65,12 +69,31 @@ var _PubSub = class _PubSub {
     if (this.mainCallback) {
       this.mainCallback(message);
     }
+    this.trySendMessageWithRetry(message, 0);
+  }
+  /**
+   * Try to send a message to a client with retry logic.
+   * Will retry up to 3 times with 1 second delay between retries.
+   * 
+   * @param message The message to send.
+   * @param retryCount The current retry count.
+   */
+  trySendMessageWithRetry(message, retryCount) {
     const subscriber = this.subscribers.get(message.to);
-    if (!subscriber) return;
-    if (subscriber.source) {
-      subscriber.source.postMessage(message, "*");
+    if (subscriber) {
+      if (subscriber.source) {
+        subscriber.source.postMessage(message, "*");
+      } else {
+        subscriber.callback(message);
+      }
+      return;
+    }
+    if (retryCount < 10) {
+      setTimeout(() => {
+        this.trySendMessageWithRetry(message, retryCount + 1);
+      }, 1e3);
     } else {
-      subscriber.callback(message);
+      console.warn(`Failed to send message to client ${message.to} after 10 retries`);
     }
   }
   async handleMessage(event) {
@@ -86,6 +109,19 @@ var _PubSub = class _PubSub {
       });
       return;
     }
+    if (data?.type === "UNREGISTER") {
+      const unregistration = data;
+      this.subscribers.delete(unregistration.pageId);
+      return;
+    }
+    if (data?.type === "CLIENT_EXISTS_CHECK") {
+      source.postMessage({
+        type: "CLIENT_EXISTS_RESPONSE",
+        requestId: data.requestId,
+        exists: this.subscribers.has(data.clientId)
+      }, "*");
+      return;
+    }
     if (!data || !data.from || !data.to) return;
     const message = data;
     if (this.mainCallback) {
@@ -99,9 +135,20 @@ var _PubSub = class _PubSub {
       await subscriber.callback(message);
     }
   }
+  /**
+   * Check if a client with the given ID exists in the PubSub system.
+   * 
+   * @param clientId The ID of the client to check.
+   * @returns True if the client exists, false otherwise.
+   */
+  isClientExists(clientId) {
+    return this.subscribers.has(clientId);
+  }
 };
 __publicField(_PubSub, "instance");
 var PubSub = _PubSub;
+
+// src/Client.ts
 var Client = class {
   /**
    * Create a new client instance.
@@ -126,6 +173,19 @@ var Client = class {
       this.pubsub.register(pageId, this.handleMessage.bind(this));
     }
   }
+  /**
+   * Unregister the client from the pubsub.
+   */
+  unregister() {
+    if (this.isIframe) {
+      window.parent.postMessage({
+        type: "UNREGISTER",
+        pageId: this.pageId
+      }, "*");
+    } else {
+      this.pubsub.unregister(this.pageId);
+    }
+  }
   /**
    * Listen for messages from the parent page with a callback.
    * 
@@ -152,6 +212,71 @@ var Client = class {
       this.pubsub.sendMessage(message);
     }
   }
+  /**
+   * Check if a client with the given ID exists in the PubSub system.
+   * 
+   * @param clientId The ID of the client to check.
+   * @param maxRetries Maximum number of retries. Default is 3.
+   * @param retryInterval Interval between retries in milliseconds. Default is 1000ms.
+   * @returns A Promise that resolves to true if the client exists, false otherwise.
+   */
+  checkClientExists(clientId, maxRetries = 3, retryInterval = 1e3) {
+    return this.checkClientExistsWithRetry(clientId, 0, maxRetries, retryInterval);
+  }
+  /**
+   * Check if a client with the given ID exists in the PubSub system with retry mechanism.
+   * Will retry up to 3 times with 1 second delay between retries.
+   * 
+   * @param clientId The ID of the client to check.
+   * @param retryCount The current retry count.
+   * @returns A Promise that resolves to true if the client exists, false otherwise.
+   * @private
+   */
+  checkClientExistsWithRetry(clientId, retryCount, maxRetries, retryInterval) {
+    return new Promise((resolve) => {
+      if (this.isIframe) {
+        const requestId = `check-client-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
+        const messageHandler = (event) => {
+          const data = event.data;
+          if (data && data.type === "CLIENT_EXISTS_RESPONSE" && data.requestId === requestId) {
+            window.removeEventListener("message", messageHandler);
+            clearTimeout(removeHandlerTimeout);
+            if (data.exists) {
+              resolve(true);
+            } else if (retryCount < maxRetries - 1) {
+              setTimeout(() => {
+                this.checkClientExistsWithRetry(clientId, retryCount + 1, maxRetries, retryInterval).then((exists) => resolve(exists));
+              }, retryInterval);
+            } else {
+              resolve(false);
+            }
+          }
+        };
+        window.addEventListener("message", messageHandler);
+        window.parent.postMessage({
+          type: "CLIENT_EXISTS_CHECK",
+          clientId,
+          requestId,
+          from: this.pageId
+        }, "*");
+        const removeHandlerTimeout = setTimeout(() => {
+          window.removeEventListener("message", messageHandler);
+          resolve(false);
+        }, retryInterval + 1e3);
+      } else {
+        const exists = this.pubsub.isClientExists(clientId);
+        if (exists) {
+          resolve(true);
+        } else if (retryCount < maxRetries - 1) {
+          setTimeout(() => {
+            this.checkClientExistsWithRetry(clientId, retryCount + 1, maxRetries, retryInterval).then((exists2) => resolve(exists2));
+          }, retryInterval);
+        } else {
+          resolve(false);
+        }
+      }
+    });
+  }
   async handleMessage(event) {
     let message;
     if (event.data) {
@@ -170,8 +295,130 @@ var Client = class {
     }
   }
 };
+
+// src/aichat/AIChatClient.ts
+var AIChatNameEnum = /* @__PURE__ */ ((AIChatNameEnum2) => {
+  AIChatNameEnum2["AI_CHAT_CLIENT_ID"] = "aichat";
+  AIChatNameEnum2["AI_CHAT_INTERNAL_COMM_TYPE"] = "pubsub";
+  AIChatNameEnum2["SET_PARENT_NAME"] = "setParentName";
+  AIChatNameEnum2["SET_USER_ID"] = "setUserId";
+  AIChatNameEnum2["SET_CUSTOM_HEADERS"] = "setCustomHeaders";
+  AIChatNameEnum2["SET_CUSTOM_PAYLOAD"] = "setCustomPayload";
+  AIChatNameEnum2["SET_SUGGESTIONS"] = "setSuggestions";
+  AIChatNameEnum2["SHOW_SUGGESTIONS"] = "showSuggestions";
+  AIChatNameEnum2["SHOW_CHAT_MESSAGE"] = "showChatMessage";
+  AIChatNameEnum2["CONFIRM_ACTION"] = "confirmAction";
+  AIChatNameEnum2["SHOW_OPTIONS"] = "showOptions";
+  AIChatNameEnum2["SEND_CHAT_PROMPT"] = "sendChatPrompt";
+  AIChatNameEnum2["DEFAULT"] = "default";
+  return AIChatNameEnum2;
+})(AIChatNameEnum || {});
+var AIChatClient = class extends Client {
+  /**
+   * Create a new client instance.
+   * 
+   * @param pageId The ID of the page or component to register to.
+   */
+  constructor(pageId) {
+    super(pageId);
+  }
+  sendAIChatMethod(methodName, arg) {
+    this.sendMessage(
+      "aichat" /* AI_CHAT_CLIENT_ID */,
+      // This name is hardcode, do not change it
+      {
+        type: "pubsub" /* AI_CHAT_INTERNAL_COMM_TYPE */,
+        // This name is hardcode, do not change it
+        methodName,
+        arg
+      }
+    );
+  }
+  /**
+   * Set the parent name then the AI Chat can know who should it communicate with.
+   * 
+   * @param parentName The name of the parent page.
+   */
+  setParentName(parentName) {
+    this.sendAIChatMethod("setParentName" /* SET_PARENT_NAME */, parentName);
+  }
+  /**
+   * Set the user ID for the AI Chat component.
+   * 
+   * @param userId The user ID.
+   */
+  setUserId(userId) {
+    this.sendAIChatMethod("setUserId" /* SET_USER_ID */, userId);
+  }
+  /**
+   * Set custom headers for the AI Chat component.
+   * 
+   * @param headers The custom headers.
+   */
+  setCustomHeaders(headers) {
+    this.sendAIChatMethod("setCustomHeaders" /* SET_CUSTOM_HEADERS */, headers);
+  }
+  /**
+   * Set custom payload for the AI Chat component.
+   * 
+   * @param payload The custom payload.
+   */
+  setCustomPayload(payload) {
+    this.sendAIChatMethod("setCustomPayload" /* SET_CUSTOM_PAYLOAD */, payload);
+  }
+  /**
+   * Set suggestions for the AI Chat component.
+   * 
+   * @param suggestions The suggestions to set.
+   */
+  setSuggestions(suggestions) {
+    this.sendAIChatMethod("setSuggestions" /* SET_SUGGESTIONS */, suggestions);
+  }
+  /**
+   * Show suggestions in the AI Chat component.
+   * 
+   * @param show Whether to show suggestions.
+   */
+  showSuggestions(show) {
+    this.sendAIChatMethod("showSuggestions" /* SHOW_SUGGESTIONS */, show);
+  }
+  /**
+   * Show a chat message in the AI Chat component.
+   * 
+   * @param message The message to show.
+   */
+  showChatMessage(message) {
+    this.sendAIChatMethod("showChatMessage" /* SHOW_CHAT_MESSAGE */, message);
+  }
+  /**
+   * Confirm an action in the AI Chat component.
+   * 
+   * @param action The action to confirm.
+   */
+  confirmAction(action) {
+    this.sendAIChatMethod("confirmAction" /* CONFIRM_ACTION */, action);
+  }
+  /**
+   * Show options in the AI Chat component.
+   * 
+   * @param options The options to show.
+   */
+  showOptions(options) {
+    this.sendAIChatMethod("showOptions" /* SHOW_OPTIONS */, options);
+  }
+  /**
+   * Send a chat prompt to the AI Chat component.
+   * 
+   * @param prompt The prompt to send.
+   */
+  sendChatPrompt(prompt) {
+    this.sendAIChatMethod("sendChatPrompt" /* SEND_CHAT_PROMPT */, prompt);
+  }
+};
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AIChatClient,
+  AIChatNameEnum,
   Client,
   PubSub
 });

package/dist/index.mjs

@@ -2,7 +2,7 @@ var __defProp = Object.defineProperty;
 var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
 var __publicField = (obj, key, value) => __defNormalProp(obj, typeof key !== "symbol" ? key + "" : key, value);
 
-// src/index.ts
+// src/PubSub.ts
 var _PubSub = class _PubSub {
   constructor() {
     __publicField(this, "subscribers");
@@ -42,12 +42,31 @@ var _PubSub = class _PubSub {
     if (this.mainCallback) {
       this.mainCallback(message);
     }
+    this.trySendMessageWithRetry(message, 0);
+  }
+  /**
+   * Try to send a message to a client with retry logic.
+   * Will retry up to 3 times with 1 second delay between retries.
+   * 
+   * @param message The message to send.
+   * @param retryCount The current retry count.
+   */
+  trySendMessageWithRetry(message, retryCount) {
     const subscriber = this.subscribers.get(message.to);
-    if (!subscriber) return;
-    if (subscriber.source) {
-      subscriber.source.postMessage(message, "*");
+    if (subscriber) {
+      if (subscriber.source) {
+        subscriber.source.postMessage(message, "*");
+      } else {
+        subscriber.callback(message);
+      }
+      return;
+    }
+    if (retryCount < 10) {
+      setTimeout(() => {
+        this.trySendMessageWithRetry(message, retryCount + 1);
+      }, 1e3);
     } else {
-      subscriber.callback(message);
+      console.warn(`Failed to send message to client ${message.to} after 10 retries`);
     }
   }
   async handleMessage(event) {
@@ -63,6 +82,19 @@ var _PubSub = class _PubSub {
       });
       return;
     }
+    if (data?.type === "UNREGISTER") {
+      const unregistration = data;
+      this.subscribers.delete(unregistration.pageId);
+      return;
+    }
+    if (data?.type === "CLIENT_EXISTS_CHECK") {
+      source.postMessage({
+        type: "CLIENT_EXISTS_RESPONSE",
+        requestId: data.requestId,
+        exists: this.subscribers.has(data.clientId)
+      }, "*");
+      return;
+    }
     if (!data || !data.from || !data.to) return;
     const message = data;
     if (this.mainCallback) {
@@ -76,9 +108,20 @@ var _PubSub = class _PubSub {
       await subscriber.callback(message);
     }
   }
+  /**
+   * Check if a client with the given ID exists in the PubSub system.
+   * 
+   * @param clientId The ID of the client to check.
+   * @returns True if the client exists, false otherwise.
+   */
+  isClientExists(clientId) {
+    return this.subscribers.has(clientId);
+  }
 };
 __publicField(_PubSub, "instance");
 var PubSub = _PubSub;
+
+// src/Client.ts
 var Client = class {
   /**
    * Create a new client instance.
@@ -103,6 +146,19 @@ var Client = class {
       this.pubsub.register(pageId, this.handleMessage.bind(this));
     }
   }
+  /**
+   * Unregister the client from the pubsub.
+   */
+  unregister() {
+    if (this.isIframe) {
+      window.parent.postMessage({
+        type: "UNREGISTER",
+        pageId: this.pageId
+      }, "*");
+    } else {
+      this.pubsub.unregister(this.pageId);
+    }
+  }
   /**
    * Listen for messages from the parent page with a callback.
    * 
@@ -129,6 +185,71 @@ var Client = class {
       this.pubsub.sendMessage(message);
     }
   }
+  /**
+   * Check if a client with the given ID exists in the PubSub system.
+   * 
+   * @param clientId The ID of the client to check.
+   * @param maxRetries Maximum number of retries. Default is 3.
+   * @param retryInterval Interval between retries in milliseconds. Default is 1000ms.
+   * @returns A Promise that resolves to true if the client exists, false otherwise.
+   */
+  checkClientExists(clientId, maxRetries = 3, retryInterval = 1e3) {
+    return this.checkClientExistsWithRetry(clientId, 0, maxRetries, retryInterval);
+  }
+  /**
+   * Check if a client with the given ID exists in the PubSub system with retry mechanism.
+   * Will retry up to 3 times with 1 second delay between retries.
+   * 
+   * @param clientId The ID of the client to check.
+   * @param retryCount The current retry count.
+   * @returns A Promise that resolves to true if the client exists, false otherwise.
+   * @private
+   */
+  checkClientExistsWithRetry(clientId, retryCount, maxRetries, retryInterval) {
+    return new Promise((resolve) => {
+      if (this.isIframe) {
+        const requestId = `check-client-${Date.now()}-${Math.random().toString(36).substr(2, 9)}`;
+        const messageHandler = (event) => {
+          const data = event.data;
+          if (data && data.type === "CLIENT_EXISTS_RESPONSE" && data.requestId === requestId) {
+            window.removeEventListener("message", messageHandler);
+            clearTimeout(removeHandlerTimeout);
+            if (data.exists) {
+              resolve(true);
+            } else if (retryCount < maxRetries - 1) {
+              setTimeout(() => {
+                this.checkClientExistsWithRetry(clientId, retryCount + 1, maxRetries, retryInterval).then((exists) => resolve(exists));
+              }, retryInterval);
+            } else {
+              resolve(false);
+            }
+          }
+        };
+        window.addEventListener("message", messageHandler);
+        window.parent.postMessage({
+          type: "CLIENT_EXISTS_CHECK",
+          clientId,
+          requestId,
+          from: this.pageId
+        }, "*");
+        const removeHandlerTimeout = setTimeout(() => {
+          window.removeEventListener("message", messageHandler);
+          resolve(false);
+        }, retryInterval + 1e3);
+      } else {
+        const exists = this.pubsub.isClientExists(clientId);
+        if (exists) {
+          resolve(true);
+        } else if (retryCount < maxRetries - 1) {
+          setTimeout(() => {
+            this.checkClientExistsWithRetry(clientId, retryCount + 1, maxRetries, retryInterval).then((exists2) => resolve(exists2));
+          }, retryInterval);
+        } else {
+          resolve(false);
+        }
+      }
+    });
+  }
   async handleMessage(event) {
     let message;
     if (event.data) {
@@ -147,7 +268,129 @@ var Client = class {
     }
   }
 };
+
+// src/aichat/AIChatClient.ts
+var AIChatNameEnum = /* @__PURE__ */ ((AIChatNameEnum2) => {
+  AIChatNameEnum2["AI_CHAT_CLIENT_ID"] = "aichat";
+  AIChatNameEnum2["AI_CHAT_INTERNAL_COMM_TYPE"] = "pubsub";
+  AIChatNameEnum2["SET_PARENT_NAME"] = "setParentName";
+  AIChatNameEnum2["SET_USER_ID"] = "setUserId";
+  AIChatNameEnum2["SET_CUSTOM_HEADERS"] = "setCustomHeaders";
+  AIChatNameEnum2["SET_CUSTOM_PAYLOAD"] = "setCustomPayload";
+  AIChatNameEnum2["SET_SUGGESTIONS"] = "setSuggestions";
+  AIChatNameEnum2["SHOW_SUGGESTIONS"] = "showSuggestions";
+  AIChatNameEnum2["SHOW_CHAT_MESSAGE"] = "showChatMessage";
+  AIChatNameEnum2["CONFIRM_ACTION"] = "confirmAction";
+  AIChatNameEnum2["SHOW_OPTIONS"] = "showOptions";
+  AIChatNameEnum2["SEND_CHAT_PROMPT"] = "sendChatPrompt";
+  AIChatNameEnum2["DEFAULT"] = "default";
+  return AIChatNameEnum2;
+})(AIChatNameEnum || {});
+var AIChatClient = class extends Client {
+  /**
+   * Create a new client instance.
+   * 
+   * @param pageId The ID of the page or component to register to.
+   */
+  constructor(pageId) {
+    super(pageId);
+  }
+  sendAIChatMethod(methodName, arg) {
+    this.sendMessage(
+      "aichat" /* AI_CHAT_CLIENT_ID */,
+      // This name is hardcode, do not change it
+      {
+        type: "pubsub" /* AI_CHAT_INTERNAL_COMM_TYPE */,
+        // This name is hardcode, do not change it
+        methodName,
+        arg
+      }
+    );
+  }
+  /**
+   * Set the parent name then the AI Chat can know who should it communicate with.
+   * 
+   * @param parentName The name of the parent page.
+   */
+  setParentName(parentName) {
+    this.sendAIChatMethod("setParentName" /* SET_PARENT_NAME */, parentName);
+  }
+  /**
+   * Set the user ID for the AI Chat component.
+   * 
+   * @param userId The user ID.
+   */
+  setUserId(userId) {
+    this.sendAIChatMethod("setUserId" /* SET_USER_ID */, userId);
+  }
+  /**
+   * Set custom headers for the AI Chat component.
+   * 
+   * @param headers The custom headers.
+   */
+  setCustomHeaders(headers) {
+    this.sendAIChatMethod("setCustomHeaders" /* SET_CUSTOM_HEADERS */, headers);
+  }
+  /**
+   * Set custom payload for the AI Chat component.
+   * 
+   * @param payload The custom payload.
+   */
+  setCustomPayload(payload) {
+    this.sendAIChatMethod("setCustomPayload" /* SET_CUSTOM_PAYLOAD */, payload);
+  }
+  /**
+   * Set suggestions for the AI Chat component.
+   * 
+   * @param suggestions The suggestions to set.
+   */
+  setSuggestions(suggestions) {
+    this.sendAIChatMethod("setSuggestions" /* SET_SUGGESTIONS */, suggestions);
+  }
+  /**
+   * Show suggestions in the AI Chat component.
+   * 
+   * @param show Whether to show suggestions.
+   */
+  showSuggestions(show) {
+    this.sendAIChatMethod("showSuggestions" /* SHOW_SUGGESTIONS */, show);
+  }
+  /**
+   * Show a chat message in the AI Chat component.
+   * 
+   * @param message The message to show.
+   */
+  showChatMessage(message) {
+    this.sendAIChatMethod("showChatMessage" /* SHOW_CHAT_MESSAGE */, message);
+  }
+  /**
+   * Confirm an action in the AI Chat component.
+   * 
+   * @param action The action to confirm.
+   */
+  confirmAction(action) {
+    this.sendAIChatMethod("confirmAction" /* CONFIRM_ACTION */, action);
+  }
+  /**
+   * Show options in the AI Chat component.
+   * 
+   * @param options The options to show.
+   */
+  showOptions(options) {
+    this.sendAIChatMethod("showOptions" /* SHOW_OPTIONS */, options);
+  }
+  /**
+   * Send a chat prompt to the AI Chat component.
+   * 
+   * @param prompt The prompt to send.
+   */
+  sendChatPrompt(prompt) {
+    this.sendAIChatMethod("sendChatPrompt" /* SEND_CHAT_PROMPT */, prompt);
+  }
+};
 export {
+  AIChatClient,
+  AIChatNameEnum,
   Client,
   PubSub
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "iframe-pubsub",
-  "version": "1.0.0",
+  "version": "1.0.2",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",
   "types": "./dist/index.d.ts",