kimun_notes/update/

mod.rs

//! Update awareness and (where permitted) self-update.
//!
//! On launch the app asks GitHub whether a newer stable `kimun-notes-v*` exists
//! and surfaces the result; on self-update-eligible channels it can also swap
//! the binary in place. All network and filesystem work here is **blocking** —
//! callers run it on `tokio::task::spawn_blocking` so the TUI never stalls.
//!
//! Self-update is offered only on the `script` and `direct` channels; on
//! `brew` and `cargo` a newer version is surfaced as the right upgrade command
//! rather than an in-app binary swap.
//! User-owned config (`update_check`) lives in `config.toml`; machine-managed
//! state (throttle, last-known version, dismissals) lives in `update_state.toml`.

mod apply;
mod channel;
mod github;
mod platform;
mod provider;
mod state;

pub use channel::InstallChannel;
pub use provider::{LatestRelease, ReleaseProvider};
pub use state::UpdateState;

/// The active release backend. **Single switch point** for *where* releases are
/// fetched from: implement [`ReleaseProvider`] elsewhere and return it here.
///
/// Scope: the trait covers release discovery and the human releases URL only.
/// Asset *naming* — the raw-binary filename ([`platform::binary_asset_name`])
/// and the `checksums-sha256.txt` name in [`apply()`] — is a property of this
/// project's CI (`build.yml`), constant across providers, and is intentionally
/// not part of the trait. A provider for a different repo layout would also
/// adjust those.
fn provider() -> impl ReleaseProvider {
    github::GitHubProvider
}

/// Human-facing releases page for the active provider (shown when self-update
/// isn't available on the current install channel).
pub fn releases_url() -> &'static str {
    provider().releases_url()
}

use chrono::{Duration, Utc};
use std::path::Path;

/// The version compiled into this binary.
pub const CURRENT_VERSION: &str = env!("CARGO_PKG_VERSION");

/// User-Agent sent on every GitHub request (the API rejects requests without
/// one). Shared by the releases query and the asset downloads.
pub(crate) const USER_AGENT: &str = concat!("kimun/", env!("CARGO_PKG_VERSION"));

/// How long a check result is reused before the next launch re-queries GitHub.
const CHECK_INTERVAL_HOURS: i64 = 24;

/// Issue a GET with kimün's standard headers. Blocking.
pub(crate) fn http_get(url: &str) -> Result<ureq::Response, UpdateError> {
    Ok(ureq::get(url)
        .set("User-Agent", USER_AGENT)
        .set("Accept", "application/vnd.github+json")
        .call()?)
}

/// The outcome of an update check, ready to drive the UI.
#[derive(Debug, Clone)]
pub struct UpdateStatus {
    /// The running version.
    pub current: String,
    /// The newest stable version available.
    pub latest: String,
    /// How this binary was installed (decides notify vs self-update).
    pub channel: InstallChannel,
    /// Whether `latest` is newer than `current`.
    pub update_available: bool,
    /// Whether the user has dismissed this exact `latest` version.
    pub dismissed: bool,
}

impl UpdateStatus {
    /// Whether the footer/dialog should nudge the user: an update exists and was
    /// not dismissed.
    pub fn should_notify(&self) -> bool {
        self.update_available && !self.dismissed
    }
}

/// Check for an update (blocking — prefer the async [`check_now`]).
///
/// When `force` is false the check is throttled: if the cached result is fresh
/// (< `CHECK_INTERVAL_HOURS`) no network call is made and the cached version
/// is reused. `force` (manual check / `kimun update`) always queries GitHub.
///
/// Returns `Ok(None)` only when throttled with no cached version yet.
pub fn check(config_dir: &Path, force: bool) -> Result<Option<UpdateStatus>, UpdateError> {
    // Force path queries immediately — no state load needed here (status_for
    // loads + persists it).
    if force {
        let release = provider().latest_stable()?;
        return Ok(Some(status_for(config_dir, &release)));
    }
    let st = UpdateState::load(config_dir);
    if st.is_stale(Utc::now(), Duration::hours(CHECK_INTERVAL_HOURS)) {
        let release = provider().latest_stable()?;
        Ok(Some(status_for(config_dir, &release)))
    } else {
        Ok(st
            .latest_version
            .as_deref()
            .map(|v| build_status(config_dir, &st, v)))
    }
}

/// Build an [`UpdateStatus`] for an already-known `version` using cached state —
/// no network, no writes.
fn build_status(config_dir: &Path, st: &UpdateState, version: &str) -> UpdateStatus {
    UpdateStatus {
        current: CURRENT_VERSION.to_string(),
        update_available: is_newer(version, CURRENT_VERSION),
        dismissed: st.dismissed_version.as_deref() == Some(version),
        channel: channel::detect(config_dir),
        latest: version.to_string(),
    }
}

/// Compute the [`UpdateStatus`] for an already-fetched `latest` release and
/// persist the check timestamp/version. Lets a caller that already holds a
/// [`LatestRelease`] (the apply path) avoid a second GitHub round-trip.
pub fn status_for(config_dir: &Path, latest: &LatestRelease) -> UpdateStatus {
    let mut st = UpdateState::load(config_dir);
    st.last_check = Some(Utc::now());
    st.latest_version = Some(latest.version.clone());
    // Best-effort persist; a write failure must not fail the status.
    if let Err(e) = st.save(config_dir) {
        tracing::warn!("could not save update state: {e}");
    }
    build_status(config_dir, &st, &latest.version)
}

/// Fetch the full latest release (with downloadable assets), needed before
/// [`apply()`]. Blocking — prefer the async [`latest_release`].
pub fn fetch_latest() -> Result<LatestRelease, UpdateError> {
    provider().latest_stable()
}

/// Download, verify, and install `latest`, replacing the running binary.
/// Blocking — prefer the async [`install`].
///
/// The caller must gate on [`InstallChannel::self_update_eligible`] first.
pub fn apply(latest: &LatestRelease) -> Result<(), UpdateError> {
    apply::self_update(latest)
}

/// Run a blocking update operation on the blocking pool, flattening the
/// `JoinError` into [`UpdateError::Task`]. The single home for the
/// `spawn_blocking` + join-error handling shared by every async caller.
async fn run_blocking<T, F>(f: F) -> Result<T, UpdateError>
where
    F: FnOnce() -> Result<T, UpdateError> + Send + 'static,
    T: Send + 'static,
{
    match tokio::task::spawn_blocking(f).await {
        Ok(result) => result,
        Err(e) => Err(UpdateError::Task(e.to_string())),
    }
}

/// Async [`check`] — runs on the blocking pool so the caller's runtime is never
/// stalled.
pub async fn check_now(
    config_dir: std::path::PathBuf,
    force: bool,
) -> Result<Option<UpdateStatus>, UpdateError> {
    run_blocking(move || check(&config_dir, force)).await
}

/// Async [`fetch_latest`].
pub async fn latest_release() -> Result<LatestRelease, UpdateError> {
    run_blocking(fetch_latest).await
}

/// Async [`apply()`] — consumes `latest` so it can move onto the blocking pool.
pub async fn install(latest: LatestRelease) -> Result<(), UpdateError> {
    run_blocking(move || apply(&latest)).await
}

/// Record that the user dismissed `version`, suppressing the notification until
/// a newer release appears. Writes only `update_state.toml`.
pub fn dismiss(config_dir: &Path, version: &str) -> std::io::Result<()> {
    let mut st = UpdateState::load(config_dir);
    st.dismissed_version = Some(version.to_string());
    st.save(config_dir)
}

/// Compare two `X.Y.Z` versions: is `candidate` strictly newer than `current`?
/// Unparseable input compares as not-newer (fail safe — never nudge on garbage).
fn is_newer(candidate: &str, current: &str) -> bool {
    match (parse_version(candidate), parse_version(current)) {
        (Some(c), Some(cur)) => c > cur,
        _ => false,
    }
}

/// Parse a plain `X.Y.Z` version into a comparable tuple. Returns `None` for
/// anything with a pre-release/build suffix or non-numeric parts — release tags
/// considered here are always plain stable triples.
fn parse_version(v: &str) -> Option<(u64, u64, u64)> {
    let mut parts = v.split('.');
    let major = parts.next()?.parse().ok()?;
    let minor = parts.next()?.parse().ok()?;
    let patch = parts.next()?.parse().ok()?;
    if parts.next().is_some() {
        return None;
    }
    Some((major, minor, patch))
}

/// Anything that can go wrong during an update check or self-update.
#[derive(Debug)]
pub enum UpdateError {
    /// Network / HTTP failure talking to GitHub.
    Http(Box<ureq::Error>),
    /// Failed to read a response body.
    Io(std::io::Error),
    /// Failed to parse the releases JSON.
    Parse(serde_json::Error),
    /// No stable `kimun-notes-v*` release found.
    NoRelease,
    /// This target has no published binary to self-update to.
    UnsupportedPlatform,
    /// A required release asset (binary or checksums) was absent.
    MissingAsset(String),
    /// No checksum line for the binary in `checksums-sha256.txt`.
    NoChecksum(String),
    /// Downloaded binary failed checksum verification.
    ChecksumMismatch { expected: String, actual: String },
    /// The in-place binary swap failed.
    Replace(std::io::Error),
    /// The blocking update task panicked or was cancelled.
    Task(String),
}

impl std::fmt::Display for UpdateError {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            Self::Http(e) => write!(f, "network error: {e}"),
            Self::Io(e) => write!(f, "I/O error: {e}"),
            Self::Parse(e) => write!(f, "could not parse GitHub response: {e}"),
            Self::NoRelease => write!(f, "no stable release found"),
            Self::UnsupportedPlatform => {
                write!(f, "no self-update binary is published for this platform")
            }
            Self::MissingAsset(name) => write!(f, "release is missing asset: {name}"),
            Self::NoChecksum(name) => write!(f, "no checksum published for {name}"),
            Self::ChecksumMismatch { expected, actual } => {
                write!(f, "checksum mismatch (expected {expected}, got {actual})")
            }
            Self::Replace(e) => write!(f, "could not replace the running binary: {e}"),
            Self::Task(e) => write!(f, "update task failed: {e}"),
        }
    }
}

impl std::error::Error for UpdateError {}

impl From<ureq::Error> for UpdateError {
    fn from(e: ureq::Error) -> Self {
        Self::Http(Box::new(e))
    }
}

impl From<std::io::Error> for UpdateError {
    fn from(e: std::io::Error) -> Self {
        Self::Io(e)
    }
}

impl From<serde_json::Error> for UpdateError {
    fn from(e: serde_json::Error) -> Self {
        Self::Parse(e)
    }
}

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

    #[test]
    fn newer_versions_compare_correctly() {
        assert!(is_newer("0.18.0", "0.17.0"));
        assert!(is_newer("1.0.0", "0.99.99"));
        assert!(is_newer("0.17.1", "0.17.0"));
        assert!(!is_newer("0.17.0", "0.17.0"));
        assert!(!is_newer("0.16.0", "0.17.0"));
    }

    #[test]
    fn unparseable_versions_never_nudge() {
        assert!(!is_newer("garbage", "0.17.0"));
        assert!(!is_newer("0.18.0-beta.1", "0.17.0"));
        assert!(!is_newer("0.18", "0.17.0"));
    }
}