cera-cli 0.2.6

//! Image-input loader shared by `cera run --image`, the line-REPL `/image`
//! command, and the TUI `/image` command.
//!
//! Resolves a user-supplied argument — filesystem path or `http(s)://` URL —
//! to bytes, with the same 50 MB cap and same non-regular-file rejection
//! across all three entry points. URL fetches use `reqwest::blocking` with a
//! streaming size guard so an adversarial / mis-Content-Length'd server can't
//! exhaust memory.

use std::io::Read;
use std::sync::OnceLock;

use anyhow::{Context, Result, bail, ensure};

/// Default byte cap applied at every `/image` and `--image` entry point.
/// LFM2-VL's 262 144-pixel input band is ~1 MB raw RGB; 50 MB rejects clearly
/// malicious / accidental multi-GB inputs without ever clipping a real image.
pub(crate) const MAX_IMAGE_BYTES: u64 = 50 * 1024 * 1024;

/// Per-fetch HTTP timeout. URL fetches block the foreground REPL/TUI; users
/// expect a few-second ceiling, not minute-long stalls behind a slow server.
const HTTP_TIMEOUT: std::time::Duration = std::time::Duration::from_secs(30);

/// Cap on redirect chain length so a redirect loop (or chain to a malicious
/// host) can't burn the entire timeout window.
const MAX_REDIRECTS: usize = 10;

/// Streaming read chunk size. 64 KiB is the standard Linux page-aligned
/// network read buffer; larger doesn't speed up TLS reads measurably and
/// smaller bloats syscall count.
const READ_CHUNK_BYTES: usize = 64 * 1024;

/// Returns true iff `arg` looks like an HTTP(S) URL. The scheme prefix match
/// is case-insensitive per RFC 3986 §3.1 ("Although schemes are case-
/// insensitive, the canonical form is lowercase…"). Anything else (including
/// `file://`, relative paths, absolute paths, and bare strings) is treated
/// as a filesystem path by [`load`].
pub(crate) fn looks_like_url(arg: &str) -> bool {
    let bytes = arg.as_bytes();
    matches!(bytes.get(..7), Some(p) if p.eq_ignore_ascii_case(b"http://"))
        || matches!(bytes.get(..8), Some(p) if p.eq_ignore_ascii_case(b"https://"))
}

/// Resolves `arg` to image bytes, capped at `max_bytes`. URL inputs fetch
/// over HTTP; everything else reads from disk.
pub(crate) fn load(arg: &str, max_bytes: u64) -> Result<Vec<u8>> {
    if looks_like_url(arg) {
        load_url(arg, max_bytes)
    } else {
        load_path(arg, max_bytes)
    }
}

fn load_path(path: &str, max_bytes: u64) -> Result<Vec<u8>> {
    let meta = std::fs::metadata(path).with_context(|| format!("stat image source {path}"))?;
    ensure!(meta.is_file(), "image source {path}: not a regular file");
    ensure!(
        meta.len() <= max_bytes,
        "image source {path}: file is {} bytes, larger than the {} byte cap",
        meta.len(),
        max_bytes
    );
    std::fs::read(path).with_context(|| format!("reading image source {path}"))
}

/// Process-wide HTTP client. Built once on first fetch; reused for every
/// subsequent URL fetch. Reqwest's blocking client owns its own tokio
/// runtime + TLS config — building it per-fetch was milliseconds-scale
/// waste that grows linearly with `cera run --image url1 --image url2 …`.
fn http_client() -> &'static reqwest::blocking::Client {
    static CLIENT: OnceLock<reqwest::blocking::Client> = OnceLock::new();
    CLIENT.get_or_init(|| {
        reqwest::blocking::Client::builder()
            .timeout(HTTP_TIMEOUT)
            .redirect(reqwest::redirect::Policy::limited(MAX_REDIRECTS))
            .user_agent(concat!("cera-cli/", env!("CARGO_PKG_VERSION")))
            .build()
            // A construction failure means the bundled rustls / tokio init
            // is broken at compile-pinned-version level; no sensible fallback.
            .expect("failed to build reqwest::blocking::Client for image fetch")
    })
}

fn load_url(url: &str, max_bytes: u64) -> Result<Vec<u8>> {
    let mut response = http_client()
        .get(url)
        .send()
        .with_context(|| format!("fetch image source {url}"))?
        .error_for_status()
        .with_context(|| format!("HTTP error from image source {url}"))?;

    // Cheap pre-flight: if the server announces a length over the cap, fail
    // before reading any body bytes. Honest servers always set this; lying
    // or omitting servers fall through to the streaming guard below.
    if let Some(content_length) = response.content_length() {
        ensure!(
            content_length <= max_bytes,
            "image source {url}: Content-Length {content_length} exceeds the {max_bytes} byte cap"
        );
    }

    // The pre-flight ensure! above guarantees `content_length <= max_bytes`
    // when present, so no second cap is needed here.
    let initial_capacity = usize::try_from(response.content_length().unwrap_or(0)).unwrap_or(0);
    let mut buf: Vec<u8> = Vec::with_capacity(initial_capacity);
    let mut chunk = [0u8; READ_CHUNK_BYTES];
    loop {
        let n = response
            .read(&mut chunk)
            .with_context(|| format!("read body from image source {url}"))?;
        if n == 0 {
            break;
        }
        let new_len = buf.len().saturating_add(n);
        if new_len as u64 > max_bytes {
            bail!("image source {url}: response exceeds the {max_bytes} byte cap");
        }
        buf.extend_from_slice(&chunk[..n]);
    }
    Ok(buf)
}

#[cfg(test)]
mod tests {
    use std::io::{Read as IoRead, Write};
    use std::net::TcpListener;
    use std::thread;

    use super::*;

    /// Bind a one-shot HTTP/1.1 server on `127.0.0.1:0`, write the supplied
    /// raw bytes (status line + headers + body) to the first incoming
    /// connection, and close. Returns the URL of the server's root, ready to
    /// pass to [`load`].
    ///
    /// The drained-request read is just enough to satisfy the client (HTTP
    /// keepalive isn't a concern — the connection closes after one
    /// response). Any test that needs to assert on the request line can
    /// extend this to a channel-returning variant.
    fn one_shot_server(response_bytes: Vec<u8>) -> String {
        let listener = TcpListener::bind("127.0.0.1:0").unwrap();
        let port = listener.local_addr().unwrap().port();
        thread::spawn(move || {
            if let Ok((mut stream, _)) = listener.accept() {
                // Read a single buffer's worth of the request — enough to
                // unblock the client's `send()`. We don't parse it.
                let mut buf = [0u8; 4096];
                let _ = stream.read(&mut buf);
                let _ = stream.write_all(&response_bytes);
            }
        });
        format!("http://127.0.0.1:{port}/")
    }

    fn build_response(status_line: &str, headers: &[&str], body: &[u8]) -> Vec<u8> {
        let mut buf: Vec<u8> = Vec::new();
        buf.extend_from_slice(status_line.as_bytes());
        buf.extend_from_slice(b"\r\n");
        for h in headers {
            buf.extend_from_slice(h.as_bytes());
            buf.extend_from_slice(b"\r\n");
        }
        buf.extend_from_slice(b"\r\n");
        buf.extend_from_slice(body);
        buf
    }

    #[test]
    fn classifier_accepts_http_and_https_any_case() {
        assert!(looks_like_url("http://example.com/a.jpg"));
        assert!(looks_like_url("https://example.com/a.jpg"));
        assert!(looks_like_url("https://"));
        assert!(looks_like_url("http://"));
        // RFC 3986 §3.1: schemes are case-insensitive.
        assert!(looks_like_url("Http://example.com/x"));
        assert!(looks_like_url("HTTP://example.com/x"));
        assert!(looks_like_url("HTTPS://example.com/x"));
        assert!(looks_like_url("HtTpS://example.com/x"));
    }

    #[test]
    fn classifier_rejects_non_http_inputs() {
        assert!(!looks_like_url(""));
        assert!(!looks_like_url("/abs/path.jpg"));
        assert!(!looks_like_url("./rel/path.jpg"));
        assert!(!looks_like_url("path.jpg"));
        assert!(!looks_like_url("file:///abs/path.jpg"));
        assert!(!looks_like_url("ftp://example.com/x"));
        assert!(!looks_like_url("httpfoo"));
        assert!(!looks_like_url("http")); // shorter than the prefix
    }

    #[test]
    fn load_path_reads_file_bytes() {
        let mut f = tempfile::NamedTempFile::new().unwrap();
        f.write_all(b"hello world").unwrap();
        let bytes = load(f.path().to_str().unwrap(), MAX_IMAGE_BYTES).unwrap();
        assert_eq!(bytes, b"hello world");
    }

    #[test]
    fn load_path_rejects_oversize() {
        let mut f = tempfile::NamedTempFile::new().unwrap();
        f.write_all(&[0u8; 32]).unwrap();
        let err = load(f.path().to_str().unwrap(), 16).unwrap_err();
        let s = format!("{err:#}");
        assert!(s.contains("byte cap"), "unexpected error: {s}");
    }

    #[test]
    fn load_path_rejects_non_regular() {
        // A directory exists on every supported target (Unix + Windows) and
        // is non-regular — exercises the same `is_file()` reject branch as
        // a Unix `/dev/zero` / FIFO without platform-gating.
        let dir = tempfile::tempdir().unwrap();
        let err = load(dir.path().to_str().unwrap(), MAX_IMAGE_BYTES).unwrap_err();
        let s = format!("{err:#}");
        assert!(s.contains("not a regular file"), "unexpected error: {s}");
    }

    #[test]
    fn load_path_missing_file_errors_with_context() {
        let err = load("/this/path/should/not/exist.jpg", MAX_IMAGE_BYTES).unwrap_err();
        let s = format!("{err:#}");
        assert!(s.contains("stat image source"), "unexpected error: {s}");
    }

    #[test]
    fn load_url_200_returns_body_bytes() {
        let body = b"hello world";
        let response = build_response(
            "HTTP/1.1 200 OK",
            &[
                &format!("Content-Length: {}", body.len()),
                "Connection: close",
            ],
            body,
        );
        let url = one_shot_server(response);
        let bytes = load(&url, MAX_IMAGE_BYTES).unwrap();
        assert_eq!(bytes, body);
    }

    #[test]
    fn load_url_4xx_surfaces_http_error() {
        let response = build_response(
            "HTTP/1.1 404 Not Found",
            &["Content-Length: 0", "Connection: close"],
            &[],
        );
        let url = one_shot_server(response);
        let err = load(&url, MAX_IMAGE_BYTES).unwrap_err();
        let s = format!("{err:#}");
        assert!(s.contains("404"), "unexpected error: {s}");
    }

    #[test]
    fn load_url_pre_flight_rejects_oversize_content_length() {
        // Server announces a body larger than the cap. Helper bails BEFORE
        // streaming any body bytes — the body in the response is irrelevant
        // (and short, to avoid blocking the test on connection buffering).
        let response = build_response(
            "HTTP/1.1 200 OK",
            &["Content-Length: 100000", "Connection: close"],
            &[],
        );
        let url = one_shot_server(response);
        let err = load(&url, 1024).unwrap_err();
        let s = format!("{err:#}");
        assert!(s.contains("Content-Length"), "unexpected error: {s}");
    }

    #[test]
    fn load_url_streaming_guard_rejects_when_content_length_missing() {
        // No Content-Length header — body is delimited by connection close.
        // Server sends 5000 bytes; helper's streaming guard with a 1024-byte
        // cap must bail on the first read iteration without OOMing.
        let response = build_response(
            "HTTP/1.1 200 OK",
            &["Connection: close"],
            &vec![0xABu8; 5000],
        );
        let url = one_shot_server(response);
        let err = load(&url, 1024).unwrap_err();
        let s = format!("{err:#}");
        assert!(
            s.contains("exceeds the 1024 byte cap"),
            "unexpected error: {s}"
        );
    }
}