1 extern crate proxmox_backup
;
4 //use std::os::unix::io::AsRawFd;
5 use chrono
::{DateTime, Local, Utc, TimeZone}
;
7 use proxmox_backup
::tools
;
8 use proxmox_backup
::cli
::*;
9 use proxmox_backup
::api_schema
::*;
10 use proxmox_backup
::api_schema
::router
::*;
11 use proxmox_backup
::client
::*;
12 use proxmox_backup
::backup
::*;
13 //use proxmox_backup::backup::image_index::*;
14 //use proxmox_backup::config::datastore;
15 //use proxmox_backup::catar::encoder::*;
16 //use proxmox_backup::backup::datastore::*;
18 use serde_json
::{Value}
;
23 use lazy_static
::lazy_static
;
26 static ref BACKUPSPEC_REGEX
: Regex
= Regex
::new(r
"^([a-zA-Z0-9_-]+):(.+)$").unwrap();
30 client
: &mut HttpClient
,
31 repo
: &BackupRepository
,
34 backup_time
: DateTime
<Utc
>,
35 chunk_size
: Option
<u64>,
36 ) -> Result
<(), Error
> {
38 let mut query
= url
::form_urlencoded
::Serializer
::new(String
::new());
41 .append_pair("archive_name", archive_name
)
42 .append_pair("type", "host")
43 .append_pair("id", &tools
::nodename())
44 .append_pair("time", &backup_time
.timestamp().to_string());
46 if let Some(size
) = chunk_size
{
47 query
.append_pair("chunk-size", &size
.to_string());
50 let query
= query
.finish();
52 let path
= format
!("api2/json/admin/datastore/{}/catar?{}", repo
.store
, query
);
54 client
.upload("application/x-proxmox-backup-catar", body
, &path
)?
;
60 fn backup_image(datastore: &DataStore, file: &std::fs::File, size: usize, target: &str, chunk_size: usize) -> Result<(), Error> {
62 let mut target = PathBuf::from(target);
64 if let Some(ext) = target.extension() {
66 bail!("got wrong file extension - expected '.fidx'");
69 target.set_extension("fidx");
72 let mut index = datastore.create_image_writer(&target, size, chunk_size)?;
74 tools::file_chunker(file, chunk_size, |pos, chunk| {
75 index.add_chunk(pos, chunk)?;
79 index.close()?; // commit changes
88 _rpcenv
: &mut RpcEnvironment
,
89 ) -> Result
<Value
, Error
> {
91 let repo_url
= tools
::required_string_param(¶m
, "repository")?
;
92 let repo
= BackupRepository
::parse(repo_url
)?
;
94 let mut client
= HttpClient
::new(&repo
.host
, &repo
.user
);
96 let path
= format
!("api2/json/admin/datastore/{}/backups", repo
.store
);
98 let result
= client
.get(&path
)?
;
100 // fixme: implement and use output formatter instead ..
101 let list
= result
["data"].as_array().unwrap();
105 let id
= item
["backup_id"].as_str(). unwrap();
106 let btype
= item
["backup_type"].as_str(). unwrap();
107 let epoch
= item
["backup_time"].as_i64(). unwrap();
109 let time_str
= Local
.timestamp(epoch
, 0).format("%c");
111 let files
= item
["files"].as_array().unwrap();
114 let filename
= file
.as_str().unwrap();
115 println
!("| {} | {} | {} | {}", btype
, id
, time_str
, filename
);
123 fn start_garbage_collection(
126 _rpcenv
: &mut RpcEnvironment
,
127 ) -> Result
<Value
, Error
> {
129 let repo_url
= tools
::required_string_param(¶m
, "repository")?
;
130 let repo
= BackupRepository
::parse(repo_url
)?
;
132 let mut client
= HttpClient
::new(&repo
.host
, &repo
.user
);
134 let path
= format
!("api2/json/admin/datastore/{}/gc", repo
.store
);
136 let result
= client
.post(&path
)?
;
141 fn parse_backupspec(value
: &str) -> Result
<(&str, &str), Error
> {
143 if let Some(caps
) = BACKUPSPEC_REGEX
.captures(value
) {
144 return Ok((caps
.get(1).unwrap().as_str(), caps
.get(2).unwrap().as_str()));
146 bail
!("unable to parse directory specification '{}'", value
);
152 _rpcenv
: &mut RpcEnvironment
,
153 ) -> Result
<Value
, Error
> {
155 let repo_url
= tools
::required_string_param(¶m
, "repository")?
;
157 let backupspec_list
= tools
::required_array_param(¶m
, "backupspec")?
;
159 let repo
= BackupRepository
::parse(repo_url
)?
;
161 let chunk_size_opt
= param
["chunk-size"].as_u64().map(|v
| v
*1024);
163 if let Some(size
) = chunk_size_opt
{
164 verify_chunk_size(size
)?
;
167 let mut upload_list
= vec
![];
169 for backupspec
in backupspec_list
{
170 let (target
, filename
) = parse_backupspec(backupspec
.as_str().unwrap())?
;
172 let stat
= match nix
::sys
::stat
::stat(filename
) {
174 Err(err
) => bail
!("unable to access '{}' - {}", filename
, err
),
177 if (stat
.st_mode
& libc
::S_IFDIR
) != 0 {
179 let target
= format
!("{}.catar", target
);
181 upload_list
.push((filename
.to_owned(), target
));
183 } else if (stat
.st_mode
& (libc
::S_IFREG
|libc
::S_IFBLK
)) != 0 {
184 if stat
.st_size
<= 0 { bail!("got strange file size '{}'
", stat.st_size); }
185 let _size = stat.st_size as usize;
187 panic!("implement me
");
189 //backup_image(&datastore, &file, size, &target, chunk_size)?;
191 // let idx = datastore.open_image_reader(target)?;
195 bail!("unsupported file
type (expected a directory
, file or block device
)");
199 let backup_time = Utc::now();
201 let mut client = HttpClient::new(&repo.host, &repo.user);
203 client.login()?; // login before starting backup
205 for (filename, target) in upload_list {
206 println!("Upload '{}' to '{:?}'
", filename, repo);
208 let stream = CaTarBackupStream::open(&filename)?;
210 let body = Body::wrap_stream(stream);
212 backup_directory(&mut client, &repo, body, &target, backup_time, chunk_size_opt)?;
215 //datastore.garbage_collection()?;
221 pub fn complete_backup_source(arg: &str) -> Vec<String> {
223 let mut result = vec![];
225 let data: Vec<&str> = arg.splitn(2, ':').collect();
227 if data.len() != 2 { return result; }
229 let files = tools::complete_file_name(data[1]);
232 result.push(format!("{}
:{}
", data[0], file));
241 _rpcenv: &mut RpcEnvironment,
242 ) -> Result<Value, Error> {
244 let repo_url = tools::required_string_param(¶m, "repository
")?;
245 let repo = BackupRepository::parse(repo_url)?;
247 let mut client = HttpClient::new(&repo.host, &repo.user);
249 let path = format!("api2
/json
/admin
/datastore
/{}
/prune
", repo.store);
251 param.as_object_mut().unwrap().remove("repository
");
253 let result = client.post_json(&path, param)?;
260 let repo_url_schema: Arc<Schema> = Arc::new(
261 StringSchema::new("Repository URL
.")
262 .format(BACKUP_REPO_URL.clone())
267 let backup_source_schema: Arc<Schema> = Arc::new(
268 StringSchema::new("Backup source
specification ([<label
>:<path
>]).")
269 .format(Arc::new(ApiStringFormat::Pattern(&BACKUPSPEC_REGEX)))
273 let create_cmd_def = CliCommand::new(
276 ObjectSchema::new("Create backup
.")
277 .required("repository
", repo_url_schema.clone())
281 "List of backup source
specifications ([<label
>:<path
>] ...)",
282 backup_source_schema,
287 IntegerSchema::new("Chunk size
in KB
. Must be a power of
2.")
293 .arg_param(vec!["repository
", "backupspec
"])
294 .completion_cb("backupspec
", complete_backup_source);
296 let list_cmd_def = CliCommand::new(
299 ObjectSchema::new("List backups
.")
300 .required("repository
", repo_url_schema.clone())
302 .arg_param(vec!["repository
"]);
304 let garbage_collect_cmd_def = CliCommand::new(
306 start_garbage_collection,
307 ObjectSchema::new("Start garbage collection
for a specific repository
.")
308 .required("repository
", repo_url_schema.clone())
310 .arg_param(vec!["repository
"]);
312 let prune_cmd_def = CliCommand::new(
315 proxmox_backup::api2::admin::datastore::add_common_prune_prameters(
316 ObjectSchema::new("Prune backup repository
.")
317 .required("repository
", repo_url_schema.clone())
320 .arg_param(vec!["repository
"]);
321 let cmd_def = CliCommandMap::new()
322 .insert("create
".to_owned(), create_cmd_def.into())
323 .insert("garbage
-collect
".to_owned(), garbage_collect_cmd_def.into())
324 .insert("list
".to_owned(), list_cmd_def.into())
325 .insert("prune
".to_owned(), prune_cmd_def.into());
327 run_cli_command(cmd_def.into());