import { create } from "zustand";
import { ExecutionStatus } from "#/types/agent-server/core/base/common";

interface ConversationStateStore {
  /**
   * Latest execution status per conversation, fed by the main and planning
   * WebSocket handlers. Scoped by conversation id so the planning helper
   * conversation's own run/idle transitions can never overwrite the main
   * conversation's status (or vice versa) — see conversation-websocket-context.tsx.
   */
  executionStatusByConversation: Record<string, ExecutionStatus>;

  /**
   * Set the agent status for a specific conversation.
   */
  setExecutionStatus: (
    conversationId: string,
    execution_status: ExecutionStatus,
  ) => void;

  /**
   * Reset the store to initial state
   */
  reset: () => void;
}

export const useConversationStateStore = create<ConversationStateStore>(
  (set) => ({
    executionStatusByConversation: {},

    setExecutionStatus: (conversationId, execution_status) =>
      set((state) => ({
        executionStatusByConversation: {
          ...state.executionStatusByConversation,
          [conversationId]: execution_status,
        },
      })),

    reset: () => set({ executionStatusByConversation: {} }),
  }),
);