rho-coding-agent 1.40.0

use pretty_assertions::assert_eq;
use rho_providers::model::{
    provider_models::{
        replace_cached_provider_models_for_tests, with_provider_models_cache_dir_for_tests,
        ProviderModel,
    },
    ReasoningCapabilities,
};

use super::*;
use crate::{
    commands::parse_command, config::InternalAgentModelConfig, tui::tests::test_app,
    tui::PickerAction,
};

fn invocation(command: &str) -> CommandInvocation {
    parse_command(command).unwrap().unwrap()
}

/// Runs `f` with a temporary OpenAI model cache so the advisor model picker can open.
///
/// OpenAI is cache-backed. Without this, `/advisor on` with no model only
/// reports "no cached provider models" and never opens a picker. The cache
/// override is thread-local, so the body stays on one thread.
fn with_cached_openai_models<T>(f: impl FnOnce() -> T) -> T {
    let cache = tempfile::tempdir().unwrap();
    with_provider_models_cache_dir_for_tests(cache.path().to_path_buf(), || {
        replace_cached_provider_models_for_tests(
            "openai",
            &[ProviderModel {
                provider: "openai".into(),
                model: "gpt-5.5".into(),
                display_name: "GPT-5.5".into(),
                context_window: None,
                max_output_tokens: None,
                reasoning_capabilities: ReasoningCapabilities::Unknown,
            }],
        )
        .unwrap();
        f()
    })
}

fn block_on<T>(f: impl std::future::Future<Output = T>) -> T {
    tokio::runtime::Builder::new_current_thread()
        .enable_all()
        .build()
        .unwrap()
        .block_on(f)
}

/// Records what the command asked the runtime to apply, so the advisor state
/// transition is observable without a live SDK runtime.
#[derive(Default)]
struct FakeAdvisorRuntime {
    applied: Vec<Option<InternalAgentModelConfig>>,
}

impl FakeAdvisorRuntime {
    fn last_applied(&self) -> Option<&Option<InternalAgentModelConfig>> {
        self.applied.last()
    }
}

impl AdvisorRuntime for FakeAdvisorRuntime {
    fn set_advisor(
        &mut self,
        model: Option<InternalAgentModelConfig>,
    ) -> impl std::future::Future<Output = anyhow::Result<Option<String>>> + Send {
        self.applied.push(model);
        std::future::ready(Ok(None))
    }

    fn tool_specs(&self) -> Vec<rho_sdk::model::ToolSpec> {
        Vec::new()
    }
}

fn app_with_advisor_model() -> App {
    let mut app = test_app();
    app.info.runtime.internal_agents.insert(
        ADVISOR_AGENT_ID.into(),
        InternalAgentModelConfig::new("openai".into(), "gpt-5.5".into(), "api-key".into()),
    );
    app
}

fn open_picker(app: &App) -> &crate::tui::UiPicker {
    match app.input_ui.composer() {
        ComposerMode::Picker(picker) => picker,
        composer => panic!("expected an open picker, found {composer:?}"),
    }
}

// Covers: /advisor on with a configured advisor model turns the mode on and saves it
// Owner: advisor command
#[tokio::test]
async fn enabling_advisor_mode_with_a_model_saves_the_setting() {
    let mut app = app_with_advisor_model();
    let mut agent = FakeAdvisorRuntime::default();

    app.execute_advisor_command_with_runtime(invocation("/advisor on"), &mut agent)
        .await
        .unwrap();

    assert!(app.info.runtime.advisor_mode);
    assert!(
        app.info
            .services
            .config_repository
            .load()
            .unwrap()
            .advisor_mode
    );
    assert_eq!(
        app.status(),
        "advisor mode is on: openai/gpt-5.5 reviews the session"
    );
    assert_eq!(
        agent.last_applied(),
        Some(&Some(InternalAgentModelConfig::new(
            "openai".into(),
            "gpt-5.5".into(),
            "api-key".into()
        )))
    );
}

// Covers: /advisor with no argument flips the current mode
// Owner: advisor command
#[tokio::test]
async fn bare_advisor_command_turns_the_mode_off_again() {
    let mut app = app_with_advisor_model();
    let mut agent = FakeAdvisorRuntime::default();
    app.execute_advisor_command_with_runtime(invocation("/advisor on"), &mut agent)
        .await
        .unwrap();

    app.execute_advisor_command_with_runtime(invocation("/advisor"), &mut agent)
        .await
        .unwrap();

    assert!(!app.info.runtime.advisor_mode);
    assert!(
        !app.info
            .services
            .config_repository
            .load()
            .unwrap()
            .advisor_mode
    );
    assert_eq!(app.status(), "advisor mode is off");
    assert_eq!(agent.last_applied(), Some(&None));
}

// Covers: /advisor off saves the mode off, and turning it on again keeps the model
// Owner: advisor command
#[tokio::test]
async fn disabling_advisor_mode_saves_the_setting() {
    let mut app = app_with_advisor_model();
    let mut agent = FakeAdvisorRuntime::default();
    app.execute_advisor_command_with_runtime(invocation("/advisor on"), &mut agent)
        .await
        .unwrap();

    app.execute_advisor_command_with_runtime(invocation("/advisor off"), &mut agent)
        .await
        .unwrap();

    assert!(!app.info.runtime.advisor_mode);
    assert!(
        !app.info
            .services
            .config_repository
            .load()
            .unwrap()
            .advisor_mode
    );

    // Turning the mode back on must reuse the model chosen before, not ask again.
    app.execute_advisor_command_with_runtime(invocation("/advisor on"), &mut agent)
        .await
        .unwrap();

    assert!(app.info.runtime.advisor_mode);
    assert!(matches!(app.input_ui.composer(), ComposerMode::Input));
    assert_eq!(
        agent.applied,
        vec![
            Some(InternalAgentModelConfig::new(
                "openai".into(),
                "gpt-5.5".into(),
                "api-key".into()
            )),
            None,
            Some(InternalAgentModelConfig::new(
                "openai".into(),
                "gpt-5.5".into(),
                "api-key".into()
            )),
        ]
    );
}

// Covers: /advisor on without an advisor model asks for one instead of turning the mode on
// Owner: advisor command
#[test]
fn enabling_advisor_mode_without_a_model_opens_the_model_picker() {
    with_cached_openai_models(|| {
        block_on(async {
            let mut app = test_app();
            let mut agent = FakeAdvisorRuntime::default();

            app.execute_advisor_command_with_runtime(invocation("/advisor on"), &mut agent)
                .await
                .unwrap();

            assert!(!app.info.runtime.advisor_mode);
            assert!(
                !app.info
                    .services
                    .config_repository
                    .load()
                    .unwrap()
                    .advisor_mode
            );
            let picker = open_picker(&app);
            assert_eq!(picker.action, PickerAction::SelectInternalAgentModel);
            assert_eq!(
                app.internal_agent_model_target
                    .as_ref()
                    .map(|target| (target.id.as_str(), target.origin)),
                Some((
                    ADVISOR_AGENT_ID,
                    InternalAgentModelPickerOrigin::AdvisorCommand
                ))
            );
            assert_eq!(
                app.status(),
                "select an advisor model to turn advisor mode on"
            );
        });
    });
}

// Covers: the advisor picker never offers the conversation model, which the advisor cannot use
// Owner: advisor model picker
#[test]
fn advisor_model_picker_omits_the_conversation_model_row() {
    with_cached_openai_models(|| {
        block_on(async {
            let mut app = test_app();
            let mut agent = FakeAdvisorRuntime::default();

            app.execute_advisor_command_with_runtime(invocation("/advisor on"), &mut agent)
                .await
                .unwrap();

            let picker = open_picker(&app);
            assert!(!picker
                .items
                .iter()
                .any(|item| item.value == super::super::model_picker::USE_CONVERSATION_MODEL));
        });
    });
}

// Covers: dismissing the advisor model prompt leaves the mode off and says why
// Owner: advisor command
#[test]
fn dismissing_the_advisor_model_prompt_leaves_the_mode_off() {
    with_cached_openai_models(|| {
        block_on(async {
            let mut app = test_app();
            let mut agent = FakeAdvisorRuntime::default();
            app.execute_advisor_command_with_runtime(invocation("/advisor on"), &mut agent)
                .await
                .unwrap();

            app.handle_picker_escape(/*running*/ false).unwrap();

            assert!(!app.info.runtime.advisor_mode);
            assert!(app.internal_agent_model_target.is_none());
            assert!(matches!(app.input_ui.composer(), ComposerMode::Input));
            assert_eq!(
                app.status(),
                "advisor mode stays off: no advisor model selected"
            );
        });
    });
}

// Covers: an active-run runtime error must not flip saved or in-memory mode.
// Owner: advisor command
#[tokio::test]
async fn advisor_mode_runtime_failure_leaves_mode_unchanged() {
    #[derive(Debug)]
    struct ActiveRunError;

    impl std::fmt::Display for ActiveRunError {
        fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
            f.write_str("advisor mode cannot change while a run is active")
        }
    }

    impl std::error::Error for ActiveRunError {}

    struct ActiveRunRuntime;

    impl AdvisorRuntime for ActiveRunRuntime {
        fn set_advisor(
            &mut self,
            _model: Option<InternalAgentModelConfig>,
        ) -> impl std::future::Future<Output = anyhow::Result<Option<String>>> + Send {
            std::future::ready(Err(anyhow::Error::new(ActiveRunError)))
        }

        fn tool_specs(&self) -> Vec<rho_sdk::model::ToolSpec> {
            Vec::new()
        }
    }

    let mut app = app_with_advisor_model();
    let mut agent = ActiveRunRuntime;
    let error = app
        .execute_advisor_command_with_runtime(invocation("/advisor on"), &mut agent)
        .await
        .expect_err("active-run failure should propagate");
    assert!(
        error.downcast_ref::<ActiveRunError>().is_some(),
        "expected typed ActiveRunError, got: {error:#}"
    );
    assert!(!app.info.runtime.advisor_mode);
    assert!(
        !app.info
            .services
            .config_repository
            .load()
            .unwrap()
            .advisor_mode
    );
    assert_eq!(app.status(), "advisor mode change failed");
}

// Covers: editing the advisor model from /config does not claim it enables mode.
// Owner: advisor command
#[test]
fn advisor_model_config_row_uses_edit_status() {
    with_cached_openai_models(|| {
        let mut app = app_with_advisor_model();
        app.open_main_config_picker_selected(super::super::config_picker::ADVISOR_MODEL_VALUE)
            .unwrap();
        app.open_advisor_model_prompt(InternalAgentModelPickerOrigin::AdvisorModelConfigRow);
        assert_eq!(app.status(), "select an advisor model");
        assert_eq!(
            app.internal_agent_model_target
                .as_ref()
                .map(|target| target.origin),
            Some(InternalAgentModelPickerOrigin::AdvisorModelConfigRow)
        );
    });
}