move lightrag_webui folder to top directory

lightrag_webui/src/stores/graph.ts

@@ -0,0 +1,130 @@
+import { create } from 'zustand'
+import { createSelectors } from '@/lib/utils'
+import { DirectedGraph } from 'graphology'
+
+export type RawNodeType = {
+  id: string
+  labels: string[]
+  properties: Record<string, any>
+
+  size: number
+  x: number
+  y: number
+  color: string
+
+  degree: number
+}
+
+export type RawEdgeType = {
+  id: string
+  source: string
+  target: string
+  type: string
+  properties: Record<string, any>
+
+  dynamicId: string
+}
+
+export class RawGraph {
+  nodes: RawNodeType[] = []
+  edges: RawEdgeType[] = []
+  nodeIdMap: Record<string, number> = {}
+  edgeIdMap: Record<string, number> = {}
+  edgeDynamicIdMap: Record<string, number> = {}
+
+  getNode = (nodeId: string) => {
+    const nodeIndex = this.nodeIdMap[nodeId]
+    if (nodeIndex !== undefined) {
+      return this.nodes[nodeIndex]
+    }
+    return undefined
+  }
+
+  getEdge = (edgeId: string, dynamicId: boolean = true) => {
+    const edgeIndex = dynamicId ? this.edgeDynamicIdMap[edgeId] : this.edgeIdMap[edgeId]
+    if (edgeIndex !== undefined) {
+      return this.edges[edgeIndex]
+    }
+    return undefined
+  }
+
+  buildDynamicMap = () => {
+    this.edgeDynamicIdMap = {}
+    for (let i = 0; i < this.edges.length; i++) {
+      const edge = this.edges[i]
+      this.edgeDynamicIdMap[edge.dynamicId] = i
+    }
+  }
+}
+
+interface GraphState {
+  selectedNode: string | null
+  focusedNode: string | null
+  selectedEdge: string | null
+  focusedEdge: string | null
+
+  rawGraph: RawGraph | null
+  sigmaGraph: DirectedGraph | null
+
+  moveToSelectedNode: boolean
+
+  setSelectedNode: (nodeId: string | null, moveToSelectedNode?: boolean) => void
+  setFocusedNode: (nodeId: string | null) => void
+  setSelectedEdge: (edgeId: string | null) => void
+  setFocusedEdge: (edgeId: string | null) => void
+  clearSelection: () => void
+  reset: () => void
+
+  setMoveToSelectedNode: (moveToSelectedNode: boolean) => void
+
+  setRawGraph: (rawGraph: RawGraph | null) => void
+  setSigmaGraph: (sigmaGraph: DirectedGraph | null) => void
+}
+
+const useGraphStoreBase = create<GraphState>()((set) => ({
+  selectedNode: null,
+  focusedNode: null,
+  selectedEdge: null,
+  focusedEdge: null,
+
+  moveToSelectedNode: false,
+
+  rawGraph: null,
+  sigmaGraph: null,
+
+  setSelectedNode: (nodeId: string | null, moveToSelectedNode?: boolean) =>
+    set({ selectedNode: nodeId, moveToSelectedNode }),
+  setFocusedNode: (nodeId: string | null) => set({ focusedNode: nodeId }),
+  setSelectedEdge: (edgeId: string | null) => set({ selectedEdge: edgeId }),
+  setFocusedEdge: (edgeId: string | null) => set({ focusedEdge: edgeId }),
+  clearSelection: () =>
+    set({
+      selectedNode: null,
+      focusedNode: null,
+      selectedEdge: null,
+      focusedEdge: null
+    }),
+  reset: () =>
+    set({
+      selectedNode: null,
+      focusedNode: null,
+      selectedEdge: null,
+      focusedEdge: null,
+      rawGraph: null,
+      sigmaGraph: null,
+      moveToSelectedNode: false
+    }),
+
+  setRawGraph: (rawGraph: RawGraph | null) =>
+    set({
+      rawGraph
+    }),
+
+  setSigmaGraph: (sigmaGraph: DirectedGraph | null) => set({ sigmaGraph }),
+
+  setMoveToSelectedNode: (moveToSelectedNode?: boolean) => set({ moveToSelectedNode })
+}))
+
+const useGraphStore = createSelectors(useGraphStoreBase)
+
+export { useGraphStore }