feat: fix search

src/components/App/ActionsToolbar/GraphClear/index.tsx

@@ -5,11 +5,11 @@ import ClearIcon from '~/components/Icons/ClearIcon'
 import { useDataStore } from '~/stores/useDataStore'
 
 export const GraphClear = () => {
-  const { resetData } = useDataStore((s) => s)
+  const { resetGraph } = useDataStore((s) => s)
 
   return (
     <Tooltip content="Clear Graph" fontSize="13px" position="left">
-      <ClearButton href="" onClick={() => resetData()} size="medium" startIcon={<ClearIcon />} />
+      <ClearButton href="" onClick={() => resetGraph()} size="medium" startIcon={<ClearIcon />} />
     </Tooltip>
   )
 }

src/components/App/SideBar/FilterSearch/__tests__/index.tsx

@@ -48,6 +48,7 @@ describe('FilterSearch Component', () => {
     ;(useDataStore as jest.Mock).mockReturnValue({
       setFilters: mockSetFilters,
       fetchData: mockFetchData,
+      resetData: jest.fn(),
       setAbortRequests: mockSetAbortRequests,
     })
 

src/components/App/SideBar/FilterSearch/index.tsx

@@ -29,7 +29,7 @@ const defaultValues = {
 
 export const FilterSearch = ({ anchorEl, setAnchorEl, onClose }: Props) => {
   const [schemaAll, setSchemaAll] = useSchemaStore((s) => [s.schemas, s.setSchemas])
-  const { abortFetchData, resetGraph, setFilters } = useDataStore((s) => s)
+  const { abortFetchData, resetGraph, setFilters, resetData } = useDataStore((s) => s)
   const [selectedTypes, setSelectedTypes] = useState<string[]>(defaultValues.selectedTypes)
   const [hops, setHops] = useState(defaultValues.hops)
   const [sourceNodes, setSourceNodes] = useState<number>(defaultValues.sourceNodes)
@@ -74,13 +74,16 @@ export const FilterSearch = ({ anchorEl, setAnchorEl, onClose }: Props) => {
   }
 
   const handleFiltersApply = async () => {
+
     setFilters({
       node_type: selectedTypes,
       limit: maxResults,
       depth: hops.toString(),
       top_node_count: sourceNodes.toString(),
     })
 
+    resetData()
+
     setAnchorEl(null)
     onClose()
   }

src/components/App/index.tsx

@@ -79,6 +79,7 @@
     runningProjectId,
     setRunningProjectMessages,
     isFetching,
+    resetData,
   } = useDataStore((s) => s)
 
   const { setAiSummaryAnswer, getKeyExist, aiRefId } = useAiSummaryStore((s) => s)
@@ -128,8 +129,10 @@
       }
     }
 
+    resetData()
+
     runSearch()
-  }, [searchTerm, fetchData, setBudget, setAbortRequests, setSidebarOpen, setSelectedNode])
+  }, [searchTerm, fetchData, setBudget, setAbortRequests, setSidebarOpen, setSelectedNode, resetData])
 
   const handleNewNode = useCallback(() => {
     setNodeCount('INCREMENT')
@@ -276,7 +279,7 @@
   ])
 
   useEffect(() => {
     if (!runningProjectId || true) {
       return
     }
 

src/components/Auth/__tests__/index.tsx

@@ -56,6 +56,7 @@ describe('Auth Component', () => {
       setCategoryFilter: jest.fn(),
       setAbortRequests: jest.fn(),
       addNewNode: jest.fn(),
+      resetData: jest.fn(),
       splashDataLoading: false,
     })
 
@@ -208,13 +209,20 @@ describe('Auth Component', () => {
   })
 
   test.skip('the unauthorized state is correctly set when the user lacks proper credentials', async () => {
-    const [setBudget, setIsAdmin, setPubKey, setIsAuthenticated] = [jest.fn(), jest.fn(), jest.fn(), jest.fn()]
+    const [setBudget, setIsAdmin, setPubKey, setIsAuthenticated, resetData] = [
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+    ]
 
     useUserStoreMock.mockReturnValue({
       setBudget,
       setIsAdmin,
       setPubKey,
       setIsAuthenticated,
+      resetData,
     })
 
     // eslint-disable-next-line @typescript-eslint/ban-ts-comment
@@ -239,13 +247,20 @@ describe('Auth Component', () => {
   })
 
   test('test unsuccessful attempts to enable Sphinx', async () => {
-    const [setBudget, setIsAdmin, setPubKey, setIsAuthenticated] = [jest.fn(), jest.fn(), jest.fn(), jest.fn()]
+    const [setBudget, setIsAdmin, setPubKey, setIsAuthenticated, resetData] = [
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+    ]
 
     useUserStoreMock.mockReturnValue({
       setBudget,
       setIsAdmin,
       setPubKey,
       setIsAuthenticated,
+      resetData,
     })
 
     // eslint-disable-next-line @typescript-eslint/ban-ts-comment
@@ -271,13 +286,20 @@ describe('Auth Component', () => {
   })
 
   test('test the public key is set correctly on successful Sphinx enablement', async () => {
-    const [setBudget, setIsAdmin, setPubKey, setIsAuthenticated] = [jest.fn(), jest.fn(), jest.fn(), jest.fn()]
+    const [setBudget, setIsAdmin, setPubKey, setIsAuthenticated, resetData] = [
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+    ]
 
     useUserStoreMock.mockReturnValue({
       setBudget,
       setIsAdmin,
       setPubKey,
       setIsAuthenticated,
+      resetData,
     })
 
     // eslint-disable-next-line @typescript-eslint/ban-ts-comment
@@ -302,13 +324,20 @@ describe('Auth Component', () => {
   })
 
   test('test the public key state is handled correctly on Sphinx enablement failure', async () => {
-    const [setBudget, setIsAdmin, setPubKey, setIsAuthenticated] = [jest.fn(), jest.fn(), jest.fn(), jest.fn()]
+    const [setBudget, setIsAdmin, setPubKey, setIsAuthenticated, resetData] = [
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+    ]
 
     useUserStoreMock.mockReturnValue({
       setBudget,
       setIsAdmin,
       setPubKey,
       setIsAuthenticated,
+      resetData,
     })
 
     // eslint-disable-next-line @typescript-eslint/ban-ts-comment
@@ -333,13 +362,20 @@ describe('Auth Component', () => {
   })
 
   test('simulate errors during the authentication process and verify that they are handled gracefully.', async () => {
-    const [setBudget, setIsAdmin, setPubKey, setIsAuthenticated] = [jest.fn(), jest.fn(), jest.fn(), jest.fn()]
+    const [setBudget, setIsAdmin, setPubKey, setIsAuthenticated, resetData] = [
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+      jest.fn(),
+    ]
 
     useUserStoreMock.mockReturnValue({
       setBudget,
       setIsAdmin,
       setPubKey,
       setIsAuthenticated,
+      resetData,
     })
 
     // eslint-disable-next-line @typescript-eslint/ban-ts-comment

src/components/Universe/Graph/index.tsx

@@ -33,6 +33,8 @@
     (s) => s,
   )
 
+  const removeSimulation = useGraphStore((s) => s.removeSimulation)
+
   useEffect(() => {
     if (!dataNew) {
       return
@@ -57,6 +59,12 @@
     resetDataNew()
   }, [setData, dataNew, simulation, simulationCreate, resetDataNew, simulationHelpers, dataInitial])
 
+  useEffect(() => {
+    if (!dataInitial) {
+      removeSimulation()
+    }
+  }, [dataInitial, removeSimulation])
+
   useEffect(() => {
     if (!simulation) {
       return
@@ -82,7 +90,7 @@
 
         const sphereRadius = Math.min(5000, boundingSphere.radius)
 
         if (false) {
           setGraphRadius(sphereRadius)
           cameraSettled.current = true
         }
@@ -174,7 +182,7 @@
 
       const boundingBox = new Box3().setFromPoints(nodesVector)
 
       console.log(boundingBox)
     })
   }, [dataInitial, simulation, setGraphRadius, normalizedSchemasByType])
 

src/stores/useDataStore/index.ts

@@ -313,9 +313,15 @@
 
     resetData: () => {
       set({
-        dataNew: { nodes: [], links: [] },
-        dataInitial: { nodes: [], links: [] },
+        dataInitial: null,
+        sidebarFilter: 'all',
+        sidebarFilters: [],
+        sidebarFilterCounts: [],
+        dataNew: null,
+        runningProjectId: '',
         nodeTypes: [],
+        nodesNormalized: new Map<string, Node>(),
+        linksNormalized: new Map<string, Link>(),
       })
     },
 
@@ -346,11 +352,11 @@
     setHideNodeDetails: (hideNodeDetails) => set({ hideNodeDetails }),
     setSeedQuestions: (questions) => set({ seedQuestions: questions }),
     updateNode: (updatedNode) => {
       console.info(updatedNode)
     },
 
     removeNode: (id) => {
       console.log(id)
     },
 
     setRunningProjectId: (runningProjectId) => set({ runningProjectId, runningProjectMessages: [] }),

src/stores/useGraphStore/index.ts

@@ -96,6 +96,7 @@
   setSelectionData: (data: GraphData) => void
   simulationCreate: (nodes: Node[], links: Link[]) => void
   setIsHovering: (isHovering: boolean) => void
+  removeSimulation: () => void
 }
 
 const defaultData: Omit<
@@ -116,6 +117,7 @@
   | 'setHideNodeDetails'
   | 'simulationCreate'
   | 'setIsHovering'
+  | 'removeSimulation'
 > = {
   data: null,
   simulation: null,
@@ -294,7 +296,7 @@
     simulationRestart: () => {
       const { simulation } = get()
 
       if (false) {
         runSimulationPhase(simulation)
       }
 
@@ -318,6 +320,7 @@
 
     set({ simulation })
   },
+  removeSimulation: () => set({ simulation: null }),
 }))
 
 export const useSelectedNode = () => useGraphStore((s) => s.selectedNode)