@type agent_capabilities() :: %{
  optional(:loadSession) => boolean(),
  optional(:promptCapabilities) => %{
    optional(:image) => boolean(),
    optional(:audio) => boolean(),
    optional(:embeddedContext) => boolean()
  },
  optional(:mcpCapabilities) => %{
    optional(:http) => boolean(),
    optional(:sse) => boolean()
  },
  optional(:sessionCapabilities) => %{
    optional(:list) => session_list_capabilities() | nil
  }
}

@type agent_info() :: %{name: String.t(), version: String.t()}

@type agent_message_chunk_update() :: %{
  sessionUpdate: :agent_message_chunk,
  content: content_block()
}

@type audio_block() :: %{type: :audio, mimeType: String.t(), data: String.t()}

@type available_commands_update() :: %{
  sessionUpdate: :available_commands_update,
  commands: [map()]
}

@type client_capabilities() :: %{
  optional(:fs) => %{
    optional(:readTextFile) => boolean(),
    optional(:writeTextFile) => boolean()
  },
  optional(:terminal) => boolean()
}

@type client_info() :: %{name: String.t(), version: String.t()}

@type config_option() :: %{
  :id => String.t(),
  :name => String.t(),
  optional(:description) => String.t(),
  optional(:type) => String.t(),
  optional(:default) => any()
}

@type config_option_update() :: %{
  sessionUpdate: :config_option_update,
  configId: String.t(),
  value: any()
}

@type content_block() ::
  text_block()
  | image_block()
  | audio_block()
  | resource_link_block()
  | resource_block()

@type current_mode_update() :: %{
  sessionUpdate: :current_mode_update,
  modeId: String.t()
}

@type file_range() :: %{
  optional(:start) => non_neg_integer(),
  optional(:end) => non_neg_integer()
}

@type file_read_request() :: %{
  :sessionId => String.t(),
  :path => String.t(),
  optional(:range) => file_range()
}

@type file_write_request() :: %{
  sessionId: String.t(),
  path: String.t(),
  content: String.t()
}

@type image_block() :: %{type: :image, mimeType: String.t(), data: String.t()}

@type initialize_request() :: %{
  :clientInfo => client_info(),
  optional(:clientCapabilities) => client_capabilities(),
  optional(:protocolVersion) => pos_integer()
}

@type initialize_response() :: %{
  :agentInfo => agent_info(),
  optional(:agentCapabilities) => agent_capabilities(),
  optional(:protocolVersion) => pos_integer()
}

@type list_sessions_request() :: %{optional(:cursor) => String.t()}

@type list_sessions_response() :: %{
  :sessions => [session_info()],
  optional(:nextCursor) => String.t()
}

@type load_session_request() :: %{
  :sessionId => String.t(),
  optional(:cwd) => String.t(),
  optional(:mcpServers) => [mcp_server()]
}

@type mcp_server() :: %{:uri => String.t(), optional(:name) => String.t()}

@type mode() :: %{
  :id => String.t(),
  :name => String.t(),
  optional(:description) => String.t()
}

@type new_session_request() :: %{
  optional(:cwd) => String.t(),
  optional(:mcpServers) => [mcp_server()]
}

@type new_session_response() :: %{sessionId: String.t()}

@type permission_option() :: %{
  :optionId => String.t(),
  :name => String.t(),
  :kind => String.t(),
  optional(:description) => String.t()
}

@type permission_outcome() :: %{
  :outcome => String.t(),
  optional(:optionId) => String.t()
}

@type permission_request() :: %{
  sessionId: String.t(),
  toolCall: tool_call_info(),
  options: [permission_option()]
}

@type plan_entry() :: %{
  content: String.t(),
  priority: :high | :medium | :low,
  status: :pending | :in_progress | :completed
}

@type plan_update() :: %{sessionUpdate: :plan_update, entries: [plan_entry()]}

@type prompt_request() :: %{sessionId: String.t(), prompt: [content_block()]}

@type prompt_response() :: %{stopReason: String.t()}

@type resource_block() :: %{
  :type => :resource,
  :uri => String.t(),
  optional(:name) => String.t(),
  optional(:mimeType) => String.t(),
  optional(:text) => String.t(),
  optional(:blob) => String.t()
}

@type resource_link_block() :: %{
  :type => :resource_link,
  :uri => String.t(),
  optional(:name) => String.t(),
  optional(:mimeType) => String.t()
}

@type session_info() :: %{
  :sessionId => String.t(),
  optional(:name) => String.t(),
  optional(:createdAt) => String.t()
}

@type session_info_update() :: %{
  :sessionUpdate => :session_info_update,
  optional(:sessionName) => String.t(),
  optional(:metadata) => map()
}

@type session_list_capabilities() :: %{optional(:supportsNameFilter) => boolean()}

@type session_update() ::
  user_message_chunk_update()
  | agent_message_chunk_update()
  | tool_call_update()
  | plan_update()
  | available_commands_update()
  | config_option_update()
  | current_mode_update()
  | session_info_update()
  | thinking_update()
  | status_update()

@type session_update_params() :: %{sessionId: String.t(), update: session_update()}

@type status_update() :: %{
  :sessionUpdate => :status,
  :status => String.t(),
  optional(:message) => String.t()
}

@type text_block() :: %{type: :text, text: String.t()}

@type thinking_update() :: %{sessionUpdate: :thinking, content: String.t()}

@type tool_call_info() :: %{
  :toolName => String.t(),
  optional(:toolCallId) => String.t(),
  optional(:arguments) => map()
}

@type tool_call_update() :: %{
  :sessionUpdate => :tool_call_update,
  :toolCallId => String.t(),
  :title => String.t(),
  :status => String.t(),
  optional(:content) => [content_block()]
}

@type user_message_chunk_update() :: %{
  sessionUpdate: :user_message_chunk,
  content: content_block()
}