fix: re-search data initialization (#817)

RainyNight9 · web-flow · commit aae6984aa731 · 2025-07-25T14:43:27.000+08:00
diff --git a/src/components/Search/DocumentList.tsx b/src/components/Search/DocumentList.tsx
@@ -50,6 +50,8 @@ export const DocumentList: React.FC<DocumentListProps> = ({
   const taskIdRef = useRef(nanoid());
   const [data, setData] = useState<Data>({ list: [] });
 
+  const loadingFromRef = useRef<number>(-1);
+
   const querySourceTimeoutRef = useRef(querySourceTimeout);
   useEffect(() => {
     querySourceTimeoutRef.current = querySourceTimeout;
@@ -107,12 +109,24 @@ export const DocumentList: React.FC<DocumentListProps> = ({
       }
     }
 
+    
     console.log("_docs", from, queryStrings, response);
     const list = response?.hits ?? [];
-    const total = response?.total_hits ?? 0;
-    setTotal(total);
+    const allTotal = response?.total_hits ?? 0;
+    // set first select hover
+    if (from === 0 && list.length > 0) {
+      setSelectedItem(0)
+      getDocDetail(list[0]?.document);
+    } 
 
     if (taskId === taskIdRef.current) {
+      // Prevent the last data from being 0
+      setTotal((prevTotal) => {
+        if (list.length === 0) {
+          return data?.list?.length === 0 ? 0 : prevTotal;
+        }
+        return allTotal;
+      });
       setData((prev) => ({
         ...prev,
         list: prev.list.concat(list),
@@ -121,17 +135,32 @@ export const DocumentList: React.FC<DocumentListProps> = ({
 
     return {
       list: list,
-      hasMore: list.length === PAGE_SIZE && from + list.length < total,
+      hasMore: list.length === PAGE_SIZE && from + list.length < allTotal,
     };
   };
 
   const { loading } = useInfiniteScroll(
     (data) => {
-      const taskId = nanoid();
+      // Prevent repeated requests for the same from value
+      const currentFrom = data?.list?.length || 0;
+
+      // If it starts from 0, it means it is a new search, reset the anti-duplicate flag
+      if (currentFrom === 0) {
+        loadingFromRef.current = -1;
+      }
+
+      if (loadingFromRef.current === currentFrom) {
+        return Promise.resolve({ list: [], hasMore: false });
+      }
+
+      loadingFromRef.current = currentFrom;
 
+      const taskId = nanoid();
       taskIdRef.current = taskId;
 
-      return getData(taskId, data);
+      return getData(taskId, data).finally(() => {
+        loadingFromRef.current = -1; // reset
+      });
     },
     {
       target: containerRef,
@@ -163,6 +192,15 @@ export const DocumentList: React.FC<DocumentListProps> = ({
     setIsKeyboardMode(false);
   }, [isChatMode, input]);
 
+  useEffect(() => {
+    setTotal(0);
+    setData((prev) => ({
+      ...prev,
+      list: [],
+    }));
+    loadingFromRef.current = -1;
+  }, [input]);
+
   const { visibleContextMenu } = useSearchStore();
 
   const handleKeyDown = useCallback(
