Improve search functionality

- Fixed: The search dialog now correctly receives focus when opened in the current block.
- Implemented direct search for selected text.

Author: lgbo-ustc

frontend/app/element/search.tsx

@@ -16,6 +16,7 @@ type SearchProps = SearchAtoms & {
     onSearch?: (search: string) => void;
     onNext?: () => void;
     onPrev?: () => void;
+    blockId?: string;
 };
 
 const SearchComponent = ({
@@ -32,6 +33,7 @@ const SearchComponent = ({
     onSearch,
     onNext,
     onPrev,
+    blockId,
 }: SearchProps) => {
     const [isOpen, setIsOpen] = useAtom<boolean>(isOpenAtom);
     const [search, setSearch] = useAtom<string>(searchAtom);
@@ -144,7 +146,7 @@ const SearchComponent = ({
         <>
             {isOpen && (
                 <FloatingPortal>
-                    <div className="search-container" style={{ ...floatingStyles }} ref={refs.setFloating}>
+                    <div className="search-container" style={{ ...floatingStyles }} ref={refs.setFloating} data-blockid={blockId}>
                         <Input
                             placeholder="Search"
                             value={search}
@@ -188,6 +190,7 @@ type SearchOptions = {
     regex?: boolean;
     caseSensitive?: boolean;
     wholeWord?: boolean;
+    blockId?: string;
 };
 
 export function useSearch(options?: SearchOptions): SearchProps {
@@ -212,7 +215,7 @@ export function useSearch(options?: SearchOptions): SearchProps {
             };
         }
     }, [options?.viewModel]);
-    return { ...searchAtoms, anchorRef };
+    return { ...searchAtoms, anchorRef, blockId: options?.blockId };
 }
 
 const createToggleButtonDecl = (

frontend/app/store/keymodel.ts

@@ -628,15 +628,86 @@ function registerGlobalKeys() {
             return true;
         });
     }
+    function getSelectedText(): string {
+        // Check for terminal selection first
+        const bcm = getBlockComponentModel(getFocusedBlockInStaticTab());
+        if (bcm?.viewModel?.viewType === "term") {
+            const termViewModel = bcm.viewModel as any;
+            if (termViewModel.termRef?.current?.terminal) {
+                const terminalSelection = termViewModel.termRef.current.terminal.getSelection();
+                if (terminalSelection && terminalSelection.length > 0) {
+                    return terminalSelection.trim();
+                }
+            }
+        }
+        // Check for regular text selection
+        const selection = window.getSelection();
+        if (selection && selection.rangeCount > 0 && !selection.isCollapsed) {
+            const selectedText = selection.toString().trim();
+            if (selectedText.length > 0) {
+                return selectedText;
+            }
+        }
+        return "";
+    }
+
+    function focusSearchInput() {
+        setTimeout(() => {
+            const blockId = getFocusedBlockInStaticTab();
+            if (!blockId) {
+                return;
+            }
+
+            // Directly find the search container by data-blockid attribute
+            const searchContainer = document.querySelector(
+                `.search-container[data-blockid="${blockId}"]`
+            ) as HTMLElement;
+            if (searchContainer) {
+                const searchInput = searchContainer.querySelector("input") as HTMLInputElement;
+                if (searchInput) {
+                    searchInput.focus();
+                    searchInput.select();
+                    return;
+                }
+            }
+
+            // Fallback: if we can't find the specific container, try to find any open search input
+            const searchInput = document.querySelector(".search-container input") as HTMLInputElement;
+            if (searchInput) {
+                searchInput.focus();
+                searchInput.select();
+            }
+        }, 0);
+    }
+
     function activateSearch(event: WaveKeyboardEvent): boolean {
         const bcm = getBlockComponentModel(getFocusedBlockInStaticTab());
         // Ctrl+f is reserved in most shells
         if (event.control && bcm.viewModel.viewType == "term") {
             return false;
         }
         if (bcm.viewModel.searchAtoms) {
-            globalStore.set(bcm.viewModel.searchAtoms.isOpen, true);
-            return true;
+            const searchAtoms = bcm.viewModel.searchAtoms;
+            const isOpen = globalStore.get(searchAtoms.isOpen);
+            const selectedText = getSelectedText();
+
+            if (isOpen) {
+                // Reset search dialog when already open
+                globalStore.set(searchAtoms.searchValue, selectedText || "");
+                globalStore.set(searchAtoms.resultsIndex, 0);
+                globalStore.set(searchAtoms.resultsCount, 0);
+                focusSearchInput();
+                return true;
+            } else {
+                // Open search dialog
+                globalStore.set(searchAtoms.isOpen, true);
+                // Set search value to selected text if available
+                if (selectedText) {
+                    globalStore.set(searchAtoms.searchValue, selectedText);
+                }
+                focusSearchInput();
+                return true;
+            }
         }
         return false;
     }

frontend/app/view/term/term.tsx

@@ -178,6 +178,7 @@ const TerminalView = ({ blockId, model }: ViewComponentProps<TermViewModel>) =>
         caseSensitive: false,
         wholeWord: false,
         regex: false,
+        blockId: blockId,
     });
     const searchIsOpen = jotai.useAtomValue<boolean>(searchProps.isOpen);
     const caseSensitive = useAtomValueSafe<boolean>(searchProps.caseSensitive);

frontend/app/view/webview/webview.tsx

@@ -821,7 +821,7 @@ const WebView = memo(({ model, onFailLoad, blockRef, initialSrc }: WebViewProps)
     }
 
     // Search
-    const searchProps = useSearch({ anchorRef: model.webviewRef, viewModel: model });
+    const searchProps = useSearch({ anchorRef: model.webviewRef, viewModel: model, blockId: model.blockId });
     const searchVal = useAtomValue<string>(searchProps.searchValue);
     const setSearchIndex = useSetAtom(searchProps.resultsIndex);
     const setNumSearchResults = useSetAtom(searchProps.resultsCount);