quiverdb-cli 0.29.0

// SPDX-License-Identifier: AGPL-3.0-only
use std::env;
use std::fs;
use std::io::{self, Write};
use std::path::PathBuf;

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

fn use_color() -> bool {
    std::env::var("NO_COLOR").is_err() && std::env::var("TERM").map_or(true, |t| t != "dumb")
}

fn banner(mode: &str) {
    if !use_color() {
        println!("  QUIVER  {mode}  v{CURRENT_VERSION}");
        return;
    }
    let b = "\x1b[38;2;205;127;50m"; // #CD7F32 bronze — theme CHROME
    let v = "\x1b[38;2;63;182;168m"; // #3FB6A8 verdigris — theme ACCENT (the V arrowhead)
    let g = "\x1b[38;2;90;90;90m"; // dark gray
    let r = "\x1b[0m";
    println!();
    println!("{b}    ██████╗ ██╗   ██╗██╗{r}{v}██╗   ██╗{r}{b}███████╗██████╗ {r}");
    println!("{b}   ██╔═══██╗██║   ██║██║{r}{v}██║   ██║{r}{b}██╔════╝██╔══██╗{r}");
    println!("{b}   ██║   ██║██║   ██║██║{r}{v}╚██╗ ██╔╝{r}{b}█████╗  ██████╔╝{r}");
    println!("{b}   ██║▄▄ ██║██║   ██║██║{r}{v} ╚████╔╝ {r}{b}██╔══╝  ██╔══██╗{r}");
    println!("{b}   ╚██████╔╝╚██████╔╝██║{r}{v}  ╚██╔╝  {r}{b}███████╗██║  ██║{r}");
    println!("{b}    ╚══▀▀═╝  ╚═════╝ ╚═╝{r}{v}   ╚═╝   {r}{b}╚══════╝╚═╝  ╚═╝{r}");
    println!("{v}        {mode}  ·  v{CURRENT_VERSION}{r}");
    println!();
    println!("{g}  ┌──────────────────────────────────────────────┐{r}");
    println!("{g}  │  encrypted · memory-frugal · self-hostable   │{r}");
    println!("{g}  └──────────────────────────────────────────────┘{r}");
    println!();
}
use sha2::{Digest, Sha256};

const REPO: &str = "achref-soua/quiver";
const CURRENT_VERSION: &str = env!("CARGO_PKG_VERSION");

fn platform() -> Result<&'static str> {
    match std::env::consts::OS {
        "linux" => Ok("linux"),
        "macos" => Ok("macos"),
        "windows" => Ok("windows"),
        other => bail!(
            "unsupported OS for self-update: {other}; \
             download manually from https://github.com/{REPO}/releases"
        ),
    }
}

fn asset_name(os: &str, arch: &str) -> String {
    // Windows binaries carry the .exe suffix; Unix binaries do not.
    if os == "windows" {
        format!("quiver-{os}-{arch}.exe")
    } else {
        format!("quiver-{os}-{arch}")
    }
}

fn prompt_yes_no(question: &str) -> Result<bool> {
    if use_color() {
        print!("  \x1b[38;2;205;127;50m?\x1b[0m  {question} \x1b[38;2;90;90;90m[Y/n]\x1b[0m: ");
    } else {
        print!("  ? {question} [Y/n]: ");
    }
    let _ = io::stdout().flush();
    let mut input = String::new();
    io::stdin()
        .read_line(&mut input)
        .context("failed to read answer")?;
    let t = input.trim().to_ascii_lowercase();
    Ok(t.is_empty() || t == "y" || t == "yes")
}

fn arch() -> Result<&'static str> {
    match std::env::consts::ARCH {
        "x86_64" => Ok("x86_64"),
        "aarch64" | "arm64" => Ok("aarch64"),
        other => bail!("unsupported architecture: {other}"),
    }
}

/// Query the GitHub Releases API for the latest tag (e.g. `"0.17.0"`, strip `v`).
fn fetch_latest_version(agent: &ureq::Agent) -> Result<String> {
    let url = format!("https://api.github.com/repos/{REPO}/releases/latest");
    let resp: serde_json::Value = agent
        .get(&url)
        .set("User-Agent", &format!("quiver-cli/{CURRENT_VERSION}"))
        .set("Accept", "application/vnd.github+json")
        .call()
        .context("failed to reach GitHub API")?
        .into_json()
        .context("failed to parse GitHub API response")?;

    resp["tag_name"]
        .as_str()
        .map(|s| s.trim_start_matches('v').to_owned())
        .context("GitHub API response missing tag_name")
}

/// Download bytes from `url`.
fn fetch_bytes(agent: &ureq::Agent, url: &str) -> Result<Vec<u8>> {
    use std::io::Read;
    let mut body = agent
        .get(url)
        .set("User-Agent", &format!("quiver-cli/{CURRENT_VERSION}"))
        .call()
        .with_context(|| format!("failed to download {url}"))?
        .into_reader();
    let mut buf = Vec::new();
    body.read_to_end(&mut buf)
        .with_context(|| format!("failed to read response from {url}"))?;
    Ok(buf)
}

/// Returns true if `latest` is a strictly higher semver than `current`.
pub fn is_newer(current: &str, latest: &str) -> bool {
    parse_semver(latest) > parse_semver(current)
}

fn parse_semver(v: &str) -> (u64, u64, u64) {
    let parts: Vec<u64> = v
        .trim_start_matches('v')
        .splitn(3, '.')
        .map(|p| p.parse().unwrap_or(0))
        .collect();
    (
        parts.first().copied().unwrap_or(0),
        parts.get(1).copied().unwrap_or(0),
        parts.get(2).copied().unwrap_or(0),
    )
}

/// Verify SHA-256 of `data` against a checksum string (hex, optionally `"hash  filename"` format).
pub fn verify_sha256(data: &[u8], checksum_line: &str) -> Result<()> {
    let expected = checksum_line
        .split_whitespace()
        .next()
        .unwrap_or("")
        .to_lowercase();
    let mut hasher = Sha256::new();
    hasher.update(data);
    let actual = format!("{:x}", hasher.finalize());
    if actual != expected {
        bail!("checksum mismatch — expected {expected}, got {actual}; aborting");
    }
    Ok(())
}

/// Atomically replace `current_exe` with `new_binary`.
///
/// On Unix a simple rename over the running binary works because the inode
/// stays open. On Windows you cannot overwrite a running `.exe` in place, but
/// you *can* rename it to a new name while it is still running — the handle
/// the OS holds is to the inode, not the path. So we rename the old binary to
/// `<name>.exe.old`, then rename the temp file into place. The `.old` file is
/// cleaned up on the next update or reinstall.
fn atomic_replace(current_exe: &PathBuf, new_binary: &[u8]) -> Result<()> {
    let parent = current_exe
        .parent()
        .context("cannot determine parent directory of the current binary")?;

    let tmp = parent.join(format!("quiver-update-{}.tmp", std::process::id()));
    fs::write(&tmp, new_binary).context("failed to write updated binary to temp file")?;

    #[cfg(unix)]
    {
        use std::os::unix::fs::PermissionsExt;
        let mut perms = fs::metadata(&tmp)?.permissions();
        perms.set_mode(0o755);
        fs::set_permissions(&tmp, perms)?;

        fs::rename(&tmp, current_exe).with_context(|| {
            format!(
                "failed to replace {} — is it write-protected?",
                current_exe.display()
            )
        })?;
    }

    #[cfg(windows)]
    {
        // Build `<stem>.exe.old` beside the current binary.
        let mut old_name = current_exe
            .file_name()
            .context("current exe has no filename")?
            .to_os_string();
        old_name.push(".old");
        let old_path = parent.join(old_name);

        // Remove any leftover .old from a previous update attempt.
        let _ = fs::remove_file(&old_path);

        // Step 1: rename the running binary out of the way (always works on Windows).
        fs::rename(current_exe, &old_path).with_context(|| {
            format!(
                "failed to rename {} — close other programs using quiver.exe and retry",
                current_exe.display()
            )
        })?;

        // Step 2: move the downloaded binary into place.
        if let Err(e) = fs::rename(&tmp, current_exe) {
            // Restore the original binary so the user is not left broken.
            let _ = fs::rename(&old_path, current_exe);
            return Err(e).with_context(|| {
                format!("failed to install new binary to {}", current_exe.display())
            });
        }
        // .old file stays — harmless, cleaned up on the next update.
    }

    Ok(())
}

fn step(msg: &str) {
    if use_color() {
        println!("  \x1b[38;2;63;182;168m⟳\x1b[0m  {msg}");
    } else {
        println!("  > {msg}");
    }
}

fn ok(msg: &str) {
    if use_color() {
        println!("  \x1b[38;2;143;179;57m✔\x1b[0m  {msg}");
    } else {
        println!("  ✔ {msg}");
    }
}

fn run_blocking(check_only: bool) -> Result<()> {
    let mode = if check_only { "update check" } else { "update" };
    banner(mode);

    let agent = ureq::AgentBuilder::new().build();

    step("Checking latest release...");
    let latest = fetch_latest_version(&agent)?;

    if !is_newer(CURRENT_VERSION, &latest) {
        ok(&format!(
            "v{CURRENT_VERSION} is already the latest release."
        ));
        return Ok(());
    }

    if use_color() {
        println!(
            "  \x1b[38;2;205;127;50m↑\x1b[0m  v{CURRENT_VERSION} → \x1b[38;2;143;179;57mv{latest}\x1b[0m"
        );
    } else {
        println!("  New version available: v{latest}");
    }

    if check_only {
        println!();
        println!("  Run  quiver update  to install v{latest}.");
        return Ok(());
    }

    if !prompt_yes_no(&format!("Install v{latest} now?"))? {
        println!();
        println!("  Skipped. Run  quiver update  any time to upgrade.");
        return Ok(());
    }

    let os = platform()?;
    let arch = arch()?;
    let name = asset_name(os, arch);
    let base_url = format!("https://github.com/{REPO}/releases/download/v{latest}/{name}");
    let checksum_url = format!("{base_url}.sha256");

    step(&format!("Downloading {name}..."));
    let binary = fetch_bytes(&agent, &base_url)?;

    step("Verifying SHA-256 checksum...");
    let checksum_bytes = fetch_bytes(&agent, &checksum_url)?;
    let checksum_str =
        String::from_utf8(checksum_bytes).context("checksum file is not valid UTF-8")?;
    verify_sha256(&binary, &checksum_str)?;
    ok("Checksum verified.");

    let current_exe = env::current_exe().context("cannot determine current binary path")?;
    step(&format!("Installing to {}...", current_exe.display()));
    atomic_replace(&current_exe, &binary)?;
    ok(&format!("Quiver updated to v{latest}."));

    #[cfg(windows)]
    println!("\n  \x1b[38;2;90;90;90mOpen a new terminal to use the updated binary.\x1b[0m");

    Ok(())
}

/// Entry point for `quiver update`.
pub async fn run(check_only: bool) -> Result<()> {
    tokio::task::spawn_blocking(move || run_blocking(check_only))
        .await
        .context("update task panicked")?
}

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

    #[test]
    fn semver_newer() {
        assert!(is_newer("0.16.0", "0.17.0"));
        assert!(is_newer("0.17.0", "1.0.0"));
        assert!(is_newer("0.9.9", "0.10.0"));
    }

    #[test]
    fn semver_not_newer() {
        assert!(!is_newer("0.17.0", "0.17.0"));
        assert!(!is_newer("0.17.0", "0.16.9"));
        assert!(!is_newer("1.0.0", "0.99.9"));
    }

    #[test]
    fn sha256_correct_hash_passes() {
        let data = b"hello quiver";
        let mut h = Sha256::new();
        h.update(data);
        let hash = format!("{:x}", h.finalize());
        verify_sha256(data, &hash).expect("correct hash should pass");
    }

    #[test]
    fn sha256_wrong_hash_fails() {
        let data = b"hello quiver";
        let wrong = "0000000000000000000000000000000000000000000000000000000000000000";
        assert!(verify_sha256(data, wrong).is_err());
    }

    #[test]
    fn sha256_checksum_file_format() {
        // checksum files often look like "abc123  filename"
        let data = b"quiver binary content";
        let mut h = Sha256::new();
        h.update(data);
        let hash = format!("{:x}", h.finalize());
        let line = format!("{hash}  quiver-linux-x86_64");
        verify_sha256(data, &line).expect("should parse hash from checksum file format");
    }
}