Spaces:
Sleeping
Sleeping
| import { create } from 'zustand'; | |
| import type { Cell, CellType } from '../api/types'; | |
| interface NotebookState { | |
| cells: Cell[]; | |
| activeCellId: string | null; | |
| addCell: (type: CellType, title?: string, config?: Record<string, any>) => string; | |
| removeCell: (id: string) => void; | |
| updateCell: (id: string, updates: Partial<Cell>) => void; | |
| setActiveCell: (id: string | null) => void; | |
| moveCell: (fromIndex: number, toIndex: number) => void; | |
| toggleCollapse: (id: string) => void; | |
| } | |
| let cellCounter = 0; | |
| export const useNotebookStore = create<NotebookState>((set) => ({ | |
| cells: [], | |
| activeCellId: null, | |
| addCell: (type, title, config = {}) => { | |
| const id = `cell-${++cellCounter}-${Date.now()}`; | |
| const cell: Cell = { | |
| id, | |
| type, | |
| title: title || `${type.charAt(0).toUpperCase() + type.slice(1)} Cell`, | |
| config, | |
| }; | |
| set((state) => ({ | |
| cells: [...state.cells, cell], | |
| activeCellId: id, | |
| })); | |
| return id; | |
| }, | |
| removeCell: (id) => | |
| set((state) => ({ | |
| cells: state.cells.filter((c) => c.id !== id), | |
| activeCellId: state.activeCellId === id ? null : state.activeCellId, | |
| })), | |
| updateCell: (id, updates) => | |
| set((state) => ({ | |
| cells: state.cells.map((c) => (c.id === id ? { ...c, ...updates } : c)), | |
| })), | |
| setActiveCell: (id) => set({ activeCellId: id }), | |
| moveCell: (fromIndex, toIndex) => | |
| set((state) => { | |
| const cells = [...state.cells]; | |
| const [moved] = cells.splice(fromIndex, 1); | |
| cells.splice(toIndex, 0, moved); | |
| return { cells }; | |
| }), | |
| toggleCollapse: (id) => | |
| set((state) => ({ | |
| cells: state.cells.map((c) => | |
| c.id === id ? { ...c, collapsed: !c.collapsed } : c, | |
| ), | |
| })), | |
| })); | |