iframe-pubsub 1.0.1 → 1.0.2

package/dist/index.d.mts

@@ -10,61 +10,18 @@ interface IMessage {
     to: string;
     payload: any;
 }
-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"
+/**
+ * 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 PubSub {
-    private static instance;
-    private subscribers;
-    private mainCallback?;
-    private constructor();
-    static getInstance(): PubSub;
-    onMessage(callback: MessageCallback): void;
-    /**
-     * Registers a callback function to be called when a message is received.
-     *
-     * @param pageId The ID of the page or component to register for.
-     * @param callback The callback function to be called when a message is received.
-     */
-    register(pageId: string, callback: MessageCallback): void;
-    /**
-     * Unregister a page or component from the pubsub.
-     * Should unregister the registered page or component if it has been removed.
-     *
-     * @param pageId The ID of the page or component to unregister from.
-     */
-    unregister(pageId: string): void;
-    sendMessage(message: IMessage): void;
-    /**
-     * 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.
-     */
-    private trySendMessageWithRetry;
-    private handleMessage;
-    /**
-     * 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: string): boolean;
-}
+
 declare class Client {
     private pageId;
     private callback?;
@@ -93,6 +50,50 @@ declare class Client {
      * @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.
@@ -154,26 +155,46 @@ declare class Client {
      * @param prompt The prompt to send.
      */
     sendChatPrompt(prompt: string): void;
+}
+
+declare class PubSub {
+    private static instance;
+    private subscribers;
+    private mainCallback?;
+    private constructor();
+    static getInstance(): PubSub;
+    onMessage(callback: MessageCallback): void;
     /**
-     * Check if a client with the given ID exists in the PubSub system.
+     * Registers a callback function to be called when a message is received.
      *
-     * @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.
+     * @param pageId The ID of the page or component to register for.
+     * @param callback The callback function to be called when a message is received.
      */
-    checkClientExists(clientId: string, maxRetries?: number, retryInterval?: number): Promise<boolean>;
+    register(pageId: string, callback: MessageCallback): void;
     /**
-     * Check if a client with the given ID exists in the PubSub system with retry mechanism.
+     * Unregister a page or component from the pubsub.
+     * Should unregister the registered page or component if it has been removed.
+     *
+     * @param pageId The ID of the page or component to unregister from.
+     */
+    unregister(pageId: string): void;
+    sendMessage(message: IMessage): void;
+    /**
+     * Try to send a message to a client with retry logic.
      * Will retry up to 3 times with 1 second delay between retries.
      *
-     * @param clientId The ID of the client to check.
+     * @param message The message to send.
      * @param retryCount The current retry count.
-     * @returns A Promise that resolves to true if the client exists, false otherwise.
-     * @private
      */
-    private checkClientExistsWithRetry;
+    private trySendMessageWithRetry;
     private handleMessage;
+    /**
+     * 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: string): boolean;
 }
 
-export { AIChatNameEnum, Client, type IMessage, type MessageCallback, PubSub };
+export { AIChatClient, AIChatNameEnum, Client, type IMessage, type IRegistrationMessage, type MessageCallback, PubSub };

package/dist/index.d.ts

@@ -10,61 +10,18 @@ interface IMessage {
     to: string;
     payload: any;
 }
-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"
+/**
+ * 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 PubSub {
-    private static instance;
-    private subscribers;
-    private mainCallback?;
-    private constructor();
-    static getInstance(): PubSub;
-    onMessage(callback: MessageCallback): void;
-    /**
-     * Registers a callback function to be called when a message is received.
-     *
-     * @param pageId The ID of the page or component to register for.
-     * @param callback The callback function to be called when a message is received.
-     */
-    register(pageId: string, callback: MessageCallback): void;
-    /**
-     * Unregister a page or component from the pubsub.
-     * Should unregister the registered page or component if it has been removed.
-     *
-     * @param pageId The ID of the page or component to unregister from.
-     */
-    unregister(pageId: string): void;
-    sendMessage(message: IMessage): void;
-    /**
-     * 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.
-     */
-    private trySendMessageWithRetry;
-    private handleMessage;
-    /**
-     * 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: string): boolean;
-}
+
 declare class Client {
     private pageId;
     private callback?;
@@ -93,6 +50,50 @@ declare class Client {
      * @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.
@@ -154,26 +155,46 @@ declare class Client {
      * @param prompt The prompt to send.
      */
     sendChatPrompt(prompt: string): void;
+}
+
+declare class PubSub {
+    private static instance;
+    private subscribers;
+    private mainCallback?;
+    private constructor();
+    static getInstance(): PubSub;
+    onMessage(callback: MessageCallback): void;
     /**
-     * Check if a client with the given ID exists in the PubSub system.
+     * Registers a callback function to be called when a message is received.
      *
-     * @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.
+     * @param pageId The ID of the page or component to register for.
+     * @param callback The callback function to be called when a message is received.
      */
-    checkClientExists(clientId: string, maxRetries?: number, retryInterval?: number): Promise<boolean>;
+    register(pageId: string, callback: MessageCallback): void;
     /**
-     * Check if a client with the given ID exists in the PubSub system with retry mechanism.
+     * Unregister a page or component from the pubsub.
+     * Should unregister the registered page or component if it has been removed.
+     *
+     * @param pageId The ID of the page or component to unregister from.
+     */
+    unregister(pageId: string): void;
+    sendMessage(message: IMessage): void;
+    /**
+     * Try to send a message to a client with retry logic.
      * Will retry up to 3 times with 1 second delay between retries.
      *
-     * @param clientId The ID of the client to check.
+     * @param message The message to send.
      * @param retryCount The current retry count.
-     * @returns A Promise that resolves to true if the client exists, false otherwise.
-     * @private
      */
-    private checkClientExistsWithRetry;
+    private trySendMessageWithRetry;
     private handleMessage;
+    /**
+     * 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: string): boolean;
 }
 
-export { AIChatNameEnum, Client, type IMessage, type MessageCallback, PubSub };
+export { AIChatClient, AIChatNameEnum, Client, type IMessage, type IRegistrationMessage, type MessageCallback, PubSub };

package/dist/index.js

@@ -22,27 +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);
-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 || {});
+
+// src/PubSub.ts
 var _PubSub = class _PubSub {
   constructor() {
     __publicField(this, "subscribers");
@@ -160,6 +147,8 @@ var _PubSub = class _PubSub {
 };
 __publicField(_PubSub, "instance");
 var PubSub = _PubSub;
+
+// src/Client.ts
 var Client = class {
   /**
    * Create a new client instance.
@@ -223,6 +212,116 @@ 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) {
+      const evt = event;
+      message = evt.data;
+    } else {
+      message = event;
+    }
+    if (!message || !message.from || !message.to || message.to !== this.pageId) return;
+    if (this.callback) {
+      try {
+        await this.callback(message);
+      } catch (error) {
+        console.error(`Client ${this.pageId} failed to process message:`, error);
+      }
+    }
+  }
+};
+
+// 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 */,
@@ -315,91 +414,10 @@ var Client = class {
   sendChatPrompt(prompt) {
     this.sendAIChatMethod("sendChatPrompt" /* SEND_CHAT_PROMPT */, prompt);
   }
-  /**
-   * 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) {
-      const evt = event;
-      message = evt.data;
-    } else {
-      message = event;
-    }
-    if (!message || !message.from || !message.to || message.to !== this.pageId) return;
-    if (this.callback) {
-      try {
-        await this.callback(message);
-      } catch (error) {
-        console.error(`Client ${this.pageId} failed to process message:`, error);
-      }
-    }
-  }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  AIChatClient,
   AIChatNameEnum,
   Client,
   PubSub

package/dist/index.mjs

@@ -2,23 +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
-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 || {});
+// src/PubSub.ts
 var _PubSub = class _PubSub {
   constructor() {
     __publicField(this, "subscribers");
@@ -136,6 +120,8 @@ var _PubSub = class _PubSub {
 };
 __publicField(_PubSub, "instance");
 var PubSub = _PubSub;
+
+// src/Client.ts
 var Client = class {
   /**
    * Create a new client instance.
@@ -199,6 +185,116 @@ 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) {
+      const evt = event;
+      message = evt.data;
+    } else {
+      message = event;
+    }
+    if (!message || !message.from || !message.to || message.to !== this.pageId) return;
+    if (this.callback) {
+      try {
+        await this.callback(message);
+      } catch (error) {
+        console.error(`Client ${this.pageId} failed to process message:`, error);
+      }
+    }
+  }
+};
+
+// 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 */,
@@ -291,90 +387,9 @@ var Client = class {
   sendChatPrompt(prompt) {
     this.sendAIChatMethod("sendChatPrompt" /* SEND_CHAT_PROMPT */, prompt);
   }
-  /**
-   * 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) {
-      const evt = event;
-      message = evt.data;
-    } else {
-      message = event;
-    }
-    if (!message || !message.from || !message.to || message.to !== this.pageId) return;
-    if (this.callback) {
-      try {
-        await this.callback(message);
-      } catch (error) {
-        console.error(`Client ${this.pageId} failed to process message:`, error);
-      }
-    }
-  }
 };
 export {
+  AIChatClient,
   AIChatNameEnum,
   Client,
   PubSub

package/package.json

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