sqlchat/store/chat.ts

import { create } from "zustand";
import { persist } from "zustand/middleware";
import { Chat, UNKNOWN_ID, User } from "../types";
import { generateUUID } from "../utils";

export const defaultChat: Chat = {
  id: generateUUID(),
  connectionId: UNKNOWN_ID,
  databaseName: "",
  assistantId: "sql-assistant",
};

interface ChatState {
  chatList: Chat[];
  currentChat: Chat;
  createChat: (user: User) => void;
  setCurrentChat: (chat: Chat) => void;
}

export const useChatStore = create<ChatState>()(
  persist(
    (set) => ({
      chatList: [defaultChat],
      currentChat: defaultChat,
      createChat: (assistant: User) => {
        const chat: Chat = {
          id: generateUUID(),
          connectionId: UNKNOWN_ID,
          databaseName: "",
          assistantId: assistant.id,
        };
        set((state) => ({
          chatList: [...state.chatList, chat],
          currentChat: chat,
        }));
      },
      setCurrentChat: (chat: Chat) => set(() => ({ currentChat: chat })),
    }),
    {
      name: "chat-storage",
    }
  )
);