Compare commits
No commits in common. "master" and "backup_error_fix" have entirely different histories.
master
...
backup_err
|
@ -10,7 +10,7 @@ trigger:
|
|||
steps:
|
||||
- name: build
|
||||
pull: always
|
||||
image: rust:1.55.0
|
||||
image: rust:1.46.0
|
||||
commands:
|
||||
- cargo build --verbose
|
||||
|
||||
|
@ -28,7 +28,7 @@ trigger:
|
|||
steps:
|
||||
- name: build
|
||||
pull: always
|
||||
image: rust:1.55.0
|
||||
image: rust:1.46.0
|
||||
commands:
|
||||
- cargo build --verbose --release
|
||||
- name: gitea-release
|
||||
|
@ -37,6 +37,6 @@ steps:
|
|||
settings:
|
||||
token:
|
||||
from_secret: gitea_token
|
||||
base: https://git.canopymc.net
|
||||
base: https://git.etztech.xyz
|
||||
files:
|
||||
- "target/release/albatross"
|
|
@ -1,7 +1,5 @@
|
|||
# This file is automatically @generated by Cargo.
|
||||
# It is not intended for manual editing.
|
||||
version = 3
|
||||
|
||||
[[package]]
|
||||
name = "addr2line"
|
||||
version = "0.13.0"
|
||||
|
@ -43,7 +41,7 @@ dependencies = [
|
|||
|
||||
[[package]]
|
||||
name = "albatross"
|
||||
version = "0.5.0"
|
||||
version = "0.4.0"
|
||||
dependencies = [
|
||||
"anvil-region",
|
||||
"chrono 0.4.19",
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
[package]
|
||||
name = "albatross"
|
||||
version = "0.5.0"
|
||||
version = "0.4.0"
|
||||
authors = ["Joey Hines <joey@ahines.net>"]
|
||||
edition = "2018"
|
||||
|
||||
|
|
14
README.md
14
README.md
|
@ -54,16 +54,14 @@ Restoring a range of chunks (from -2,-2 to 2,2):
|
|||
[backup]
|
||||
# Minecraft sever directory
|
||||
minecraft_dir = "/home/mc/server"
|
||||
# Optional Discord webhook
|
||||
discord_webhook = "https://discordapp.com/api/webhooks/"
|
||||
# Directory to place backups
|
||||
output_dir = "/home/mc/backups"
|
||||
# Number of backups to keep
|
||||
backups_to_keep = 10
|
||||
# Optional Discord webhook
|
||||
discord_webhook = "https://discordapp.com/api/webhooks/"
|
||||
|
||||
[backup.output_config]
|
||||
# Directory to place backups
|
||||
path = "/home/mc/backups"
|
||||
|
||||
# Optional remote_backup backup config
|
||||
# Optional remote backup config
|
||||
[remote]
|
||||
# SFTP server host:port
|
||||
sftp_server_addr = "localhost:22"
|
||||
|
@ -76,7 +74,7 @@ password = "cooluser123"
|
|||
# Key Auth
|
||||
#public_key = /home/user/.ssh/id_rsa.pub"
|
||||
#private_key = /home/user/.ssh/id_rsa"
|
||||
# Backups to keep on the remote_backup host
|
||||
# Backups to keep on the remote host
|
||||
backups_to_keep = 3
|
||||
|
||||
# World config options
|
||||
|
|
|
@ -137,11 +137,11 @@ pub fn backup_overworld(
|
|||
world_config: &WorldConfig,
|
||||
) -> Result<(u64, u64)> {
|
||||
backup_dir("data", &world_path, &backup_path)?;
|
||||
backup_dir("stats", &world_path, &backup_path).ok();
|
||||
backup_dir("stats", &world_path, &backup_path)?;
|
||||
|
||||
backup_file("level.dat", world_path.clone(), backup_path.clone())?;
|
||||
backup_file("level.dat_old", world_path.clone(), backup_path.clone()).ok();
|
||||
backup_file("session.lock", world_path.clone(), backup_path.clone()).ok();
|
||||
backup_file("level.dat_old", world_path.clone(), backup_path.clone())?;
|
||||
backup_file("session.lock", world_path.clone(), backup_path.clone())?;
|
||||
backup_file("uid.dat", world_path.clone(), backup_path.clone())?;
|
||||
|
||||
let player_count = backup_dir("playerdata", &world_path, &backup_path)?;
|
||||
|
@ -249,25 +249,20 @@ pub fn convert_backup_to_sp(
|
|||
Ok(())
|
||||
}
|
||||
|
||||
/// Preform a remote_backup backup, if configured
|
||||
/// Preform a remote backup, if configured
|
||||
pub fn do_remote_backup(
|
||||
remote_backup_cfg: &RemoteBackupConfig,
|
||||
backup_path: PathBuf,
|
||||
) -> Result<()> {
|
||||
if let Some(config) = &remote_backup_cfg.sftp {
|
||||
let mut sftp_backup = SFTPBackup::new(config, remote_backup_cfg.backups_to_keep)?;
|
||||
if remote_backup_cfg.sftp.is_some() {
|
||||
let mut sftp_backup = SFTPBackup::new(&remote_backup_cfg)?;
|
||||
sftp_backup.backup_to_remote(backup_path)?;
|
||||
sftp_backup.cleanup()?;
|
||||
} else if let Some(config) = &remote_backup_cfg.ftp {
|
||||
let mut ftps_backup = FTPBackup::new(config, remote_backup_cfg.backups_to_keep)?;
|
||||
} else if remote_backup_cfg.ftp.is_some() {
|
||||
let mut ftps_backup = FTPBackup::new(&remote_backup_cfg)?;
|
||||
ftps_backup.backup_to_remote(backup_path)?;
|
||||
ftps_backup.cleanup()?;
|
||||
}
|
||||
else if let Some(config) = &remote_backup_cfg.file {
|
||||
let mut file_backup = FileBackup::new(config, remote_backup_cfg.backups_to_keep)?;
|
||||
file_backup.backup_to_remote(backup_path)?;
|
||||
file_backup.cleanup()?;
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
@ -283,10 +278,10 @@ pub fn do_backup(cfg: AlbatrossConfig, output: Option<PathBuf>) -> Result<()> {
|
|||
let backup_name = format!("{}_backup.tar.gz", time_str);
|
||||
let mut output_archive = match output {
|
||||
Some(out_path) => out_path,
|
||||
None => cfg.backup.output_config.path.clone(),
|
||||
None => cfg.backup.output_dir.clone(),
|
||||
};
|
||||
output_archive.push(backup_name);
|
||||
let mut tmp_dir = cfg.backup.output_config.path.clone();
|
||||
let mut tmp_dir = cfg.backup.output_dir.clone();
|
||||
tmp_dir.push("tmp");
|
||||
remove_dir_all(&tmp_dir).ok();
|
||||
|
||||
|
@ -310,7 +305,7 @@ pub fn do_backup(cfg: AlbatrossConfig, output: Option<PathBuf>) -> Result<()> {
|
|||
|
||||
remove_dir_all(&tmp_dir)?;
|
||||
|
||||
let mut local_backup = FileBackup::new(&cfg.backup.output_config, cfg.backup.backups_to_keep).unwrap();
|
||||
let mut local_backup = FileBackup::new(&cfg.backup).unwrap();
|
||||
|
||||
match local_backup.cleanup() {
|
||||
Ok(backups_removed) => {
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
pub(crate) mod remote;
|
||||
mod remote;
|
||||
|
||||
use crate::config::remote::{FTPConfig, SFTPConfig, FileConfig};
|
||||
use crate::config::remote::{FTPConfig, SFTPConfig};
|
||||
use config::{Config, ConfigError, File};
|
||||
use serde::Deserialize;
|
||||
use std::path::PathBuf;
|
||||
|
@ -39,18 +39,17 @@ pub struct WorldConfig {
|
|||
#[derive(Debug, Deserialize, Clone)]
|
||||
pub struct BackupConfig {
|
||||
pub minecraft_dir: PathBuf,
|
||||
pub backups_to_keep: usize,
|
||||
pub output_dir: PathBuf,
|
||||
pub backups_to_keep: u64,
|
||||
pub discord_webhook: Option<String>,
|
||||
pub output_config: FileConfig,
|
||||
}
|
||||
|
||||
/// Config for remote_backup backups
|
||||
/// Config for remote backups
|
||||
#[derive(Debug, Deserialize, Clone)]
|
||||
pub struct RemoteBackupConfig {
|
||||
pub backups_to_keep: usize,
|
||||
pub backups_to_keep: u64,
|
||||
pub sftp: Option<SFTPConfig>,
|
||||
pub ftp: Option<FTPConfig>,
|
||||
pub file: Option<FileConfig>
|
||||
}
|
||||
|
||||
/// Configs
|
||||
|
|
|
@ -30,10 +30,3 @@ pub struct FTPConfig {
|
|||
/// Password
|
||||
pub password: String,
|
||||
}
|
||||
|
||||
/// File Config
|
||||
#[derive(Debug, Deserialize, Clone)]
|
||||
pub struct FileConfig {
|
||||
/// Path to backup to
|
||||
pub path: PathBuf,
|
||||
}
|
||||
|
|
|
@ -10,6 +10,7 @@ pub enum AlbatrossError {
|
|||
RegionParseError(crate::region::RegionParseError),
|
||||
ChronoParseError(chrono::ParseError),
|
||||
NoSSHAuth,
|
||||
RemoteConfigError(String),
|
||||
FTPError(ftp::FtpError),
|
||||
}
|
||||
|
||||
|
@ -26,6 +27,7 @@ impl std::fmt::Display for AlbatrossError {
|
|||
AlbatrossError::RegionParseError(e) => write!(f, "Unable to parse region name: {}", e),
|
||||
AlbatrossError::ChronoParseError(e) => write!(f, "Unable to parse time: {}", e),
|
||||
AlbatrossError::NoSSHAuth => write!(f, "No SSH auth methods provided in the config"),
|
||||
AlbatrossError::RemoteConfigError(e) => write!(f, "Invalid configuration for {}", e),
|
||||
AlbatrossError::FTPError(e) => write!(f, "FTP error: {}", e),
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,4 +1,4 @@
|
|||
use crate::config::remote::FileConfig;
|
||||
use crate::config::BackupConfig;
|
||||
use crate::error::Result;
|
||||
use crate::remote::{PathLocation, RemoteBackupSite};
|
||||
use std::path::PathBuf;
|
||||
|
@ -12,10 +12,10 @@ pub struct FileBackup {
|
|||
|
||||
impl FileBackup {
|
||||
/// New FileBackup
|
||||
pub fn new(config: &FileConfig, backups_to_keep: usize) -> Result<Self> {
|
||||
pub fn new(config: &BackupConfig) -> Result<Self> {
|
||||
Ok(Self {
|
||||
target_dir: config.path.clone(),
|
||||
backups_to_keep,
|
||||
target_dir: config.output_dir.clone(),
|
||||
backups_to_keep: config.backups_to_keep as usize,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,9 +1,10 @@
|
|||
use ftp::FtpStream;
|
||||
use std::path::PathBuf;
|
||||
|
||||
use crate::config::RemoteBackupConfig;
|
||||
use crate::error;
|
||||
use crate::error::AlbatrossError;
|
||||
use crate::remote::{PathLocation, RemoteBackupSite};
|
||||
use crate::config::remote::FTPConfig;
|
||||
|
||||
/// FTP Remote Site
|
||||
pub struct FTPBackup {
|
||||
|
@ -17,15 +18,20 @@ pub struct FTPBackup {
|
|||
|
||||
impl FTPBackup {
|
||||
/// New FTPBackup
|
||||
pub fn new(config: &FTPConfig, backups_to_keep: usize) -> error::Result<Self> {
|
||||
let mut ftp_stream = FtpStream::connect(&config.server_addr)?;
|
||||
pub fn new(config: &RemoteBackupConfig) -> error::Result<Self> {
|
||||
let ftp_config = config
|
||||
.ftp
|
||||
.as_ref()
|
||||
.ok_or_else(|| AlbatrossError::RemoteConfigError("FTP".to_string()))?;
|
||||
|
||||
ftp_stream.login(&config.username, &config.password)?;
|
||||
let mut ftp_stream = FtpStream::connect(&ftp_config.server_addr)?;
|
||||
|
||||
ftp_stream.login(&ftp_config.username, &ftp_config.password)?;
|
||||
|
||||
Ok(Self {
|
||||
stream: ftp_stream,
|
||||
target_dir: config.remote_dir.clone(),
|
||||
backups_to_keep,
|
||||
target_dir: ftp_config.remote_dir.clone(),
|
||||
backups_to_keep: config.backups_to_keep as usize,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
|
|
@ -9,13 +9,13 @@ pub mod ftp;
|
|||
pub mod sftp;
|
||||
|
||||
pub trait RemoteBackupFile {
|
||||
/// Type containing the location of the remote_backup backup
|
||||
/// Type containing the location of the remote backup
|
||||
type LocationType;
|
||||
|
||||
/// Get the underlying location type
|
||||
fn location(&self) -> Self::LocationType;
|
||||
|
||||
/// Get the time the remote_backup file was created
|
||||
/// Get the time the remote file was created
|
||||
fn time_created(&self) -> chrono::NaiveDateTime;
|
||||
|
||||
/// Parse the time created from the file name
|
||||
|
@ -34,10 +34,10 @@ pub trait RemoteBackupSite {
|
|||
/// Struct representing the location of a backup on the site
|
||||
type FileType: RemoteBackupFile;
|
||||
|
||||
/// Backup a file to the the remote_backup site
|
||||
/// Backup a file to the the remote site
|
||||
fn backup_to_remote(&mut self, file: PathBuf) -> Result<()>;
|
||||
|
||||
/// Get the locations backups contained on the remote_backup site
|
||||
/// Get the locations backups contained on the remote site
|
||||
fn get_backups(&mut self) -> Result<Vec<Self::FileType>>;
|
||||
|
||||
/// Remove a backup from the side
|
||||
|
@ -46,7 +46,7 @@ pub trait RemoteBackupSite {
|
|||
/// Number of backups to keep on the site
|
||||
fn backups_to_keep(&self) -> usize;
|
||||
|
||||
/// Cleanup old backups on the remote_backup site
|
||||
/// Cleanup old backups on the remote site
|
||||
fn cleanup(&mut self) -> Result<usize> {
|
||||
let mut backups = self.get_backups()?;
|
||||
|
||||
|
|
|
@ -3,10 +3,10 @@ use std::path::PathBuf;
|
|||
|
||||
use ssh2::Session;
|
||||
|
||||
use crate::config::RemoteBackupConfig;
|
||||
use crate::error;
|
||||
use crate::error::AlbatrossError;
|
||||
use crate::remote::{PathLocation, RemoteBackupSite};
|
||||
use crate::config::remote::SFTPConfig;
|
||||
|
||||
/// SFTP Remote Site
|
||||
pub struct SFTPBackup {
|
||||
|
@ -20,25 +20,30 @@ pub struct SFTPBackup {
|
|||
|
||||
impl SFTPBackup {
|
||||
/// New SFTPBackup
|
||||
pub fn new(config: &SFTPConfig, backups_to_keep: usize) -> error::Result<Self> {
|
||||
let tcp = TcpStream::connect(&config.server_addr)?;
|
||||
pub fn new(config: &RemoteBackupConfig) -> error::Result<Self> {
|
||||
let sftp_config = config
|
||||
.sftp
|
||||
.as_ref()
|
||||
.ok_or_else(|| AlbatrossError::RemoteConfigError("SFTP".to_string()))?;
|
||||
|
||||
let tcp = TcpStream::connect(&sftp_config.server_addr)?;
|
||||
let mut sess = Session::new()?;
|
||||
sess.set_tcp_stream(tcp);
|
||||
sess.handshake().unwrap();
|
||||
|
||||
if let Some(password) = &config.password {
|
||||
sess.userauth_password(&config.username, password)?;
|
||||
} else if let Some(key) = &config.private_key {
|
||||
let public_key = config.public_key.as_deref();
|
||||
sess.userauth_pubkey_file(&config.username, public_key, key, None)?;
|
||||
if let Some(password) = &sftp_config.password {
|
||||
sess.userauth_password(&sftp_config.username, password)?;
|
||||
} else if let Some(key) = &sftp_config.private_key {
|
||||
let public_key = sftp_config.public_key.as_deref();
|
||||
sess.userauth_pubkey_file(&sftp_config.username, public_key, key, None)?;
|
||||
} else {
|
||||
return Err(AlbatrossError::NoSSHAuth);
|
||||
}
|
||||
|
||||
Ok(Self {
|
||||
session: sess,
|
||||
target_dir: config.remote_dir.clone(),
|
||||
backups_to_keep,
|
||||
target_dir: sftp_config.remote_dir.clone(),
|
||||
backups_to_keep: config.backups_to_keep as usize,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
|
Loading…
Reference in New Issue