diff options
author | Matthias Beyer <mail@beyermatthias.de> | 2018-04-26 11:19:14 +0200 |
---|---|---|
committer | Matthias Beyer <mail@beyermatthias.de> | 2018-05-04 11:09:20 +0200 |
commit | 61d0136c262517d7c5fab6c7bdfdfdb67f905bfb (patch) | |
tree | e413d7583bc501234726a59dac6021b962a38485 /bin | |
parent | 2e2bce77a08636d19625a58c02d266271d18c58f (diff) |
imag-category: Initial import
Diffstat (limited to 'bin')
-rw-r--r-- | bin/core/imag-category/Cargo.toml | 39 | ||||
l--------- | bin/core/imag-category/README.md | 1 | ||||
-rw-r--r-- | bin/core/imag-category/src/main.rs | 240 | ||||
-rw-r--r-- | bin/core/imag-category/src/ui.rs | 118 |
4 files changed, 398 insertions, 0 deletions
diff --git a/bin/core/imag-category/Cargo.toml b/bin/core/imag-category/Cargo.toml new file mode 100644 index 00000000..b5ec21ef --- /dev/null +++ b/bin/core/imag-category/Cargo.toml @@ -0,0 +1,39 @@ +[package] +name = "imag-category" +version = "0.8.0" +authors = ["Matthias Beyer <mail@beyermatthias.de>"] + +description = "Part of the imag core distribution: imag-category command" + +keywords = ["imag", "PIM", "personal", "information", "management"] +readme = "../../../README.md" +license = "LGPL-2.1" + +documentation = "https://imag-pim.org/doc/" +repository = "https://github.com/matthiasbeyer/imag" +homepage = "http://imag-pim.org" + +build = "../../../build.rs" + +[badges] +travis-ci = { repository = "matthiasbeyer/imag" } +is-it-maintained-issue-resolution = { repository = "matthiasbeyer/imag" } +is-it-maintained-open-issues = { repository = "matthiasbeyer/imag" } +maintenance = { status = "actively-developed" } + +[dependencies] +log = "0.4.0" +toml = "0.4" +toml-query = "0.6" + +libimagstore = { version = "0.8.0", path = "../../../lib/core/libimagstore" } +libimagrt = { version = "0.8.0", path = "../../../lib/core/libimagrt" } +libimagerror = { version = "0.8.0", path = "../../../lib/core/libimagerror" } +libimagentrycategory = { version = "0.8.0", path = "../../../lib/entry/libimagentrycategory" } +libimaginteraction = { version = "0.8.0", path = "../../../lib/etc/libimaginteraction" } + +[dependencies.clap] +version = "^2.29" +default-features = false +features = ["color", "suggestions", "wrap_help"] + diff --git a/bin/core/imag-category/README.md b/bin/core/imag-category/README.md new file mode 120000 index 00000000..94db65a6 --- /dev/null +++ b/bin/core/imag-category/README.md @@ -0,0 +1 @@ +../../../doc/src/04020-module-category.md
\ No newline at end of file diff --git a/bin/core/imag-category/src/main.rs b/bin/core/imag-category/src/main.rs new file mode 100644 index 00000000..5b68eeb5 --- /dev/null +++ b/bin/core/imag-category/src/main.rs @@ -0,0 +1,240 @@ +// +// imag - the personal information management suite for the commandline +// Copyright (C) 2015-2018 Matthias Beyer <mail@beyermatthias.de> and contributors +// +// This library is free software; you can redistribute it and/or +// modify it under the terms of the GNU Lesser General Public +// License as published by the Free Software Foundation; version +// 2.1 of the License. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License along with this library; if not, write to the Free Software +// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA +// + +#![deny( + non_camel_case_types, + non_snake_case, + path_statements, + trivial_numeric_casts, + unstable_features, + unused_allocation, + unused_import_braces, + unused_imports, + unused_must_use, + unused_mut, + unused_qualifications, + while_true, +)] + +extern crate clap; +#[macro_use] +extern crate log; + +extern crate libimagentrycategory; +extern crate libimagerror; +#[macro_use] extern crate libimagrt; +extern crate libimagstore; +extern crate libimaginteraction; + +use libimagerror::trace::MapErrTrace; +use libimagerror::exit::ExitUnwrap; +use libimagerror::io::ToExitCode; +use libimagrt::runtime::Runtime; +use libimagrt::setup::generate_runtime_setup; +use libimagstore::storeid::IntoStoreId; + +mod ui; + +use std::io::Write; +use std::io::Read; +use std::path::PathBuf; + +use libimagentrycategory::store::CategoryStore; +use libimagstore::storeid::StoreIdIterator; +use libimagstore::iter::get::StoreIdGetIteratorExtension; +use libimagerror::iter::TraceIterator; +use libimagentrycategory::entry::EntryCategory; +use libimagentrycategory::category::Category; + +fn main() { + let version = make_imag_version!(); + let rt = generate_runtime_setup("imag-category", + &version, + "Add a category to entries and manage categories", + ui::build_ui); + + rt.cli() + .subcommand_name() + .map(|name| { + match name { + "set" => set(&rt), + "get" => get(&rt), + "list-category" => list_category(&rt), + "create-category" => create_category(&rt), + "delete-category" => delete_category(&rt), + "list-categories" => list_categories(&rt), + other => { + debug!("Unknown command"); + let _ = rt.handle_unknown_subcommand("imag-category", other, rt.cli()) + .map_err_trace_exit_unwrap(1) + .code() + .map(::std::process::exit); + }, + } + }); +} + +fn set(rt: &Runtime) { + let scmd = rt.cli().subcommand_matches("set").unwrap(); // safed by main() + let name = scmd.value_of("set-name").map(String::from).unwrap(); // safed by clap + let sids = match scmd.value_of("set-ids") { + Some(path) => vec![PathBuf::from(path).into_storeid().map_err_trace_exit_unwrap(1)], + None => if rt.cli().is_present("entries-from-stdin") { + let stdin = rt.stdin().unwrap_or_else(|| { + error!("Cannot get handle to stdin"); + ::std::process::exit(1) + }); + + let mut buf = String::new(); + let _ = stdin.lock().read_to_string(&mut buf).unwrap_or_else(|_| { + error!("Failed to read from stdin"); + ::std::process::exit(1) + }); + + buf.lines() + .map(PathBuf::from) + .map(|p| p.into_storeid().map_err_trace_exit_unwrap(1)) + .collect() + } else { + error!("Something weird happened. I was not able to find the path of the entries to edit"); + ::std::process::exit(1) + } + }; + + StoreIdIterator::new(Box::new(sids.into_iter().map(Ok))) + .into_get_iter(rt.store()) + .trace_unwrap_exit(1) + .map(|o| o.unwrap_or_else(|| { + error!("Did not find one entry"); + ::std::process::exit(1) + })) + .for_each(|mut entry| { + let _ = entry + .set_category_checked(rt.store(), &name) + .map_err_trace_exit_unwrap(1); + }) +} + +fn get(rt: &Runtime) { + let scmd = rt.cli().subcommand_matches("get").unwrap(); // safed by main() + let sids = match scmd.value_of("get-ids") { + Some(path) => vec![PathBuf::from(path).into_storeid().map_err_trace_exit_unwrap(1)], + None => if rt.cli().is_present("entries-from-stdin") { + let stdin = rt.stdin().unwrap_or_else(|| { + error!("Cannot get handle to stdin"); + ::std::process::exit(1) + }); + + let mut buf = String::new(); + let _ = stdin.lock().read_to_string(&mut buf).unwrap_or_else(|_| { + error!("Failed to read from stdin"); + ::std::process::exit(1) + }); + + buf.lines() + .map(PathBuf::from) + .map(|p| p.into_storeid().map_err_trace_exit_unwrap(1)) + .collect() + } else { + error!("Something weird happened. I was not able to find the path of the entries to edit"); + ::std::process::exit(1) + } + }; + + let out = rt.stdout(); + let mut outlock = out.lock(); + + StoreIdIterator::new(Box::new(sids.into_iter().map(Ok))) + .into_get_iter(rt.store()) + .trace_unwrap_exit(1) + .map(|o| o.unwrap_or_else(|| { + error!("Did not find one entry"); + ::std::process::exit(1) + })) + .map(|entry| entry.get_category().map_err_trace_exit_unwrap(1)) + .for_each(|name| { + let _ = writeln!(outlock, "{}", name).to_exit_code().unwrap_or_exit(); + }) +} + +fn list_category(rt: &Runtime) { + let scmd = rt.cli().subcommand_matches("list-category").unwrap(); // safed by main() + let name = scmd.value_of("list-category-name").map(String::from).unwrap(); // safed by clap + + if let Some(category) = rt.store().get_category_by_name(&name).map_err_trace_exit_unwrap(1) { + let out = rt.stdout(); + let mut outlock = out.lock(); + + category + .get_entries(rt.store()) + .map_err_trace_exit_unwrap(1) + .for_each(|entry| { + writeln!(outlock, "{}", entry.map_err_trace_exit_unwrap(1).get_location()) + .to_exit_code() + .unwrap_or_exit(); + }) + } else { + info!("No category named '{}'", name); + ::std::process::exit(1) + } +} + +fn create_category(rt: &Runtime) { + let scmd = rt.cli().subcommand_matches("create-category").unwrap(); // safed by main() + let name = scmd.value_of("create-category-name").map(String::from).unwrap(); // safed by clap + + let _ = rt + .store() + .create_category(&name) + .map_err_trace_exit_unwrap(1); +} + +fn delete_category(rt: &Runtime) { + use libimaginteraction::ask::ask_bool; + + let scmd = rt.cli().subcommand_matches("delete-category").unwrap(); // safed by main() + let name = scmd.value_of("delete-category-name").map(String::from).unwrap(); // safed by clap + let ques = format!("Do you really want to delete category '{}' and remove links to all categorized enties?", name); + let answer = ask_bool(&ques, Some(false)); + + if answer { + info!("Deleting category '{}'", name); + let _ = rt + .store() + .delete_category(&name) + .map_err_trace_exit_unwrap(1); + } else { + info!("Not doing anything"); + } +} + +fn list_categories(rt: &Runtime) { + let out = rt.stdout(); + let mut outlock = out.lock(); + + rt.store() + .all_category_names() + .map_err_trace_exit_unwrap(1) + .for_each(|name| { + writeln!(outlock, "{}", name.map_err_trace_exit_unwrap(1)) + .to_exit_code() + .unwrap_or_exit(); + }) +} + diff --git a/bin/core/imag-category/src/ui.rs b/bin/core/imag-category/src/ui.rs new file mode 100644 index 00000000..6d32711c --- /dev/null +++ b/bin/core/imag-category/src/ui.rs @@ -0,0 +1,118 @@ +// +// imag - the personal information management suite for the commandline +// Copyright (C) 2015-2018 Matthias Beyer <mail@beyermatthias.de> and contributors +// +// This library is free software; you can redistribute it and/or +// modify it under the terms of the GNU Lesser General Public +// License as published by the Free Software Foundation; version +// 2.1 of the License. +// +// This library is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU +// Lesser General Public License for more details. +// +// You should have received a copy of the GNU Lesser General Public +// License along with this library; if not, write to the Free Software +// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA +// + +use clap::{Arg, ArgGroup, App, SubCommand}; + +pub fn build_ui<'a>(app: App<'a, 'a>) -> App<'a, 'a> { + app + .subcommand(SubCommand::with_name("create-category") + .about("Create a new category") + .version("0.1") + .arg(Arg::with_name("create-category-name") + .index(1) + .takes_value(true) + .required(true) + .multiple(false) + .help("The name of the new category") + .value_name("NAME")) + ) + + .subcommand(SubCommand::with_name("delete-category") + .about("Delete a new category") + .version("0.1") + .arg(Arg::with_name("delete-category-name") + .index(1) + .takes_value(true) + .required(true) + .multiple(false) + .help("The name of the category to delete") + .value_name("NAME")) + ) + + .subcommand(SubCommand::with_name("list-categories") + .about("Show all category names") + .version("0.1")) + + .subcommand(SubCommand::with_name("list-category") + .about("List all entries for a category") + .version("0.1") + .arg(Arg::with_name("list-category-name") + .index(1) + .takes_value(true) + .required(true) + .multiple(false) + .help("The name of the category to list all entries for") + .value_name("NAME")) + ) + + .subcommand(SubCommand::with_name("set") + .about("Set the category of entries") + .version("0.1") + .arg(Arg::with_name("set-name") + .index(1) + .takes_value(true) + .required(true) + .multiple(false) + .help("The name of the category to list all entries for") + .value_name("NAME")) + + .arg(Arg::with_name("set-ids") + .index(2) + .takes_value(true) + .required(false) + .multiple(true) + .help("The entries to set the category for") + .value_name("ID")) + .arg(Arg::with_name("entries-from-stdin") + .long("ids-from-stdin") + .short("I") + .takes_value(false) + .required(false) + .multiple(false) + .help("Read the ids for the entries from stdin")) + + .group(ArgGroup::with_name("input-method") + .args(&["set-ids", "entries-from-stdin"]) + .required(true)) + ) + + .subcommand(SubCommand::with_name("get") + .about("Get the category of the entry") + .version("0.1") + .arg(Arg::with_name("get-ids") + .index(1) + .takes_value(true) + .required(false) + .multiple(true) + .help("The id of the Entry to get the category for") + .value_name("ID")) + .arg(Arg::with_name("entries-from-stdin") + .long("ids-from-stdin") + .short("I") + .takes_value(false) + .required(false) + .multiple(false) + .help("Read the ids for the entries from stdin")) + + .group(ArgGroup::with_name("input-method") + .args(&["get-ids", "entries-from-stdin"]) + .required(true)) + ) +} + |