import { createPersistStore } from "../utils/store";

export const useMindMapStore = createPersistStore<
  { newMessages: { role: string; content: string }[]; content: string },
  {
    setMindMapData: (
      newMessages: { role: string; content: string }[],
      content: string,
    ) => void;
    clearMindMapData: () => void;
  }
>(
  {
    newMessages: [],
    content: "",
  },
  (set, get) => ({
    setMindMapData: (newMessages, content) => {
      set(() => ({
        newMessages,
        content,
      }));
    },

    clearMindMapData: () => {
      set(() => ({
        newMessages: [],
        content: "",
      }));
    },
  }),
  {
    name: "mind-map-store",
    version: 1,
  },
);