use crate::endpoint::realtime_websocket::methods_v1::conversation_handoff_append_message as v1_conversation_handoff_append_message;
use crate::endpoint::realtime_websocket::methods_v1::conversation_item_create_message as v1_conversation_item_create_message;
use crate::endpoint::realtime_websocket::methods_v1::session_update_session as v1_session_update_session;
use crate::endpoint::realtime_websocket::methods_v1::websocket_intent as v1_websocket_intent;
use crate::endpoint::realtime_websocket::methods_v2::conversation_function_call_output_message as v2_conversation_function_call_output_message;
use crate::endpoint::realtime_websocket::methods_v2::conversation_item_create_message as v2_conversation_item_create_message;
use crate::endpoint::realtime_websocket::methods_v2::session_update_session as v2_session_update_session;
use crate::endpoint::realtime_websocket::methods_v2::websocket_intent as v2_websocket_intent;
use crate::endpoint::realtime_websocket::protocol::RealtimeEventParser;
use crate::endpoint::realtime_websocket::protocol::RealtimeOutboundMessage;
use crate::endpoint::realtime_websocket::protocol::RealtimeOutputModality;
use crate::endpoint::realtime_websocket::protocol::RealtimeSessionConfig;
use crate::endpoint::realtime_websocket::protocol::RealtimeSessionMode;
use crate::endpoint::realtime_websocket::protocol::RealtimeVoice;
use crate::endpoint::realtime_websocket::protocol::SessionUpdateSession;
use codex_protocol::protocol::ConversationTextRole;
use serde_json::Result as JsonResult;
use serde_json::Value;
use serde_json::to_value;

pub(super) const REALTIME_AUDIO_SAMPLE_RATE: u32 = 24_000;
const AGENT_FINAL_MESSAGE_PREFIX: &str = "\"Agent Final Message\":\n\n";

pub(super) fn normalized_session_mode(
    event_parser: RealtimeEventParser,
    session_mode: RealtimeSessionMode,
) -> RealtimeSessionMode {
    match event_parser {
        RealtimeEventParser::V1 => RealtimeSessionMode::Conversational,
        RealtimeEventParser::RealtimeV2 => session_mode,
    }
}

pub(super) fn conversation_item_create_message(
    event_parser: RealtimeEventParser,
    text: String,
    role: ConversationTextRole,
) -> RealtimeOutboundMessage {
    match event_parser {
        RealtimeEventParser::V1 => v1_conversation_item_create_message(text, role),
        RealtimeEventParser::RealtimeV2 => v2_conversation_item_create_message(text, role),
    }
}

pub(super) fn conversation_function_call_output_message(
    event_parser: RealtimeEventParser,
    call_id: String,
    output_text: String,
) -> RealtimeOutboundMessage {
    match event_parser {
        RealtimeEventParser::V1 => v1_conversation_handoff_append_message(
            call_id,
            format!("{AGENT_FINAL_MESSAGE_PREFIX}{output_text}"),
        ),
        RealtimeEventParser::RealtimeV2 => {
            v2_conversation_function_call_output_message(call_id, output_text)
        }
    }
}

pub(super) fn session_update_session(
    event_parser: RealtimeEventParser,
    instructions: String,
    session_mode: RealtimeSessionMode,
    output_modality: RealtimeOutputModality,
    voice: RealtimeVoice,
) -> SessionUpdateSession {
    let session_mode = normalized_session_mode(event_parser, session_mode);
    match event_parser {
        RealtimeEventParser::V1 => v1_session_update_session(instructions, voice),
        RealtimeEventParser::RealtimeV2 => {
            v2_session_update_session(instructions, session_mode, output_modality, voice)
        }
    }
}

pub fn session_update_session_json(config: RealtimeSessionConfig) -> JsonResult<Value> {
    let mut session = session_update_session(
        config.event_parser,
        config.instructions,
        config.session_mode,
        config.output_modality,
        config.voice,
    );
    session.id = config.session_id;
    session.model = config.model;
    to_value(session)
}

pub(super) fn websocket_intent(event_parser: RealtimeEventParser) -> Option<&'static str> {
    match event_parser {
        RealtimeEventParser::V1 => v1_websocket_intent(),
        RealtimeEventParser::RealtimeV2 => v2_websocket_intent(),
    }
}
