Rewrite main()
This commit is contained in:
parent
9c994b2011
commit
60863d2fe3
1 changed files with 32 additions and 114 deletions
|
|
@ -24,147 +24,65 @@ extern crate libimagstore;
|
|||
extern crate libimagentryview;
|
||||
#[macro_use] extern crate libimagerror;
|
||||
|
||||
use std::result::Result as RResult;
|
||||
use std::process::exit;
|
||||
use std::path::PathBuf;
|
||||
|
||||
use libimagrt::runtime::Runtime;
|
||||
use libimagrt::setup::generate_runtime_setup;
|
||||
use libimagstore::store::FileLockEntry;
|
||||
use libimagerror::trace::{trace_error, trace_error_exit};
|
||||
use libimagerror::trace::trace_error;
|
||||
use libimagentryview::builtin::stdout::StdoutViewer;
|
||||
use libimagentryview::builtin::versions::VersionsViewer;
|
||||
use libimagentryview::viewer::Viewer;
|
||||
|
||||
mod ui;
|
||||
|
||||
use error::{ViewError, ViewErrorKind};
|
||||
use ui::build_ui;
|
||||
|
||||
type Result<T> = RResult<T, ViewError>;
|
||||
|
||||
fn main() {
|
||||
let rt = generate_runtime_setup( "imag-view",
|
||||
&version!()[..],
|
||||
"View entries (readonly)",
|
||||
build_ui);
|
||||
|
||||
let entry_id = rt.cli().value_of("id").unwrap(); // enforced by clap
|
||||
let entry_id = rt.cli().value_of("id").unwrap(); // enforced by clap
|
||||
let view_header = rt.cli().is_present("view-header");
|
||||
let view_content = rt.cli().is_present("view-content");
|
||||
|
||||
if entry_id.contains("~") {
|
||||
error!("The --id argument does not need the version part");
|
||||
exit(1);
|
||||
}
|
||||
|
||||
if rt.cli().is_present("versions") {
|
||||
if let Err(e) = view_versions_of(entry_id, &rt) {
|
||||
trace_error_exit(&e, 1);
|
||||
}
|
||||
} else {
|
||||
let entry_version = rt.cli().value_of("version");
|
||||
let view_header = rt.cli().is_present("view-header");
|
||||
let view_content = rt.cli().is_present("view-content");
|
||||
|
||||
let scmd = rt.cli().subcommand_matches("view-in");
|
||||
if scmd.is_none() {
|
||||
let scmd = match rt.cli().subcommand_matches("view-in") {
|
||||
None => {
|
||||
debug!("No commandline call");
|
||||
exit(1); // we can afford not-executing destructors here
|
||||
}
|
||||
let scmd = scmd.unwrap();
|
||||
Some(s) => s,
|
||||
};
|
||||
|
||||
let viewer = {
|
||||
if scmd.is_present("view-in-stdout") {
|
||||
} else if scmd.is_present("view-in-ui") {
|
||||
warn!("Viewing in UI is currently not supported, switch to stdout");
|
||||
} else if scmd.is_present("view-in-browser") {
|
||||
warn!("Viewing in browser is currently not supported, switch to stdout");
|
||||
} else if scmd.is_present("view-in-texteditor") {
|
||||
warn!("Viewing in texteditor is currently not supported, switch to stdout");
|
||||
} else if scmd.is_present("view-in-custom") {
|
||||
warn!("Viewing in custom is currently not supported, switch to stdout");
|
||||
}
|
||||
|
||||
StdoutViewer::new(view_header, view_content)
|
||||
};
|
||||
|
||||
let entry = load_entry(entry_id, entry_version, &rt);
|
||||
if entry.is_err() {
|
||||
trace_error_exit(&entry.unwrap_err(), 1);
|
||||
}
|
||||
let entry = entry.unwrap();
|
||||
|
||||
if let Err(e) = viewer.view_entry(&entry) {
|
||||
let entry = match rt.store().retrieve(PathBuf::from(entry_id)) {
|
||||
Ok(fle) => fle,
|
||||
Err(e) => {
|
||||
trace_error(&e);
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// TODO: This is a shameless adaption of imag-store/src/util.rs
|
||||
fn load_entry<'a>(id: &str,
|
||||
version: Option<&str>,
|
||||
rt: &'a Runtime)
|
||||
-> Result<FileLockEntry<'a>>
|
||||
{
|
||||
debug!("Checking path element for version");
|
||||
|
||||
let version = {
|
||||
if version.is_none() {
|
||||
let r = id.split('~').last();
|
||||
if r.is_none() {
|
||||
warn!("No version");
|
||||
return Err(ViewError::new(ViewErrorKind::NoVersion, None));
|
||||
} else {
|
||||
r.unwrap()
|
||||
}
|
||||
} else {
|
||||
version.unwrap()
|
||||
exit(1); // we can afford not-executing destructors here
|
||||
}
|
||||
};
|
||||
|
||||
debug!("Building path from {:?} and {:?}", id, version);
|
||||
let mut path = rt.store().path().clone();
|
||||
|
||||
if id.starts_with('/') {
|
||||
path.push(format!("{}~{}", &id[1..id.len()], version));
|
||||
let res = if rt.cli().is_present("versions") {
|
||||
VersionsViewer::new(rt.store()).view_entry(&entry)
|
||||
} else {
|
||||
path.push(format!("{}~{}", id, version));
|
||||
}
|
||||
|
||||
// the above is the adaption...
|
||||
|
||||
rt.store().retrieve(path)
|
||||
.map_err(|e| ViewError::new(ViewErrorKind::StoreError, Some(Box::new(e))))
|
||||
}
|
||||
|
||||
fn view_versions_of(id: &str, rt: &Runtime) -> Result<()> {
|
||||
use glob::glob;
|
||||
|
||||
let mut path = rt.store().path().clone();
|
||||
|
||||
if id.starts_with('/') {
|
||||
path.push(format!("{}~*", &id[1..id.len()]));
|
||||
} else {
|
||||
path.push(format!("{}~*", id));
|
||||
}
|
||||
|
||||
if let Some(path) = path.to_str() {
|
||||
match glob(path) {
|
||||
Ok(paths) => {
|
||||
for entry in paths {
|
||||
match entry {
|
||||
Ok(path) => println!("{}", path.file_name().and_then(|s| s.to_str()).unwrap()),
|
||||
Err(e) => trace_error(e.error()),
|
||||
}
|
||||
}
|
||||
Ok(())
|
||||
},
|
||||
Err(e) => {
|
||||
debug!("Error in pattern");
|
||||
Err(ViewError::new(ViewErrorKind::PatternError, Some(Box::new(e))))
|
||||
},
|
||||
if scmd.is_present("view-in-stdout") {
|
||||
} else if scmd.is_present("view-in-ui") {
|
||||
warn!("Viewing in UI is currently not supported, switch to stdout");
|
||||
} else if scmd.is_present("view-in-browser") {
|
||||
warn!("Viewing in browser is currently not supported, switch to stdout");
|
||||
} else if scmd.is_present("view-in-texteditor") {
|
||||
warn!("Viewing in texteditor is currently not supported, switch to stdout");
|
||||
} else if scmd.is_present("view-in-custom") {
|
||||
warn!("Viewing in custom is currently not supported, switch to stdout");
|
||||
}
|
||||
} else {
|
||||
warn!("Could not build glob() argument!");
|
||||
Err(ViewError::new(ViewErrorKind::GlobBuildError, None))
|
||||
|
||||
StdoutViewer::new(view_header, view_content).view_entry(&entry)
|
||||
};
|
||||
|
||||
if let Err(e) = res {
|
||||
trace_error(&e);
|
||||
exit(1);
|
||||
}
|
||||
}
|
||||
|
||||
|
|
|
|||
Loading…
Reference in a new issue