fix search

Author: Anchel123

app/components/code-graph.tsx

@@ -38,8 +38,6 @@ interface Props {
     setSearchNode: Dispatch<SetStateAction<any>>
     cooldownTicks: number | undefined
     setCooldownTicks: Dispatch<SetStateAction<number | undefined>>
-    cooldownTime: number
-    setCooldownTime: Dispatch<SetStateAction<number>>
     onCategoryClick: (name: string, show: boolean) => void
     handleDownloadImage: () => void
     zoomedNodes: Node[]
@@ -67,8 +65,6 @@ export function CodeGraph({
     setSearchNode,
     cooldownTicks,
     setCooldownTicks,
-    cooldownTime,
-    setCooldownTime,
     onCategoryClick,
     handleDownloadImage,
     zoomedNodes,
@@ -337,8 +333,6 @@ export function CodeGraph({
                                     setSelectedPathId={setSelectedPathId}
                                     cooldownTicks={cooldownTicks}
                                     setCooldownTicks={setCooldownTicks}
-                                    cooldownTime={cooldownTime}
-                                    setCooldownTime={setCooldownTime}
                                     setZoomedNodes={setZoomedNodes}
                                     zoomedNodes={zoomedNodes}
                                 />

app/components/graphView.tsx

@@ -30,8 +30,6 @@ interface Props {
     setSelectedPathId: (selectedPathId: number) => void
     cooldownTicks: number | undefined
     setCooldownTicks: Dispatch<SetStateAction<number | undefined>>
-    cooldownTime: number | undefined
-    setCooldownTime: Dispatch<SetStateAction<number>>
     setZoomedNodes: Dispatch<SetStateAction<Node[]>>
     zoomedNodes: Node[]
 }
@@ -56,9 +54,7 @@ export default function GraphView({
     selectedPathId,
     setSelectedPathId,
     cooldownTicks,
-    cooldownTime,
     setCooldownTicks,
-    setCooldownTime,
     zoomedNodes,
     setZoomedNodes
 }: Props) {
@@ -105,7 +101,6 @@ export default function GraphView({
     }, [parentRef])
 
     useEffect(() => {
-        setCooldownTime(2000)
         setCooldownTicks(undefined)
     }, [graph.Id, graph.getElements().length])
 
@@ -357,12 +352,12 @@ export default function GraphView({
                 onZoom={() => unsetSelectedObjects()}
                 onEngineStop={() => {
                     setCooldownTicks(0)
-                    setCooldownTime(0)
+                    debugger
                     handleZoomToFit(chartRef, zoomedNodes.length === 1 ? 4 : 1, (n: NodeObject<Node>) => zoomedNodes.some(node => node.id === n.id))
                     setZoomedNodes([])
                 }}
                 cooldownTicks={cooldownTicks}
-                cooldownTime={cooldownTime}
+                cooldownTime={6000}
             />
         </div>
     )

app/page.tsx

@@ -18,7 +18,7 @@ import Input from './components/Input';
 import { ForceGraphMethods, NodeObject } from 'react-force-graph-2d';
 import { Labels } from './components/labels';
 import { Toolbar } from './components/toolbar';
-import { cn, Message, Path, PathData, PathNode } from '@/lib/utils';
+import { cn, handleZoomToFit, Message, Path, PathData, PathNode } from '@/lib/utils';
 
 type Tip = {
   title: string
@@ -70,7 +70,6 @@ export default function Home() {
   const [chatOpen, setChatOpen] = useState(false)
   const [searchNode, setSearchNode] = useState<PathNode>({});
   const [cooldownTicks, setCooldownTicks] = useState<number | undefined>(0)
-  const [cooldownTime, setCooldownTime] = useState<number>(0)
   const [optionsOpen, setOptionsOpen] = useState(false)
   const [messages, setMessages] = useState<Message[]>([]);
   const [query, setQuery] = useState('');
@@ -191,22 +190,25 @@ export default function Home() {
     const chart = chartRef.current
 
     if (chart) {
-
       let chartNode = graph.Elements.nodes.find(n => n.id == node.id)
 
       if (!chartNode?.visible) {
         if (!chartNode) {
           chartNode = graph.extend({ nodes: [node], edges: [] }).nodes[0]
-        } else {
-          chartNode.visible = true
           setCooldownTicks(undefined)
-          setCooldownTime(1000)
+          setZoomedNodes([chartNode])
+          graph.visibleLinks(true, [chartNode!.id])
+          setData({ ...graph.Elements })
+          return
         }
+        chartNode.visible = true
         graph.visibleLinks(true, [chartNode!.id])
         setData({ ...graph.Elements })
       }
-      
-      setZoomedNodes([chartNode])
+
+      setTimeout(() => {
+        handleZoomToFit(chartRef, 4, (n: NodeObject<Node>) => n.id === chartNode!.id)
+      }, 0)
       setSearchNode(chartNode)
       setOptionsOpen(false)
     }
@@ -402,8 +404,6 @@ export default function Home() {
               setSearchNode={setSearchNode}
               cooldownTicks={cooldownTicks}
               setCooldownTicks={setCooldownTicks}
-              cooldownTime={cooldownTime}
-              setCooldownTime={setCooldownTime}
               onCategoryClick={onCategoryClick}
               handleDownloadImage={handleDownloadImage}
               zoomedNodes={zoomedNodes}
@@ -522,8 +522,6 @@ export default function Home() {
             searchNode={searchNode}
             cooldownTicks={cooldownTicks}
             setCooldownTicks={setCooldownTicks}
-            cooldownTime={cooldownTime}
-            setCooldownTime={setCooldownTime}
             onCategoryClick={onCategoryClick}
             handleDownloadImage={handleDownloadImage}
             zoomedNodes={zoomedNodes}