loopflow 0.12.0

use std::sync::atomic::{AtomicBool, AtomicU32, Ordering};
use std::sync::Arc;
use std::time::{Duration, Instant};

use anyhow::{anyhow, Result};
use async_trait::async_trait;
use reqwest::Method;
use serde_json::{json, Value};
use tokio::process::{Child, Command};
use tokio::sync::mpsc;
use tokio::task::JoinHandle;

use crate::chat::types::{ConversationEvent, Lifecycle};
use crate::engine::agent::{register_interrupt_cleanup, AgentConfig};
use crate::engine::config::parse_agent;
use crate::harness::common::{spawn_stderr_logger, TurnInProgressGuard};
use crate::harness::{
    opencode_mapping, opencode_runtime, ApprovalPolicy, Capabilities, Harness, HarnessError,
    RawProviderEvent,
};

pub(crate) const OPENCODE_DISCONNECTED_CODE: &str = "opencode_disconnected";

/// SIGKILL an entire process group. `opencode serve` spawns descendants
/// (MCP servers, model proxies, npm-shim grandchildren) that a bare
/// `start_kill` of the direct child leaves running as orphans. The harness
/// spawns the server in its own group (`process_group(0)`), so the group id is
/// the child pid and this reaches the whole tree. Shared by `stop()` and the
/// interrupt-cleanup hook.
fn kill_process_group(pid: u32) {
    #[cfg(unix)]
    // SAFETY: a negative pid targets the process group we created for the
    // child at spawn (process_group(0)); plain syscall, no pointers.
    unsafe {
        libc::kill(-(pid as i32), libc::SIGKILL)
    };
    #[cfg(not(unix))]
    let _ = pid;
}

pub struct OpenCodeHarness {
    events: mpsc::UnboundedSender<ConversationEvent>,
    raw_provider: Option<mpsc::UnboundedSender<RawProviderEvent>>,
    client: reqwest::Client,
    approval: ApprovalPolicy,
    config: Option<AgentConfig>,
    should_seed_prompt: bool,
    turn_in_progress: Arc<AtomicBool>,
    shutdown_requested: Arc<AtomicBool>,
    interrupt_requested: Arc<AtomicBool>,
    child: Option<Child>,
    /// The spawned server's process-group id (== its pid under
    /// `process_group(0)`). 0 while no server is running.
    child_group: Arc<AtomicU32>,
    interrupt_hook_registered: bool,
    stderr_task: Option<JoinHandle<()>>,
    sse_task: Option<JoinHandle<()>>,
    server_base_url: Option<String>,
    provider_session_id: Option<String>,
}

impl std::fmt::Debug for OpenCodeHarness {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.debug_struct("OpenCodeHarness").finish()
    }
}

impl OpenCodeHarness {
    pub fn new(events: mpsc::UnboundedSender<ConversationEvent>, approval: ApprovalPolicy) -> Self {
        Self {
            events,
            raw_provider: None,
            client: reqwest::Client::new(),
            approval,
            config: None,
            should_seed_prompt: true,
            turn_in_progress: Arc::new(AtomicBool::new(false)),
            shutdown_requested: Arc::new(AtomicBool::new(false)),
            interrupt_requested: Arc::new(AtomicBool::new(false)),
            child: None,
            child_group: Arc::new(AtomicU32::new(0)),
            interrupt_hook_registered: false,
            stderr_task: None,
            sse_task: None,
            server_base_url: None,
            provider_session_id: None,
        }
    }

    async fn start_inner(&mut self, config: &AgentConfig) -> Result<()> {
        let port = allocate_port()?;
        let mut command = Command::new("opencode");
        command
            .arg("serve")
            .arg("--port")
            .arg(port.to_string())
            .stdin(std::process::Stdio::null())
            .stdout(std::process::Stdio::null())
            .stderr(std::process::Stdio::piped())
            // Dropping the harness (e.g. a run task is aborted) must not leak a
            // live server. The direct-child kill this fires is a backstop; the
            // group kill in `stop()` and the interrupt hook are what reach the
            // descendants.
            .kill_on_drop(true);
        if let Some(cwd) = &config.cwd {
            command.current_dir(cwd);
        }
        // Own process group so `stop()` and the interrupt hook can kill the
        // whole tree — `opencode serve` spawns descendants (MCP servers, model
        // proxies, npm-shim grandchildren) that a direct-child kill orphans.
        #[cfg(unix)]
        command.process_group(0);
        super::configure_vendor_tokio_env(&mut command)?;

        let mut child = command
            .spawn()
            .map_err(|err| anyhow!("failed to spawn opencode serve: {err}"))?;
        let stderr = child
            .stderr
            .take()
            .ok_or_else(|| anyhow!("missing opencode stderr"))?;

        let base_url = format!("http://127.0.0.1:{port}");
        if let Err(err) = wait_for_server(&self.client, &base_url, &mut child).await {
            shutdown_child(&mut child).await;
            return Err(err);
        }

        let provider_session_id = match create_provider_session(&self.client, &base_url).await {
            Ok(provider_session_id) => provider_session_id,
            Err(err) => {
                shutdown_child(&mut child).await;
                return Err(err);
            }
        };

        let event_tx = self.events.clone();
        let raw_provider = self.raw_provider.clone();
        let client = self.client.clone();
        let shutdown_requested = self.shutdown_requested.clone();
        let turn_in_progress = self.turn_in_progress.clone();
        let interrupt_requested = self.interrupt_requested.clone();
        let approval = self.approval;
        let reader_base_url = base_url.clone();
        let reader_session_id = provider_session_id.clone();

        let sse_task = tokio::spawn(async move {
            let stream_url = format!("{reader_base_url}/event");
            let request = client
                .get(&stream_url)
                .header(reqwest::header::ACCEPT, "text/event-stream");
            let mut response = match request.send().await {
                Ok(response) => response,
                Err(err) => {
                    send_disconnect_error(
                        &event_tx,
                        &shutdown_requested,
                        format!("failed to connect to OpenCode SSE stream: {err}"),
                    );
                    return;
                }
            };
            if let Err(err) = response.error_for_status_ref() {
                send_disconnect_error(
                    &event_tx,
                    &shutdown_requested,
                    format!("OpenCode SSE stream failed: {err}"),
                );
                return;
            }

            let mut parser = SseParser::default();
            let mut state = opencode_mapping::ReaderState::new(reader_session_id.clone());

            loop {
                if shutdown_requested.load(Ordering::Relaxed) {
                    break;
                }

                let chunk = match response.chunk().await {
                    Ok(chunk) => chunk,
                    Err(err) => {
                        tracing::warn!(error = %err, "opencode SSE chunk read failed");
                        break;
                    }
                };
                let Some(chunk) = chunk else {
                    break;
                };

                for payload in parser.push(&chunk) {
                    if payload.trim().is_empty() || payload.trim() == "[DONE]" {
                        continue;
                    }
                    if let Some(raw_provider) = &raw_provider {
                        let _ = raw_provider.send(RawProviderEvent {
                            stream: "sse",
                            line: payload.clone(),
                        });
                    }

                    let parsed = serde_json::from_str::<Value>(&payload);
                    let raw = match parsed {
                        Ok(raw) => raw,
                        Err(err) => {
                            tracing::debug!(error = %err, payload = %payload, "invalid SSE data");
                            continue;
                        }
                    };

                    let mapped = opencode_mapping::map_event(&raw, &mut state);
                    for event in mapped.events {
                        let event = match event {
                            ConversationEvent::TurnStarted { turn_id } => {
                                turn_in_progress.store(true, Ordering::SeqCst);
                                // An abort that raced turn completion must not
                                // stamp the next turn.
                                interrupt_requested.store(false, Ordering::SeqCst);
                                ConversationEvent::TurnStarted { turn_id }
                            }
                            ConversationEvent::TurnCompleted { turn_id, status } => {
                                turn_in_progress.store(false, Ordering::SeqCst);
                                // opencode's event stream does not distinguish
                                // aborted turns (it reports idle or error), so
                                // the first turn boundary after an abort
                                // request is stamped Interrupted here.
                                let status = if interrupt_requested.swap(false, Ordering::SeqCst) {
                                    Lifecycle::Interrupted
                                } else {
                                    status
                                };
                                ConversationEvent::TurnCompleted { turn_id, status }
                            }
                            other => other,
                        };
                        let _ = event_tx.send(event);
                    }

                    for request_id in mapped.permission_requests {
                        if let Err(err) = answer_permission(
                            &client,
                            &reader_base_url,
                            &reader_session_id,
                            &request_id,
                            approval,
                        )
                        .await
                        {
                            tracing::warn!(
                                request_id = %request_id,
                                error = %err,
                                "failed to answer OpenCode permission request"
                            );
                        }
                    }
                }
            }

            turn_in_progress.store(false, Ordering::SeqCst);

            // Capture phase before closing the orphaned turn — close_orphaned_turn
            // takes the turn id, so turn_is_open() would read false afterwards.
            let turn_was_open = state.turn_is_open();
            let turn_had_content = state.turn_has_content();

            // Close any turn left open when the stream died so every TurnStarted
            // gets a terminal TurnCompleted and the journal never carries an open
            // turn past a disconnect. The body still fails once via the Error
            // handler below — this is ledger honesty, not a second failure.
            if turn_was_open {
                for event in state.close_orphaned_turn() {
                    let _ = event_tx.send(event);
                }
            }

            // Phase-aware reason so the durable Failed record names where the
            // stream died, not just that it did. A mid-turn disconnect is
            // already failed by the consumer's Error handler; this makes the
            // evidence actionable (pre-content vs mid-stream truncation).
            let reason = if turn_was_open {
                if turn_had_content {
                    "OpenCode event stream disconnected mid-stream after partial output"
                } else {
                    "OpenCode event stream disconnected before the turn produced any output"
                }
            } else {
                "OpenCode event stream disconnected"
            };
            send_disconnect_error(&event_tx, &shutdown_requested, reason);
        });

        let stderr_task = spawn_stderr_logger(stderr, "harness::opencode");

        let opencode_pid = child.id();
        if let Some(pid) = opencode_pid {
            self.child_group.store(pid, Ordering::Release);
            if let Err(err) = opencode_runtime::register_opencode_server(pid) {
                tracing::warn!(
                    opencode_pid = pid,
                    error = %err,
                    "failed to register OpenCode server runtime metadata"
                );
            }
        }

        // The signal handler (SIGINT/SIGTERM/SIGHUP — see bin/lf.rs) exits the
        // process before destructors run, so `kill_on_drop` never fires on
        // that path. This hook is what keeps `tmux kill-session` (SIGHUP) and a
        // Ctrl+C from orphaning the server's descendant tree. Registered once
        // per harness; restarts just update the atomic.
        if !self.interrupt_hook_registered {
            self.interrupt_hook_registered = true;
            let group = Arc::clone(&self.child_group);
            register_interrupt_cleanup(move || {
                let pid = group.swap(0, Ordering::AcqRel);
                if pid != 0 {
                    kill_process_group(pid);
                }
            });
        }

        self.child = Some(child);
        self.stderr_task = Some(stderr_task);
        self.sse_task = Some(sse_task);
        self.server_base_url = Some(base_url);
        self.provider_session_id = Some(provider_session_id);
        Ok(())
    }
}

#[async_trait]
impl Harness for OpenCodeHarness {
    fn set_raw_provider_sender(
        &mut self,
        raw_provider: Option<mpsc::UnboundedSender<RawProviderEvent>>,
    ) {
        self.raw_provider = raw_provider;
    }

    async fn start(&mut self, config: &AgentConfig) -> Result<()> {
        if self.child.is_some() {
            return Ok(());
        }

        self.shutdown_requested.store(false, Ordering::SeqCst);
        self.config = Some(config.clone());
        self.should_seed_prompt = true;

        let start_result = self.start_inner(config).await;
        if let Err(err) = start_result {
            let _ = self.stop().await;
            return Err(err);
        }
        Ok(())
    }

    async fn send_input(&mut self, content: &str) -> Result<()> {
        let config = self
            .config
            .as_ref()
            .ok_or_else(|| anyhow!("opencode harness not started"))?;

        let first_turn = self.should_seed_prompt;
        let turn_content = build_turn_content(content, config, first_turn);
        let Some(turn_content) = turn_content else {
            return Ok(());
        };

        self.interrupt_requested.store(false, Ordering::SeqCst);

        if self
            .turn_in_progress
            .compare_exchange(false, true, Ordering::SeqCst, Ordering::SeqCst)
            .is_err()
        {
            return Err(HarnessError::TurnAlreadyInProgress.into());
        }
        let mut turn_guard = TurnInProgressGuard::new(self.turn_in_progress.clone());

        let base_url = self
            .server_base_url
            .clone()
            .ok_or_else(|| anyhow!("opencode server not started"))?;
        let provider_session_id = self
            .provider_session_id
            .clone()
            .ok_or_else(|| anyhow!("opencode provider session id is not available"))?;

        let payload = build_turn_payload(&turn_content, config, first_turn);

        let message_url = format!("{base_url}/session/{provider_session_id}/message");
        send_request_with_retry(&self.client, Method::POST, &message_url, Some(payload)).await?;

        self.should_seed_prompt = false;
        turn_guard.disarm();
        Ok(())
    }

    async fn interrupt(&mut self) -> Result<()> {
        // opencode cancels via its session abort endpoint; the server and
        // session stay up for the next turn. The turn's terminal status is
        // stamped Interrupted by the SSE task (see the abort note there),
        // since opencode itself reports the boundary as idle/error.
        if !self.turn_in_progress.load(Ordering::SeqCst) {
            return Ok(());
        }
        let base_url = self
            .server_base_url
            .clone()
            .ok_or_else(|| anyhow!("opencode server not started"))?;
        let provider_session_id = self
            .provider_session_id
            .clone()
            .ok_or_else(|| anyhow!("opencode provider session id is not available"))?;

        self.interrupt_requested.store(true, Ordering::SeqCst);
        let abort_url = format!("{base_url}/session/{provider_session_id}/abort");
        send_request_with_retry(&self.client, Method::POST, &abort_url, Some(json!({}))).await?;
        Ok(())
    }

    async fn stop(&mut self) -> Result<()> {
        self.shutdown_requested.store(true, Ordering::SeqCst);

        if let (Some(base_url), Some(provider_session_id)) =
            (&self.server_base_url, &self.provider_session_id)
        {
            let abort_url = format!("{base_url}/session/{provider_session_id}/abort");
            let _ =
                send_request_with_retry(&self.client, Method::POST, &abort_url, Some(json!({})))
                    .await;

            let delete_url = format!("{base_url}/session/{provider_session_id}");
            let _ = send_request_with_retry(&self.client, Method::DELETE, &delete_url, None).await;
        }

        let opencode_pid = self.child.as_ref().and_then(|child| child.id());
        if let Some(pid) = opencode_pid {
            // Kill the whole process group before awaiting the direct child,
            // so descendants (MCP servers, model proxies, npm-shim
            // grandchildren) come down with the server instead of orphaning.
            kill_process_group(pid);
        }
        if let Some(child) = self.child.as_mut() {
            shutdown_child(child).await;
        }
        self.child = None;
        self.child_group.store(0, Ordering::Release);
        if let Some(pid) = opencode_pid {
            if let Err(err) = opencode_runtime::unregister_opencode_server(pid) {
                tracing::warn!(
                    opencode_pid = pid,
                    error = %err,
                    "failed to unregister OpenCode server runtime metadata"
                );
            }
        }

        if let Some(task) = self.sse_task.take() {
            task.abort();
            let _ = task.await;
        }
        if let Some(task) = self.stderr_task.take() {
            task.abort();
            let _ = task.await;
        }

        self.turn_in_progress.store(false, Ordering::SeqCst);
        self.provider_session_id = None;
        self.server_base_url = None;

        Ok(())
    }

    fn capabilities(&self) -> Capabilities {
        Capabilities {
            // opencode queues mid-turn messages server-side rather than
            // steering the running turn; send_input reports
            // TurnAlreadyInProgress and the caller queues.
            supports_steer: false,
        }
    }

    fn provider_session_id(&self) -> Option<String> {
        self.provider_session_id.clone()
    }

    fn process_group_id(&self) -> Option<u32> {
        let group = self.child_group.load(Ordering::Acquire);
        (group > 1).then_some(group)
    }

    fn set_provider_session_id(&mut self, provider_session_id: Option<String>) {
        self.provider_session_id = provider_session_id;
    }
}

async fn shutdown_child(child: &mut Child) {
    let _ = child.start_kill();
    let _ = child.wait().await;
}

async fn create_provider_session(client: &reqwest::Client, base_url: &str) -> Result<String> {
    let session_url = format!("{base_url}/session");
    let response =
        send_request_with_retry(client, Method::POST, &session_url, Some(json!({}))).await?;
    let body: Value = response
        .json()
        .await
        .map_err(|err| anyhow!("failed to parse opencode session response: {err}"))?;

    parse_session_id(&body).ok_or_else(|| {
        anyhow!(
            "opencode session response did not include session id: {}",
            body
        )
    })
}

fn send_disconnect_error(
    event_tx: &mpsc::UnboundedSender<ConversationEvent>,
    shutdown_requested: &AtomicBool,
    message: impl Into<String>,
) {
    if shutdown_requested.load(Ordering::Relaxed) {
        return;
    }

    let _ = event_tx.send(ConversationEvent::Error {
        code: OPENCODE_DISCONNECTED_CODE.to_string(),
        message: message.into(),
    });
}

fn allocate_port() -> Result<u16> {
    let listener = std::net::TcpListener::bind("127.0.0.1:0")
        .map_err(|err| anyhow!("failed to allocate port for OpenCode: {err}"))?;
    let port = listener
        .local_addr()
        .map_err(|err| anyhow!("failed to read allocated OpenCode port: {err}"))?
        .port();
    Ok(port)
}

async fn wait_for_server(
    client: &reqwest::Client,
    base_url: &str,
    child: &mut Child,
) -> Result<()> {
    let deadline = Instant::now() + Duration::from_secs(15);
    let mut delay = Duration::from_millis(100);

    loop {
        if let Some(exit_status) = child
            .try_wait()
            .map_err(|err| anyhow!("failed to poll opencode serve process: {err}"))?
        {
            return Err(anyhow!(
                "opencode serve exited before becoming ready: {exit_status}"
            ));
        }

        if client.get(base_url).send().await.is_ok() {
            return Ok(());
        }

        if Instant::now() >= deadline {
            return Err(anyhow!(
                "timed out waiting for opencode serve health check at {base_url}"
            ));
        }

        tokio::time::sleep(delay).await;
        delay = std::cmp::min(delay * 2, Duration::from_secs(1));
    }
}

async fn send_request_with_retry(
    client: &reqwest::Client,
    method: Method,
    url: &str,
    payload: Option<Value>,
) -> Result<reqwest::Response> {
    let mut attempt = 0;

    loop {
        let mut request = client.request(method.clone(), url);
        if let Some(body) = payload.clone() {
            request = request.json(&body);
        }

        match request.send().await {
            Ok(response) if response.status().is_server_error() && attempt == 0 => {
                attempt += 1;
                tokio::time::sleep(Duration::from_millis(200)).await;
            }
            Ok(response) => {
                return response
                    .error_for_status()
                    .map_err(|err| anyhow!("OpenCode request failed ({method} {url}): {err}"));
            }
            Err(err) if attempt == 0 && (err.is_timeout() || err.is_connect()) => {
                attempt += 1;
                tokio::time::sleep(Duration::from_millis(200)).await;
            }
            Err(err) => {
                return Err(anyhow!("OpenCode request failed ({method} {url}): {err}"));
            }
        }
    }
}

async fn answer_permission(
    client: &reqwest::Client,
    base_url: &str,
    session_id: &str,
    request_id: &str,
    approval: ApprovalPolicy,
) -> Result<()> {
    let url = format!("{base_url}/session/{session_id}/permissions/{request_id}");
    let response = match approval {
        ApprovalPolicy::AutoApprove => "always",
    };
    let payload = json!({ "response": response });
    let _ = send_request_with_retry(client, Method::POST, &url, Some(payload)).await?;
    Ok(())
}

fn parse_session_id(value: &Value) -> Option<String> {
    value
        .get("id")
        .and_then(Value::as_str)
        .map(ToString::to_string)
}

fn build_turn_content(content: &str, config: &AgentConfig, first_turn: bool) -> Option<String> {
    if first_turn {
        let mut parts = Vec::new();
        if !config.task_prompt.trim().is_empty() {
            parts.push(config.task_prompt.trim().to_string());
        }
        if !content.trim().is_empty() {
            parts.push(content.trim().to_string());
        }
        if parts.is_empty() {
            None
        } else {
            Some(parts.join("\n\n"))
        }
    } else {
        let text = content.trim();
        if text.is_empty() {
            None
        } else {
            Some(text.to_string())
        }
    }
}

fn build_turn_payload(content: &str, config: &AgentConfig, first_turn: bool) -> Value {
    let mut payload = json!({
        "parts": [
            { "type": "text", "text": content }
        ]
    });

    if first_turn && !config.system_prompt.trim().is_empty() {
        payload["system"] = Value::String(config.system_prompt.trim().to_string());
    }

    if let Some((provider_id, model_id)) = opencode_model(config) {
        payload["model"] = json!({
            "providerID": provider_id,
            "modelID": model_id,
        });
    }

    payload
}

fn opencode_model(config: &AgentConfig) -> Option<(String, String)> {
    let agent_str = config.agent.as_deref()?;
    let (harness, variant) = parse_agent(agent_str);
    if harness != "opencode" {
        return None;
    }
    let variant = variant?;
    let (provider_id, model_id) = variant.split_once('/')?;
    if provider_id.is_empty() || model_id.is_empty() {
        return None;
    }
    Some((provider_id.to_string(), model_id.to_string()))
}

#[derive(Debug, Default)]
struct SseParser {
    buffer: String,
}

impl SseParser {
    fn push(&mut self, chunk: &[u8]) -> Vec<String> {
        self.buffer.push_str(&String::from_utf8_lossy(chunk));
        if self.buffer.contains('\r') {
            self.buffer = self.buffer.replace("\r\n", "\n").replace('\r', "\n");
        }

        let mut events = Vec::new();
        while let Some(separator) = self.buffer.find("\n\n") {
            let frame = self.buffer[..separator].to_string();
            self.buffer.drain(..separator + 2);
            if let Some(data) = parse_data_frame(&frame) {
                events.push(data);
            }
        }
        events
    }
}

fn parse_data_frame(frame: &str) -> Option<String> {
    let mut data_lines = Vec::new();
    for line in frame.lines() {
        if let Some(data) = line.strip_prefix("data:") {
            data_lines.push(data.trim_start().to_string());
        }
    }
    if data_lines.is_empty() {
        None
    } else {
        Some(data_lines.join("\n"))
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use serde_json::json;

    #[test]
    fn build_turn_content_includes_task_prompt_on_first_turn() {
        let content = build_turn_content(
            "",
            &AgentConfig {
                task_prompt: "task".to_string(),
                ..Default::default()
            },
            true,
        );
        assert_eq!(content.as_deref(), Some("task"));
    }

    #[test]
    fn sse_parser_collects_data_lines() {
        let mut parser = SseParser::default();
        let events = parser.push(b"event: message\ndata: {\"a\":1}\n\n");
        assert_eq!(events, vec!["{\"a\":1}".to_string()]);
    }

    #[test]
    fn sse_parser_handles_split_crlf_frames() {
        let mut parser = SseParser::default();
        assert!(parser.push(b"data: {\"a\":").is_empty());
        let events = parser.push(b"1}\r\n\r\n");
        assert_eq!(events, vec!["{\"a\":1}".to_string()]);
    }

    #[test]
    fn parse_session_id_requires_canonical_top_level_id() {
        assert_eq!(
            parse_session_id(&json!({"id": "session_1"})),
            Some("session_1".to_string())
        );
        assert_eq!(
            parse_session_id(&json!({"session": {"id": "session_2"}})),
            None
        );
        assert_eq!(parse_session_id(&json!({"sessionID": "session_3"})), None);
    }

    #[test]
    fn build_turn_payload_includes_explicit_opencode_model() {
        let payload = build_turn_payload(
            "hello",
            &AgentConfig {
                agent: Some("opencode:moonshotai/kimi-k2".to_string()),
                ..Default::default()
            },
            false,
        );
        assert_eq!(
            payload.get("model"),
            Some(&json!({
                "providerID": "moonshotai",
                "modelID": "kimi-k2"
            }))
        );
    }

    #[test]
    fn build_turn_payload_omits_model_for_non_opencode_agent_model() {
        let payload = build_turn_payload(
            "hello",
            &AgentConfig {
                agent: Some("claude:sonnet".to_string()),
                ..Default::default()
            },
            false,
        );
        assert!(payload.get("model").is_none());
    }

    #[test]
    fn build_turn_payload_uses_loopflow_default_for_bare_opencode() {
        let payload = build_turn_payload(
            "hello",
            &AgentConfig {
                agent: Some("opencode".to_string()),
                ..Default::default()
            },
            false,
        );
        assert_eq!(
            payload.get("model"),
            Some(&json!({
                "providerID": "opencode",
                "modelID": "glm-5.2"
            }))
        );
    }

    // The orphaned-leader shape (the reaper's Dead + group-alive arm at the
    // harness level): the direct child backgrounds a grandchild (same process
    // group) and exits. The grandchild touches a flag file after a short sleep;
    // the group kill `stop()` fires must take it down before the sleep
    // finishes, so the flag never appears. This is the same kill the interrupt
    // hook fires on SIGINT/SIGTERM/SIGHUP.
    #[cfg(unix)]
    #[tokio::test]
    async fn kill_process_group_reaches_the_grandchild() {
        let tmp = tempfile::tempdir().unwrap();
        let flag = tmp.path().join("survived");
        let mut command = Command::new("sh");
        command
            .arg("-c")
            .arg(format!("(sleep 1 && touch {}) &", flag.display()));
        command.process_group(0);
        let mut child = command.spawn().unwrap();
        let pid = child.id().unwrap();
        // Let the shell fork the grandchild and exit.
        let _ = child.wait().await;

        kill_process_group(pid);

        // Past the grandchild's sleep: if it leaked, the flag would exist.
        tokio::time::sleep(Duration::from_millis(1500)).await;
        assert!(!flag.exists(), "grandchild outlived the group kill");
    }

    // The W2-225 interrupt shape: the leader (`opencode serve`) is still alive
    // when the signal handler fires the group kill, with a descendant (MCP
    // server / model proxy / npm-shim grandchild) running in the same group.
    // Both must come down — the parent leader AND the provider-child — not just
    // the orphaned grandchild. This is the kill `stop()` and the
    // interrupt-cleanup hook fire on SIGINT/SIGTERM/SIGHUP.
    #[cfg(unix)]
    #[tokio::test]
    async fn kill_process_group_reaps_live_parent_and_provider_child() {
        let tmp = tempfile::tempdir().unwrap();
        let parent_flag = tmp.path().join("parent_alive");
        let child_flag = tmp.path().join("descendant_survived");
        let mut command = Command::new("sh");
        command.arg("-c").arg(format!(
            "touch {parent} && (sleep 2 && touch {child}) & sleep 30",
            parent = parent_flag.display(),
            child = child_flag.display(),
        ));
        command.process_group(0);
        let mut child = command.spawn().unwrap();
        let pid = child.id().unwrap();

        // Wait for the leader to prove it is alive before we kill the group.
        let deadline = Instant::now() + Duration::from_secs(2);
        loop {
            if parent_flag.exists() {
                break;
            }
            if Instant::now() >= deadline {
                panic!("parent leader never started");
            }
            tokio::time::sleep(Duration::from_millis(20)).await;
        }

        kill_process_group(pid);

        // The parent leader must be dead: the direct child exited (killed).
        let leader_dead = tokio::time::timeout(Duration::from_secs(2), child.wait())
            .await
            .is_ok();
        assert!(leader_dead, "live parent leader outlived the group kill");

        // Past the descendant's sleep: if it leaked, the flag would exist.
        tokio::time::sleep(Duration::from_millis(2500)).await;
        assert!(
            !child_flag.exists(),
            "provider-child descendant outlived the group kill"
        );
    }

    // -- Fake-SSE disconnect matrix --

    use crate::chat::types::ConversationItem;
    use tokio::io::{AsyncReadExt, AsyncWriteExt};
    use tokio::net::TcpListener;

    /// Where the fake SSE stream dies relative to the turn lifecycle.
    #[derive(Clone, Copy, Debug, PartialEq, Eq)]
    enum DisconnectCase {
        /// Stream closes before any event — no TurnStarted.
        PreContent,
        /// Stream closes after `session.status: active` but before any content.
        AfterActive,
        /// Stream closes after a tool call starts but before it completes.
        MidTool,
        /// Stream closes after a durable tool completion (Command) — the
        /// side-effecting case that is NOT replay-safe.
        AfterDurable,
    }

    const SESSION_ID: &str = "test_session";

    fn scripted_sse_events(case: DisconnectCase) -> Vec<&'static str> {
        let active = r#"data: {"type":"session.status","properties":{"sessionID":"test_session","status":"active"}}"#;
        let text = r#"data: {"type":"message.part.updated","properties":{"sessionID":"test_session","part":{"id":"p1","type":"TextPart","delta":"working"}}}"#;
        let tool_running = r#"data: {"type":"message.part.updated","properties":{"sessionID":"test_session","part":{"id":"tool_1","type":"ToolPart","state":"running","name":"Bash","command":["echo","ok"]}}}"#;
        let tool_completed = r#"data: {"type":"message.part.updated","properties":{"sessionID":"test_session","part":{"id":"tool_1","type":"ToolPart","state":"completed","name":"Bash","command":["echo","ok"],"output":"ok"}}}"#;

        match case {
            DisconnectCase::PreContent => vec![],
            DisconnectCase::AfterActive => vec![active],
            DisconnectCase::MidTool => vec![active, tool_running],
            DisconnectCase::AfterDurable => vec![active, text, tool_completed],
        }
    }

    /// Drive the fake SSE stream through the same pipeline the harness SSE task
    /// uses: SseParser → map_event → close_orphaned_turn + Error on disconnect.
    async fn process_fake_sse(case: DisconnectCase) -> Vec<ConversationEvent> {
        let listener = TcpListener::bind("127.0.0.1:0").await.unwrap();
        let port = listener.local_addr().unwrap().port();
        let url = format!("http://127.0.0.1:{port}/event");

        tokio::spawn(async move {
            let (mut socket, _) = listener.accept().await.unwrap();
            let mut buf = vec![0u8; 4096];
            let _ = socket.read(&mut buf).await;
            socket
                .write_all(b"HTTP/1.1 200 OK\r\nContent-Type: text/event-stream\r\n\r\n")
                .await
                .unwrap();
            for event in scripted_sse_events(case) {
                socket
                    .write_all(format!("{event}\n\n").as_bytes())
                    .await
                    .unwrap();
            }
            drop(socket);
        });

        let client = reqwest::Client::new();
        let mut response = client
            .get(&url)
            .header(reqwest::header::ACCEPT, "text/event-stream")
            .send()
            .await
            .unwrap();

        let (tx, mut rx) = mpsc::unbounded_channel::<ConversationEvent>();
        let mut state = opencode_mapping::ReaderState::new(SESSION_ID.to_string());
        let mut parser = SseParser::default();

        while let Ok(Some(chunk)) = response.chunk().await {
            for payload in parser.push(&chunk) {
                if payload.trim().is_empty() || payload.trim() == "[DONE]" {
                    continue;
                }
                if let Ok(raw) = serde_json::from_str::<Value>(&payload) {
                    let mapped = opencode_mapping::map_event(&raw, &mut state);
                    for event in mapped.events {
                        let _ = tx.send(event);
                    }
                }
            }
        }

        let turn_was_open = state.turn_is_open();
        let turn_had_content = state.turn_has_content();
        if turn_was_open {
            for event in state.close_orphaned_turn() {
                let _ = tx.send(event);
            }
        }
        let reason = if turn_was_open {
            if turn_had_content {
                "OpenCode event stream disconnected mid-stream after partial output"
            } else {
                "OpenCode event stream disconnected before the turn produced any output"
            }
        } else {
            "OpenCode event stream disconnected"
        };
        let _ = tx.send(ConversationEvent::Error {
            code: OPENCODE_DISCONNECTED_CODE.to_string(),
            message: reason.to_string(),
        });

        let mut events = Vec::new();
        while let Ok(event) = rx.try_recv() {
            events.push(event);
        }
        events
    }

    fn assert_no_false_completed(events: &[ConversationEvent]) {
        let false_completed = events.iter().any(|event| {
            matches!(
                event,
                ConversationEvent::TurnCompleted {
                    status: Lifecycle::Completed,
                    ..
                }
            )
        });
        assert!(
            !false_completed,
            "a disconnect must never produce a Completed turn; events: {events:?}"
        );
    }

    fn assert_every_started_turn_closed(events: &[ConversationEvent]) {
        let started = events
            .iter()
            .filter(|e| matches!(e, ConversationEvent::TurnStarted { .. }))
            .count();
        let closed = events
            .iter()
            .filter(|e| matches!(e, ConversationEvent::TurnCompleted { .. }))
            .count();
        assert_eq!(
            started, closed,
            "every TurnStarted must get a terminal TurnCompleted; events: {events:?}"
        );
    }

    fn assert_disconnect_error_present(events: &[ConversationEvent]) {
        let has_error = events.iter().any(|event| {
            matches!(
                event,
                ConversationEvent::Error { code, .. }
                    if code == OPENCODE_DISCONNECTED_CODE
            )
        });
        assert!(
            has_error,
            "a disconnect must emit an Error with opencode_disconnected; events: {events:?}"
        );
    }

    #[tokio::test]
    async fn fake_sse_pre_content_disconnect() {
        let events = process_fake_sse(DisconnectCase::PreContent).await;
        assert_no_false_completed(&events);
        assert_every_started_turn_closed(&events);
        assert_disconnect_error_present(&events);
        // No turn was started, so no TurnCompleted.
        assert!(
            !events
                .iter()
                .any(|e| matches!(e, ConversationEvent::TurnCompleted { .. })),
            "pre-content disconnect should not close a turn that never started"
        );
    }

    #[tokio::test]
    async fn fake_sse_after_active_disconnect() {
        let events = process_fake_sse(DisconnectCase::AfterActive).await;
        assert_no_false_completed(&events);
        assert_every_started_turn_closed(&events);
        assert_disconnect_error_present(&events);
        // Turn started but produced no content → pre-content reason.
        let error = events.iter().find_map(|e| match e {
            ConversationEvent::Error { message, .. } => Some(message.as_str()),
            _ => None,
        });
        assert!(
            error.is_some_and(|m| m.contains("before the turn")),
            "pre-content disconnect reason, got: {error:?}"
        );
    }

    #[tokio::test]
    async fn fake_sse_mid_tool_disconnect() {
        let events = process_fake_sse(DisconnectCase::MidTool).await;
        assert_no_false_completed(&events);
        assert_every_started_turn_closed(&events);
        assert_disconnect_error_present(&events);
        // Tool started → content seen → mid-stream reason.
        let error = events.iter().find_map(|e| match e {
            ConversationEvent::Error { message, .. } => Some(message.as_str()),
            _ => None,
        });
        assert!(
            error.is_some_and(|m| m.contains("mid-stream")),
            "mid-stream disconnect reason, got: {error:?}"
        );
        // The tool item was started but not completed.
        assert!(
            events
                .iter()
                .any(|e| matches!(e, ConversationEvent::ItemStarted { .. })),
            "tool start should be visible before the disconnect"
        );
        assert!(
            !events
                .iter()
                .any(|e| matches!(e, ConversationEvent::ItemCompleted { .. })),
            "tool should not have completed before the mid-tool disconnect"
        );
    }

    #[tokio::test]
    async fn fake_sse_after_durable_disconnect() {
        let events = process_fake_sse(DisconnectCase::AfterDurable).await;
        assert_no_false_completed(&events);
        assert_every_started_turn_closed(&events);
        assert_disconnect_error_present(&events);
        // A durable tool (Command) completed before the disconnect.
        let has_completed_command = events.iter().any(|e| {
            matches!(
                e,
                ConversationEvent::ItemCompleted {
                    item: ConversationItem::Command { .. },
                    ..
                }
            )
        });
        assert!(
            has_completed_command,
            "the durable Command completion must be visible before the disconnect; events: {events:?}"
        );
        // Still no false Completed — the turn closes Failed.
        let turn_close = events.iter().find_map(|e| match e {
            ConversationEvent::TurnCompleted { status, .. } => Some(*status),
            _ => None,
        });
        assert_eq!(
            turn_close,
            Some(Lifecycle::Failed),
            "turn must close Failed even after a durable completion"
        );
    }
}