mnote/wolai-frontend/src/store/ai-agent-ui.ts

"use client";

import { create } from "zustand";

type AiAgentUiState = {
  globalAgentOpen: boolean;
  documentAgentAvailable: boolean;
  documentAgentOpen: boolean;
  setGlobalAgentOpen: (open: boolean) => void;
  toggleGlobalAgentOpen: () => void;
  setDocumentAgentAvailable: (available: boolean) => void;
  setDocumentAgentOpen: (open: boolean) => void;
  toggleDocumentAgentOpen: () => void;
};

export const useAiAgentUiStore = create<AiAgentUiState>((set, get) => ({
  globalAgentOpen: false,
  documentAgentAvailable: false,
  documentAgentOpen: false,
  setGlobalAgentOpen: (open) => set({ globalAgentOpen: open }),
  toggleGlobalAgentOpen: () => {
    const s = get();
    set({ globalAgentOpen: !s.globalAgentOpen });
  },
  setDocumentAgentAvailable: (available) => set({ documentAgentAvailable: available }),
  setDocumentAgentOpen: (open) => set({ documentAgentOpen: open }),
  toggleDocumentAgentOpen: () => {
    const s = get();
    if (!s.documentAgentAvailable) return;
    set({ documentAgentOpen: !s.documentAgentOpen });
  },
}));