imag/imag-notes/src/main.rs

130 lines
3.2 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 libimagnotes;
extern crate libimagrt;
extern crate libimagentryedit;
extern crate libimagentrytag;
extern crate libimagerror;
2016-03-19 15:47:56 +00:00
use std::process::exit;
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::{trace_error, trace_error_exit};
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())
2016-03-19 15:47:56 +00:00
.map_err(|e| trace_error(&e))
2016-03-19 19:23:09 +00:00
.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")))
2016-03-19 15:47:56 +00:00
.map_err(|e| trace_error(&e))
2016-03-19 19:23:09 +00:00
.map(|_| println!("Ok"))
.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
2016-03-26 15:06:12 +00:00
if let Err(e) = note.edit_content(rt) {
trace_error(&e);
warn!("Editing failed");
return false
}
true
2016-03-19 15:47:56 +00:00
}
fn list(rt: &Runtime) {
use std::cmp::Ordering;
let iter = Note::all_notes(rt.store());
if iter.is_err() {
trace_error_exit(&iter.unwrap_err(), 1);
2016-03-19 15:47:56 +00:00
}
let mut iter = iter.unwrap()
.filter_map(|note| {
match note {
Err(e) => {
trace_error(&e);
None
},
Ok(e) => Some(e)
}
})
.collect::<Vec<Note>>();
iter.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 iter {
2016-03-19 19:23:09 +00:00
note.get_name()
.map(|name| println!("{}", name))
.map_err(|e| trace_error(&e))
.ok();
2016-03-19 15:47:56 +00:00
}
}