121 lines
3.5 KiB
Rust
121 lines
3.5 KiB
Rust
#[macro_use] extern crate log;
|
|
#[macro_use] extern crate version;
|
|
extern crate semver;
|
|
extern crate clap;
|
|
|
|
extern crate libimagstore;
|
|
extern crate libimagrt;
|
|
extern crate libimagref;
|
|
extern crate libimagerror;
|
|
extern crate libimagentrylist;
|
|
extern crate libimaginteraction;
|
|
|
|
mod ui;
|
|
use ui::build_ui;
|
|
|
|
use std::path::PathBuf;
|
|
|
|
use libimagref::reference::Ref;
|
|
use libimagref::flags::RefFlags;
|
|
use libimagerror::trace::trace_error;
|
|
use libimagrt::setup::generate_runtime_setup;
|
|
use libimagrt::runtime::Runtime;
|
|
|
|
fn main() {
|
|
let rt = generate_runtime_setup("imag-ref",
|
|
&version!()[..],
|
|
"Reference files outside of the store",
|
|
build_ui);
|
|
rt.cli()
|
|
.subcommand_name()
|
|
.map(|name| {
|
|
debug!("Call: {}", name);
|
|
match name {
|
|
"add" => add(&rt),
|
|
"remove" => remove(&rt),
|
|
"list" => list(&rt),
|
|
_ => {
|
|
debug!("Unknown command"); // More error handling
|
|
},
|
|
};
|
|
});
|
|
}
|
|
|
|
fn add(rt: &Runtime) {
|
|
let cmd = rt.cli().subcommand_matches("add").unwrap();
|
|
let path = cmd.value_of("path").map(PathBuf::from).unwrap(); // saved by clap
|
|
|
|
let flags = RefFlags::default()
|
|
.with_content_hashing(cmd.is_present("track-content"))
|
|
.with_permission_tracking(cmd.is_present("track-permissions"));
|
|
|
|
match Ref::create(rt.store(), path, flags) {
|
|
Ok(r) => {
|
|
debug!("Reference created: {:?}", r);
|
|
info!("Ok");
|
|
},
|
|
Err(e) => {
|
|
trace_error(&e);
|
|
warn!("Failed to create reference");
|
|
},
|
|
}
|
|
}
|
|
|
|
fn remove(rt: &Runtime) {
|
|
use libimagref::error::RefErrorKind;
|
|
use libimagerror::into::IntoError;
|
|
use libimaginteraction::ask::ask_bool;
|
|
|
|
let cmd = rt.cli().subcommand_matches("remove").unwrap();
|
|
let hash = cmd.value_of("hash").map(String::from).unwrap(); // saved by clap
|
|
let yes = cmd.is_present("yes");
|
|
|
|
if yes || ask_bool(&format!("Delete Ref with hash '{}'", hash)[..], None) {
|
|
match Ref::delete_by_hash(rt.store(), hash) {
|
|
Err(e) => trace_error(&e),
|
|
Ok(_) => info!("Ok"),
|
|
}
|
|
} else {
|
|
info!("Aborted");
|
|
}
|
|
|
|
}
|
|
|
|
fn list(rt: &Runtime) {
|
|
use std::process::exit;
|
|
use std::ops::Deref;
|
|
|
|
use libimagentrylist::lister::Lister;
|
|
use libimagentrylist::listers::core::CoreLister;
|
|
use libimagref::lister::RefLister;
|
|
|
|
let cmd = rt.cli().subcommand_matches("list").unwrap();
|
|
let do_check_dead = cmd.is_present("check-dead");
|
|
let do_check_changed = cmd.is_present("check-changed");
|
|
let do_check_changed_content = cmd.is_present("check-changed-content");
|
|
let do_check_changed_permiss = cmd.is_present("check-changed-permissions");
|
|
|
|
let iter = match rt.store().retrieve_for_module("ref") {
|
|
Ok(iter) => iter.filter_map(|id| {
|
|
match Ref::get(rt.store(), id) {
|
|
Ok(r) => Some(r),
|
|
Err(e) => {
|
|
trace_error(&e);
|
|
None
|
|
},
|
|
}
|
|
}),
|
|
Err(e) => {
|
|
trace_error(&e);
|
|
exit(1);
|
|
}
|
|
};
|
|
|
|
RefLister::new()
|
|
.check_dead(do_check_dead)
|
|
.check_changed(do_check_changed)
|
|
.check_changed_content(do_check_changed_content)
|
|
.check_changed_permiss(do_check_changed_permiss)
|
|
.list(iter.map(|e| e.into()));
|
|
}
|
|
|