koh 0.2.0

//! Tier 1 end-to-end: the full koh stack over a **real iroh connection**, headless.
//!
//! Two iroh endpoints on loopback (no relay, no second machine), a real PTY-hosted shell on
//! the server, and the real client session loop driven through a mock terminal — exercising
//! the entire path: scripted keystroke → client → iroh datagram → server → PTY → shell echo →
//! vt100 → iroh datagram → client render. This is the slice the in-process `SimHarness` tests
//! (Tier 0) deliberately cannot cover: that the genuine iroh accept/connect/datagram API
//! actually carries our protocol.

// Integration test: a failed unwrap/expect/assert IS the test failing.
#![allow(
    clippy::unwrap_used,
    clippy::expect_used,
    clippy::panic,
    clippy::indexing_slicing,
    clippy::string_slice,
    clippy::unwrap_in_result,
    reason = "integration test code; panics are assertion failures"
)]

use std::sync::{Arc, Mutex};
use std::time::Duration;

use koh::client::{run_client, ClientTerminal, IrohConnector};
use koh::predict::{DisplayPreference, Overlay};
use koh::server::run_session;
use koh::transport_iroh::{
    bind_endpoint_local, generate_secret_key, loopback_addr, IrohChannel, ALPN,
};
use tokio::sync::mpsc;

/// A terminal backend that captures the latest authoritative screen as plain text, so a test
/// can assert on what the user would see — no real TTY involved.
struct MockTerminal {
    latest: Arc<Mutex<String>>,
}

impl ClientTerminal for MockTerminal {
    fn render(
        &mut self,
        screen: &vt100::Screen,
        _overlay: &Overlay,
        _status: Option<&str>,
        _title: &str,
        _bell_count: u64,
    ) -> std::io::Result<()> {
        *self.latest.lock().unwrap() = screen.contents();
        Ok(())
    }

    fn size(&self) -> std::io::Result<(u16, u16)> {
        Ok((24, 80))
    }
}

#[tokio::test]
async fn full_session_over_loopback_pty() {
    // --- two real iroh endpoints on loopback ---
    let server_ep = bind_endpoint_local(generate_secret_key(), true)
        .await
        .expect("bind server endpoint");
    let client_ep = bind_endpoint_local(generate_secret_key(), false)
        .await
        .expect("bind client endpoint");
    let server_addr = loopback_addr(&server_ep);

    // --- server: accept one connection and run a real PTY shell session ---
    let server_task = tokio::spawn(async move {
        let incoming = server_ep.accept().await.expect("server accept");
        let conn = incoming.await.expect("server handshake");
        // `sh` is portable and quiet; scrollback 0.
        let _ = run_session(conn, Some("sh".into()), 0).await;
    });

    // --- client: connect and run the real session loop against a mock terminal ---
    // A connector for the single connection this test uses. Reconnect is never triggered here:
    // dropping the input sender ends the session via Quit before any link loss.
    let connector = IrohConnector::new(
        client_ep.clone(),
        server_addr.clone(),
        std::sync::Arc::new(None),
    );
    let conn = client_ep
        .connect(server_addr, ALPN)
        .await
        .expect("client connect over loopback");
    let channel = IrohChannel::new(conn);

    let latest = Arc::new(Mutex::new(String::new()));
    let term = MockTerminal {
        latest: latest.clone(),
    };

    let (input_tx, input_rx) = mpsc::channel::<Vec<u8>>(64);
    // Keep the resize sender alive for the session (never resize here).
    let (resize_tx, resize_rx) = mpsc::channel::<()>(8);

    let client_task = tokio::spawn(async move {
        let _ = run_client(
            channel,
            connector,
            DisplayPreference::Never, // predictions are overlay-only; assert on the real grid
            (24, 80),
            input_rx,
            resize_rx,
            term,
        )
        .await;
    });

    // Let the shell start and the initial screen sync to the client.
    tokio::time::sleep(Duration::from_millis(700)).await;

    // Type a command with a distinctive marker; the PTY echoes it and `sh` runs it.
    input_tx
        .send(b"echo koh_e2e_marker\r".to_vec())
        .await
        .expect("send keystrokes");

    // Poll the client's rendered screen until the marker round-trips back.
    let mut seen = false;
    for _ in 0..120 {
        if latest.lock().unwrap().contains("koh_e2e_marker") {
            seen = true;
            break;
        }
        tokio::time::sleep(Duration::from_millis(100)).await;
    }

    let final_screen = latest.lock().unwrap().clone();
    assert!(
        seen,
        "marker never round-tripped to the client screen; last frame was:\n{final_screen}"
    );

    // Clean teardown: dropping the input sender ends the client loop, which closes the
    // connection, which ends the server session (and kills the child shell).
    drop(input_tx);
    drop(resize_tx);
    let _ = tokio::time::timeout(Duration::from_secs(3), client_task).await;
    let _ = tokio::time::timeout(Duration::from_secs(3), server_task).await;
}