vzcode 2.15.0 → 2.17.0

package/src/client/VZSidebar/AIChat/ThinkingScratchpad.tsx

@@ -1,10 +1,4 @@
-import {
-  memo,
-  useRef,
-  useEffect,
-  useCallback,
-  useState,
-} from 'react';
+import { memo } from 'react';
 import Markdown from 'react-markdown';
 import remarkGfm from 'remark-gfm';
 
@@ -17,115 +11,6 @@ const ThinkingScratchpadComponent = ({
   content,
   isVisible,
 }: ThinkingScratchpadProps) => {
-  const contentRef = useRef<HTMLDivElement>(null);
-  const [isUserScrolled, setIsUserScrolled] =
-    useState(false);
-  const [autoScrollEnabled, setAutoScrollEnabled] =
-    useState(true);
-  const scrollTimeoutRef =
-    useRef<ReturnType<typeof setTimeout>>();
-
-  // Check if the user is scrolled to the bottom
-  const isScrolledToBottom = useCallback(() => {
-    const container = contentRef.current;
-    if (!container) return true;
-
-    const threshold = 10; // Allow 10px tolerance for "at bottom"
-    const { scrollTop, scrollHeight, clientHeight } =
-      container;
-    return (
-      scrollHeight - scrollTop - clientHeight < threshold
-    );
-  }, []);
-
-  // Smooth scroll to bottom
-  const scrollToBottom = useCallback(() => {
-    if (!autoScrollEnabled) return;
-
-    const container = contentRef.current;
-    if (container) {
-      container.scrollTo({
-        top: container.scrollHeight,
-        behavior: 'smooth',
-      });
-    }
-  }, [autoScrollEnabled]);
-
-  // Handle scroll events to detect user manual scrolling
-  const handleScroll = useCallback(() => {
-    const container = contentRef.current;
-    if (!container) return;
-
-    const isAtBottom = isScrolledToBottom();
-
-    // Clear any existing timeout
-    if (scrollTimeoutRef.current) {
-      clearTimeout(scrollTimeoutRef.current);
-    }
-
-    // If user scrolled up from bottom, disable auto-scroll
-    if (!isAtBottom && !isUserScrolled) {
-      setIsUserScrolled(true);
-      setAutoScrollEnabled(false);
-    }
-
-    // If user scrolled back to bottom, re-enable auto-scroll after a brief delay
-    if (isAtBottom && isUserScrolled) {
-      scrollTimeoutRef.current = setTimeout(() => {
-        setIsUserScrolled(false);
-        setAutoScrollEnabled(true);
-      }, 300); // 300ms delay to prevent flickering
-    }
-  }, [isUserScrolled, isScrolledToBottom]);
-
-  // Auto-scroll when content changes, but only if auto-scroll is enabled
-  useEffect(() => {
-    if (
-      autoScrollEnabled &&
-      !isUserScrolled &&
-      isVisible &&
-      content
-    ) {
-      // Use a short debounce to prevent multiple scroll calls during rapid updates
-      if (scrollTimeoutRef.current) {
-        clearTimeout(scrollTimeoutRef.current);
-      }
-
-      scrollTimeoutRef.current = setTimeout(() => {
-        scrollToBottom();
-      }, 50); // 50ms debounce
-    }
-
-    return () => {
-      if (scrollTimeoutRef.current) {
-        clearTimeout(scrollTimeoutRef.current);
-      }
-    };
-  }, [
-    content,
-    autoScrollEnabled,
-    isUserScrolled,
-    isVisible,
-    scrollToBottom,
-  ]);
-
-  // Clean up timeout on unmount
-  useEffect(() => {
-    return () => {
-      if (scrollTimeoutRef.current) {
-        clearTimeout(scrollTimeoutRef.current);
-      }
-    };
-  }, []);
-
-  // Reset scroll state when component becomes visible
-  useEffect(() => {
-    if (isVisible) {
-      setIsUserScrolled(false);
-      setAutoScrollEnabled(true);
-    }
-  }, [isVisible]);
-
   if (!isVisible || !content) {
     return null;
   }
@@ -140,8 +25,9 @@ const ThinkingScratchpadComponent = ({
       </div>
       <div
         className="thinking-scratchpad-content"
-        ref={contentRef}
-        onScroll={handleScroll}
+        role="log"
+        aria-live="polite"
+        aria-relevant="additions"
       >
         <Markdown remarkPlugins={[remarkGfm]}>
           {content}

package/src/client/VZSidebar/AIChat/index.tsx

@@ -7,6 +7,9 @@ import {
 import { VZCodeContext } from '../../VZCodeContext';
 import { MessageList } from './MessageList';
 import { ChatInput } from './ChatInput';
+import { ExtendedVizChat } from '../../../types.js';
+import { useAutoScroll } from '../../hooks/useAutoScroll';
+import { JumpToLatestButton } from './JumpToLatestButton';
 import './styles.scss';
 
 const DEBUG = false;
@@ -50,13 +53,11 @@ export const AIChat = () => {
     content,
     aiChatMode,
     aiChatMessage,
-    isLoading,
     currentChatId,
     selectedChatId,
     setSelectedChatId,
     aiErrorMessage,
     setAIChatMode,
-    clearStoredAIPrompt,
     getStoredAIPrompt,
     setAIChatMessage,
     handleSendMessage,
@@ -64,17 +65,42 @@ export const AIChat = () => {
     navigateMessageHistoryUp,
     navigateMessageHistoryDown,
     resetMessageHistoryNavigation,
+    enableMinimalEditFlow,
   } = useContext(VZCodeContext);
 
+  // Use the new simplified auto-scroll hook
+  const {
+    containerRef: messagesContainerRef,
+    showJumpButton,
+    onNewEvent,
+    onJumpToLatest,
+    beforeRender,
+    afterRender,
+  } = useAutoScroll({ threshold: 24 });
+
   // Get the active chat ID and chat data
   // If selectedChatId is set, use it; otherwise, if no chat selected, don't default to currentChatId
   const activeChatId = selectedChatId;
   const currentChat = activeChatId
     ? content?.chats?.[activeChatId]
     : null;
-  const rawMessages = currentChat?.messages || [];
+  const rawMessages = useMemo(
+    () => currentChat?.messages || [],
+    [currentChat?.messages],
+  );
   const aiStatus = currentChat?.aiStatus;
   const aiScratchpad = currentChat?.aiScratchpad;
+  const currentStatus = (currentChat as ExtendedVizChat)
+    ?.currentStatus;
+
+  // Debug logging for AI status
+  DEBUG && console.log('AIChat: currentChat:', currentChat);
+  DEBUG && console.log('AIChat: aiStatus:', aiStatus);
+  DEBUG &&
+    console.log(
+      'AIChat: enableMinimalEditFlow:',
+      enableMinimalEditFlow,
+    );
 
   // Transform messages to ensure they have required id field - memoized to avoid recreation
   const messages = useMemo(
@@ -166,7 +192,10 @@ export const AIChat = () => {
   return (
     <div className="ai-chat-container">
       <div className="ai-chat-content">
-        <div className="ai-chat-messages-container">
+        <div
+          className="ai-chat-messages-container"
+          ref={messagesContainerRef}
+        >
           {isEmptyState ? (
             <div className="ai-chat-empty">
               <div className="ai-chat-empty-icon">✨</div>
@@ -198,7 +227,7 @@ export const AIChat = () => {
                             )
                           }
                         >
-                          "Explain how this works"
+                          &quot;Explain how this works&quot;
                         </button>
                         <button
                           className="ai-chat-suggested-prompt"
@@ -208,8 +237,8 @@ export const AIChat = () => {
                             )
                           }
                         >
-                          "How could I change it so that the
-                          circles are bigger?"
+                          &quot;How could I change it so
+                          that the circles are bigger?&quot;
                         </button>
                         <button
                           className="ai-chat-suggested-prompt"
@@ -219,7 +248,8 @@ export const AIChat = () => {
                             )
                           }
                         >
-                          "What does this function do?"
+                          &quot;What does this function
+                          do?&quot;
                         </button>
                         <button
                           className="ai-chat-suggested-prompt"
@@ -229,8 +259,8 @@ export const AIChat = () => {
                             )
                           }
                         >
-                          "How can I make this more
-                          accessible?"
+                          &quot;How can I make this more
+                          accessible?&quot;
                         </button>
                       </div>
                     </>
@@ -246,7 +276,8 @@ export const AIChat = () => {
                             )
                           }
                         >
-                          "Change the circles to squares"
+                          &quot;Change the circles to
+                          squares&quot;
                         </button>
                         <button
                           className="ai-chat-suggested-prompt"
@@ -256,8 +287,8 @@ export const AIChat = () => {
                             )
                           }
                         >
-                          "Add a button that toggles the
-                          animation"
+                          &quot;Add a button that toggles
+                          the animation&quot;
                         </button>
                         <button
                           className="ai-chat-suggested-prompt"
@@ -267,8 +298,8 @@ export const AIChat = () => {
                             )
                           }
                         >
-                          "Fix the CSS so the layout is
-                          responsive"
+                          &quot;Fix the CSS so the layout is
+                          responsive&quot;
                         </button>
                         <button
                           className="ai-chat-suggested-prompt"
@@ -278,8 +309,8 @@ export const AIChat = () => {
                             )
                           }
                         >
-                          "Refactor this function to use
-                          async/await"
+                          &quot;Refactor this function to
+                          use async/await&quot;
                         </button>
                       </div>
                     </>
@@ -297,9 +328,14 @@ export const AIChat = () => {
           ) : (
             <MessageList
               messages={messages}
-              isLoading={isLoading}
+              isLoading={false}
               chatId={selectedChatId || currentChatId}
               aiScratchpad={aiScratchpad}
+              currentStatus={currentStatus}
+              onNewEvent={onNewEvent}
+              onJumpToLatest={onJumpToLatest}
+              beforeRender={beforeRender}
+              afterRender={afterRender}
             />
           )}
           {aiErrorMessage && (
@@ -321,6 +357,11 @@ export const AIChat = () => {
               </div>
             </div>
           )}
+          {/* Jump to Latest Button */}
+          <JumpToLatestButton
+            visible={showJumpButton}
+            onClick={onJumpToLatest}
+          />
         </div>
       </div>
       <div className="ai-chat-input-fixed">
@@ -328,7 +369,6 @@ export const AIChat = () => {
           aiChatMessage={aiChatMessage}
           setAIChatMessage={setAIChatMessage}
           onSendMessage={handleSendMessageWithSelection}
-          isLoading={isLoading}
           focused={aiChatFocused}
           aiChatMode={aiChatMode}
           setAIChatMode={setAIChatMode}

package/src/client/VZSidebar/AIChat/styles.scss

@@ -10,6 +10,7 @@
   overflow: hidden;
   display: flex;
   flex-direction: column;
+  position: relative;
 }
 
 .ai-chat-messages-container {
@@ -19,6 +20,21 @@
   padding-bottom: 0;
   display: flex;
   flex-direction: column;
+  scroll-behavior: smooth;
+
+  &::-webkit-scrollbar {
+    width: 6px;
+  }
+
+  &::-webkit-scrollbar-track {
+    background: var(--vh-color-neutral-02);
+    border-radius: 3px;
+  }
+
+  &::-webkit-scrollbar-thumb {
+    background: var(--vh-color-neutral-04);
+    border-radius: 3px;
+  }
 }
 
 .ai-chat-input-fixed {
@@ -111,6 +127,125 @@
   }
 }
 
+// ============================================================================
+// Streaming Message Components
+// ============================================================================
+
+.streaming-message {
+  .text-chunk {
+    margin-bottom: 1rem;
+  }
+}
+
+.file-editing-indicator {
+  display: flex;
+  align-items: center;
+  margin: 0.25rem 0;
+  background-color: var(--vscode-editor-background);
+  border: 1px solid var(--vscode-panel-border);
+  border-radius: 4px;
+
+  .file-editing-header {
+    display: flex;
+    align-items: center;
+    gap: 0.5rem;
+    width: 100%;
+  }
+
+  .file-editing-icon {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+  }
+
+  .file-editing-text {
+    color: var(--vscode-foreground);
+    font-size: 0.9rem;
+    flex: 1;
+
+    code {
+      background-color: var(
+        --vscode-textCodeBlock-background
+      );
+      padding: 0.2rem 0.4rem;
+      border-radius: 3px;
+      font-family: var(--vscode-editor-font-family);
+    }
+
+    .file-editing-done-container {
+      display: flex;
+      align-items: center;
+      justify-content: space-between;
+      width: 100%;
+      gap: 0.5rem;
+    }
+
+    .file-editing-done-status {
+      display: flex;
+      align-items: center;
+      gap: 8px;
+      font-size: 16px;
+      font-weight: 500;
+      margin-left: 4px;
+    }
+
+    .file-editing-done-widgets {
+      display: flex;
+      align-items: center;
+      gap: 0.5rem;
+      flex-shrink: 0;
+
+      // // Override the margin-top for undo-button-container when inside file editing indicator
+      // .undo-button-container {
+      //   margin-top: 0;
+      //   margin-bottom: 0;
+      // }
+    }
+  }
+}
+
+.status-indicator {
+  display: flex;
+  align-items: center;
+  padding: 0.5rem;
+  margin: 0.5rem 0;
+  background-color: var(--vscode-notifications-background);
+  border-left: 3px solid
+    var(--vscode-progressBar-foreground);
+  border-radius: 0 4px 4px 0;
+
+  .status-content {
+    display: flex;
+    align-items: center;
+    gap: 0.5rem;
+  }
+
+  .status-icon {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+
+    .done-icon {
+      font-size: 1rem;
+    }
+  }
+
+  .status-text {
+    color: var(--vscode-foreground);
+    font-size: 0.9rem;
+    font-weight: 500;
+  }
+}
+
+@keyframes spin {
+  0% {
+    transform: rotate(0deg);
+  }
+  100% {
+    transform: rotate(360deg);
+  }
+}
+
 .ai-chat-list {
   margin-top: 20px;
 
@@ -126,24 +261,8 @@
 
 .ai-chat-messages {
   flex: 1;
-  overflow-y: auto;
   margin-bottom: 10px;
   padding-right: 5px;
-  scroll-behavior: smooth;
-
-  &::-webkit-scrollbar {
-    width: 6px;
-  }
-
-  &::-webkit-scrollbar-track {
-    background: var(--vh-color-neutral-02);
-    border-radius: 3px;
-  }
-
-  &::-webkit-scrollbar-thumb {
-    background: var(--vh-color-neutral-04);
-    border-radius: 3px;
-  }
 }
 
 .ai-chat-message {
@@ -200,33 +319,6 @@
   line-height: 1.4;
   word-wrap: break-word;
 
-  .undo-button-container {
-    display: flex;
-    justify-content: flex-end;
-    gap: 8px;
-    margin-top: 16px;
-  }
-
-  .undo-button {
-    background-color: #dc3545;
-    color: white;
-    border: none;
-    border-radius: 4px;
-    padding: 4px 8px;
-    font-size: 12px;
-    cursor: pointer;
-    transition: background-color 0.2s;
-
-    &:hover:not(:disabled) {
-      background-color: #c82333;
-    }
-
-    &:disabled {
-      background-color: #6c757d;
-      cursor: not-allowed;
-    }
-  }
-
   // Markdown styling for dark backgrounds
   pre {
     background-color: rgba(0, 0, 0, 0.3);
@@ -549,6 +641,61 @@
   }
 }
 
+// Jump to latest button styles
+.jump-to-latest-button {
+  position: absolute;
+  left: 50%;
+  transform: translateX(-50%);
+  bottom: 20px; // Position above the input area
+  width: 44px;
+  height: 44px;
+  border-radius: 50%;
+  background: var(--vh-color-primary-01);
+  border: none;
+  color: white;
+  box-shadow: 0 2px 8px rgba(0, 0, 0, 0.3);
+  cursor: pointer;
+  display: flex;
+  align-items: center;
+  justify-content: center;
+  z-index: 10;
+  transition: all 0.2s ease;
+
+  &:hover {
+    background: var(--vh-color-primary-02);
+    box-shadow: 0 4px 12px rgba(0, 0, 0, 0.4);
+    transform: translateX(-50%) scale(1.05);
+  }
+
+  &:focus {
+    outline: 2px solid var(--vh-color-primary-01);
+    outline-offset: 2px;
+  }
+
+  &:active {
+    transform: translateX(-50%) scale(0.95);
+  }
+
+  .jump-to-latest-icon {
+    font-size: 16px;
+    font-weight: bold;
+    line-height: 1;
+  }
+
+  // Respect prefers-reduced-motion
+  @media (prefers-reduced-motion: reduce) {
+    transition: none;
+
+    &:hover {
+      transform: translateX(-50%);
+    }
+
+    &:active {
+      transform: translateX(-50%);
+    }
+  }
+}
+
 // Thinking scratchpad styles
 .thinking-scratchpad {
   background: var(--vh-color-neutral-02);

package/src/client/VZSidebar/Item.tsx

@@ -140,11 +140,11 @@ export const Item = ({
 
   const handleMouseEnter = useCallback(() => {
     setIsHovered(true);
-  }, []);
+  }, [setIsHovered]);
 
   const handleMouseLeave = useCallback(() => {
     setIsHovered(false);
-  }, []);
+  }, [setIsHovered]);
 
   const onChange = useCallback(() => {
     setRenameValue(renameInputRef.current.value);

package/src/client/VZSidebar/Search.tsx

@@ -98,7 +98,7 @@ export const Search = () => {
     } else {
       setIsMounted(true);
     }
-  }, [pattern]);
+  }, [pattern, isMounted, setSearchResults, shareDBDoc]);
 
   const flattenResult = useCallback(
     (fileId: string, file: SearchFile) => {
@@ -111,19 +111,26 @@ export const Search = () => {
           : 'open',
       );
     },
-    [focusedIndex, focusedChildIndex],
+    [
+      setSearchFileVisibility,
+      shareDBDoc,
+      focusedChildIndex,
+    ],
   );
 
-  const closeResult = useCallback((fileId: VizFileId) => {
-    setSearchFileVisibility(shareDBDoc, fileId, 'closed');
-  }, []);
+  const closeResult = useCallback(
+    (fileId: VizFileId) => {
+      setSearchFileVisibility(shareDBDoc, fileId, 'closed');
+    },
+    [setSearchFileVisibility, shareDBDoc],
+  );
 
   const focusFileElement = useCallback(
     (fileId: VizFileId, index: number) => {
       setActiveFileId(fileId);
       setSearchFocusedIndex(index, null);
     },
-    [],
+    [setActiveFileId, setSearchFocusedIndex],
   );
 
   const handleKeyDown = (event) => {