src/bin/backup-client.rs

   1 extern crate proxmox_backup;
   2
   3 use failure::*;
   4 use std::os::unix::io::AsRawFd;
   5
   6 use proxmox_backup::tools;
   7 use proxmox_backup::cli::command::*;
   8 use proxmox_backup::api::schema::*;
   9 use proxmox_backup::api::router::*;
  10 //use proxmox_backup::backup::chunk_store::*;
  11 //use proxmox_backup::backup::image_index::*;
  12 //use proxmox_backup::config::datastore;
  13 use proxmox_backup::backup::datastore::*;
  14 use serde_json::{Value};
  15
  16 fn required_string_param<'a>(param: &'a Value, name: &str) -> &'a str {
  17     param[name].as_str().expect(&format!("missing parameter '{}'", name))
  18 }
  19
  20
  21 fn backup_file(param: Value, _info: &ApiMethod) -> Result<Value, Error> {
  22
  23     let filename = required_string_param(&param, "filename");
  24     let store = required_string_param(&param, "store");
  25
  26     let mut datastore = DataStore::open(store)?;
  27
  28     println!("Backup file '{}' to '{}'", filename, store);
  29
  30     let target = "test1.idx";
  31
  32     {
  33         let file = std::fs::File::open(filename)?;
  34         let stat = nix::sys::stat::fstat(file.as_raw_fd())?;
  35         if stat.st_size <= 0 { bail!("got strange file size '{}'", stat.st_size); }
  36         let size = stat.st_size as usize;
  37
  38         let mut index = datastore.create_image_writer(target, size)?;
  39
  40         tools::file_chunker(file, 64*1024, |pos, chunk| {
  41             index.add_chunk(pos, chunk)?;
  42             Ok(true)
  43         })?;
  44
  45         index.close()?; // commit changes
  46     }
  47
  48     datastore.garbage_collection()?;
  49
  50     let idx = datastore.open_image_reader(target)?;
  51     idx.print_info();
  52
  53     Ok(Value::Null)
  54 }
  55
  56
  57 fn main() {
  58
  59     let cmd_def = CliCommand::new(
  60         ApiMethod::new(
  61             backup_file,
  62             ObjectSchema::new("Create backup from file.")
  63                 .required("filename", StringSchema::new("Source file name."))
  64                 .required("store", StringSchema::new("Datastore name."))
  65         ))
  66         .arg_param(vec!["filename"])
  67         .completion_cb("store", proxmox_backup::config::datastore::complete_datastore_name);
  68
  69
  70     if let Err(err) = run_cli_command(&cmd_def.into()) {
  71         eprintln!("Error: {}", err);
  72         print_cli_usage();
  73         std::process::exit(-1);
  74     }
  75
  76 }