//! E2E tests for LSP code action modal behavior.
//!
//! Tests for issue #1405: pressing numbers in the code action modal should
//! select and apply the corresponding action.

use crate::common::fake_lsp::FakeLspServer;
use crate::common::harness::EditorTestHarness;
use crate::common::tracing::init_tracing_from_env;
use crossterm::event::{KeyCode, KeyModifiers};

/// Issue #1405: pressing a number key should select, dismiss the popup,
/// and apply the workspace edit from the code action.
///
/// The fake LSP returns "Add missing import" (action 3) with a workspace edit
/// that inserts "use std::io;\n" at the top of the file.
#[test]
#[cfg_attr(
    target_os = "windows",
    ignore = "FakeLspServer uses a Bash script which is not available on Windows"
)]
fn test_code_action_number_key_selects_and_applies() -> anyhow::Result<()> {
    // Initialize tracing + signal handlers so CI timeouts surface a backtrace
    // instead of a silent hang (diagnoses the macOS CI 180s timeout).
    init_tracing_from_env();
    fresh::services::signal_handler::install_signal_handlers();

    let temp_dir = tempfile::tempdir()?;
    let _fake_server = FakeLspServer::spawn_with_code_actions(temp_dir.path())?;

    let test_file = temp_dir.path().join("test.rs");
    std::fs::write(&test_file, "fn main() {\n    let x = 5;\n}\n")?;

    let mut config = fresh::config::Config::default();
    config.lsp.insert(
        "rust".to_string(),
        fresh::types::LspLanguageConfig::Multi(vec![fresh::services::lsp::LspServerConfig {
            command: FakeLspServer::code_actions_script_path(temp_dir.path())
                .to_string_lossy()
                .to_string(),
            args: vec![],
            enabled: true,
            auto_start: true,
            process_limits: fresh::services::process_limits::ProcessLimits::default(),
            initialization_options: None,
            env: Default::default(),
            language_id_overrides: Default::default(),
            root_markers: Default::default(),
            name: None,
            only_features: None,
            except_features: None,
        }]),
    );

    let mut harness = EditorTestHarness::create(
        120,
        24,
        crate::common::harness::HarnessOptions::new()
            .with_config(config)
            .with_working_dir(temp_dir.path().to_path_buf()),
    )?;

    harness.open_file(&test_file)?;
    harness.render()?;

    // Wait for the LSP server to reach `Running` state before we ask it
    // for code actions.  The status-bar `LSP (on)` indicator flips on as
    // soon as the server is `Starting` / `Initializing`, so the older
    // `wait_for_screen_contains("LSP (on)")` would return before the
    // handshake had completed and any `textDocument/codeAction` request
    // would be dropped on the floor (empirically reproduced: code-actions
    // request fires, LSP initialize arrives ~170ms later, the popup never
    // materialises and the test hits nextest's 180s ceiling).
    harness.wait_until(|h| h.editor().is_lsp_server_ready("rust"))?;

    // Position cursor on "let x = 5;" (line 2)
    harness.send_key(KeyCode::Down, KeyModifiers::NONE)?;
    harness.render()?;

    // Trigger code actions via the command palette. We avoid Alt+. because
    // the macOS CI terminal was not delivering the Alt modifier through
    // crossterm reliably — the keystroke never reached the action
    // dispatcher, the popup never opened, and the test ran out its 180s
    // nextest budget (see claude/fix-macos-ci-timeout-OZgNX diagnostic
    // run).  The command-palette path exercises the same
    // `Action::LspCodeActions` handler without depending on modifier
    // decoding.
    harness.send_key(KeyCode::Char('p'), KeyModifiers::CONTROL)?;
    harness.type_text("Code Actions")?;
    harness.send_key(KeyCode::Enter, KeyModifiers::NONE)?;
    harness.render()?;

    // Wait for code action popup
    harness.wait_for_screen_contains("Extract function")?;

    // Verify the popup shows numbered code actions
    harness.assert_screen_contains("1. Extract function");
    harness.assert_screen_contains("3. Add missing import");

    // Press '3' to select "Add missing import" which has a real workspace edit
    harness.send_key(KeyCode::Char('3'), KeyModifiers::NONE)?;
    harness.render()?;

    // The popup should be dismissed
    harness.assert_screen_not_contains("Code Actions");

    // The workspace edit should have been applied: "use std::io;\n" inserted at top
    let buffer = harness.get_buffer_content().unwrap();
    assert_eq!(
        buffer, "use std::io;\nfn main() {\n    let x = 5;\n}\n",
        "Expected 'use std::io;' to be inserted at the top of the file by the code action"
    );

    Ok(())
}

/// Arrow-down + Enter should navigate to an action and apply it.
#[test]
#[cfg_attr(
    target_os = "windows",
    ignore = "FakeLspServer uses a Bash script which is not available on Windows"
)]
fn test_code_action_arrow_enter_applies() -> anyhow::Result<()> {
    // Initialize tracing + signal handlers so CI timeouts surface a backtrace
    // instead of a silent hang (diagnoses the macOS CI 180s timeout).
    init_tracing_from_env();
    fresh::services::signal_handler::install_signal_handlers();

    let temp_dir = tempfile::tempdir()?;
    let _fake_server = FakeLspServer::spawn_with_code_actions(temp_dir.path())?;

    let test_file = temp_dir.path().join("test.rs");
    std::fs::write(&test_file, "fn main() {\n    let x = 5;\n}\n")?;

    let mut config = fresh::config::Config::default();
    config.lsp.insert(
        "rust".to_string(),
        fresh::types::LspLanguageConfig::Multi(vec![fresh::services::lsp::LspServerConfig {
            command: FakeLspServer::code_actions_script_path(temp_dir.path())
                .to_string_lossy()
                .to_string(),
            args: vec![],
            enabled: true,
            auto_start: true,
            process_limits: fresh::services::process_limits::ProcessLimits::default(),
            initialization_options: None,
            env: Default::default(),
            language_id_overrides: Default::default(),
            root_markers: Default::default(),
            name: None,
            only_features: None,
            except_features: None,
        }]),
    );

    let mut harness = EditorTestHarness::create(
        120,
        24,
        crate::common::harness::HarnessOptions::new()
            .with_config(config)
            .with_working_dir(temp_dir.path().to_path_buf()),
    )?;

    harness.open_file(&test_file)?;
    harness.render()?;

    // Wait for the LSP server to reach `Running` state — see sibling test
    // for the rationale (the status-bar `LSP (on)` label turns on during
    // `Starting`, before the handshake completes).
    harness.wait_until(|h| h.editor().is_lsp_server_ready("rust"))?;

    // Position cursor
    harness.send_key(KeyCode::Down, KeyModifiers::NONE)?;
    harness.render()?;

    // Trigger code actions via the command palette rather than Alt+. — see
    // sibling test for the rationale (Alt modifier decoding is unreliable
    // on macOS CI terminals).
    harness.send_key(KeyCode::Char('p'), KeyModifiers::CONTROL)?;
    harness.type_text("Code Actions")?;
    harness.send_key(KeyCode::Enter, KeyModifiers::NONE)?;
    harness.render()?;

    harness.wait_for_screen_contains("Extract function")?;

    // Navigate down twice to "Add missing import"
    harness.send_key(KeyCode::Down, KeyModifiers::NONE)?;
    harness.send_key(KeyCode::Down, KeyModifiers::NONE)?;
    harness.render()?;

    // Press Enter to confirm
    harness.send_key(KeyCode::Enter, KeyModifiers::NONE)?;
    harness.render()?;

    // Popup should be dismissed and edit applied
    harness.assert_screen_not_contains("Code Actions");

    let buffer = harness.get_buffer_content().unwrap();
    assert_eq!(
        buffer, "use std::io;\nfn main() {\n    let x = 5;\n}\n",
        "Expected 'use std::io;' to be inserted at the top of the file by the code action"
    );

    Ok(())
}
