bin/proxmox-backup-client.rs: implement file name completions - first try ...

This commit is contained in:
Dietmar Maurer
2019-01-17 14:24:20 +01:00
parent 83bdac1e3b
commit 244d9b17a8
3 changed files with 56 additions and 3 deletions

View File

@ -111,6 +111,58 @@ fn create_backup(param: Value, _info: &ApiMethod) -> Result<Value, Error> {
}
pub fn complete_file_name(arg: &str) -> Vec<String> {
let mut result = vec![];
use nix::fcntl::OFlag;
use nix::sys::stat::Mode;
use nix::fcntl::AtFlags;
let mut dirname = std::path::PathBuf::from(arg);
if let Ok(stat) = nix::sys::stat::fstatat(libc::AT_FDCWD, &dirname, AtFlags::empty()) {
} else {
if let Some(parent) = dirname.parent() {
dirname = parent.to_owned();
}
}
let mut dir = match nix::dir::Dir::openat(libc::AT_FDCWD, &dirname, OFlag::O_DIRECTORY, Mode::empty()) {
Ok(d) => d,
Err(err) => {
return result;
}
};
for item in dir.iter() {
if let Ok(entry) = item {
if let Ok(name) = entry.file_name().to_str() {
if name == "." || name == ".." { continue; }
let mut newpath = dirname.clone();
newpath.push(name);
if let Ok(stat) = nix::sys::stat::fstatat(libc::AT_FDCWD, &newpath, AtFlags::empty()) {
if (stat.st_mode & libc::S_IFMT) == libc::S_IFDIR {
newpath.push("");
if let Some(newpath) = newpath.to_str() {
result.push(newpath.to_owned());
}
continue;
}
}
if let Some(newpath) = newpath.to_str() {
result.push(newpath.to_owned());
}
}
}
}
result
}
fn main() {
let cmd_def = CliCommand::new(
@ -129,6 +181,7 @@ fn main() {
)
))
.arg_param(vec!["filename", "target"])
.completion_cb("filename", complete_file_name)
.completion_cb("store", proxmox_backup::config::datastore::complete_datastore_name);