netget-tool 0.3.4

//! HLS server (HTTP Live Streaming, RFC 8216).
//!
//! Serves an `.m3u8` playlist and its media segments over HTTP/1.1. The model decides the playlist
//! (variants and the segment list — structurally, or as verbatim m3u8 text) and what each segment
//! contains. Real MPEG-TS is binary, so a segment body is either model-supplied text or, for
//! genuine binary, an explicit `encoding: "hex"` field this server decodes for real — never
//! sniffed, never base64-guessed.
//!
//! A self-contained minimal HTTP/1.1 request reader lives here rather than sharing the hyper-based
//! `http` server: HLS needs nothing more than method + path routing, and the `http` server's
//! event/response model is a single `http_request` event, not the two distinct playlist/segment
//! events HLS wants. The framing written here is standard HTTP a real client (curl, ffplay) reads.

pub mod actions;

use anyhow::Result;
use std::net::SocketAddr;
use std::sync::Arc;
use tokio::io::{AsyncReadExt, AsyncWriteExt};
use tokio::net::TcpStream;
use tokio::sync::mpsc;
use tracing::{error, warn};

use crate::llm::action_helper::call_llm;
use crate::llm::ollama_client::OllamaClient;
use crate::logging::emit::Log;
use crate::protocol::Event;
use crate::server::connection::ConnectionId;
use crate::server::HlsProtocol;
use crate::state::app_state::AppState;
use crate::utils::WireFailure;
use crate::{console_error, console_trace};
use actions::{HLS_PLAYLIST_EVENT, HLS_SEGMENT_EVENT};

/// One HTTP response, assembled before it is framed.
///
/// `retry_after` exists so the two failure categories stay distinguishable on the wire: an
/// overloaded backend is transient and a player should back off and re-request, while anything
/// else is a hard 500 it should not hammer.
struct HlsResponse {
    status: u16,
    content_type: String,
    body: Vec<u8>,
    retry_after: bool,
}

impl HlsResponse {
    fn new(status: u16, content_type: impl Into<String>, body: Vec<u8>) -> Self {
        Self {
            status,
            content_type: content_type.into(),
            body,
            retry_after: false,
        }
    }

    /// The peer-visible reply for an internal failure.
    ///
    /// The body is [`WireFailure::prefixed_text`] — a `&'static str` category, never the error.
    /// The error goes to `tracing` and the status stream, which is where an operator looks.
    /// `Overloaded` becomes 503 + `Retry-After` and `Unavailable` becomes 500, so a client backs
    /// off on a transient saturation instead of recording a permanent fault.
    fn failure(failure: WireFailure) -> Self {
        Self {
            status: if failure.is_overloaded() { 503 } else { 500 },
            content_type: "text/plain; charset=utf-8".to_string(),
            body: failure.prefixed_text().as_bytes().to_vec(),
            retry_after: failure.is_overloaded(),
        }
    }
}

/// How long a peer may take to finish sending its request headers.
///
/// HLS clients send one short GET and nothing else, so this is generous for every legitimate
/// case and still bounds a connection that sends a byte at a time.
const HEADER_READ_TIMEOUT: std::time::Duration = std::time::Duration::from_secs(30);

/// Longest request path carried into the log, the status stream and the model's prompt.
const MAX_PATH_LEN: usize = 512;

/// The largest request head — request line, headers and the blank line ending them — this
/// server buffers, and its declared `max_inbound_bytes`.
///
/// The head is the only thing an HLS peer sends that is read: HLS clients send bodiless GETs,
/// and a request that declares a body is refused with 413 before any of it is read (see
/// [`declares_body`]). The read loop never asks for more than one byte past this, so a head
/// that has not ended within it is refused with 431 and the buffer never grows further.
pub const MAX_REQUEST_HEAD_BYTES: usize = 64 * 1024;

/// How long the unread remainder of a refused request is read and discarded before the socket
/// drops. Closing with bytes unread makes the kernel reset the connection, which can destroy
/// the refusal before the client reads it; the bytes go into the reused read chunk, so this is
/// bounded in time and allocates nothing.
const LINGER_AFTER_REFUSAL: std::time::Duration = std::time::Duration::from_secs(2);

/// Body of the 431 for a request head over [`MAX_REQUEST_HEAD_BYTES`].
const HEAD_TOO_LARGE_BODY: &[u8] = b"netget: request head too large\n";

/// Body of the 413 for a request that declares a body. Fixed text: the declared length stays
/// in the log.
const BODY_NOT_ACCEPTED_BODY: &[u8] = b"netget: this server accepts no request body\n";

/// Concurrent connections this server admits before it starts refusing.
///
/// [`crate::server::accept_bounded::DEFAULT_MAX_CONNECTIONS`]. [`HEADER_READ_TIMEOUT`] bounds
/// how long one peer can hold a slot saying nothing; only a cap bounds how many of them there
/// can be at once. HLS is one GET per connection with `Connection: close`, so a player's
/// working set is a handful of sockets and the house default is far above anything real
/// playback produces.
const MAX_CONNECTIONS: usize = crate::server::accept_bounded::DEFAULT_MAX_CONNECTIONS;

/// The body of [`CONNECTION_CAP_REFUSAL`].
const CONNECTION_CAP_BODY: &[u8] = b"Too many connections, try again later.\n";

/// What a peer over [`MAX_CONNECTIONS`] is told before the socket closes.
///
/// HLS rides HTTP/1.1, and 503 + `Retry-After` is HTTP's own vocabulary for exactly this —
/// the same pair this server already sends when the model backend is saturated, for the same
/// reason given at [`build_http_response`]: it tells a player to back off and re-request
/// rather than treat the stream as dead. Built through this server's own response builder so
/// the `Content-Length` cannot drift from the body, which a hand-written literal invites.
///
/// It is written before any request has been read, which is deliberate and safe: HTTP/1.1
/// §3.3 allows a server to send a final response before the request is complete, and the
/// response says `Connection: close`, so a client that was still sending stops and reads.
static CONNECTION_CAP_REFUSAL: std::sync::LazyLock<Vec<u8>> =
    std::sync::LazyLock::new(|| build_http_response(503, "text/plain", CONNECTION_CAP_BODY, true));

/// Why the peer got what it got, for the log only.
///
/// The three failure cases a reader of `netget.log` must be able to tell apart: the model
/// deliberately answered with an error status, the model answered nothing usable, and the LLM
/// call itself failed. Collapsing them makes an outage look like a policy decision.
const DECISION_MODEL_ANSWER: &str = "model_answer";
const DECISION_MODEL_REJECT: &str = "model_reject";
const DECISION_NO_ANSWER: &str = "fail_closed_no_answer";
const DECISION_BAD_ACTION: &str = "fail_closed_bad_action";
const DECISION_LLM_ERROR: &str = "fail_closed_llm_error";
const DECISION_LLM_OVERLOADED: &str = "fail_closed_llm_overloaded";

/// `model_reject` when the model chose a 4xx/5xx itself, `model_answer` otherwise.
fn decision_for_model_status(status: u16) -> &'static str {
    if status >= 400 {
        DECISION_MODEL_REJECT
    } else {
        DECISION_MODEL_ANSWER
    }
}

pub struct HlsServer;

impl HlsServer {
    /// Spawn the HLS server. Awaits the TCP bind so failure is reported as `Err`, and registers
    /// the accept loop so `stop_server` can abort it.
    pub async fn spawn_with_llm_actions(
        listen_addr: SocketAddr,
        llm_client: OllamaClient,
        app_state: Arc<AppState>,
        status_tx: mpsc::UnboundedSender<String>,
        server_id: crate::state::ServerId,
    ) -> Result<SocketAddr> {
        let listener =
            crate::server::socket_helpers::create_reusable_tcp_listener(listen_addr).await?;
        let local_addr = listener.local_addr()?;
        Log::new(Some(&status_tx)).info(format!("HLS server listening on {}", local_addr));

        let protocol = Arc::new(HlsProtocol::new());
        let task_registrar = app_state.clone();

        let limiter = crate::server::accept_bounded::ConnectionLimiter::new(MAX_CONNECTIONS);
        let accept_handle = tokio::spawn(async move {
            loop {
                match crate::server::accept_bounded::accept_bounded(
                    &listener,
                    &limiter,
                    &CONNECTION_CAP_REFUSAL,
                    "HLS",
                    Some(&status_tx),
                )
                .await
                {
                    Ok((stream, remote_addr, permit)) => {
                        let connection_id =
                            ConnectionId::new(app_state.get_next_unified_id().await);
                        let local_addr_conn = stream.local_addr().unwrap_or(local_addr);

                        use crate::state::server::{
                            ConnectionState as ServerConnectionState, ConnectionStatus,
                            ProtocolConnectionInfo,
                        };
                        let now = crate::utils::clock::Instant::now();
                        let conn_state = ServerConnectionState {
                            id: connection_id,
                            remote_addr,
                            local_addr: local_addr_conn,
                            bytes_sent: 0,
                            bytes_received: 0,
                            packets_sent: 0,
                            packets_received: 0,
                            last_activity: now,
                            status: ConnectionStatus::Active,
                            status_changed_at: now,
                            protocol_info: ProtocolConnectionInfo::empty(),
                        };
                        app_state
                            .add_connection_to_server(server_id, conn_state)
                            .await;
                        let _ = status_tx.send("__UPDATE_UI__".to_string());

                        let llm = llm_client.clone();
                        let state = app_state.clone();
                        let stx = status_tx.clone();
                        let proto = protocol.clone();
                        // Tracked, not detached: stop_server must abort this task too.
                        let task_owner = app_state.clone();
                        task_owner
                            .spawn_server_task(server_id, async move {
                                // Released when this task ends, and this task is the whole of
                                // the connection — HLS spawns nothing else per peer — so
                                // `MAX_CONNECTIONS` caps live connections, not accepts.
                                let _permit = permit;
                                if let Err(e) = Self::handle_connection(
                                    stream,
                                    remote_addr,
                                    connection_id,
                                    server_id,
                                    llm,
                                    state,
                                    stx.clone(),
                                    proto,
                                )
                                .await
                                {
                                    Log::new(Some(&stx))
                                        .debug(format!("HLS connection ended: {}", e));
                                }
                            })
                            .await;
                    }
                    Err(e) => {
                        console_error!(status_tx, "HLS accept error: {}", e);
                    }
                }
            }
        });

        task_registrar
            .register_server_task(server_id, accept_handle)
            .await;
        Ok(local_addr)
    }

    #[allow(clippy::too_many_arguments)]
    async fn handle_connection(
        stream: TcpStream,
        remote_addr: SocketAddr,
        connection_id: ConnectionId,
        server_id: crate::state::ServerId,
        llm: OllamaClient,
        state: Arc<AppState>,
        status_tx: mpsc::UnboundedSender<String>,
        protocol: Arc<HlsProtocol>,
    ) -> Result<()> {
        let (mut read_half, mut write_half) = tokio::io::split(stream);
        let mut buffer = Vec::new();
        let mut chunk = [0u8; 8192];

        // Read one HTTP request (headers up to the blank line; HLS clients send GETs with no body).
        //
        // Bounded by a deadline as well as by size. Without it a peer that connects and sends
        // one byte — or nothing at all — parks this task and its socket for as long as it cares
        // to keep the connection open, which is the whole of slowloris. 64 KiB caps how much a
        // peer can make us buffer; the deadline caps how long it can make us wait.
        //
        // Each read asks for at most one byte past `MAX_REQUEST_HEAD_BYTES`, so the buffer can
        // never exceed the bound by more than that byte, and a head that ends inside the bound
        // is told apart exactly from one that does not.
        let deadline = tokio::time::Instant::now() + HEADER_READ_TIMEOUT;
        let (method, path) = loop {
            let want = (MAX_REQUEST_HEAD_BYTES + 1 - buffer.len()).min(chunk.len());
            let n =
                match tokio::time::timeout_at(deadline, read_half.read(&mut chunk[..want])).await {
                    Ok(r) => r?,
                    Err(_) => anyhow::bail!(
                        "HLS request headers not complete within {}s",
                        HEADER_READ_TIMEOUT.as_secs()
                    ),
                };
            if n == 0 {
                return Ok(());
            }
            buffer.extend_from_slice(&chunk[..n]);
            match head_len(&buffer) {
                Some(len) if len <= MAX_REQUEST_HEAD_BYTES => {
                    let head = &buffer[..len];
                    if let Some(declared) = declares_body(head) {
                        warn!(
                            "HLS {} declared a request body ({}) decision=fail_closed_body_not_accepted; \
                             answering 413 without reading it",
                            remote_addr, declared
                        );
                        refuse(
                            &mut read_half,
                            &mut write_half,
                            &mut chunk,
                            413,
                            BODY_NOT_ACCEPTED_BODY,
                        )
                        .await;
                        return Ok(());
                    }
                    if let Some(req) = parse_request_line(head) {
                        break req;
                    }
                    // A complete head whose request line is unusable: nothing to route.
                    warn!(
                        "HLS {} sent an unparseable request line; closing",
                        remote_addr
                    );
                    refuse(&mut read_half, &mut write_half, &mut chunk, 400, b"").await;
                    return Ok(());
                }
                Some(_) => {
                    warn!(
                        "HLS {} request head over {} bytes decision=fail_closed_head_too_large",
                        remote_addr, MAX_REQUEST_HEAD_BYTES
                    );
                    refuse(
                        &mut read_half,
                        &mut write_half,
                        &mut chunk,
                        431,
                        HEAD_TOO_LARGE_BODY,
                    )
                    .await;
                    return Ok(());
                }
                None if buffer.len() > MAX_REQUEST_HEAD_BYTES => {
                    warn!(
                        "HLS {} request head over {} bytes decision=fail_closed_head_too_large",
                        remote_addr, MAX_REQUEST_HEAD_BYTES
                    );
                    refuse(
                        &mut read_half,
                        &mut write_half,
                        &mut chunk,
                        431,
                        HEAD_TOO_LARGE_BODY,
                    )
                    .await;
                    return Ok(());
                }
                None => {}
            }
        };
        // The path is peer-supplied and can be the better part of 64 KiB. It reaches the log,
        // the status stream and — as event data — the model's prompt, so it is bounded once
        // here rather than at each of those.
        let path = crate::utils::truncate_for_log(&path, MAX_PATH_LEN);
        console_trace!(status_tx, "HLS {} {}", method, path);

        // Refresh connection stats (bytes/packets in) so the dashboard rail shows real traffic
        // and a fresh last_activity rather than ↓0 ↑0. This is a one-shot HTTP request/response
        // connection (`Connection: close`): exactly one read of the request, one write of the
        // response, then the connection returns and closes.
        state
            .update_connection_stats(
                server_id,
                connection_id,
                Some(buffer.len() as u64),
                None,
                Some(1),
                None,
            )
            .await;

        let is_playlist = path.contains(".m3u8");
        let base = serde_json::json!({
            "peer_addr": remote_addr.to_string(),
            "connection_id": connection_id.to_string(),
            "path": path,
            "method": method,
        });
        let event = if is_playlist {
            Event {
                event_type: &HLS_PLAYLIST_EVENT,
                data: base,
            }
        } else {
            Event {
                event_type: &HLS_SEGMENT_EVENT,
                data: base,
            }
        };

        let (reply, decision): (HlsResponse, &str) = match call_llm(
            &llm,
            &state,
            server_id,
            Some(connection_id),
            &event,
            protocol.as_ref(),
        )
        .await
        {
            Ok(result) => {
                let action = result.raw_actions.into_iter().next();
                if is_playlist {
                    render_playlist(&action)
                } else {
                    render_segment(&action)
                }
            }
            Err(e) => {
                // Fail closed. The peer gets a category and a status it can act on; the error
                // itself — backend URL, model name, anyhow chain — stays in the log.
                let failure = WireFailure::classify(&e);
                let decision = if failure.is_overloaded() {
                    DECISION_LLM_OVERLOADED
                } else {
                    DECISION_LLM_ERROR
                };
                error!("HLS {} decision={} error={}", path, decision, e);
                Log::new(Some(&status_tx)).error(format!(
                    "HLS fail-closed for {} (decision={}): {}",
                    path, decision, e
                ));
                (HlsResponse::failure(failure), decision)
            }
        };

        let HlsResponse {
            status,
            content_type,
            body,
            retry_after,
        } = reply;
        let response = build_http_response(status, &content_type, &body, retry_after);
        write_half.write_all(&response).await?;
        write_half.flush().await?;

        state
            .update_connection_stats(
                server_id,
                connection_id,
                None,
                Some(response.len() as u64),
                None,
                Some(1),
            )
            .await;
        let _ = status_tx.send(format!(
            "→ HLS {} {} ({} bytes, decision={})",
            status,
            path,
            body.len(),
            decision
        ));
        Ok(())
    }
}

/// Render an `.m3u8` playlist response from the model's action.
///
/// Accepts either a verbatim `playlist` string, or a structured `segments` array
/// (`[{"uri":"seg0.ts","duration":6.0}, …]`) plus optional `target_duration`/`version` which are
/// assembled into a valid media playlist here.
fn render_playlist(action: &Option<serde_json::Value>) -> (HlsResponse, &'static str) {
    let ct = "application/vnd.apple.mpegurl".to_string();
    let Some(action) = action else {
        warn!(
            "HLS playlist decision={} (no action returned)",
            DECISION_NO_ANSWER
        );
        return (
            HlsResponse::failure(WireFailure::Unavailable),
            DECISION_NO_ANSWER,
        );
    };
    let status = action
        .get("status_code")
        .and_then(|v| v.as_u64())
        .unwrap_or(200) as u16;

    if let Some(playlist) = action.get("playlist").and_then(|v| v.as_str()) {
        return (
            HlsResponse::new(status, ct, playlist.as_bytes().to_vec()),
            decision_for_model_status(status),
        );
    }

    if let Some(segments) = action.get("segments").and_then(|v| v.as_array()) {
        let version = action.get("version").and_then(|v| v.as_u64()).unwrap_or(3);
        let target = action
            .get("target_duration")
            .and_then(|v| v.as_u64())
            .unwrap_or_else(|| {
                segments
                    .iter()
                    .filter_map(|s| s.get("duration").and_then(|d| d.as_f64()))
                    .fold(0.0_f64, f64::max)
                    .ceil() as u64
            })
            .max(1);
        let media_sequence = action
            .get("media_sequence")
            .and_then(|v| v.as_u64())
            .unwrap_or(0);
        let mut m = String::new();
        m.push_str("#EXTM3U\n");
        m.push_str(&format!("#EXT-X-VERSION:{}\n", version));
        m.push_str(&format!("#EXT-X-TARGETDURATION:{}\n", target));
        m.push_str(&format!("#EXT-X-MEDIA-SEQUENCE:{}\n", media_sequence));
        for seg in segments {
            let dur = seg
                .get("duration")
                .and_then(|d| d.as_f64())
                .unwrap_or(target as f64);
            let uri = seg
                .get("uri")
                .and_then(|u| u.as_str())
                .unwrap_or("segment.ts");
            m.push_str(&format!("#EXTINF:{:.3},\n{}\n", dur, uri));
        }
        let ended = action
            .get("ended")
            .and_then(|v| v.as_bool())
            .unwrap_or(true);
        if ended {
            m.push_str("#EXT-X-ENDLIST\n");
        }
        return (
            HlsResponse::new(status, ct, m.into_bytes()),
            decision_for_model_status(status),
        );
    }

    // No usable field: fail closed rather than fabricate a playlist. The peer gets a category;
    // what the action actually lacked is a log line for the operator, not for a stranger.
    warn!(
        "HLS playlist decision={} (action lacked both 'playlist' and 'segments')",
        DECISION_NO_ANSWER
    );
    (
        HlsResponse::failure(WireFailure::Unavailable),
        DECISION_NO_ANSWER,
    )
}

/// Render a media segment response. Body is either model text (`content`, utf8) or explicit
/// hex-encoded binary (`encoding: "hex"`, `data`) which is decoded here.
fn render_segment(action: &Option<serde_json::Value>) -> (HlsResponse, &'static str) {
    let Some(action) = action else {
        warn!(
            "HLS segment decision={} (no action returned)",
            DECISION_NO_ANSWER
        );
        return (
            HlsResponse::failure(WireFailure::Unavailable),
            DECISION_NO_ANSWER,
        );
    };
    let status = action
        .get("status_code")
        .and_then(|v| v.as_u64())
        .unwrap_or(200) as u16;
    let content_type = action
        .get("content_type")
        .and_then(|v| v.as_str())
        .unwrap_or("video/mp2t")
        .to_string();

    // Explicit binary path: decode hex for real (the send_tcp_data lesson).
    if let Some(data) = action.get("data").and_then(|v| v.as_str()) {
        let encoding = action
            .get("encoding")
            .and_then(|v| v.as_str())
            .unwrap_or("utf8");
        match encoding {
            "hex" => match hex::decode(data.trim()) {
                Ok(bytes) => {
                    return (
                        HlsResponse::new(status, content_type, bytes),
                        decision_for_model_status(status),
                    )
                }
                Err(e) => {
                    // The decoder's message names offsets in the model's own string; the peer
                    // gets the category, the operator gets the detail.
                    warn!(
                        "HLS segment decision={} (hex decode failed): {}",
                        DECISION_BAD_ACTION, e
                    );
                    return (
                        HlsResponse::failure(WireFailure::Unavailable),
                        DECISION_BAD_ACTION,
                    );
                }
            },
            "utf8" => {
                return (
                    HlsResponse::new(status, content_type, data.as_bytes().to_vec()),
                    decision_for_model_status(status),
                )
            }
            other => {
                warn!(
                    "HLS segment decision={} (unknown encoding {:?}; expected \"utf8\" or \"hex\")",
                    DECISION_BAD_ACTION, other
                );
                return (
                    HlsResponse::failure(WireFailure::Unavailable),
                    DECISION_BAD_ACTION,
                );
            }
        }
    }

    if let Some(content) = action.get("content").and_then(|v| v.as_str()) {
        return (
            HlsResponse::new(status, content_type, content.as_bytes().to_vec()),
            decision_for_model_status(status),
        );
    }

    // An action carrying neither `data` nor `content` is the model answering nothing usable.
    // Serving it as 200 with an empty body would be fail-open: a player accepts a zero-length
    // segment as a valid one and the stream silently plays nothing.
    warn!(
        "HLS segment decision={} (action carried neither 'data' nor 'content')",
        DECISION_NO_ANSWER
    );
    (
        HlsResponse::failure(WireFailure::Unavailable),
        DECISION_NO_ANSWER,
    )
}

/// Build an HTTP/1.1 response. `Connection: close` keeps this a clean one-request-per-connection
/// server, which HLS clients (a new GET per playlist/segment) handle fine.
fn build_http_response(status: u16, content_type: &str, body: &[u8], retry_after: bool) -> Vec<u8> {
    let reason = reason_phrase(status);
    // Only a 503 from the overload path carries Retry-After: it is the signal that tells a
    // player to back off and re-request rather than treat the stream as dead.
    let retry_header = if retry_after {
        "Retry-After: 5\r\n"
    } else {
        ""
    };
    let mut resp = format!(
        "HTTP/1.1 {} {}\r\nContent-Type: {}\r\nContent-Length: {}\r\n{}Cache-Control: no-cache\r\nConnection: close\r\n\r\n",
        status,
        reason,
        content_type,
        body.len(),
        retry_header
    )
    .into_bytes();
    resp.extend_from_slice(body);
    resp
}

/// Reason phrase for a status code.
///
/// The fallback is class-correct rather than a blanket `"OK"`: a model picking 403 used to be
/// framed as `HTTP/1.1 403 OK`, which is a contradiction on the wire.
fn reason_phrase(status: u16) -> &'static str {
    match status {
        200 => "OK",
        204 => "No Content",
        206 => "Partial Content",
        301 => "Moved Permanently",
        302 => "Found",
        304 => "Not Modified",
        400 => "Bad Request",
        401 => "Unauthorized",
        403 => "Forbidden",
        404 => "Not Found",
        410 => "Gone",
        413 => "Content Too Large",
        416 => "Range Not Satisfiable",
        429 => "Too Many Requests",
        431 => "Request Header Fields Too Large",
        500 => "Internal Server Error",
        502 => "Bad Gateway",
        503 => "Service Unavailable",
        504 => "Gateway Timeout",
        100..=199 => "Informational",
        200..=299 => "Success",
        300..=399 => "Redirection",
        400..=499 => "Client Error",
        500..=599 => "Server Error",
        _ => "Unknown",
    }
}

/// Length of the request head — through the `\r\n\r\n` that ends it — once it has arrived.
fn head_len(buf: &[u8]) -> Option<usize> {
    buf.windows(4).position(|w| w == b"\r\n\r\n").map(|p| p + 4)
}

/// The `Content-Length` / `Transfer-Encoding` header a request uses to declare a body, if any.
///
/// An HLS client sends bodiless GETs. A request that declares a body is refused rather than
/// answered: its body would be left unread on the socket, and the model would be asked about a
/// request whose content it never saw. `Content-Length: 0` declares no body and is accepted.
fn declares_body(head: &[u8]) -> Option<String> {
    let head = String::from_utf8_lossy(head);
    for line in head.split("\r\n").skip(1) {
        let Some((name, value)) = line.split_once(':') else {
            continue;
        };
        let name = name.trim();
        let value = value.trim();
        if name.eq_ignore_ascii_case("transfer-encoding") {
            return Some(format!(
                "Transfer-Encoding: {}",
                crate::utils::truncate_for_log(value, 64)
            ));
        }
        if name.eq_ignore_ascii_case("content-length") && value != "0" {
            return Some(format!(
                "Content-Length: {}",
                crate::utils::truncate_for_log(value, 32)
            ));
        }
    }
    None
}

/// Answer a request that will not reach the model, then close.
///
/// The response goes out first; whatever the peer is still sending is then read into the reused
/// chunk and discarded for at most [`LINGER_AFTER_REFUSAL`], so the close does not become a
/// reset that destroys the response before the peer reads it.
async fn refuse<R, W>(
    read_half: &mut R,
    write_half: &mut W,
    chunk: &mut [u8],
    status: u16,
    body: &[u8],
) where
    R: tokio::io::AsyncRead + Unpin,
    W: tokio::io::AsyncWrite + Unpin,
{
    let response = build_http_response(status, "text/plain; charset=utf-8", body, false);
    if write_half.write_all(&response).await.is_err() {
        return;
    }
    let _ = write_half.flush().await;
    let _ = write_half.shutdown().await;
    let deadline = tokio::time::Instant::now() + LINGER_AFTER_REFUSAL;
    while let Ok(Ok(n)) = tokio::time::timeout_at(deadline, read_half.read(chunk)).await {
        if n == 0 {
            break;
        }
    }
}

/// Parse the method and path from a partial HTTP request. Returns None until the request line and
/// header terminator (`\r\n\r\n`) are present.
fn parse_request_line(buf: &[u8]) -> Option<(String, String)> {
    // Terminator located on the bytes, and only the request line decoded. Running
    // `from_utf8` over the whole buffer rejected a request whose headers were valid because
    // something later in the buffer was not — including a read that merely happened to stop in
    // the middle of a multi-byte character — and the loop then waited for bytes that would
    // never make it valid. The request line itself must still be UTF-8; a path that is not is
    // a request this server cannot route.
    let header_end = buf.windows(4).position(|w| w == b"\r\n\r\n")?;
    let head = std::str::from_utf8(&buf[..header_end]).ok()?;
    let first = head.lines().next()?;
    let mut parts = first.split_whitespace();
    let method = parts.next()?.to_string();
    let path = parts.next()?.to_string();
    Some((method, path))
}