🎉 initial commit

src/renderer/store/editorStore.ts

@@ -0,0 +1,119 @@
+import { create } from 'zustand'
+import type { FileNode, ChatMessage, TextAnnotation, AnalysisMode } from '../types/editor'
+
+interface EditorState {
+  // File tree
+  fileTree: FileNode[]
+  setFileTree: (tree: FileNode[]) => void
+
+  // Active file
+  activeFilePath: string | null
+  activeFileContent: string
+  isDirty: boolean
+  setActiveFile: (path: string, content: string) => void
+  setContent: (content: string) => void
+  markSaved: () => void
+
+  // Chat - persisted per file path
+  chatHistoryByFile: Record<string, ChatMessage[]>
+  chatHistory: ChatMessage[]
+  isAILoading: boolean
+  aiError: string | null
+  addUserMessage: (text: string) => void
+  startAssistantMessage: () => void
+  appendToLastAssistantMessage: (chunk: string) => void
+  setAILoading: (loading: boolean) => void
+  setAIError: (error: string | null) => void
+  clearChat: () => void
+
+  // Annotations (highlights in editor)
+  annotations: TextAnnotation[]
+  setAnnotations: (annotations: TextAnnotation[]) => void
+  clearAnnotations: () => void
+
+  // Analysis mode
+  analysisMode: AnalysisMode
+  setAnalysisMode: (mode: AnalysisMode) => void
+}
+
+export const useEditorStore = create<EditorState>((set, get) => ({
+  fileTree: [],
+  setFileTree: (fileTree) => set({ fileTree }),
+
+  activeFilePath: null,
+  activeFileContent: '',
+  isDirty: false,
+  setActiveFile: (path, content) => {
+    const existing = get().chatHistoryByFile[path] ?? []
+    set({
+      activeFilePath: path,
+      activeFileContent: content,
+      isDirty: false,
+      chatHistory: existing,
+      annotations: [],
+      analysisMode: 'none'
+    })
+  },
+  setContent: (content) => set({ activeFileContent: content, isDirty: true }),
+  markSaved: () => set({ isDirty: false }),
+
+  chatHistoryByFile: {},
+  chatHistory: [],
+  isAILoading: false,
+  aiError: null,
+
+  addUserMessage: (text) => {
+    const msg: ChatMessage = { id: `user-${Date.now()}`, role: 'user', content: text }
+    set((s) => {
+      const history = [...s.chatHistory, msg]
+      const byFile = s.activeFilePath
+        ? { ...s.chatHistoryByFile, [s.activeFilePath]: history }
+        : s.chatHistoryByFile
+      return { chatHistory: history, chatHistoryByFile: byFile }
+    })
+  },
+
+  startAssistantMessage: () => {
+    const msg: ChatMessage = { id: `asst-${Date.now()}`, role: 'assistant', content: '' }
+    set((s) => {
+      const history = [...s.chatHistory, msg]
+      const byFile = s.activeFilePath
+        ? { ...s.chatHistoryByFile, [s.activeFilePath]: history }
+        : s.chatHistoryByFile
+      return { chatHistory: history, chatHistoryByFile: byFile }
+    })
+  },
+
+  appendToLastAssistantMessage: (chunk) => {
+    set((s) => {
+      const history = [...s.chatHistory]
+      const last = history[history.length - 1]
+      if (last?.role === 'assistant') {
+        history[history.length - 1] = { ...last, content: last.content + chunk }
+      }
+      const byFile = s.activeFilePath
+        ? { ...s.chatHistoryByFile, [s.activeFilePath]: history }
+        : s.chatHistoryByFile
+      return { chatHistory: history, chatHistoryByFile: byFile }
+    })
+  },
+
+  setAILoading: (isAILoading) => set({ isAILoading }),
+  setAIError: (aiError) => set({ aiError }),
+
+  clearChat: () => {
+    set((s) => {
+      const byFile = s.activeFilePath
+        ? { ...s.chatHistoryByFile, [s.activeFilePath]: [] }
+        : s.chatHistoryByFile
+      return { chatHistory: [], chatHistoryByFile: byFile }
+    })
+  },
+
+  annotations: [],
+  setAnnotations: (annotations) => set({ annotations }),
+  clearAnnotations: () => set({ annotations: [], analysisMode: 'none' }),
+
+  analysisMode: 'none',
+  setAnalysisMode: (analysisMode) => set({ analysisMode })
+}))