@xpert-ai/chatkit-web-component 0.4.1 → 0.5.1

package/dist/xpert-chatkit.js

@@ -47,7 +47,7 @@ var __privateGet = (obj, member, getter) => (__accessCheck(obj, member, "read fr
 var __privateAdd = (obj, member, value) => member.has(obj) ? __typeError("Cannot add the same private member more than once") : member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
 var __privateSet = (obj, member, value, setter) => (__accessCheck(obj, member, "write to private field"), setter ? setter.call(obj, value) : member.set(obj, value), value);
 var __privateMethod = (obj, member, method) => (__accessCheck(obj, member, "access private method"), method);
-var _a, _b, _setTrainingOptOut_dec, _hideHistory_dec, _showHistory_dec, _sendCustomAction_dec, _shareThread_dec, _setThreadId_dec, _setRuntimeCapabilities_dec, _setComposerValue_dec, _sendUserMessage_dec, _fetchUpdates_dec, _focusComposer_dec, _c, _opts, _frameUrl, _frame, _wrapper, _launcherCloseButton, _launcherOpen, _chatMinimizedToPet, _framePetOptionsOverride, _petClosedByContextMenu, _shadow, _petOverlay, _resolveLoaded, _loaded, _messenger, _ChatKitElementBase_instances, emitAndThrow_fn, setOptionsDataAttributes_fn, getDisplayMode_fn, getConfiguredPetOptions_fn, mergeConfiguredPetPositionDefaults_fn, getOverlayPetOptions_fn, resolvePetAssetUrl_fn, resolveOverlayPetOptions_fn, getFrameOptions_fn, setLauncherOpen_fn, setChatMinimizedToPet_fn, syncPetOverlayOptions_fn, handlePetActivate_fn, handlePetClose_fn, handlePetReply_fn, handlePetThreadSummaryActivate_fn, _handleLauncherClose, getFrameUrl_fn, setFrameUrl_fn, consumeFrameUrl_fn, _handleFrameLoad, _initialized, maybeInit_fn, _init;
+var _a, _b, _setTrainingOptOut_dec, _hideHistory_dec, _showHistory_dec, _sendCustomAction_dec, _shareThread_dec, _setThreadId_dec, _setRuntimeCapabilities_dec, _setComposerValue_dec, _sendUserMessage_dec, _fetchUpdates_dec, _focusComposer_dec, _c, _opts, _frameUrl, _frame, _wrapper, _launcherCloseButton, _launcherOpen, _chatMinimizedToPet, _framePetOptionsOverride, _petClosedByContextMenu, _shadow, _petOverlay, _resolveLoaded, _loaded, _channelId, _messenger, _ChatKitElementBase_instances, emitAndThrow_fn, setOptionsDataAttributes_fn, getDisplayMode_fn, getConfiguredPetOptions_fn, mergeConfiguredPetPositionDefaults_fn, getOverlayPetOptions_fn, resolvePetAssetUrl_fn, resolveOverlayPetOptions_fn, getFrameOptions_fn, setLauncherOpen_fn, setChatMinimizedToPet_fn, syncPetOverlayOptions_fn, handlePetActivate_fn, handlePetClose_fn, handlePetReply_fn, handlePetThreadSummaryActivate_fn, _handleLauncherClose, getFrameUrl_fn, setFrameUrl_fn, consumeFrameUrl_fn, _handleFrameLoad, _initialized, maybeInit_fn, _init;
 class EventEmitter {
   constructor() {
     __publicField(this, "callbacks", /* @__PURE__ */ new Map());
@@ -366,7 +366,7 @@ const fetchEventSourceWithRetry = async (url, params) => {
     }
   });
 };
-function safeRandomUUID() {
+function createSecureChannelId() {
   const cryptoRef = globalThis.crypto;
   if (typeof (cryptoRef == null ? void 0 : cryptoRef.randomUUID) === "function") {
     return cryptoRef.randomUUID();
@@ -378,7 +378,30 @@ function safeRandomUUID() {
     bytes[8] = bytes[8] & 63 | 128;
     return [...bytes].map((byte) => byte.toString(16).padStart(2, "0")).join("").replace(/^(.{8})(.{4})(.{4})(.{4})(.{12})$/, "$1-$2-$3-$4-$5");
   }
-  return `ck_${Date.now()}_${Math.random().toString(16).slice(2)}`;
+  return null;
+}
+function safeRandomUUID() {
+  return createSecureChannelId() ?? `ck_${Date.now()}_${Math.random().toString(16).slice(2)}`;
+}
+function isTrustedChatKitMessageEvent(event, { channelId, expectedOrigin, expectedSource }) {
+  const payload = event.data;
+  if (!payload || typeof payload !== "object" || !("__xpaiChatKit" in payload)) {
+    return false;
+  }
+  if (payload.__xpaiChatKit !== true) {
+    return false;
+  }
+  if (expectedOrigin !== "*" && event.origin !== expectedOrigin) {
+    return false;
+  }
+  if (event.source === expectedSource) {
+    return true;
+  }
+  const normalizedChannelId = channelId == null ? void 0 : channelId.trim();
+  if (normalizedChannelId) {
+    return "channelId" in payload && payload.channelId === normalizedChannelId;
+  }
+  return false;
 }
 let IntegrationError$1 = class IntegrationError extends Error {
   constructor(message) {
@@ -412,10 +435,12 @@ class BaseMessenger {
     handlers,
     target,
     targetOrigin,
+    channelId,
     fetch: fetch2 = window.fetch
   }) {
     __publicField(this, "targetOrigin");
     __publicField(this, "target");
+    __publicField(this, "channelId");
     __publicField(this, "commandHandlers");
     __publicField(this, "_fetch");
     __publicField(this, "emitter", new EventEmitter());
@@ -446,7 +471,11 @@ class BaseMessenger {
     ));
     __publicField(this, "handleMessage", async (event) => {
       var _a2, _b2, _c2, _d;
-      if (!event.data || event.data.__xpaiChatKit !== true || event.origin !== this.targetOrigin || event.source !== this.target()) {
+      if (!isTrustedChatKitMessageEvent(event, {
+        channelId: this.channelId,
+        expectedOrigin: this.targetOrigin,
+        expectedSource: this.target()
+      })) {
         return;
       }
       const data = event.data;
@@ -587,6 +616,7 @@ class BaseMessenger {
     this.commandHandlers = handlers;
     this.target = target;
     this.targetOrigin = targetOrigin;
+    this.channelId = (channelId == null ? void 0 : channelId.trim()) || void 0;
     this._fetch = (...args) => fetch2(...args);
   }
   setTargetOrigin(targetOrigin) {
@@ -596,6 +626,7 @@ class BaseMessenger {
     var _a2;
     const message = {
       __xpaiChatKit: true,
+      ...this.channelId ? { channelId: this.channelId } : {},
       ...data
     };
     (_a2 = this.target()) == null ? void 0 : _a2.postMessage(message, this.targetOrigin, transfer);
@@ -2603,7 +2634,9 @@ class ChatKitElementBase extends (_c = HTMLElement, _focusComposer_dec = [requir
     __privateAdd(this, _loaded, new Promise((resolve) => {
       __privateSet(this, _resolveLoaded, resolve);
     }));
+    __privateAdd(this, _channelId, createSecureChannelId());
     __privateAdd(this, _messenger, new ChatFrameMessenger({
+      channelId: __privateGet(this, _channelId) ?? void 0,
       fetch: (...args) => {
         var _a2;
         const customFetch = ((_a2 = __privateGet(this, _opts)) == null ? void 0 : _a2.api) && "fetch" in __privateGet(this, _opts).api && __privateGet(this, _opts).api.fetch;
@@ -2650,6 +2683,16 @@ class ChatKitElementBase extends (_c = HTMLElement, _focusComposer_dec = [requir
           }
           return onClientTool({ name, params, id, tool_call_id });
         },
+        onToolOutputAttachmentPreview: async (request) => {
+          var _a2, _b2;
+          const onRequestPreview = (_b2 = (_a2 = __privateGet(this, _opts)) == null ? void 0 : _a2.toolOutputAttachments) == null ? void 0 : _b2.onRequestPreview;
+          if (!onRequestPreview) {
+            __privateMethod(this, _ChatKitElementBase_instances, emitAndThrow_fn).call(this, new IntegrationError2(
+              "No handler for tool-output attachment previews. Add toolOutputAttachments.onRequestPreview to your ChatKit options."
+            ));
+          }
+          return onRequestPreview(request);
+        },
         onWorkbenchClientCommand: async ({
           commandKey,
           payload,
@@ -3092,6 +3135,7 @@ _shadow = new WeakMap();
 _petOverlay = new WeakMap();
 _resolveLoaded = new WeakMap();
 _loaded = new WeakMap();
+_channelId = new WeakMap();
 _messenger = new WeakMap();
 _ChatKitElementBase_instances = new WeakSet();
 emitAndThrow_fn = function(error) {
@@ -3302,7 +3346,8 @@ maybeInit_fn = function() {
   frameURL.hash = encodeBase64({
     options: getInnerOptions(__privateMethod(this, _ChatKitElementBase_instances, getFrameOptions_fn).call(this, __privateGet(this, _opts))),
     referrer: window.location.origin,
-    profile: this.profile
+    profile: this.profile,
+    ...__privateGet(this, _channelId) ? { channelId: __privateGet(this, _channelId) } : {}
   });
   __privateGet(this, _messenger).connect();
   __privateGet(this, _frame).src = frameURL.toString();