mushroomdb-backup 0.10.5

//! Restore a store from a backup prefix: `mushroomdb restore --from s3://…`.
//!
//! A restore reads one restore point (W4 §3.1):
//!
//! 1. `{prefix}/LATEST` names the newest manifest. A point in time
//!    (`--at`) uses the earliest manifest of the same lineage that holds the
//!    target commit instead; a timestamp is first resolved to a commit
//!    through the newest manifest's `commit_times.bin`, over the commits the
//!    restore can see. When the instant is at or after the newest recorded
//!    time and commits with no recorded time follow, it is refused
//!    ([`RestoreS3Error::CommitTimesEnd`]) rather than placed before them.
//! 2. Every file the manifest names is downloaded and its size and CRC-32
//!    checked; its segments are checked the same way, must be of the
//!    manifest's generation and contiguous from offset 0, and are
//!    concatenated into `wal.bin`. The manifest, its entries and the store's
//!    `backup.lineage` file must all name one lineage. A restore of the
//!    newest point also takes the segments of its generation shipped after
//!    the manifest, up to the last contiguous one. No manifest names those,
//!    so they are checked by their own frame checksums, size and offsets,
//!    not against an entry; a restore loses at most the segment cadence.
//! 3. With `--at`, the copy is cut back to the state right after the target
//!    commit ([`mushroomdb::restore::cut_at_commit`]): what `open_at` shows
//!    there, as an ordinary store. Either way `commit_times.bin` is trimmed
//!    to the commits restored, so the store stamps its own next commit.
//! 4. The copy gets a **new backup lineage**. A restored store is a new
//!    history: its next backup can never resume into the prefix it came from
//!    or delete anything there (defect-ledger row 138). Back it up to a new
//!    prefix.
//! 5. The copy is opened, read-only, and must hold the commits the restore
//!    point says it does. Only then is it moved into place, with one
//!    directory rename.
//!
//! Everything up to that rename happens in a staging directory beside the
//! target, every file fsynced. A restore that fails at any point, or a
//! crash, leaves the target exactly as it was — absent, or an empty
//! directory — never a part of a store. A failed restore removes the staging
//! directory; one killed outright can leave it behind, as a hidden
//! `.<name>.restore-…` directory and lock file beside the target, and the
//! next restore into the same target removes such leftovers no running
//! restore holds. A target that holds anything is refused before anything is
//! downloaded.
//!
//! A restore on boot ([`crate::boot`]) installs in place instead: its target
//! is a server's data directory, often a volume's mount point, which no
//! rename can replace and whose parent may not be writable. It stages inside
//! the data directory, records the restore point it came from
//! ([`crate::disk::PARENT_FILE`]), and moves the staged files into the
//! directory under an install marker ([`INSTALL_MARKER`]) naming them. The
//! marker is removed last: a data directory holding it holds a part of a
//! store, and the next boot removes exactly the files it names
//! ([`undo_interrupted_install`]) before restoring again.

use crate::disk;
use crate::lease::new_instance_id;
use crate::manifest::{crc32_hex, latest_key, object_key, parse_segment_key, Manifest};
use crate::segment::complete_frames;
use crate::store::{build_store_with, BackupConfig};
use crate::{redact, BackupError};
use mushroomdb::restore::cut_at_commit;
use object_store::path::Path as ObjectPath;
use object_store::ObjectStore;
use std::path::{Path, PathBuf};
use std::sync::Arc;
use time::format_description::well_known::Rfc3339;
use time::OffsetDateTime;
use tokio::io::AsyncWriteExt as _;

/// The point in time to restore (`--at`).
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum RestoreTarget {
    /// The store as it was right after this commit: the 0-based commit index
    /// `asof --commit` and `open_at` take. The restored store holds this
    /// many commits plus one.
    CommitSeq(u64),
    /// The store as it was at this instant: the last commit recorded at or
    /// before it, resolved through `commit_times.bin`.
    Timestamp(OffsetDateTime),
}

impl RestoreTarget {
    /// A commit index (digits only) or an RFC 3339 timestamp
    /// (`2026-10-01T12:00:00Z`).
    pub fn parse(s: &str) -> Result<Self, String> {
        let s = s.trim();
        if !s.is_empty() && s.bytes().all(|b| b.is_ascii_digit()) {
            return s
                .parse()
                .map(RestoreTarget::CommitSeq)
                .map_err(|e| format!("--at {s:?}: {e}"));
        }
        OffsetDateTime::parse(s, &Rfc3339)
            .map(RestoreTarget::Timestamp)
            .map_err(|_| {
                format!(
                    "--at {s:?} is neither a commit number nor an RFC 3339 timestamp \
                     (for example 2026-10-01T12:00:00Z)"
                )
            })
    }
}

impl std::fmt::Display for RestoreTarget {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            RestoreTarget::CommitSeq(n) => write!(f, "commit {n}"),
            RestoreTarget::Timestamp(t) => write!(f, "{}", rfc3339(*t)),
        }
    }
}

/// What a restore did.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct RestoreS3Outcome {
    /// The directory the store was restored into.
    pub into: PathBuf,
    /// The key of the manifest restored from.
    pub manifest: String,
    /// Commits the restored store holds (its `wal_total_commits`).
    pub commits: u64,
    /// The lineage of the backup restored from.
    pub source_lineage: String,
    /// The restored store's own, new lineage.
    pub lineage: String,
    /// The store files written, by name.
    pub files: Vec<String>,
    /// WAL segments concatenated into `wal.bin`.
    pub segments: usize,
    /// Bytes downloaded.
    pub bytes: u64,
    /// Why the segments after the newest manifest stop where they do, when a
    /// later one exists but could not be used (missing, short, not whole
    /// frames, or after a gap). The restore holds the commits up to there.
    pub tail_note: Option<String>,
    /// The store is in place, but something after the final rename failed
    /// (syncing the parent directory): its directory entry may not be
    /// durable yet.
    pub install_note: Option<String>,
    /// Staging directories (and their lock files) that earlier, killed
    /// restores into the same target left beside it, removed at the start.
    pub swept: Vec<PathBuf>,
}

/// Why a restore failed. The target is in every case as it was before the
/// restore — absent, or an empty directory. (Once the store is in place
/// nothing is reported as a failure; see [`RestoreS3Outcome::install_note`].)
#[derive(Debug)]
pub enum RestoreS3Error {
    /// The `--from` URL or the `AWS_*` environment cannot be used.
    Config(String),
    /// The object store refused or failed a request.
    Store(object_store::Error),
    /// Reading or writing local files failed.
    Io(std::io::Error),
    /// The prefix holds no backup: there is no `LATEST`.
    NoBackup { from: String },
    /// A manifest is missing or does not parse.
    Manifest(String),
    /// An object does not match its manifest entry (size, CRC-32, lineage,
    /// generation or offsets), or the restored copy does not hold what the
    /// manifest says.
    Verify(String),
    /// The target directory already holds something.
    IntoNotEmpty { into: PathBuf },
    /// A restore on boot: the data directory holds no store, but holds files
    /// a restore must not mix with (`entries`, by name).
    DataDirNotEmpty { dir: PathBuf, entries: Vec<String> },
    /// `--at` names a commit below the oldest one the backup still holds.
    BeforeOldestArchive { target: u64, oldest: u64 },
    /// `--at` names a commit past the newest one in the backup, which holds
    /// commits `0..commits`.
    AfterLatest {
        target: u64,
        commits: u64,
        /// Why the backup holds no more: segments after the newest manifest
        /// that exist but could not be used.
        not_restorable: Option<String>,
    },
    /// `--at` is a timestamp, and the backup holds no `commit_times.bin` (or
    /// one that records no commit) to resolve it with.
    CommitTimesAbsent,
    /// `--at` is a timestamp at or after the newest commit time the backup
    /// records, and the backup holds commits after that one with no recorded
    /// time: which of them the instant falls after cannot be told.
    CommitTimesEnd {
        /// The newest commit with a recorded time.
        recorded_through: u64,
        /// That commit's recorded time, in Unix milliseconds.
        recorded_through_ms: i64,
        /// The commits the backup holds (`0..commits`).
        commits: u64,
    },
    /// The target cannot be rebuilt from what the backup holds.
    Unreachable(String),
}

impl std::fmt::Display for RestoreS3Error {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            // A client builder's error can quote configuration; nothing
            // credential-shaped reaches the terminal.
            RestoreS3Error::Config(m) => write!(f, "{}", redact(m)),
            RestoreS3Error::Store(e) => write!(f, "object store: {}", redact(&e.to_string())),
            RestoreS3Error::Io(e) => write!(f, "{e}"),
            RestoreS3Error::NoBackup { from } => {
                write!(f, "{from} holds no backup: there is no LATEST under it")
            }
            RestoreS3Error::Manifest(m) => write!(f, "manifest: {m}"),
            RestoreS3Error::Verify(m) => write!(f, "the restore point does not verify: {m}"),
            RestoreS3Error::IntoNotEmpty { into } => write!(
                f,
                "{} is not empty; restore into a new or empty directory",
                into.display()
            ),
            RestoreS3Error::DataDirNotEmpty { dir, entries } => write!(
                f,
                "{} holds no store but is not empty ({}); a restore on boot installs only \
                 into an empty data directory. Move those away and start again",
                dir.display(),
                entries.join(", ")
            ),
            RestoreS3Error::BeforeOldestArchive { target, oldest } => write!(
                f,
                "commit {target} is before the oldest commit the backup holds ({oldest}); \
                 restore commit {oldest} or later"
            ),
            RestoreS3Error::AfterLatest {
                target,
                commits,
                not_restorable,
            } => {
                match commits.checked_sub(1) {
                    Some(newest) => write!(
                        f,
                        "commit {target} is past the newest commit in the backup ({newest})"
                    )?,
                    None => write!(f, "commit {target}: the backup holds no commits")?,
                }
                match not_restorable {
                    Some(why) => write!(f, "; {why}"),
                    None => Ok(()),
                }
            }
            RestoreS3Error::CommitTimesAbsent => write!(
                f,
                "the backup records no commit times (no commit_times.bin), so a timestamp \
                 cannot be resolved; restore by commit number instead"
            ),
            RestoreS3Error::CommitTimesEnd {
                recorded_through,
                recorded_through_ms,
                commits,
            } => write!(
                f,
                "the backup records commit times only up to commit {recorded_through}, at {at}, \
                 and commits {} to {} have none, so an instant at or after {at} cannot be \
                 placed among them. Restore an instant before it, or by commit index \
                 (--at <commit index>, for example --at {recorded_through}), or without --at \
                 for the newest",
                recorded_through + 1,
                commits.saturating_sub(1),
                at = ms_rfc3339(*recorded_through_ms),
            ),
            RestoreS3Error::Unreachable(m) => write!(f, "{m}"),
        }
    }
}

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

impl From<object_store::Error> for RestoreS3Error {
    fn from(e: object_store::Error) -> Self {
        RestoreS3Error::Store(e)
    }
}

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

impl From<BackupError> for RestoreS3Error {
    fn from(e: BackupError) -> Self {
        match e {
            BackupError::Store(e) => RestoreS3Error::Store(e),
            BackupError::Io(e) => RestoreS3Error::Io(e),
            BackupError::Manifest(m) => RestoreS3Error::Manifest(m),
            other => RestoreS3Error::Config(other.to_string()),
        }
    }
}

/// Restore the backup at `from` (`s3://bucket/prefix`, `r2://…` or
/// `minio://…`) into `into`, at `at` or at the newest restore point.
///
/// Credentials and endpoint come from the `AWS_*` variables, as for backup
/// ([`crate::build_store`]). See the module docs for what is verified and in
/// which order.
pub async fn restore_from_s3(
    from: &str,
    at: Option<RestoreTarget>,
    into: &Path,
) -> Result<RestoreS3Outcome, RestoreS3Error> {
    restore_from_s3_with(from, at, into, |k| std::env::var(k).ok()).await
}

/// [`restore_from_s3`] over any variable lookup.
pub async fn restore_from_s3_with(
    from: &str,
    at: Option<RestoreTarget>,
    into: &Path,
    get: impl Fn(&str) -> Option<String>,
) -> Result<RestoreS3Outcome, RestoreS3Error> {
    let config = BackupConfig::new(from.trim());
    let location = config
        .location()
        .map_err(|e| RestoreS3Error::Config(config_message(e, from)))?;
    check_into(into)?;
    let store = build_store_with(&config, get)
        .map_err(|e| RestoreS3Error::Config(config_message(e, from)))?;
    restore_from_store(store, &location.prefix, at, into).await
}

fn config_message(e: BackupError, from: &str) -> String {
    match e {
        BackupError::Config(m) => m.replace("MUSHROOMDB_BACKUP_S3_URL", "--from"),
        other => format!("{from}: {other}"),
    }
}

/// Restore from `prefix` in `store` into `into`: the object-store half of
/// [`restore_from_s3`], for any [`ObjectStore`].
pub async fn restore_from_store(
    store: Arc<dyn ObjectStore>,
    prefix: &str,
    at: Option<RestoreTarget>,
    into: &Path,
) -> Result<RestoreS3Outcome, RestoreS3Error> {
    restore_with(store, prefix, at, into, Mode::Beside).await
}

/// Restore the newest restore point of `prefix` into the data directory
/// `dir`, in place (see the [module docs](self)), recording where it came
/// from in [`crate::disk::PARENT_FILE`]. `dir` may be absent, empty, or hold
/// only `lost+found`, a `LOCK` file, an empty `wal.bin` and the leftovers of
/// killed restores into it.
pub async fn restore_in_place(
    store: Arc<dyn ObjectStore>,
    prefix: &str,
    dir: &Path,
) -> Result<RestoreS3Outcome, RestoreS3Error> {
    restore_with(store, prefix, None, dir, Mode::InPlace).await
}

/// Where a restore stages and how it installs.
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
enum Mode {
    /// `restore --from`: stage beside the target, install with one rename.
    Beside,
    /// Restore on boot: stage inside the data directory, install under the
    /// marker, and record the parent restore point.
    InPlace,
}

impl Mode {
    /// The directory the staging directory and its lock file go in.
    fn staging_root(self, into: &Path) -> PathBuf {
        match self {
            Mode::Beside => parent_of(into),
            Mode::InPlace => into.to_path_buf(),
        }
    }
}

async fn restore_with(
    store: Arc<dyn ObjectStore>,
    prefix: &str,
    at: Option<RestoreTarget>,
    into: &Path,
    mode: Mode,
) -> Result<RestoreS3Outcome, RestoreS3Error> {
    if mode == Mode::Beside {
        check_into(into)?;
    }
    let bucket = Bucket { store, prefix };
    let Some(latest_target) = bucket.get_text(&latest_key(prefix)).await? else {
        return Err(RestoreS3Error::NoBackup {
            from: if prefix.is_empty() {
                "the bucket".into()
            } else {
                prefix.to_string()
            },
        });
    };
    // Only once there is something to restore: a restore that cannot start
    // leaves what killed ones left alone. The sweep can delete gigabytes, so
    // not on a runtime thread.
    let (root, name) = (mode.staging_root(into), name_of(into));
    let swept = blocking(move || Ok(sweep_stale_staging(&root, &name))).await?;
    if mode == Mode::InPlace {
        let dir = into.to_path_buf();
        blocking(move || check_in_place(&dir)).await?;
    }
    let latest = bucket.manifest(&latest_target).await?;
    // The segments after the newest manifest matter unless the target is a
    // commit the manifest already holds; only then are they skipped.
    let needs_tail = !matches!(at, Some(RestoreTarget::CommitSeq(n)) if n < latest.commit_seq);
    let (tail, tail_note) = if needs_tail {
        bucket.tail_segments(&latest).await?
    } else {
        (Vec::new(), None)
    };
    let available = latest.commit_seq + tail.iter().map(|s| s.frames).sum::<u64>();
    let latest_commit_seq = latest.commit_seq;

    let target = match at {
        None => None,
        Some(RestoreTarget::CommitSeq(n)) => Some(n),
        Some(RestoreTarget::Timestamp(t)) => {
            Some(bucket.resolve_time(&latest, t, available).await?)
        }
    };
    let (key, manifest, tail, expect) = match target {
        None => (latest_target, latest, tail, available),
        Some(n) if n >= available => {
            return Err(RestoreS3Error::AfterLatest {
                target: n,
                commits: available,
                not_restorable: tail_note,
            })
        }
        Some(n) if n >= latest.commit_seq => (latest_target, latest, tail, n + 1),
        Some(n) => {
            let (key, m) = bucket.manifest_holding(n, &latest_target, &latest).await?;
            if n < m.oldest_commit_seq {
                return Err(RestoreS3Error::BeforeOldestArchive {
                    target: n,
                    oldest: m.oldest_commit_seq,
                });
            }
            // A restore point older than the newest was asked for: what
            // follows the newest does not concern it.
            (key, m, Vec::new(), n + 1)
        }
    };
    let tail_note = if manifest.commit_seq < latest_commit_seq {
        None
    } else {
        tail_note
    };
    if let Some(note) = &tail_note {
        tracing::warn!(note = %note, "the segments after the newest manifest stop early");
    }

    let staging = {
        let into = into.to_path_buf();
        blocking(move || Staging::new(&into, mode)).await?
    };
    let mut out = RestoreS3Outcome {
        into: into.to_path_buf(),
        manifest: key,
        commits: expect,
        source_lineage: manifest.lineage.clone(),
        lineage: new_instance_id(),
        files: Vec::new(),
        segments: 0,
        bytes: 0,
        tail_note,
        install_note: None,
        swept,
    };
    bucket
        .download(&manifest, &tail, staging.path(), &mut out)
        .await?;

    let dir = staging.path().to_path_buf();
    let lineage = out.lineage.clone();
    let cut = target;
    let manifest_key = out.manifest.clone();
    let parent = (mode == Mode::InPlace).then(|| disk::ParentRecord {
        lineage: out.source_lineage.clone(),
        manifest: out.manifest.clone(),
        commit_seq: expect,
    });
    blocking(move || -> Result<(), RestoreS3Error> {
        if let Some(n) = cut {
            cut_at_commit(&dir, n).map_err(|e| match e {
                mushroomdb::GraphError::CommitOutOfRange { floor, .. } if n < floor => {
                    RestoreS3Error::BeforeOldestArchive {
                        target: n,
                        oldest: floor,
                    }
                }
                mushroomdb::GraphError::CommitOutOfRange { .. } => {
                    RestoreS3Error::Unreachable(format!(
                        "commit {n} cannot be rebuilt from the restore point {manifest_key}: \
                         the history it needs before that restore point's snapshot is not \
                         whole in the backup (a truncating snapshot or archive retention \
                         removed some of it)"
                    ))
                }
                other => RestoreS3Error::Io(std::io::Error::other(other.to_string())),
            })?;
        }
        // The side files were read after the WAL they describe, so the
        // commit times can run past the commits restored; the restored store
        // must stamp its own next commit.
        mushroomdb::restore::trim_commit_times(&dir, expect)
            .map_err(|e| RestoreS3Error::Io(std::io::Error::other(e.to_string())))?;
        disk::create_lineage(&dir, &lineage)?;
        if let Some(parent) = &parent {
            disk::create_parent(&dir, parent)?;
        }
        let opened = mushroomdb::GraphDb::open_with_options(
            &dir,
            mushroomdb::OpenOptions {
                auto_migrate: false,
                repair_wal: false,
                read_only: true,
            },
        )
        .map_err(|e| RestoreS3Error::Verify(format!("the restored copy does not open: {e}")))?;
        let commits = opened
            .wal_total_commits()
            .map_err(|e| RestoreS3Error::Verify(e.to_string()))?;
        if commits != expect {
            return Err(RestoreS3Error::Verify(format!(
                "the restored copy holds {commits} commits; {manifest_key} says {expect}"
            )));
        }
        Ok(())
    })
    .await?;
    if !out.files.iter().any(|f| f == disk::LINEAGE_FILE) {
        out.files.push(disk::LINEAGE_FILE.to_string());
    }
    if mode == Mode::InPlace {
        out.files.push(disk::PARENT_FILE.to_string());
    }
    let dir = staging.path().to_path_buf();
    let target_dir = into.to_path_buf();
    out.install_note = match mode {
        Mode::Beside => blocking(move || install(&dir, &target_dir)).await?,
        Mode::InPlace => blocking(move || install_in_place(&dir, &target_dir)).await?,
    };
    staging.installed();
    Ok(out)
}

// ── The bucket ──────────────────────────────────────────────────────────────

/// Bytes per ranged GET when downloading a file: the most a restore holds in
/// memory at once, whatever the size of the snapshot.
const DOWNLOAD_CHUNK: usize = 8 * 1024 * 1024;

struct Bucket<'a> {
    store: Arc<dyn ObjectStore>,
    prefix: &'a str,
}

/// A segment of the newest manifest's generation shipped after it.
struct TailSegment {
    key: String,
    bytes: Vec<u8>,
    frames: u64,
}

impl Bucket<'_> {
    async fn get(&self, key: &str) -> Result<Option<Vec<u8>>, RestoreS3Error> {
        match self.store.get(&ObjectPath::from(key)).await {
            Ok(r) => Ok(Some(r.bytes().await?.to_vec())),
            Err(object_store::Error::NotFound { .. }) => Ok(None),
            Err(e) => Err(e.into()),
        }
    }

    async fn get_text(&self, key: &str) -> Result<Option<String>, RestoreS3Error> {
        match self.get(key).await? {
            None => Ok(None),
            Some(b) => String::from_utf8(b)
                .map(|s| Some(s.trim().to_string()))
                .map_err(|_| RestoreS3Error::Manifest(format!("{key} is not UTF-8"))),
        }
    }

    async fn manifest(&self, key: &str) -> Result<Manifest, RestoreS3Error> {
        let bytes = self.get(key).await?.ok_or_else(|| {
            RestoreS3Error::Manifest(format!("{key} is named but does not exist"))
        })?;
        Manifest::from_json(&bytes).map_err(|e| RestoreS3Error::Manifest(format!("{key}: {e}")))
    }

    /// The earliest manifest of `latest`'s lineage whose restore point holds
    /// commit `n`: the first one past it in commit order. Manifests of any
    /// other lineage, and ones that do not parse, are passed over. `latest`
    /// itself holds `n`, so there always is one.
    async fn manifest_holding(
        &self,
        n: u64,
        latest_key: &str,
        latest: &Manifest,
    ) -> Result<(String, Manifest), RestoreS3Error> {
        let dir = ObjectPath::from(object_key(self.prefix, "manifest"));
        let mut keyed: Vec<(u64, String)> = self
            .store
            .list_with_delimiter(Some(&dir))
            .await?
            .objects
            .into_iter()
            .filter_map(|o| {
                let key = o.location.to_string();
                let seq = key
                    .rsplit('/')
                    .next()?
                    .strip_suffix(".json")?
                    .parse::<u64>()
                    .ok()?;
                (seq > n).then_some((seq, key))
            })
            .collect();
        keyed.sort();
        for (_, key) in keyed {
            match self.manifest(&key).await {
                Ok(m) if m.lineage == latest.lineage && m.commit_seq > n => return Ok((key, m)),
                Ok(_) | Err(RestoreS3Error::Manifest(_)) => continue,
                Err(e) => return Err(e),
            }
        }
        Ok((latest_key.to_string(), latest.clone()))
    }

    /// The segments of `latest`'s generation after its last one, up to the
    /// last contiguous one whose bytes are whole, valid WAL frames.
    async fn tail_segments(
        &self,
        latest: &Manifest,
    ) -> Result<(Vec<TailSegment>, Option<String>), RestoreS3Error> {
        let mut end = match latest.segments.last() {
            Some(s) => {
                parse_segment_key(self.prefix, &s.key)
                    .ok_or_else(|| {
                        RestoreS3Error::Verify(format!("{} is not a segment key", s.key))
                    })?
                    .2
            }
            None => 0,
        };
        let dir = ObjectPath::from(object_key(
            self.prefix,
            &format!("wal/{}", latest.wal_generation),
        ));
        let mut later: Vec<(u64, u64, String)> = self
            .store
            .list_with_delimiter(Some(&dir))
            .await?
            .objects
            .into_iter()
            .filter_map(|o| {
                let key = o.location.to_string();
                let (g, start, stop) = parse_segment_key(self.prefix, &key)?;
                (g == latest.wal_generation && start >= end).then_some((start, stop, key))
            })
            .collect();
        later.sort();
        let mut tail = Vec::new();
        for (i, (start, stop, key)) in later.iter().enumerate() {
            let why = if *start != end {
                Some(format!(
                    "the segment that should start at byte {end} is missing \
                     ({key} starts at {start})"
                ))
            } else {
                match self.get(key).await? {
                    None => Some(format!("{key} was listed but is gone")),
                    Some(bytes) => {
                        let (whole, frames) = complete_frames(&bytes);
                        if bytes.len() as u64 != stop - start || whole != bytes.len() {
                            Some(format!(
                                "{key} is damaged: not {} bytes of whole, valid frames",
                                stop - start
                            ))
                        } else {
                            end = *stop;
                            tail.push(TailSegment {
                                key: key.clone(),
                                bytes,
                                frames,
                            });
                            None
                        }
                    }
                }
            };
            if let Some(why) = why {
                let unrestored = &later[i..];
                let bytes = unrestored.last().map_or(end, |l| l.1) - end;
                // A gap is at least one more segment, missing from the list.
                let (at_least, count) = if *start != end {
                    ("at least ", unrestored.len() + 1)
                } else {
                    ("", unrestored.len())
                };
                return Ok((
                    tail,
                    Some(format!(
                        "{at_least}{count} segment(s) uploaded after the newest manifest, \
                         {bytes} bytes of WAL from byte {end} on, were not restored, so the \
                         commits they hold are not in the restored store: {why}"
                    )),
                ));
            }
        }
        Ok((tail, None))
    }

    /// The commit `t` resolves to through `latest`'s `commit_times.bin`.
    async fn resolve_time(
        &self,
        latest: &Manifest,
        t: OffsetDateTime,
        available: u64,
    ) -> Result<u64, RestoreS3Error> {
        let entry = latest
            .files
            .iter()
            .find(|f| f.name == "commit_times.bin")
            .ok_or(RestoreS3Error::CommitTimesAbsent)?;
        let bytes = self
            .get(&entry.key)
            .await?
            .ok_or_else(|| RestoreS3Error::Verify(format!("{} does not exist", entry.key)))?;
        if bytes.len() as u64 != entry.size || crc32_hex(&bytes) != entry.crc32 {
            return Err(RestoreS3Error::Verify(format!(
                "{} does not match its manifest entry",
                entry.key
            )));
        }
        let ms = (t.unix_timestamp_nanos().div_euclid(1_000_000)) as i64;
        let at =
            mushroomdb::restore::commit_at_instant(&bytes, latest.oldest_commit_seq, ms, available)
                .map_err(|e| match e {
                    mushroomdb::GraphError::NoRecordedTime => RestoreS3Error::CommitTimesAbsent,
                    mushroomdb::GraphError::TimeBeforeFloor {
                        floor_ms,
                        floor_commit,
                    } => RestoreS3Error::Unreachable(format!(
                        "{} is before the oldest recorded commit time ({}, commit {floor_commit})",
                        rfc3339(t),
                        ms_rfc3339(floor_ms)
                    )),
                    mushroomdb::GraphError::CommitOutOfRange { commit, floor, .. } => {
                        RestoreS3Error::BeforeOldestArchive {
                            target: commit,
                            oldest: floor,
                        }
                    }
                    other => RestoreS3Error::Verify(format!("commit_times.bin: {other}")),
                })?;
        // The instant is at or after the newest recorded time. If commits
        // follow that one with no time of their own — shipped after the
        // manifest, or stamped after its side files were read — the instant
        // may fall after some of them: a date query on the source would see
        // them. Refused rather than answered with an earlier commit.
        if at.commit == at.newest_recorded && at.newest_recorded + 1 < available {
            return Err(RestoreS3Error::CommitTimesEnd {
                recorded_through: at.newest_recorded,
                recorded_through_ms: at.newest_recorded_ms,
                commits: available,
            });
        }
        Ok(at.commit)
    }

    /// Download and verify every object of `m` (and `tail`) into `dir`.
    async fn download(
        &self,
        m: &Manifest,
        tail: &[TailSegment],
        dir: &Path,
        out: &mut RestoreS3Outcome,
    ) -> Result<(), RestoreS3Error> {
        for f in &m.files {
            if !is_store_file_name(&f.name) {
                return Err(RestoreS3Error::Verify(format!(
                    "{} restores to {:?}, which is not a store file",
                    f.key, f.name
                )));
            }
            if out.files.contains(&f.name) {
                return Err(RestoreS3Error::Verify(format!(
                    "the manifest names {} twice",
                    f.name
                )));
            }
            let crc = self
                .download_file(&f.key, f.size, &dir.join(&f.name))
                .await?;
            if crc != f.crc32 {
                return Err(RestoreS3Error::Verify(format!(
                    "{} has CRC-32 {crc}; the manifest says {}",
                    f.key, f.crc32
                )));
            }
            if f.name == disk::LINEAGE_FILE {
                let named = tokio::fs::read_to_string(dir.join(&f.name)).await?;
                if named.trim() != m.lineage {
                    return Err(RestoreS3Error::Verify(format!(
                        "{} names lineage {}, its manifest lineage {}",
                        f.key,
                        named.trim(),
                        m.lineage
                    )));
                }
            }
            out.bytes += f.size;
            out.files.push(f.name.clone());
        }

        let mut wal = tokio::fs::File::create(dir.join("wal.bin")).await?;
        let mut offset = 0u64;
        for s in &m.segments {
            let (g, start, end) = parse_segment_key(self.prefix, &s.key)
                .ok_or_else(|| RestoreS3Error::Verify(format!("{} is not a segment key", s.key)))?;
            if g != m.wal_generation {
                return Err(RestoreS3Error::Verify(format!(
                    "{} is of generation {g}, the manifest's is {}",
                    s.key, m.wal_generation
                )));
            }
            if start != offset {
                return Err(RestoreS3Error::Verify(format!(
                    "{} starts at {start}; the segments before it end at {offset}",
                    s.key
                )));
            }
            let bytes = self.get(&s.key).await?.ok_or_else(|| {
                RestoreS3Error::Verify(format!("{} is named but does not exist", s.key))
            })?;
            let (whole, frames) = complete_frames(&bytes);
            if bytes.len() as u64 != end - start
                || crc32_hex(&bytes) != s.crc32
                || whole != bytes.len()
                || frames != s.frames
            {
                return Err(RestoreS3Error::Verify(format!(
                    "{} does not match its manifest entry (size, CRC-32 or frames)",
                    s.key
                )));
            }
            wal.write_all(&bytes).await?;
            offset = end;
            out.bytes += bytes.len() as u64;
            out.segments += 1;
        }
        for s in tail {
            wal.write_all(&s.bytes).await?;
            out.bytes += s.bytes.len() as u64;
            out.segments += 1;
            tracing::debug!(key = %s.key, "segment after the newest manifest");
        }
        wal.sync_all().await?;
        out.files.push("wal.bin".into());
        Ok(())
    }

    /// Download `key` (`size` bytes) to `path` by ranged GETs, returning its
    /// CRC-32.
    async fn download_file(
        &self,
        key: &str,
        size: u64,
        path: &Path,
    ) -> Result<String, RestoreS3Error> {
        let location = ObjectPath::from(key);
        let meta = match self.store.head(&location).await {
            Ok(m) => m,
            Err(object_store::Error::NotFound { .. }) => {
                return Err(RestoreS3Error::Verify(format!(
                    "{key} is named but does not exist"
                )))
            }
            Err(e) => return Err(e.into()),
        };
        if meta.size as u64 != size {
            return Err(RestoreS3Error::Verify(format!(
                "{key} is {} bytes; the manifest says {size}",
                meta.size
            )));
        }
        let mut file = tokio::fs::File::create(path).await?;
        let mut crc = crc32fast::Hasher::new();
        let mut at = 0usize;
        let size = size as usize;
        while at < size {
            let stop = (at + DOWNLOAD_CHUNK).min(size);
            let chunk = self.store.get_range(&location, at..stop).await?;
            if chunk.len() != stop - at {
                return Err(RestoreS3Error::Verify(format!(
                    "{key} changed while it was read"
                )));
            }
            crc.update(&chunk);
            file.write_all(&chunk).await?;
            at = stop;
        }
        file.sync_all().await?;
        Ok(crate::manifest::format_crc32(crc.finalize()))
    }
}

/// A name a manifest may restore a file to: one plain file name, and none
/// the restore writes itself or must never restore.
fn is_store_file_name(name: &str) -> bool {
    is_plain_name(name)
        && !matches!(
            name,
            "wal.bin" | "snapshot.intent" | "LOCK" | disk::PARENT_FILE | INSTALL_MARKER
        )
}

/// One plain file name: not empty, not `.` or `..`, no separator.
fn is_plain_name(name: &str) -> bool {
    !name.is_empty() && name != "." && name != ".." && !name.contains(['/', '\\'])
}

// ── The target ──────────────────────────────────────────────────────────────

/// Refuse a target that holds anything, or is not a directory.
fn check_into(into: &Path) -> Result<(), RestoreS3Error> {
    match std::fs::read_dir(into) {
        Ok(mut entries) => {
            if entries.next().is_some() {
                Err(RestoreS3Error::IntoNotEmpty {
                    into: into.to_path_buf(),
                })
            } else {
                Ok(())
            }
        }
        Err(e) if e.kind() == std::io::ErrorKind::NotFound => Ok(()),
        Err(e) => Err(RestoreS3Error::Io(std::io::Error::new(
            e.kind(),
            format!("{}: {e}", into.display()),
        ))),
    }
}

/// Refuse a data directory to restore into in place unless it holds nothing
/// but what an empty data directory may hold: `lost+found` (a new ext4
/// volume has one), a `LOCK` file and an empty `wal.bin` (an open that
/// stopped before its first commit), and staging directories and lock files
/// of restores into it.
fn check_in_place(dir: &Path) -> Result<(), RestoreS3Error> {
    let entries = match std::fs::read_dir(dir) {
        Ok(e) => e,
        Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(()),
        Err(e) => {
            return Err(RestoreS3Error::Io(std::io::Error::new(
                e.kind(),
                format!("{}: {e}", dir.display()),
            )))
        }
    };
    let name = name_of(dir);
    let mut others = Vec::new();
    for e in entries {
        let e = e?;
        let entry = e.file_name().to_string_lossy().into_owned();
        let empty_file =
            |n: &str| entry == n && e.metadata().is_ok_and(|m| m.is_file() && m.len() == 0);
        let allowed = entry == "lost+found"
            || entry == "LOCK"
            || empty_file("wal.bin")
            || staging_stem(&entry, &name).is_some();
        if !allowed {
            others.push(entry);
        }
    }
    if others.is_empty() {
        return Ok(());
    }
    others.sort();
    Err(RestoreS3Error::DataDirNotEmpty {
        dir: dir.to_path_buf(),
        entries: others,
    })
}

/// The parent directory a target sits in; `.` for a bare relative name.
fn parent_of(into: &Path) -> PathBuf {
    match into.parent() {
        Some(p) if !p.as_os_str().is_empty() => p.to_path_buf(),
        _ => PathBuf::from("."),
    }
}

/// The target's file name, as the staging names use it.
fn name_of(into: &Path) -> String {
    into.file_name()
        .map(|n| n.to_string_lossy().into_owned())
        .unwrap_or_else(|| "restored".into())
}

/// The staging stem (`.<name>.restore-<pid>-<8 hex>`) an entry of the parent
/// directory belongs to, for a restore into `name`: the staging directory
/// itself, or its lock file (the same name plus `.lock`). Exact: nothing
/// else in the parent ever matches.
fn staging_stem(entry: &str, name: &str) -> Option<String> {
    let rest = entry.strip_prefix(&format!(".{name}.restore-"))?;
    let stem = rest.strip_suffix(".lock").unwrap_or(rest);
    let (pid, tag) = stem.split_once('-')?;
    let ok = !pid.is_empty()
        && pid.bytes().all(|b| b.is_ascii_digit())
        && tag.len() == 8
        && tag.bytes().all(|b| matches!(b, b'0'..=b'9' | b'a'..=b'f'));
    ok.then(|| format!(".{name}.restore-{stem}"))
}

/// How old an unlocked lock file, or a staging directory without one, must
/// be before a sweep counts it as a killed restore's. A restore creates its
/// lock file, then locks it, then creates its directory: one younger than
/// this may be a restore between those steps.
pub const STALE_STAGING_AFTER: std::time::Duration = std::time::Duration::from_secs(10);

/// Whether `path` was last modified at least `age` ago. Unknown is no.
fn older_than(path: &Path, age: std::time::Duration) -> bool {
    std::fs::metadata(path)
        .and_then(|m| m.modified())
        .ok()
        .and_then(|t| t.elapsed().ok())
        .is_some_and(|elapsed| elapsed >= age)
}

/// Remove the staging directories, and their lock files, that earlier
/// restores into `name` left in `root` when they were killed. Removed: one
/// whose lock file no process holds and is older than
/// [`STALE_STAGING_AFTER`], and one with no lock file that old itself. Left
/// alone: one whose lock is held (a restore still running), a young one, and
/// one whose lock cannot be tried (a filesystem without advisory locks: a
/// running restore there cannot be told from a killed one). Returns what was
/// removed.
fn sweep_stale_staging(root: &Path, name: &str) -> Vec<PathBuf> {
    let Ok(entries) = std::fs::read_dir(root) else {
        return Vec::new();
    };
    let mut stems = std::collections::BTreeSet::new();
    for e in entries.flatten() {
        if let Some(stem) = staging_stem(&e.file_name().to_string_lossy(), name) {
            stems.insert(stem);
        }
    }
    let mut removed = Vec::new();
    for stem in stems {
        let dir = root.join(&stem);
        let lock = root.join(format!("{stem}.lock"));
        let stale = match std::fs::OpenOptions::new().write(true).open(&lock) {
            Ok(f) => match f.try_lock() {
                Ok(()) => older_than(&lock, STALE_STAGING_AFTER),
                Err(std::fs::TryLockError::WouldBlock) => false,
                Err(std::fs::TryLockError::Error(e)) => {
                    tracing::warn!(
                        path = %dir.display(),
                        error = %e,
                        "cannot tell whether a restore staging directory is in use \
                         (no advisory locks here); left in place"
                    );
                    false
                }
            },
            Err(e) if e.kind() == std::io::ErrorKind::NotFound => {
                older_than(&dir, STALE_STAGING_AFTER)
            }
            Err(_) => false,
        };
        if !stale {
            continue;
        }
        if dir.is_dir() && std::fs::remove_dir_all(&dir).is_ok() {
            tracing::info!(path = %dir.display(), "removed a stale restore staging directory");
            removed.push(dir);
        }
        if std::fs::remove_file(&lock).is_ok() {
            removed.push(lock);
        }
    }
    removed
}

/// Sweep what killed restores on boot left inside the data directory `dir`
/// ([`sweep_stale_staging`] with the in-place staging root).
pub(crate) fn sweep_in_place(dir: &Path) -> Vec<PathBuf> {
    sweep_stale_staging(dir, &name_of(dir))
}

/// Staging directories and lock files of restores into `dir` still inside
/// it: ones a sweep had to leave (too young, or no advisory locks).
pub(crate) fn staging_leftovers(dir: &Path) -> Vec<PathBuf> {
    let name = name_of(dir);
    let Ok(entries) = std::fs::read_dir(dir) else {
        return Vec::new();
    };
    let mut out: Vec<PathBuf> = entries
        .flatten()
        .filter(|e| staging_stem(&e.file_name().to_string_lossy(), &name).is_some())
        .map(|e| e.path())
        .collect();
    out.sort();
    out
}

/// The staging directory, `.<name>.restore-<pid>-<tag>` in the mode's
/// staging root, with a lock file of the same name plus `.lock` that this
/// restore holds locked for as long as it runs, so a later restore can tell
/// a killed restore's leftovers from a running one's. Dropped without
/// [`installed`](Self::installed), the directory and its lock file are
/// removed, and so are the directories above it that the restore created.
struct Staging {
    path: PathBuf,
    lock_path: PathBuf,
    _lock: Option<std::fs::File>,
    /// Directories this restore created, deepest first.
    created: Vec<PathBuf>,
}

impl Staging {
    fn new(into: &Path, mode: Mode) -> Result<Self, RestoreS3Error> {
        let root = mode.staging_root(into);
        let stem = format!(
            ".{}.restore-{}-{}",
            name_of(into),
            std::process::id(),
            &new_instance_id()[..8]
        );
        // The struct first: whatever fails below, Drop removes what was made.
        let mut staging = Self {
            path: root.join(&stem),
            lock_path: root.join(format!("{stem}.lock")),
            _lock: None,
            created: root
                .ancestors()
                .take_while(|a| !a.as_os_str().is_empty() && !a.exists())
                .map(Path::to_path_buf)
                .collect(),
        };
        std::fs::create_dir_all(&root)?;
        // The lock before the directory: a sweep never sees the directory
        // unlocked, and leaves a lock file alone while it is young.
        let lock = std::fs::File::create(&staging.lock_path)?;
        let locked = if take_fault(into, Fault::Lock) {
            Err(std::io::Error::other("injected lock failure"))
        } else {
            lock.lock()
        };
        if let Err(e) = locked {
            // Some network and FUSE filesystems have no advisory locks. The
            // restore still works there; a concurrent restore into the same
            // target just cannot tell this one from a killed one, and leaves
            // it alone.
            tracing::warn!(
                path = %staging.lock_path.display(),
                error = %e,
                "cannot lock the restore's lock file; restoring without it"
            );
        }
        staging._lock = Some(lock);
        std::fs::create_dir(&staging.path)?;
        Ok(staging)
    }

    fn path(&self) -> &Path {
        &self.path
    }

    /// The restore is in place: the staging directory became the target, or
    /// was emptied into it, and the directories above it stay.
    fn installed(mut self) {
        self.created.clear();
    }
}

impl Drop for Staging {
    fn drop(&mut self) {
        let _ = std::fs::remove_dir_all(&self.path);
        let _ = std::fs::remove_file(&self.lock_path);
        for dir in &self.created {
            // Only an empty one: anything another process put there stays.
            let _ = std::fs::remove_dir(dir);
        }
    }
}

/// Put the staged store in place with one directory rename, so the target
/// is either what it was (absent or empty) or the whole restored store, and
/// never a part of one — a crash included.
///
/// Every file in `staging` was fsynced as it was written; the staging
/// directory is fsynced before the rename and its parent after it. A target
/// that exists must be an empty directory, which the rename replaces (on
/// Unix in the same step; elsewhere it is removed first, so a crash in
/// between leaves it absent, never partial). A mount point cannot be
/// replaced: the rename fails and the target is left as it was.
fn install(staging: &Path, into: &Path) -> Result<Option<String>, RestoreS3Error> {
    // The read-only verification open may have left its lock file.
    remove_if_present(&staging.join("LOCK"))?;
    #[cfg(unix)]
    std::fs::File::open(staging)?.sync_all()?;
    // The target may have been filled since the start; it is checked again.
    check_into(into)?;
    if take_fault(into, Fault::BeforeRename) {
        return Err(injected(into));
    }
    #[cfg(not(unix))]
    let replaced_empty = if into.exists() {
        std::fs::remove_dir(into)?;
        true
    } else {
        false
    };
    if let Err(e) = std::fs::rename(staging, into) {
        #[allow(unused_mut)]
        let mut recreated = String::new();
        #[cfg(not(unix))]
        if replaced_empty {
            recreated = match std::fs::create_dir(into) {
                Ok(()) => "; the empty directory it replaced was created again".into(),
                Err(_) => "; the empty directory it was to replace was removed and could \
                           not be created again"
                    .into(),
            };
        }
        return Err(RestoreS3Error::Io(std::io::Error::new(
            e.kind(),
            format!(
                "moving the restored store into {} (it must be a new or empty directory, \
                 not a mount point; restore into a directory inside one): {e}{recreated}",
                into.display()
            ),
        )));
    }
    // From here the store is in place: nothing below is a failure of the
    // restore, only a doubt about how durable its directory entry is.
    let parent = parent_of(into);
    Ok(sync_dir(into, &parent).err().map(|e| {
        let note = format!(
            "the restored store is in place in {}, but syncing its parent directory {} \
             failed ({e}): its directory entry may not survive a power loss",
            into.display(),
            parent.display()
        );
        tracing::warn!(note = %note, "restore installed");
        note
    }))
}

/// The file in a data directory that says a restore on boot is moving a
/// store into it: one line per file it moves, `<name>\t<size>\t<mtime ns>`,
/// as the file was staged. [`mushroomdb::GraphDb::open`] refuses a directory
/// holding it.
pub use mushroomdb::restore::INSTALL_MARKER;

/// Move the staged store into the data directory `dir`, in place.
///
/// The marker naming every staged file (with its size and modification
/// time) is written and fsynced first; the files are renamed in
/// (`snapshot.bin` and `wal.bin`, which make the directory a store, last),
/// the directory fsynced, the marker removed and the directory fsynced
/// again. The install has succeeded only once that last fsync has: until
/// then a power loss can bring the marker back, and the next boot would undo
/// the install, so a failed last fsync fails the restore and the server does
/// not start. A failure before the marker's removal removes what it moved,
/// makes that durable, and only then removes the marker, so the directory is
/// as it was. A crash leaves the marker, and the next restore on boot
/// removes the files it names ([`undo_interrupted_install`]).
fn install_in_place(staging: &Path, dir: &Path) -> Result<Option<String>, RestoreS3Error> {
    remove_if_present(&staging.join("LOCK"))?;
    #[cfg(unix)]
    std::fs::File::open(staging)?.sync_all()?;
    check_in_place(dir)?;
    let mut names = Vec::new();
    for e in std::fs::read_dir(staging)? {
        names.push(e?.file_name().to_string_lossy().into_owned());
    }
    // The files that make a directory a store go in last.
    names.sort_by_key(|n| (matches!(n.as_str(), "snapshot.bin" | "wal.bin"), n.clone()));
    if take_fault(dir, Fault::BeforeRename) {
        return Err(injected(dir));
    }
    let mut lines = Vec::new();
    for name in &names {
        let (size, mtime) = disk::ident_of(&std::fs::metadata(staging.join(name))?);
        lines.push(format!("{name}\t{size}\t{mtime}"));
    }
    let marker = dir.join(INSTALL_MARKER);
    {
        use std::io::Write as _;
        let mut f = std::fs::File::create(&marker)?;
        f.write_all(lines.join("\n").as_bytes())?;
        f.sync_all()?;
    }
    let mut moved: Vec<&str> = Vec::new();
    let result = (|| -> Result<(), RestoreS3Error> {
        sync_dir(dir, dir)?;
        for name in &names {
            if take_fault(dir, Fault::AfterMoving(moved.len())) {
                return Err(injected(dir));
            }
            std::fs::rename(staging.join(name), dir.join(name)).map_err(|e| {
                std::io::Error::new(
                    e.kind(),
                    format!("moving {name} into {}: {e}", dir.display()),
                )
            })?;
            moved.push(name);
        }
        sync_dir(dir, dir)?;
        Ok(())
    })();
    if let Err(e) = result {
        for name in &moved {
            let _ = std::fs::remove_file(dir.join(name));
        }
        // The marker goes only once the removals are durable: otherwise a
        // power loss could keep a part of a store and lose the marker.
        if sync_dir(dir, dir).is_ok() {
            let _ = std::fs::remove_file(&marker);
            let _ = sync_dir(dir, dir);
        }
        return Err(e);
    }
    if !take_fault(dir, Fault::KeepMarker) {
        std::fs::remove_file(&marker)?;
    }
    if let Err(e) = sync_marker_removal(dir) {
        return Err(RestoreS3Error::Io(std::io::Error::new(
            e.kind(),
            format!(
                "the restored store is in {}, but syncing the directory after removing the \
                 install marker failed ({e}); not served, since a power loss could bring the \
                 marker back. Start again: the next boot serves the store, or restores it \
                 again if the marker came back",
                dir.display()
            ),
        )));
    }
    Ok(None)
}

fn sync_marker_removal(dir: &Path) -> std::io::Result<()> {
    if take_fault(dir, Fault::MarkerRemovalSync) {
        return Err(std::io::Error::other("injected fsync failure"));
    }
    sync_dir(dir, dir)
}

/// Whether `dir` holds a restore on boot that stopped while it moved its
/// files in ([`INSTALL_MARKER`]): what the directory holds is then a part of
/// a store, and must not be opened.
pub fn install_interrupted(dir: &Path) -> bool {
    dir.join(INSTALL_MARKER).exists()
}

/// Undo a restore on boot that stopped while it moved its files into `dir`:
/// remove the files its [`INSTALL_MARKER`] names, then the marker. Returns
/// the files removed; nothing when there is no marker. Only plain file names
/// directly in `dir` are touched.
///
/// # Errors
///
/// Nothing is removed when a file the marker names is no longer the file the
/// restore staged (another size or modification time) or a store file it
/// does not name has appeared: the store has changed since the install, so
/// it may have served writes, and deleting it could lose them. The error says
/// so and how to go on.
pub fn undo_interrupted_install(dir: &Path) -> std::io::Result<Vec<String>> {
    let marker = dir.join(INSTALL_MARKER);
    let listed = match std::fs::read_to_string(&marker) {
        Ok(s) => s,
        Err(e) if e.kind() == std::io::ErrorKind::NotFound => return Ok(Vec::new()),
        Err(e) => return Err(e),
    };
    let mut entries: Vec<(&str, Option<disk::Ident>)> = Vec::new();
    for line in listed.lines() {
        let mut fields = line.split('\t').map(str::trim);
        let Some(name) = fields.next().filter(|n| is_plain_name(n)) else {
            continue;
        };
        let size = fields.next().and_then(|v| v.parse::<u64>().ok());
        let mtime = fields.next().and_then(|v| v.parse::<u64>().ok());
        entries.push((name, size.zip(mtime)));
    }
    let changed = |why: String| {
        std::io::Error::other(format!(
            "{} holds the install marker of a restore on boot ({}), but {why} since that \
             install: the store may have served writes, so nothing was removed. If the store \
             is the one you want, delete {}; to restore again, empty the directory",
            dir.display(),
            marker.display(),
            marker.display()
        ))
    };
    for (name, ident) in &entries {
        let Ok(meta) = std::fs::symlink_metadata(dir.join(name)) else {
            continue;
        };
        if let Some(staged) = ident {
            // An empty file is what an empty data directory may hold before
            // the install (an empty `wal.bin`): it holds no writes.
            if meta.is_file() && meta.len() > 0 && disk::ident_of(&meta) != *staged {
                return Err(changed(format!("{name} has changed")));
            }
        }
    }
    for written in ["snapshot.bin", "wal.bin"] {
        if !entries.iter().any(|(n, _)| *n == written) && dir.join(written).exists() {
            return Err(changed(format!("{written} has appeared")));
        }
    }
    let mut removed = Vec::new();
    for (name, _) in &entries {
        let path = dir.join(name);
        if std::fs::symlink_metadata(&path).is_ok_and(|m| m.is_file()) {
            std::fs::remove_file(&path)?;
            removed.push(name.to_string());
        }
    }
    sync_dir(dir, dir)?;
    std::fs::remove_file(&marker)?;
    sync_dir(dir, dir)?;
    Ok(removed)
}

fn remove_if_present(path: &Path) -> std::io::Result<()> {
    match std::fs::remove_file(path) {
        Ok(()) => Ok(()),
        Err(e) if e.kind() == std::io::ErrorKind::NotFound => Ok(()),
        Err(e) => Err(e),
    }
}

/// fsync the directory `dir` (a no-op off Unix). `target` keys the test
/// hook's injected failure.
fn sync_dir(target: &Path, dir: &Path) -> std::io::Result<()> {
    if take_fault(target, Fault::DirSync) {
        return Err(std::io::Error::other("injected fsync failure"));
    }
    #[cfg(unix)]
    std::fs::File::open(dir)?.sync_all()?;
    #[cfg(not(unix))]
    let _ = dir;
    Ok(())
}

fn injected(into: &Path) -> RestoreS3Error {
    RestoreS3Error::Io(std::io::Error::other(format!(
        "installing into {}: injected failure",
        into.display()
    )))
}

/// A failure a test can make the next restore into a target hit.
#[doc(hidden)]
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Fault {
    /// Just before the install moves anything, the store staged and fsynced.
    BeforeRename,
    /// An in-place install, after it moved this many files in.
    AfterMoving(usize),
    /// The next directory fsync.
    DirSync,
    /// Locking the staging lock file.
    Lock,
    /// An in-place install: the fsync after the marker's removal.
    MarkerRemovalSync,
    /// An in-place install: the marker is not removed, yet the install
    /// reports success, as after a power loss that undid the removal.
    KeepMarker,
}

/// Faults to inject, by target (tests only).
static FAULTS: std::sync::Mutex<Vec<(PathBuf, Fault)>> = std::sync::Mutex::new(Vec::new());

/// Make the next restore into `into` fail at the last step, with the whole
/// store staged and fsynced and nothing yet moved. For tests of what a failed
/// install leaves behind.
#[doc(hidden)]
pub fn fail_next_install_into(into: &Path) {
    inject_fault(into, Fault::BeforeRename);
}

/// Make the next restore into `into` hit `fault` (tests only).
#[doc(hidden)]
pub fn inject_fault(into: &Path, fault: Fault) {
    FAULTS
        .lock()
        .unwrap_or_else(|p| p.into_inner())
        .push((into.to_path_buf(), fault));
}

fn take_fault(into: &Path, fault: Fault) -> bool {
    let mut faults = FAULTS.lock().unwrap_or_else(|p| p.into_inner());
    match faults.iter().position(|(p, f)| p == into && *f == fault) {
        Some(i) => {
            faults.remove(i);
            true
        }
        None => false,
    }
}

async fn blocking<T: Send + 'static>(
    f: impl FnOnce() -> Result<T, RestoreS3Error> + Send + 'static,
) -> Result<T, RestoreS3Error> {
    tokio::task::spawn_blocking(f)
        .await
        .map_err(|e| RestoreS3Error::Io(std::io::Error::other(e.to_string())))?
}

fn ms_rfc3339(ms: i64) -> String {
    OffsetDateTime::from_unix_timestamp_nanos(ms as i128 * 1_000_000)
        .map(rfc3339)
        .unwrap_or_else(|_| format!("{ms} ms"))
}

fn rfc3339(t: OffsetDateTime) -> String {
    t.format(&Rfc3339).unwrap_or_else(|_| t.to_string())
}