src/bin/proxmox-backup-client.rs: implement completion for backup groups

This commit is contained in:
Dietmar Maurer 2019-03-13 11:56:37 +01:00
parent edd3c8c605
commit 024f11bb55
2 changed files with 76 additions and 0 deletions

View File

@ -581,6 +581,42 @@ fn prune(
Ok(result) Ok(result)
} }
fn complete_backup_group(arg: &str, param: &HashMap<String, String>) -> Vec<String> {
let mut result = vec![];
let repo_url = match param.get("repository") {
Some(v) => v,
_ => return result,
};
let repo: BackupRepository = match repo_url.parse() {
Ok(v) => v,
_ => return result,
};
let mut client = HttpClient::new(repo.host(), repo.user());
let path = format!("api2/json/admin/datastore/{}/groups", repo.store());
let resp = match client.try_get(&path) {
Ok(v) => v,
_ => return result,
};
if let Some(list) = resp["data"].as_array() {
for item in list {
if let Some(backup_id) = item["backup-id"].as_str() {
if let Some(backup_type) = item["backup-type"].as_str() {
result.push(format!("{}/{}", backup_type, backup_id));
}
}
}
}
result
}
fn main() { fn main() {
let repo_url_schema: Arc<Schema> = Arc::new( let repo_url_schema: Arc<Schema> = Arc::new(
@ -643,6 +679,7 @@ fn main() {
.required("group", StringSchema::new("Backup group.")) .required("group", StringSchema::new("Backup group."))
)) ))
.arg_param(vec!["repository", "group"]) .arg_param(vec!["repository", "group"])
.completion_cb("group", complete_backup_group)
.completion_cb("repository", complete_repository); .completion_cb("repository", complete_repository);
let forget_cmd_def = CliCommand::new( let forget_cmd_def = CliCommand::new(

View File

@ -257,6 +257,45 @@ impl HttpClient {
Self::run_request(request) Self::run_request(request)
} }
/// like get(), but use existing credentials (never asks for password).
/// this simply fails when there is no ticket
pub fn try_get(&mut self, path: &str) -> Result<Value, Error> {
let path = path.trim_matches('/');
let url: Uri = format!("https://{}:8007/{}", self.server, path).parse()?;
let mut credentials = None;
if let Some(ref ticket) = self.ticket {
if let Some(ref token) = self.token {
credentials = Some((ticket.clone(), token.clone()));
}
}
if credentials == None {
if let Some((ticket, token)) = load_ticket_info(&self.server, &self.username) {
credentials = Some((ticket.clone(), token.clone()));
}
}
if credentials == None {
bail!("unable to get credentials");
}
let (ticket, _token) = credentials.unwrap();
let enc_ticket = percent_encode(ticket.as_bytes(), DEFAULT_ENCODE_SET).to_string();
let request = Request::builder()
.method("GET")
.uri(url)
.header("User-Agent", "proxmox-backup-client/1.0")
.header("Cookie", format!("PBSAuthCookie={}", enc_ticket))
.body(Body::empty())?;
Self::run_request(request)
}
pub fn delete(&mut self, path: &str) -> Result<Value, Error> { pub fn delete(&mut self, path: &str) -> Result<Value, Error> {
let path = path.trim_matches('/'); let path = path.trim_matches('/');