imag/imag-notes/src/main.rs

118 lines
3.1 KiB
Rust
Raw Normal View History

2016-03-19 15:47:56 +00:00
extern crate clap;
#[macro_use] extern crate log;
extern crate semver;
#[macro_use] extern crate version;
extern crate itertools;
2016-03-19 15:47:56 +00:00
extern crate libimagnotes;
extern crate libimagrt;
extern crate libimagentryedit;
extern crate libimagentrytag;
extern crate libimagerror;
extern crate libimagutil;
2016-03-19 15:47:56 +00:00
use std::process::exit;
use itertools::Itertools;
use libimagentryedit::edit::Edit;
2016-03-19 15:47:56 +00:00
use libimagrt::runtime::Runtime;
use libimagrt::setup::generate_runtime_setup;
2016-03-19 15:47:56 +00:00
use libimagnotes::note::Note;
use libimagerror::trace::{MapErrTrace, trace_error};
use libimagutil::info_result::*;
use libimagutil::warn_result::WarnResult;
2016-03-19 15:47:56 +00:00
mod ui;
use ui::build_ui;
fn main() {
let rt = generate_runtime_setup("imag-notes",
&version!()[..],
"Note taking helper",
build_ui);
2016-03-19 15:47:56 +00:00
rt.cli()
.subcommand_name()
.map(|name| {
debug!("Call: {}", name);
match name {
"create" => create(&rt),
"delete" => delete(&rt),
"edit" => edit(&rt),
"list" => list(&rt),
_ => {
debug!("Unknown command"); // More error handling
},
};
});
}
fn name_from_cli(rt: &Runtime, subcmd: &str) -> String {
rt.cli().subcommand_matches(subcmd).unwrap().value_of("name").map(String::from).unwrap()
}
2016-03-19 15:47:56 +00:00
fn create(rt: &Runtime) {
2016-03-26 15:06:12 +00:00
let name = name_from_cli(rt, "create");
Note::new(rt.store(), name.clone(), String::new()).map_err_trace().ok();
2016-03-26 15:06:12 +00:00
if rt.cli().subcommand_matches("create").unwrap().is_present("edit") &&
!edit_entry(rt, name) {
exit(1);
2016-03-26 15:06:12 +00:00
}
2016-03-19 15:47:56 +00:00
}
fn delete(rt: &Runtime) {
Note::delete(rt.store(), String::from(name_from_cli(rt, "delete")))
.map_err_trace()
.map_info_str("Ok")
2016-03-19 19:23:09 +00:00
.ok();
2016-03-19 15:47:56 +00:00
}
2016-03-25 15:38:17 +00:00
fn edit(rt: &Runtime) {
2016-03-26 15:06:12 +00:00
edit_entry(rt, name_from_cli(rt, "edit"));
}
fn edit_entry(rt: &Runtime, name: String) -> bool {
let mut note = match Note::get(rt.store(), name) {
Ok(Some(note)) => note,
Ok(None) => {
warn!("Cannot edit nonexistent Note");
return false
},
Err(e) => {
trace_error(&e);
warn!("Cannot edit nonexistent Note");
return false
},
};
2016-03-25 15:38:17 +00:00
note.edit_content(rt).map_err_trace().map_warn_err_str("Editing failed").is_ok()
2016-03-19 15:47:56 +00:00
}
fn list(rt: &Runtime) {
use std::cmp::Ordering;
Note::all_notes(rt.store())
.map_err_trace_exit(1)
.map(|iter| {
let notes = iter.filter_map(|note| note.map_err_trace().ok())
.sort_by(|note_a, note_b| {
if let (Ok(a), Ok(b)) = (note_a.get_name(), note_b.get_name()) {
return a.cmp(&b)
} else {
return Ordering::Greater;
}
});
for note in notes.iter() {
note.get_name()
.map(|name| println!("{}", name))
.map_err_trace()
.ok();
2016-03-19 15:47:56 +00:00
}
})
.ok();
2016-03-19 15:47:56 +00:00
}