#![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, )] #[macro_use] extern crate log; extern crate clap; #[macro_use] extern crate semver; extern crate toml; extern crate url; #[macro_use] extern crate version; extern crate libimagentrylink; extern crate libimagrt; extern crate libimagstore; extern crate libimagerror; extern crate libimagutil; use std::ops::Deref; use libimagrt::runtime::Runtime; use libimagrt::setup::generate_runtime_setup; use libimagstore::error::StoreError; use libimagstore::store::Entry; use libimagstore::store::FileLockEntry; use libimagstore::store::Store; use libimagerror::trace::{MapErrTrace, trace_error, trace_error_exit}; use libimagentrylink::external::ExternalLinker; use libimagutil::warn_result::*; use libimagutil::warn_exit::warn_exit; use libimagutil::info_result::*; use clap::ArgMatches; use url::Url; mod ui; use ui::build_ui; fn main() { let rt = generate_runtime_setup("imag-link", &version!()[..], "Link entries", build_ui); rt.cli() .subcommand_name() .map(|name| { match name { "internal" => handle_internal_linking(&rt), "external" => handle_external_linking(&rt), _ => warn_exit("No commandline call", 1) } }); } fn handle_internal_linking(rt: &Runtime) { use libimagentrylink::internal::InternalLinker; debug!("Handle internal linking call"); let cmd = rt.cli().subcommand_matches("internal").unwrap(); if cmd.is_present("list") { debug!("List..."); for entry in cmd.value_of("list").unwrap().split(',') { debug!("Listing for '{}'", entry); match get_entry_by_name(rt, entry) { Ok(Some(e)) => { e.get_internal_links() .map(|links| { let i = links .iter() .filter_map(|l| { l.to_str() .map_warn_err(|e| format!("Failed to convert StoreId to string: {:?}", e)) .ok() }) .enumerate(); for (i, link) in i { println!("{: <3}: {}", i, link); } }) .map_err_trace() .ok(); }, Ok(None) => { warn!("Entry not found: {:?}", entry); break; } Err(e) => { trace_error(&e); break; }, } } debug!("Listing ready!"); } else { let mut from = { let from = get_from_entry(&rt); if from.is_none() { warn_exit("No 'from' entry", 1); } from.unwrap() }; debug!("Link from = {:?}", from.deref()); let to = { let to = get_to_entries(&rt); if to.is_none() { warn_exit("No 'to' entry", 1); } to.unwrap() }; debug!("Link to = {:?}", to.iter().map(|f| f.deref()).collect::>()); match cmd.subcommand_name() { Some("add") => { for mut to_entry in to { if let Err(e) = to_entry.add_internal_link(&mut from) { trace_error_exit(&e, 1); } } }, Some("remove") => { for mut to_entry in to { if let Err(e) = to_entry.remove_internal_link(&mut from) { trace_error_exit(&e, 1); } } }, _ => unreachable!(), }; } } fn get_from_entry<'a>(rt: &'a Runtime) -> Option> { rt.cli() .subcommand_matches("internal") .unwrap() // safe, we know there is an "internal" subcommand" .subcommand_matches("add") .unwrap() // safe, we know there is an "add" subcommand .value_of("from") .and_then(|from_name| { match get_entry_by_name(rt, from_name) { Err(e) => { debug!("We couldn't get the entry from name: '{:?}'", from_name); trace_error(&e); None }, Ok(Some(e)) => Some(e), Ok(None) => None, } }) } fn get_to_entries<'a>(rt: &'a Runtime) -> Option>> { rt.cli() .subcommand_matches("internal") .unwrap() // safe, we know there is an "internal" subcommand" .subcommand_matches("add") .unwrap() // safe, we know there is an "add" subcommand .values_of("to") .map(|values| { let mut v = vec![]; for entry in values.map(|v| get_entry_by_name(rt, v)) { match entry { Err(e) => trace_error(&e), Ok(Some(e)) => v.push(e), Ok(None) => warn!("Entry not found: {:?}", v), } } v }) } fn get_entry_by_name<'a>(rt: &'a Runtime, name: &str) -> Result>, StoreError> { use std::path::PathBuf; use libimagstore::storeid::StoreId; StoreId::new(Some(rt.store().path().clone()), PathBuf::from(name)) .and_then(|id| rt.store().get(id)) } fn handle_external_linking(rt: &Runtime) { let scmd = rt.cli().subcommand_matches("external").unwrap(); let entry_name = scmd.value_of("id").unwrap(); // enforced by clap let entry = get_entry_by_name(rt, entry_name); if entry.is_err() { trace_error_exit(&entry.unwrap_err(), 1); } let entry = entry.unwrap(); if entry.is_none() { warn!("Entry not found: {:?}", entry_name); return; } let mut entry = entry.unwrap(); if scmd.is_present("add") { debug!("Adding link to entry!"); add_link_to_entry(rt.store(), scmd, &mut entry); return; } if scmd.is_present("remove") { debug!("Removing link from entry!"); remove_link_from_entry(rt.store(), scmd, &mut entry); return; } if scmd.is_present("set") { debug!("Setting links in entry!"); set_links_for_entry(rt.store(), scmd, &mut entry); return; } if scmd.is_present("list") { debug!("Listing links in entry!"); list_links_for_entry(rt.store(), &mut entry); return; } panic!("Clap failed to enforce one of 'add', 'remove', 'set' or 'list'"); } fn add_link_to_entry(store: &Store, matches: &ArgMatches, entry: &mut FileLockEntry) { Url::parse(matches.value_of("add").unwrap()) .map_err_trace_exit(1) .map(|link| entry.add_external_link(store, link).map_err_trace().map_info_str("Ok")) .ok(); } fn remove_link_from_entry(store: &Store, matches: &ArgMatches, entry: &mut FileLockEntry) { Url::parse(matches.value_of("remove").unwrap()) .map_err_trace_exit(1) .map(|link| entry.remove_external_link(store, link).map_err_trace().map_info_str("Ok")) .ok(); } fn set_links_for_entry(store: &Store, matches: &ArgMatches, entry: &mut FileLockEntry) { let links = matches .value_of("links") .map(String::from) .unwrap() .split(',') .map(|uri| { match Url::parse(uri) { Err(e) => { warn!("Could not parse '{}' as URL, ignoring", uri); trace_error(&e); None }, Ok(u) => Some(u), } }) .filter_map(|x| x) .collect(); entry.set_external_links(store, links) .map_err_trace() .map_info_str("Ok") .ok(); } fn list_links_for_entry(store: &Store, entry: &mut FileLockEntry) { entry.get_external_links(store) .and_then(|links| { for (i, link) in links.iter().enumerate() { println!("{: <3}: {}", i, link); } Ok(()) }) .map_err_trace() .map_info_str("Ok") .ok(); }