@wabbit-dashboard/embed 1.0.17 → 1.1.1

package/dist/wabbit-embed.umd.js

@@ -122,6 +122,143 @@
         };
     }
 
+    /**
+     * localStorage wrapper with type safety
+     */
+    /**
+     * Memory storage implementation (clears on page reload)
+     */
+    class MemoryStorage {
+        constructor() {
+            this.store = new Map();
+        }
+        getItem(key) {
+            return this.store.get(key) ?? null;
+        }
+        setItem(key, value) {
+            this.store.set(key, value);
+        }
+        removeItem(key) {
+            this.store.delete(key);
+        }
+        clear() {
+            this.store.clear();
+        }
+        get length() {
+            return this.store.size;
+        }
+        key(index) {
+            const keys = Array.from(this.store.keys());
+            return keys[index] ?? null;
+        }
+    }
+    /**
+     * Safe storage wrapper
+     */
+    class SafeStorage {
+        constructor(storage = localStorage, prefix = 'wabbit_') {
+            this.storage = storage;
+            this.prefix = prefix;
+        }
+        /**
+         * Get item from storage
+         */
+        get(key) {
+            try {
+                const item = this.storage.getItem(this.prefix + key);
+                if (item === null) {
+                    return null;
+                }
+                return JSON.parse(item);
+            }
+            catch (error) {
+                console.error(`[Wabbit] Failed to get storage item "${key}":`, error);
+                return null;
+            }
+        }
+        /**
+         * Set item in storage
+         */
+        set(key, value) {
+            try {
+                this.storage.setItem(this.prefix + key, JSON.stringify(value));
+                return true;
+            }
+            catch (error) {
+                console.error(`[Wabbit] Failed to set storage item "${key}":`, error);
+                return false;
+            }
+        }
+        /**
+         * Remove item from storage
+         */
+        remove(key) {
+            this.storage.removeItem(this.prefix + key);
+        }
+        /**
+         * Clear all items with prefix
+         */
+        clear() {
+            if (this.storage.length !== undefined && this.storage.key) {
+                const keys = [];
+                for (let i = 0; i < this.storage.length; i++) {
+                    const key = this.storage.key(i);
+                    if (key && key.startsWith(this.prefix)) {
+                        keys.push(key);
+                    }
+                }
+                keys.forEach((key) => this.storage.removeItem(key));
+            }
+            else {
+                // Fallback: try to clear common keys
+                const commonKeys = ['session_id', 'email_capture_dismissed'];
+                commonKeys.forEach((key) => this.remove(key));
+            }
+        }
+    }
+    // Export singleton instance (for backward compatibility)
+    const storage = new SafeStorage();
+    /**
+     * Create a storage instance with the specified backend
+     *
+     * @param persistSession - Whether to persist sessions across browser sessions
+     * @returns SafeStorage instance with appropriate backend
+     */
+    function createStorage(persistSession = true) {
+        let storageBackend;
+        if (persistSession) {
+            // Use localStorage (persists across browser sessions)
+            storageBackend = typeof window !== 'undefined' ? window.localStorage : new MemoryStorage();
+        }
+        else {
+            // Use sessionStorage (clears when browser closes)
+            storageBackend = typeof window !== 'undefined' ? window.sessionStorage : new MemoryStorage();
+        }
+        return new SafeStorage(storageBackend);
+    }
+    /**
+     * Get item from storage (simple string getter)
+     */
+    function getStorageItem(key) {
+        try {
+            return localStorage.getItem('wabbit_' + key);
+        }
+        catch {
+            return null;
+        }
+    }
+    /**
+     * Set item in storage (simple string setter)
+     */
+    function setStorageItem(key, value) {
+        try {
+            localStorage.setItem('wabbit_' + key, value);
+        }
+        catch (error) {
+            console.error(`[Wabbit] Failed to set storage item "${key}":`, error);
+        }
+    }
+
     /**
      * Main Wabbit SDK class
      *
@@ -138,6 +275,8 @@
             this.formsWidget = null; // FormWidget | null
             this.emailCaptureWidget = null; // EmailCapture | null
             this.config = config;
+            // Create storage instance with appropriate backend based on persistSession
+            this.storage = createStorage(config.persistSession ?? true);
         }
         /**
          * Initialize the Wabbit SDK
@@ -217,7 +356,7 @@
                             }
                         }
                     };
-                    const chatWidget = new ChatWidget(chatConfig);
+                    const chatWidget = new ChatWidget(chatConfig, Wabbit.instance.storage);
                     chatWidget.init();
                     Wabbit.instance.chatWidget = chatWidget;
                 });
@@ -540,110 +679,21 @@
     }
     Wabbit.instance = null;
 
-    /**
-     * localStorage wrapper with type safety
-     */
-    /**
-     * Safe storage wrapper
-     */
-    class SafeStorage {
-        constructor(storage = localStorage, prefix = 'wabbit_') {
-            this.storage = storage;
-            this.prefix = prefix;
-        }
-        /**
-         * Get item from storage
-         */
-        get(key) {
-            try {
-                const item = this.storage.getItem(this.prefix + key);
-                if (item === null) {
-                    return null;
-                }
-                return JSON.parse(item);
-            }
-            catch (error) {
-                console.error(`[Wabbit] Failed to get storage item "${key}":`, error);
-                return null;
-            }
-        }
-        /**
-         * Set item in storage
-         */
-        set(key, value) {
-            try {
-                this.storage.setItem(this.prefix + key, JSON.stringify(value));
-                return true;
-            }
-            catch (error) {
-                console.error(`[Wabbit] Failed to set storage item "${key}":`, error);
-                return false;
-            }
-        }
-        /**
-         * Remove item from storage
-         */
-        remove(key) {
-            this.storage.removeItem(this.prefix + key);
-        }
-        /**
-         * Clear all items with prefix
-         */
-        clear() {
-            if (this.storage.length !== undefined && this.storage.key) {
-                const keys = [];
-                for (let i = 0; i < this.storage.length; i++) {
-                    const key = this.storage.key(i);
-                    if (key && key.startsWith(this.prefix)) {
-                        keys.push(key);
-                    }
-                }
-                keys.forEach((key) => this.storage.removeItem(key));
-            }
-            else {
-                // Fallback: try to clear common keys
-                const commonKeys = ['session_id', 'email_capture_dismissed'];
-                commonKeys.forEach((key) => this.remove(key));
-            }
-        }
-    }
-    // Export singleton instance
-    const storage = new SafeStorage();
-    /**
-     * Get item from storage (simple string getter)
-     */
-    function getStorageItem(key) {
-        try {
-            return localStorage.getItem('wabbit_' + key);
-        }
-        catch {
-            return null;
-        }
-    }
-    /**
-     * Set item in storage (simple string setter)
-     */
-    function setStorageItem(key, value) {
-        try {
-            localStorage.setItem('wabbit_' + key, value);
-        }
-        catch (error) {
-            console.error(`[Wabbit] Failed to set storage item "${key}":`, error);
-        }
-    }
-
     /**
      * WebSocket client for chat functionality
      *
      * Based on demo-website/src/lib/websocket.ts but without React dependencies
      */
     class ChatWebSocketClient {
-        constructor(apiKey, wsUrl, sessionId = null) {
+        constructor(apiKey, wsUrl, sessionId = null, storage) {
             this.ws = null;
             this.status = 'disconnected';
             this.reconnectAttempts = 0;
             this.maxReconnectAttempts = 3;
             this.reconnectDelay = 1000;
+            // Message queue for handling messages during disconnection
+            this.messageQueue = [];
+            this.MAX_QUEUE_SIZE = 10;
             // Event handlers
             this.onStatusChange = null;
             this.onWelcome = null;
@@ -651,8 +701,14 @@
             this.onMessageHistory = null;
             this.onError = null;
             this.onDisconnect = null;
+            this.onQueueOverflow = null;
+            // Streaming event handlers
+            this.onMessageStart = null;
+            this.onMessageChunk = null;
+            this.onMessageEnd = null;
             this.apiKey = apiKey;
             this.wsUrl = wsUrl;
+            this.storage = storage;
             this.sessionId = sessionId || this.getStoredSessionId();
         }
         setStatus(status) {
@@ -682,6 +738,8 @@
                     this.setStatus('connected');
                     this.reconnectAttempts = 0;
                     console.log('[Wabbit] WebSocket connected');
+                    // Flush any queued messages
+                    this.flushMessageQueue();
                 };
                 this.ws.onmessage = (event) => {
                     try {
@@ -719,9 +777,9 @@
             switch (data.type) {
                 case 'welcome':
                     this.sessionId = data.session_id;
-                    // Store session ID in localStorage
+                    // Store session ID in storage (localStorage or sessionStorage)
                     if (this.sessionId) {
-                        storage.set('session_id', this.sessionId);
+                        this.storage.set('session_id', this.sessionId);
                     }
                     if (this.onWelcome) {
                         this.onWelcome(data.session_id, data.collection_id, data.message || 'Connected');
@@ -740,7 +798,26 @@
                         this.onMessageHistory(messages);
                     }
                     break;
+                case 'assistant_message_start':
+                    // Streaming: Start of assistant message
+                    if (this.onMessageStart) {
+                        this.onMessageStart(data.message_id);
+                    }
+                    break;
+                case 'assistant_message_chunk':
+                    // Streaming: Chunk of assistant message
+                    if (this.onMessageChunk) {
+                        this.onMessageChunk(data.message_id, data.content);
+                    }
+                    break;
+                case 'assistant_message_end':
+                    // Streaming: End of assistant message
+                    if (this.onMessageEnd) {
+                        this.onMessageEnd(data.message_id, data.metadata);
+                    }
+                    break;
                 case 'assistant_message':
+                    // Non-streaming: Complete assistant message (backward compatibility)
                     if (this.onMessage) {
                         const message = {
                             id: data.message_id || crypto.randomUUID(),
@@ -762,18 +839,27 @@
             }
         }
         sendMessage(content, metadata) {
-            if (!this.ws || this.ws.readyState !== WebSocket.OPEN) {
-                if (this.onError) {
-                    this.onError('Not connected to chat service');
-                }
-                return;
-            }
             const message = {
                 type: 'message',
                 content,
                 metadata: metadata || {},
             };
-            this.ws.send(JSON.stringify(message));
+            // If connected, send immediately
+            if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+                this.ws.send(JSON.stringify(message));
+                return;
+            }
+            // If disconnected or reconnecting, queue the message
+            if (this.messageQueue.length >= this.MAX_QUEUE_SIZE) {
+                console.warn('[Wabbit] Message queue full, cannot queue message');
+                if (this.onQueueOverflow) {
+                    this.onQueueOverflow('Message queue full. Please wait for reconnection.');
+                }
+                return;
+            }
+            // Add to queue
+            this.messageQueue.push({ content, metadata });
+            console.log(`[Wabbit] Message queued (${this.messageQueue.length}/${this.MAX_QUEUE_SIZE})`);
         }
         disconnect() {
             if (this.ws) {
@@ -795,12 +881,50 @@
                 this.connect();
             }, delay);
         }
+        /**
+         * Flush queued messages after connection is established
+         */
+        flushMessageQueue() {
+            if (this.messageQueue.length === 0) {
+                return;
+            }
+            console.log(`[Wabbit] Flushing ${this.messageQueue.length} queued message(s)`);
+            // Send all queued messages in order
+            const queuedMessages = [...this.messageQueue];
+            this.messageQueue = [];
+            queuedMessages.forEach(({ content, metadata }) => {
+                if (this.ws && this.ws.readyState === WebSocket.OPEN) {
+                    const message = {
+                        type: 'message',
+                        content,
+                        metadata: metadata || {},
+                    };
+                    this.ws.send(JSON.stringify(message));
+                }
+            });
+        }
         clearSession() {
             this.sessionId = null;
-            storage.remove('session_id');
+            this.storage.remove('session_id');
+            // Clear message queue only when user explicitly starts new session
+            // Queue should persist across reconnection attempts
+            this.messageQueue = [];
+        }
+        /**
+         * Clear the message queue
+         * Only call this when explicitly abandoning queued messages
+         */
+        clearMessageQueue() {
+            this.messageQueue = [];
         }
         getStoredSessionId() {
-            return storage.get('session_id') || null;
+            return this.storage.get('session_id') || null;
+        }
+        /**
+         * Get current message queue size
+         */
+        getQueueSize() {
+            return this.messageQueue.length;
         }
     }
 
@@ -930,6 +1054,9 @@
             this.isWaitingForResponse = false;
             this.closeButton = null;
             this.eventCleanup = [];
+            this.streamingMessages = new Map();
+            this.streamingCleanupInterval = null;
+            this.STREAMING_TIMEOUT_MS = 30000; // 30 seconds timeout for stale streams
             this.options = options;
         }
         /**
@@ -1161,13 +1288,150 @@
         formatMessage(content) {
             // Escape HTML first
             let formatted = escapeHtml(content);
+            // Markdown links [text](url) - must come before URL auto-linking
+            formatted = formatted.replace(/\[([^\]]+)\]\(([^)]+)\)/g, '<a href="$2" target="_blank" rel="noopener noreferrer">$1</a>');
             // Simple markdown-like formatting
             formatted = formatted.replace(/\*\*(.+?)\*\*/g, '<strong>$1</strong>');
             formatted = formatted.replace(/\*(.+?)\*/g, '<em>$1</em>');
             formatted = formatted.replace(/`(.+?)`/g, '<code style="background: rgba(0,0,0,0.1); padding: 2px 4px; border-radius: 3px;">$1</code>');
+            // Auto-link URLs (not already inside an href attribute)
+            formatted = formatted.replace(/(?<!href=")(https?:\/\/[^\s<]+)/g, '<a href="$1" target="_blank" rel="noopener noreferrer">$1</a>');
+            // Auto-link email addresses
+            formatted = formatted.replace(/(?<!["\/])([a-zA-Z0-9._%+-]+@[a-zA-Z0-9.-]+\.[a-zA-Z]{2,})/g, '<a href="mailto:$1">$1</a>');
+            // Auto-link phone numbers (international format with +)
+            formatted = formatted.replace(/(?<!["\/])(\+\d[\d\s-]{7,})/g, (_match, phone) => {
+                const cleanPhone = phone.replace(/[\s-]/g, '');
+                return `<a href="tel:${cleanPhone}">${phone}</a>`;
+            });
             formatted = formatted.replace(/\n/g, '<br>');
             return formatted;
         }
+        /**
+         * Start a streaming assistant message
+         */
+        startStreamingMessage(messageId) {
+            if (!this.messagesContainer)
+                return;
+            // Create message container
+            const messageDiv = createElement('div', {
+                class: 'wabbit-chat-message wabbit-chat-message-assistant wabbit-chat-message-streaming',
+                'data-message-id': messageId,
+            });
+            const bubble = createElement('div', { class: 'wabbit-chat-message-bubble' });
+            const content = createElement('div', { class: 'wabbit-chat-message-content' });
+            // Add streaming cursor
+            content.innerHTML = '<span class="wabbit-chat-cursor">▋</span>';
+            bubble.appendChild(content);
+            messageDiv.appendChild(bubble);
+            this.messagesContainer.appendChild(messageDiv);
+            // Store reference with timestamp for timeout tracking
+            this.streamingMessages.set(messageId, {
+                element: messageDiv,
+                content: '',
+                startTime: Date.now()
+            });
+            // Start cleanup interval if not already running
+            if (!this.streamingCleanupInterval) {
+                this.streamingCleanupInterval = window.setInterval(() => {
+                    this.cleanupStaleStreamingMessages();
+                }, 5000); // Check every 5 seconds
+            }
+            this.scrollToBottom();
+        }
+        /**
+         * Append chunk to streaming message
+         */
+        appendToStreamingMessage(messageId, chunk) {
+            const streaming = this.streamingMessages.get(messageId);
+            if (!streaming) {
+                console.warn('[ChatPanel] No streaming message found for ID:', messageId);
+                return;
+            }
+            // Append to content
+            streaming.content += chunk;
+            // Update DOM
+            const contentDiv = streaming.element.querySelector('.wabbit-chat-message-content');
+            if (contentDiv) {
+                // Format the content and add cursor
+                contentDiv.innerHTML = this.formatMessage(streaming.content) + '<span class="wabbit-chat-cursor">▋</span>';
+            }
+            this.scrollToBottom();
+        }
+        /**
+         * Finish streaming message
+         */
+        finishStreamingMessage(messageId, metadata) {
+            const streaming = this.streamingMessages.get(messageId);
+            if (!streaming) {
+                // Don't warn - this is expected if cleanup already removed it or on error
+                return;
+            }
+            // Remove streaming class and cursor
+            streaming.element.classList.remove('wabbit-chat-message-streaming');
+            const contentDiv = streaming.element.querySelector('.wabbit-chat-message-content');
+            if (contentDiv) {
+                // Remove cursor, keep formatted content
+                contentDiv.innerHTML = this.formatMessage(streaming.content);
+            }
+            // Add to messages array
+            const message = {
+                id: messageId,
+                role: 'assistant',
+                content: streaming.content,
+                timestamp: new Date(),
+                metadata,
+            };
+            this.messages.push(message);
+            // Clean up
+            this.streamingMessages.delete(messageId);
+            this.scrollToBottom();
+        }
+        /**
+         * Cancel a streaming message (e.g., on error)
+         * Cleans up the streaming state without adding to message history
+         */
+        cancelStreamingMessage(messageId) {
+            const streaming = this.streamingMessages.get(messageId);
+            if (!streaming) {
+                return;
+            }
+            // Remove the streaming message element from DOM
+            streaming.element.remove();
+            // Clean up from map
+            this.streamingMessages.delete(messageId);
+        }
+        /**
+         * Cancel all active streaming messages
+         * Useful when connection drops or on error
+         */
+        cancelAllStreamingMessages() {
+            this.streamingMessages.forEach((streaming) => {
+                streaming.element.remove();
+            });
+            this.streamingMessages.clear();
+        }
+        /**
+         * Cleanup stale streaming messages that have exceeded timeout
+         * Runs periodically to prevent memory leaks from abandoned streams
+         */
+        cleanupStaleStreamingMessages() {
+            const now = Date.now();
+            const staleIds = [];
+            this.streamingMessages.forEach((streaming, messageId) => {
+                if (now - streaming.startTime > this.STREAMING_TIMEOUT_MS) {
+                    console.warn('[ChatPanel] Cleaning up stale streaming message:', messageId);
+                    streaming.element.remove();
+                    staleIds.push(messageId);
+                }
+            });
+            // Remove stale entries from map
+            staleIds.forEach(id => this.streamingMessages.delete(id));
+            // Stop cleanup interval if no more streaming messages
+            if (this.streamingMessages.size === 0 && this.streamingCleanupInterval) {
+                clearInterval(this.streamingCleanupInterval);
+                this.streamingCleanupInterval = null;
+            }
+        }
         /**
          * Handle send message
          */
@@ -1252,6 +1516,11 @@
             if (this.element) {
                 this.element.style.display = 'none';
             }
+            // Stop streaming cleanup interval when hidden to prevent memory leaks
+            if (this.streamingCleanupInterval) {
+                clearInterval(this.streamingCleanupInterval);
+                this.streamingCleanupInterval = null;
+            }
         }
         /**
          * Remove the panel from DOM and cleanup event listeners
@@ -1260,6 +1529,11 @@
             // Run all event cleanup functions
             this.eventCleanup.forEach((cleanup) => cleanup());
             this.eventCleanup = [];
+            // Clear streaming cleanup interval
+            if (this.streamingCleanupInterval) {
+                clearInterval(this.streamingCleanupInterval);
+                this.streamingCleanupInterval = null;
+            }
             if (this.element) {
                 this.element.remove();
                 this.element = null;
@@ -1581,6 +1855,20 @@
       line-height: 1.5;
     }
 
+    .wabbit-chat-message-content a {
+      color: var(--wabbit-primary);
+      text-decoration: underline;
+      text-underline-offset: 2px;
+    }
+
+    .wabbit-chat-message-content a:hover {
+      opacity: 0.8;
+    }
+
+    .wabbit-chat-message-user .wabbit-chat-message-content a {
+      color: inherit;
+    }
+
     /* Typing Indicator */
     .wabbit-chat-typing {
       display: flex;
@@ -1723,6 +2011,24 @@
       }
     }
 
+    /* Streaming Cursor */
+    .wabbit-chat-cursor {
+      display: inline-block;
+      color: var(--wabbit-primary);
+      font-weight: bold;
+      animation: wabbit-cursor-blink 1s infinite;
+      margin-left: 2px;
+    }
+
+    @keyframes wabbit-cursor-blink {
+      0%, 50% {
+        opacity: 1;
+      }
+      51%, 100% {
+        opacity: 0;
+      }
+    }
+
     .wabbit-chat-panel {
       animation: wabbit-fade-in 0.3s ease-out;
     }
@@ -1737,17 +2043,17 @@
 
     /* Inline Chat Panel - renders inside container instead of fixed position */
     .wabbit-chat-panel.wabbit-chat-panel-inline {
-      position: relative !important;
+      position: absolute !important;
+      top: 0;
+      left: 0;
+      right: 0;
+      bottom: 0;
       width: 100%;
       height: 100%;
-      min-height: 400px;
       max-height: none;
       max-width: none;
-      bottom: auto !important;
-      right: auto !important;
-      left: auto !important;
-      border-radius: 12px;
-      box-shadow: 0 4px 12px rgba(0, 0, 0, 0.15);
+      border-radius: 0;
+      box-shadow: none;
       animation: none; /* Disable slide-in animation for inline */
     }
 
@@ -1798,7 +2104,7 @@
      * Chat Widget - Main class that integrates all chat components
      */
     class ChatWidget {
-        constructor(config) {
+        constructor(config, storage) {
             this.wsClient = null;
             this.bubble = null;
             this.panel = null;
@@ -1808,6 +2114,7 @@
             this.onChatReadyCallback = null;
             this.chatReadyFired = false;
             this.config = config;
+            this.storage = storage;
             this.onChatReadyCallback = config.onChatReady || null;
         }
         /**
@@ -1836,8 +2143,8 @@
             this.setupThemeWatcher();
             // Create WebSocket client
             const wsUrl = this.getWebSocketUrl();
-            this.wsClient = new ChatWebSocketClient(this.config.apiKey || '', wsUrl, null // Will use stored session if available
-            );
+            this.wsClient = new ChatWebSocketClient(this.config.apiKey || '', wsUrl, null, // Will use stored session if available
+            this.storage);
             // Set up event handlers
             this.setupWebSocketHandlers();
             // Only create bubble for widget mode (not inline)
@@ -1930,6 +2237,26 @@
                     this.panel.setMessages(messages);
                 }
             };
+            // Streaming message handlers
+            this.wsClient.onMessageStart = (messageId) => {
+                console.log('[Wabbit] Streaming message started:', messageId);
+                if (this.panel) {
+                    this.panel.startStreamingMessage(messageId);
+                }
+            };
+            this.wsClient.onMessageChunk = (messageId, chunk) => {
+                if (this.panel) {
+                    this.panel.appendToStreamingMessage(messageId, chunk);
+                }
+            };
+            this.wsClient.onMessageEnd = (messageId, metadata) => {
+                console.log('[Wabbit] Streaming message ended:', messageId);
+                if (this.panel) {
+                    this.panel.finishStreamingMessage(messageId, metadata);
+                    this.panel.setWaitingForResponse(false);
+                }
+            };
+            // Non-streaming message handler (backward compatibility)
             this.wsClient.onMessage = (message) => {
                 if (this.panel) {
                     this.panel.addMessage(message);
@@ -1939,6 +2266,8 @@
             this.wsClient.onError = (error) => {
                 console.error('[Wabbit] Chat error:', error);
                 if (this.panel) {
+                    // Clean up any active streaming messages on error
+                    this.panel.cancelAllStreamingMessages();
                     this.panel.addSystemMessage(`Error: ${error}`);
                     this.panel.setWaitingForResponse(false);
                 }
@@ -1950,10 +2279,24 @@
             };
             this.wsClient.onDisconnect = () => {
                 if (this.panel) {
-                    this.panel.addSystemMessage('Disconnected from chat service');
+                    // Clean up any active streaming messages on disconnect
+                    this.panel.cancelAllStreamingMessages();
+                    // Check if there are queued messages
+                    const queueSize = this.wsClient.getQueueSize();
+                    if (queueSize > 0) {
+                        this.panel.addSystemMessage(`Disconnected from chat service. ${queueSize} message(s) will be sent when reconnected.`);
+                    }
+                    else {
+                        this.panel.addSystemMessage('Disconnected from chat service');
+                    }
                     this.panel.setDisabled(true);
                 }
             };
+            this.wsClient.onQueueOverflow = (message) => {
+                if (this.panel) {
+                    this.panel.addSystemMessage(message);
+                }
+            };
         }
         /**
          * Handle trigger type configuration