use anyhow::Result;
use serde::Serialize;
use crate::cli::Format;
use crate::progress::Step;
use crate::{output, remote};
fn newest(listing: &str) -> Option<&str> {
listing.lines().map(str::trim).filter(|line| is_backup_name(line)).max()
}
fn is_backup_name(entry: &str) -> bool {
let Some(stem) = entry.strip_suffix(".tar.gz") else {
return false;
};
let bytes = stem.as_bytes();
bytes.len() == 15 && bytes[8] == b'-' && stem.chars().enumerate().all(|(i, c)| if i == 8 { c == '-' } else { c.is_ascii_digit() })
}
pub fn backup(target_name: Option<String>, overrides: remote::Overrides) -> Result<()> {
let target = remote::resolve(target_name, overrides)?;
let dir = &target.path.dir;
let step = Step::start(format!("Connecting to {} ...", target.connection_label()));
let session = remote::connect(&target.server, &target.credential)?;
let dialect = remote::dialect(&session, &target.server);
step.update(format!("Backing up {dir} ..."));
let name = remote::run_backup(&session, dialect, dir)?;
step.done(format!("Backup {} created in {}", name.trim(), remote::backups_dir(dir)));
crate::journal::record("backup", Some(&target.app.name), Some(&target.server.name), Some(name.trim()));
Ok(())
}
#[derive(Serialize)]
struct Backups {
dir: String,
backups: Vec<String>,
}
fn archives(listing: &str) -> Vec<String> {
let mut names: Vec<String> = listing.lines().map(str::trim).filter(|line| is_backup_name(line)).map(str::to_string).collect();
names.sort_unstable_by(|a, b| b.cmp(a));
names
}
pub fn restore(target_name: Option<String>, overrides: remote::Overrides, from: Option<String>, list: Option<Format>) -> Result<()> {
let target = remote::resolve(target_name, overrides)?;
let deploy_dir = target.path.dir.clone();
let backups_raw = remote::backups_dir(&deploy_dir);
let step = Step::start(format!("Connecting to {} ...", target.connection_label()));
let session = remote::connect(&target.server, &target.credential)?;
let dialect = remote::dialect(&session, &target.server);
remote::check_quotable(dialect, &[&deploy_dir, &backups_raw])?;
step.clear();
if let Some(format) = list {
let listing = remote::exec(&session, &dialect.list_dir(&backups_raw))?;
let backups = Backups {
backups: archives(&listing),
dir: backups_raw,
};
let app = target.app.name.clone();
return output::emit(format, &backups, |backups| {
if backups.backups.is_empty() {
return format!("No backups yet in {} - create one with `turnout backup {app}`.", backups.dir);
}
backups.backups.join("\n")
});
}
let name = match from {
Some(name) => name,
None => {
let listing = remote::exec(&session, &dialect.list_dir(&backups_raw))?;
newest(&listing)
.ok_or_else(|| anyhow::anyhow!("no backups in {backups_raw} - create one with `turnout backup {}`", target.app.name))?
.to_string()
}
};
let step = Step::start(format!("Restoring {name} into {deploy_dir} ..."));
remote::restore_backup(&session, dialect, &deploy_dir, &name)?;
step.done(format!("Restored {name} into {deploy_dir}"));
crate::journal::record("restore", Some(&target.app.name), Some(&target.server.name), Some(&name));
if let Some(restart) = &target.path.restart {
let mut session = session;
let connect = || remote::connect(&target.server, &target.credential);
let step = Step::start(format!("Running: {restart}"));
let output = remote::restart(&mut session, dialect, &target.path, restart, &connect, |line| {
step.update(format!("Running: {restart} ยท {line}"))
})?;
step.done(format!("Restarted: {restart}"));
if !output.trim().is_empty() {
println!("{}", output.trim_end());
}
}
Ok(())
}
#[cfg(test)]
mod tests {
use super::{is_backup_name, newest};
#[test]
fn the_newest_backup_wins_regardless_of_listing_order() {
let shuffled = "20260810-120000.tar.gz\n20260812-181500.tar.gz\n20260811-090000.tar.gz\n";
assert_eq!(newest(shuffled), Some("20260812-181500.tar.gz"));
let crlf = "20260810-120000.tar.gz\r\n20260812-181500.tar.gz\r\n";
assert_eq!(newest(crlf), Some("20260812-181500.tar.gz"));
}
#[test]
fn backups_from_the_same_day_compare_by_time() {
let listing = "20260812-090000.tar.gz\n20260812-181500.tar.gz\n20260812-093000.tar.gz";
assert_eq!(newest(listing), Some("20260812-181500.tar.gz"));
}
#[test]
fn only_our_own_archives_are_candidates() {
assert!(is_backup_name("20260812-181500.tar.gz"));
assert!(!is_backup_name("notes.txt"));
assert!(!is_backup_name("site-final.tar.gz"), "not a timestamp");
assert!(!is_backup_name("20260812-181500.zip"));
assert!(!is_backup_name("2026081-181500.tar.gz"), "wrong width");
let mixed = "readme.txt\n20260812-181500.tar.gz\nold-backup.tar.gz\n";
assert_eq!(newest(mixed), Some("20260812-181500.tar.gz"));
}
#[test]
fn nothing_to_restore_is_not_a_backup() {
assert_eq!(newest(""), None);
assert_eq!(newest(" \n\n"), None);
assert_eq!(newest("readme.txt\n"), None);
}
}