vzcode 2.30.0 → 2.31.0

package/src/client/aiChatButtonEvent.ts

@@ -0,0 +1,61 @@
+/**
+ * Host-facing event emitted when the VZCode sidebar "Edit with AI"
+ * button is clicked in "emit" mode.
+ *
+ * A host opts in by passing `aiChatButtonEvent` to `VZSidebar` /
+ * `VZLeft`. When set, clicking `#ai-chat-icon` dispatches this event
+ * instead of opening VZCode's built-in AI chat panel, so the host can
+ * route the click to its own AI editing entry point (e.g. VizHub
+ * Studio's top-bar "AI Edit" action).
+ *
+ * The constant is exported so hosts don't hardcode the event name.
+ */
+export const AI_CHAT_BUTTON_EVENT =
+  'vzcode:ai-chat-button-click';
+
+export interface AIChatButtonEventDetail {
+  source: 'vzcode-sidebar';
+}
+
+/**
+ * Dispatch the host-facing AI chat button event.
+ * Kept as a standalone function so it is unit-testable without a DOM.
+ */
+export function emitAIChatButtonEvent(
+  eventName: string,
+  target: EventTarget,
+): void {
+  target.dispatchEvent(
+    new CustomEvent<AIChatButtonEventDetail>(eventName, {
+      bubbles: true,
+      composed: true,
+      detail: { source: 'vzcode-sidebar' },
+    }),
+  );
+}
+
+/**
+ * Decide what clicking the sidebar AI chat button should do.
+ *
+ * - When `aiChatButtonEvent` is set, emit the event and leave all
+ *   VZCode state untouched (the host owns the action).
+ * - Otherwise, fall back to opening VZCode's built-in AI chat.
+ *
+ * Extracted from the component so the branching is unit-testable
+ * without rendering or a real DOM.
+ */
+export function handleAIChatButtonClick({
+  aiChatButtonEvent,
+  target,
+  openAIChat,
+}: {
+  aiChatButtonEvent?: string;
+  target: EventTarget;
+  openAIChat: () => void;
+}): void {
+  if (aiChatButtonEvent) {
+    emitAIChatButtonEvent(aiChatButtonEvent, target);
+    return;
+  }
+  openAIChat();
+}

package/src/client/index.ts

@@ -11,6 +11,12 @@ export { VZKeyboardShortcutsDoc } from './VZKeyboardShortcutsDoc';
 export { VZResizer } from './VZResizer';
 export { VZLeft } from './VZLeft';
 export { VZMiddle } from './VZMiddle';
+export {
+  AI_CHAT_BUTTON_EVENT,
+  emitAIChatButtonEvent,
+  handleAIChatButtonClick,
+} from './aiChatButtonEvent';
+export type { AIChatButtonEventDetail } from './aiChatButtonEvent';
 export {
   SplitPaneResizeProvider,
   SplitPaneResizeContext,

package/src/client/main.jsx

@@ -3,6 +3,28 @@ import * as ReactDOM from 'react-dom/client';
 import App from './App';
 import { BrowserRouter } from 'react-router-dom';
 
+// Global error handler to catch CodeMirror DOM position errors
+// These can occur when CodeMirror instances are created but not yet
+// attached to the DOM, or when presence indicators try to render
+// on offscreen editor instances.
+window.addEventListener('error', (event) => {
+  // Check if this is the specific CodeMirror posFromDOM/posAtDOM error
+  if (
+    event.error instanceof RangeError &&
+    event.error.message.includes(
+      'Trying to find position for a DOM position outside of the document',
+    )
+  ) {
+    // Log the error for debugging purposes
+    console.log(
+      'Suppressed CodeMirror DOM position error:',
+      event.error,
+    );
+    // Prevent the error from being logged to console by returning true
+    return true;
+  }
+});
+
 ReactDOM.createRoot(document.getElementById('root')).render(
   <BrowserRouter>
     <App />

package/src/llm-streaming-server/chatOperations.ts

@@ -335,30 +335,66 @@ export const finalizeAIMessage = (
 };
 
 /**
- * Adds diff data to the most recent AI message
+ * Adds diff data to a specific AI message by ID
+ * Phase 3 Fix: Use explicit message ID instead of assuming
+ * last message is AI, preventing bugs with rapid message sends
  */
 export const addDiffToAIMessage = (
   shareDBDoc: ShareDBDoc<VizContent>,
   chatId: VizChatId,
   diffData: any,
-  beforeCommitId?: string, // Commit ID before AI changes for VizHub integration
+  beforeCommitId?: string, // Commit ID before AI changes
+  messageId?: string, // Explicit AI message ID to update
 ) => {
   const chat = shareDBDoc.data.chats[chatId];
+  if (!chat || !chat.messages) {
+    console.warn(
+      `Chat ${chatId} not found or has no messages`,
+    );
+    return;
+  }
+
   const messages = [...chat.messages];
+  let targetMessageIndex = -1;
 
-  // Find the most recent AI message
-  const lastAIMessageIndex = messages.length - 1;
-  if (
-    lastAIMessageIndex >= 0 &&
-    messages[lastAIMessageIndex].role === 'assistant'
-  ) {
+  // Phase 3 Fix: Use explicit message ID if provided
+  if (messageId) {
+    targetMessageIndex = messages.findIndex(
+      (msg) => msg.id === messageId,
+    );
+    if (
+      targetMessageIndex === -1 ||
+      messages[targetMessageIndex].role !== 'assistant'
+    ) {
+      console.warn(
+        `AI message with id ${messageId} not found in chat ${chatId}`,
+      );
+      return;
+    }
+  } else {
+    // Fallback: Find the most recent AI message (safer than
+    // just assuming last message)
+    for (let i = messages.length - 1; i >= 0; i--) {
+      if (messages[i].role === 'assistant') {
+        targetMessageIndex = i;
+        break;
+      }
+    }
+
+    if (targetMessageIndex === -1) {
+      console.warn(`No AI message found in chat ${chatId}`);
+      return;
+    }
+  }
+
+  // Update the target message with diff data
+  if (targetMessageIndex >= 0) {
     const newMessage = {
-      ...messages[lastAIMessageIndex],
+      ...messages[targetMessageIndex],
       diffData,
-      ...(beforeCommitId && { beforeCommitId }), // Add beforeCommitId for VizHub integration
+      ...(beforeCommitId && { beforeCommitId }),
     };
-    // Use type assertion to extend the message with diffData
-    (messages[lastAIMessageIndex] as any) = newMessage;
+    (messages[targetMessageIndex] as any) = newMessage;
 
     const messageOp = diff(shareDBDoc.data, {
       ...shareDBDoc.data,
@@ -373,6 +409,11 @@ export const addDiffToAIMessage = (
     });
 
     shareDBDoc.submitOp(messageOp);
+
+    DEBUG &&
+      console.log(
+        `addDiffToAIMessage: Successfully added diff to message ${targetMessageIndex} in chat ${chatId}`,
+      );
   }
 };
 

package/src/llm-streaming-server/llmStreaming.ts

@@ -78,7 +78,6 @@ export const createLLMFunction = ({
       },
     });
 
-
     let fullContent = '';
     let generationId = '';
     let currentEditingFileName = null;
@@ -253,8 +252,6 @@ export const createLLMFunction = ({
     const chunks = [];
     let reasoningContent = '';
 
-
-
     // Configure reasoning tokens based on enableReasoningTokens flag
     const requestConfig: any = {
       model: modelName,

package/src/llm-streaming-ui/components/Message.tsx

@@ -13,8 +13,6 @@ import { DiffView, DiffViewRef } from './DiffView';
 import { UnifiedFilesDiff } from '../../utils/fileDiff';
 import { enableDiffView } from '../../client/featureFlags';
 
-const DEBUG = true;
-
 interface MessageProps {
   id: string;
   role: 'user' | 'assistant';