import { create } from "zustand"

interface GraphHighlightsState {
	documentIds: string[]
	lastUpdated: number
	setDocumentIds: (ids: string[]) => void
	clear: () => void
}

export const useGraphHighlightsStore = create<GraphHighlightsState>()(
	(set, get) => ({
		documentIds: [],
		lastUpdated: 0,
		setDocumentIds: (ids) => {
			const next = Array.from(new Set(ids))
			const prev = get().documentIds
			if (
				prev.length === next.length &&
				prev.every((id) => next.includes(id))
			) {
				return
			}
			set({ documentIds: next, lastUpdated: Date.now() })
		},
		clear: () => set({ documentIds: [], lastUpdated: Date.now() }),
	}),
)

export function useGraphHighlights() {
	const documentIds = useGraphHighlightsStore((s) => s.documentIds)
	const lastUpdated = useGraphHighlightsStore((s) => s.lastUpdated)
	const setDocumentIds = useGraphHighlightsStore((s) => s.setDocumentIds)
	const clear = useGraphHighlightsStore((s) => s.clear)
	return { documentIds, lastUpdated, setDocumentIds, clear }
}