#![cfg(unix)]

use zellij_integration_tests::{
    claim_first_terminal_and_wait_for_prompt, split_right_and_wait_for_prompt, start_zellij,
    FakePtyHandle, TestSession,
};
use zellij_utils::cli::CliAction;

fn replace_pane_in_place_with_held_command(
    zellij: &TestSession,
    pane_id: &str,
    command: &str,
) -> FakePtyHandle {
    zellij.run_cli_action(CliAction::NewPane {
        direction: None,
        command: vec![command.to_string()],
        plugin: None,
        cwd: None,
        floating: false,
        in_place: true,
        close_replaced_pane: true,
        pane_id: Some(pane_id.to_string()),
        name: None,
        close_on_exit: false,
        start_suspended: false,
        configuration: None,
        skip_plugin_cache: false,
        x: None,
        y: None,
        width: None,
        height: None,
        pinned: None,
        stacked: false,
        blocking: false,
        block_until_exit_success: false,
        block_until_exit_failure: false,
        block_until_exit: false,
        unblock_condition: None,
        near_current_pane: false,
        no_focus: false,
        borderless: None,
        tab_id: None,
    });
    zellij.expect_pty_spawn()
}

#[test]
fn lone_pane_title_becomes_the_tab_name() {
    let mut zellij = start_zellij();
    let terminal = claim_first_terminal_and_wait_for_prompt(&zellij);

    terminal.output(b"\x1b]0;my-title\x07");
    let grid_snapshot = zellij.wait_until("tab borrows the pane title", |grid_snapshot| {
        grid_snapshot.status_bar_appears() && grid_snapshot.contains("my-title")
    });
    assert!(
        !grid_snapshot.contains("Tab #1"),
        "the borrowed pane title replaces the default tab name:\n{}",
        grid_snapshot.text
    );

    zellij.quit();
}

#[test]
fn lone_default_pane_keeps_the_default_tab_name() {
    let mut zellij = start_zellij();
    claim_first_terminal_and_wait_for_prompt(&zellij);

    let grid_snapshot = zellij.wait_until("default tab name is shown", |grid_snapshot| {
        grid_snapshot.status_bar_appears() && grid_snapshot.contains("Tab #1")
    });
    assert!(
        !grid_snapshot.contains("Pane #1"),
        "the default pane name never leaks into the tab name:\n{}",
        grid_snapshot.text
    );

    zellij.quit();
}

#[test]
fn tab_name_reverts_when_a_second_pane_is_opened() {
    let mut zellij = start_zellij();
    let terminal = claim_first_terminal_and_wait_for_prompt(&zellij);

    terminal.output(b"\x1b]0;my-title\x07");
    zellij.wait_until("tab borrows the pane title", |grid_snapshot| {
        grid_snapshot.contains("my-title") && !grid_snapshot.contains("Tab #1")
    });

    split_right_and_wait_for_prompt(&zellij);
    let grid_snapshot = zellij.wait_until(
        "tab name reverts once a second pane exists",
        |grid_snapshot| {
            grid_snapshot
                .lines()
                .first()
                .is_some_and(|tab_bar| tab_bar.contains("Tab #1"))
        },
    );
    assert!(
        grid_snapshot.contains("my-title"),
        "the original pane keeps its title in its frame:\n{}",
        grid_snapshot.text
    );

    zellij.quit();
}

#[test]
fn lone_held_pane_tab_name_shows_exit_status() {
    let mut zellij = start_zellij();
    let shell_terminal = claim_first_terminal_and_wait_for_prompt(&zellij);

    let command_pane_id = format!("terminal_{}", shell_terminal.terminal_id());
    let command_terminal =
        replace_pane_in_place_with_held_command(&zellij, &command_pane_id, "held-command");
    zellij.wait_until("command pane took over the lone tab", |grid_snapshot| {
        grid_snapshot.status_bar_appears() && grid_snapshot.tab_bar_appears()
    });

    command_terminal.exit(Some(0));
    zellij.wait_until("tab name carries the exit status", |grid_snapshot| {
        grid_snapshot.contains("EXIT CODE: 0")
    });

    zellij.quit();
}