bookshelf 1.1.4

pub mod config;
pub mod fsio;
pub mod module_handler;
pub mod shelf;
pub mod tui;

use clap::{load_yaml, App};
use std::fs::File;
use std::io::BufReader;
use std::path::{Path, PathBuf};
use std::{
    collections::{BTreeMap, BTreeSet},
    io::BufRead,
};

use crate::tui::ui::TUI;
use config::*;
use fsio::*;
use module_handler::*;
use shelf::*;

/// Derive the directory for an item
/// # Example:
/// ```
/// let data_root: String = String::from("/tmp/data");
/// let module: &str = "myMod";
/// let code: &str = "12345";
/// let out_dir: String = get_item_dir(data_root, module, code);
///
/// assert_eq!("/tmp/data/myMod/12345", out_dir);
/// ```
fn get_item_dir(data_root: &PathBuf, module: &str, code: &str) -> PathBuf {
    let mut pb = data_root.clone();
    pb.push(module);
    pb.push(code);
    pb
}

/// Given a URL, derive the module and code then add item to shelf, optionally
/// downloading said item
fn add_by_url(
    shelf: &mut Shelf,
    module_handler: &ModuleHandler,
    url: &str,
    data_root: Option<&PathBuf>,
    verbose: bool,
) -> Result<(), ModuleError> {
    match module_handler.derive_module(url) {
        Ok(module) => match module_handler.derive_code(module, url) {
            Ok(code) => {
                match add_by_code(
                    shelf,
                    &module_handler,
                    module,
                    code.as_str(),
                    data_root,
                    verbose,
                ) {
                    Ok(()) => Ok(()),
                    Err(e) => Err(e),
                }
            }
            Err(e) => Err(e),
        },
        Err(e) => Err(e),
    }
}

/// Given a module and code, add item to shelf, optionally downloading said item
fn add_by_code(
    shelf: &mut Shelf,
    module_handler: &ModuleHandler,
    module: &str,
    code: &str,
    data_root: Option<&PathBuf>,
    verbose: bool,
) -> Result<(), ModuleError> {
    if shelf.has_item(&module, &code) {
        if verbose {
            println!("Item {}/{} already indexed", &module, &code);
        }
        Ok(())
    } else {
        // Get metadata
        match module_handler.get_metadata(&module, &code) {
            // Print item (verbose)
            Ok(metadata) => {
                if verbose {
                    println!(
                        "Adding item: {}/{}\n\tTitle: {}\n\tAuthors: {}\n\tgenres: {}",
                        &module, &code, &metadata.0, &metadata.1, metadata.2
                    );
                }
                // title
                let title = metadata.0;
                // authors
                let mut authors: BTreeSet<String> = BTreeSet::new();
                for author in metadata.1.split(",") {
                    authors.insert(author.to_string());
                }
                // genres
                let mut genres: BTreeSet<String> = BTreeSet::new();
                for genre in metadata.2.split(",") {
                    genres.insert(genre.to_string());
                }
                // Construct item
                shelf.add_item(&module, &code, title, authors, genres);
                // Download if data_root is set
                if let Some(data_root) = data_root {
                    let dest_dir: PathBuf = get_item_dir(data_root, &module, &code);
                    match module_handler.download(&module, &code, &dest_dir) {
                        Ok(()) => Ok(()),
                        Err(e) => Err(e),
                    };
                }
                Ok(())
            }
            Err(e) => Err(e),
        }
    }
}

/// Add item to index and optionally download it.
/// Set data_root to None to skip download, or set it to Some() to download
fn add_item(
    shelf: &mut Shelf,
    module_handler: &ModuleHandler,
    url: Option<&str>,
    url_file: Option<&str>,
    module: Option<&str>,
    code: Option<&str>,
    code_file: Option<&str>,
    data_root: Option<&PathBuf>,
    verbose: bool,
) -> Result<(), BTreeMap<String, ModuleError>> {
    let mut errors: BTreeMap<String, ModuleError> = BTreeMap::new();
    if let Some(url) = url {
        // shelf add|download -u
        match add_by_url(shelf, &module_handler, url, data_root, verbose) {
            Ok(()) => {
                return Ok(());
            }
            Err(e) => {
                errors.insert(url.to_string(), e);
                return Err(errors);
            }
        }
    } else if let Some(url_file) = url_file {
        // shelf add|download -U
        match File::open(url_file) {
            Ok(file) => {
                for line in BufReader::new(file).lines() {
                    match line {
                        Ok(url) => {
                            match add_by_url(
                                shelf,
                                &module_handler,
                                url.as_str(),
                                data_root,
                                verbose,
                            ) {
                                Ok(()) => {}
                                Err(e) => {
                                    errors.insert(url, e);
                                }
                            }
                        }
                        Err(_e) => {}
                    }
                }
            }
            Err(_e) => {}
        }
    } else if let Some(module) = module {
        if let Some(code) = code {
            // shelf add|download -m MODULE -c CODE
            match add_by_code(shelf, &module_handler, module, code, data_root, verbose) {
                Ok(()) => {
                    return Ok(());
                }
                Err(e) => {
                    errors.insert(format!("{} {}", module, code), e);
                    return Err(errors);
                }
            }
        } else if let Some(code_file) = code_file {
            // shelf add|download -m MODULE -C CODE_FILE
            match File::open(code_file) {
                Ok(file) => {
                    for line in BufReader::new(file).lines() {
                        match line {
                            Ok(code) => {
                                match add_by_code(
                                    shelf,
                                    &module_handler,
                                    module,
                                    code.as_str(),
                                    data_root,
                                    verbose,
                                ) {
                                    Ok(()) => {}
                                    Err(e) => {
                                        errors.insert(format!("{} {}", module, code), e);
                                    }
                                }
                            }
                            Err(_e) => {}
                        }
                    }
                }
                Err(_e) => {}
            }
        }
    }
    if errors.is_empty() {
        Ok(())
    } else {
        Err(errors)
    }
}

/// Print item info to stdout. To be used in CLI (single command) mode.
fn cli_print_item(shelf: &Shelf, module: &str, code: &str) {
    if let Some(item) = shelf.get_item(module, code) {
        let (i_title, i_authors, i_genres) = item.export();
        println!("Title: {}", i_title);
        let mut authors = String::new();
        for author in i_authors {
            authors.push_str(author);
            authors.push_str(", ");
        }
        authors.pop();
        authors.pop();
        println!("Authors: {}", authors);
        let mut genres = String::new();
        for genre in i_genres {
            genres.push_str(genre);
            genres.push_str(", ");
        }
        genres.pop();
        genres.pop();
        println!("Authors: {}", genres);
    } else {
    }
}

fn main() {
    /***** Parse arguments and load config *****/
    let arg_file = load_yaml!("args.yaml");
    let args = App::from(arg_file).get_matches();
    let mut config = Config::default();
    {
        let config = &mut config;
        if let Some(c) = args.value_of("config") {
            let path_to_config = PathBuf::from(c);
            if Path::new(&path_to_config).exists() {
                match config.update(&path_to_config) {
                    Ok(()) => {}
                    Err(e) => println!("Error loading config file: {:?}", e),
                }
            }
        } else {
            let mut home_dir = dirs_next::home_dir();
            match &mut home_dir {
                Some(h) => {
                    h.push(".config/bookshelf/bookshelf.yaml");
                    match config.update(&h) {
                        Ok(()) => {}
                        Err(e) => println!("Error loading config file: {:?}", e),
                    }
                }
                _ => println!("Error getting home dir"),
            }
        }
    }
    // Create necessary directories
    create_dirs(&config.data_dir, &config.modules_dir);
    let verbose: bool = { args.is_present("verbose") };

    /***** Initialize shelf and handlers *****/
    // These can be unwrap'd safely because load_config guarantees the entries
    let mut shelf: Shelf = load_shelf(&config.index_file);
    let module_handler = ModuleHandler::new(&config.modules_dir);

    /***** main *****/
    match args.subcommand() {
        Some(("modules", _args)) => {
            for i in module_handler.list_modules().iter() {
                println!("{}", i);
            }
        }
        Some(("add", args)) => {
            match add_item(
                &mut shelf,
                &module_handler,
                args.value_of("url"),
                args.value_of("url_file"),
                args.value_of("module"),
                args.value_of("code"),
                args.value_of("code_file"),
                None,
                verbose,
            ) {
                Ok(()) => {
                    println!("All items added sucessfully");
                }
                Err(errors) => {
                    println!("Some items failed to be added:");
                    for (item, error) in errors {
                        println!("{}: {:?}", item, error);
                    }
                }
            }
        }

        Some(("download", args)) => {
            match add_item(
                &mut shelf,
                &module_handler,
                args.value_of("url"),
                args.value_of("url_file"),
                args.value_of("module"),
                args.value_of("code"),
                args.value_of("code_file"),
                Some(&config.data_dir),
                verbose,
            ) {
                Ok(()) => {
                    println!("All items have been downloaded sucessfully");
                }
                Err(errors) => {
                    println!("Some items failed to be downloaded:");
                    for (item, error) in errors {
                        println!("{}: {:?}", item, error);
                    }
                }
            }
        }

        Some(("search", args)) => {
            match shelf.search_item(
                args.value_of("module"),
                args.value_of("title"),
                args.value_of("authors"),
                args.value_of("genres"),
                args.value_of("blacklist"),
                args.is_present("broad_search"),
                args.is_present("favorite"),
            ) {
                Ok(result) => {
                    for (m, c) in result {
                        println!("{} {}", &m, &c);
                        if verbose {
                            cli_print_item(&shelf, &m, &c);
                        }
                    }
                }
                Err(e) => {
                    println!("Error searching items: {}", e);
                }
            }
        }

        Some(("rm", args)) => {
            match shelf.search_item(
                args.value_of("module"),
                args.value_of("title"),
                args.value_of("authors"),
                args.value_of("genres"),
                args.value_of("blacklist"),
                args.is_present("broad_search"),
                args.is_present("favorite"),
            ) {
                Ok(result) => {
                    for (m, c) in result.iter() {
                        shelf.remove_item(m, c);
                    }
                }
                Err(e) => {
                    println!("Error removing items: {}", e);
                }
            }
        }

        Some(("pull", args)) => {
            match shelf.search_item(
                args.value_of("module"),
                args.value_of("title"),
                args.value_of("authors"),
                args.value_of("genres"),
                args.value_of("blacklist"),
                args.is_present("broad_search"),
                args.is_present("favorite"),
            ) {
                Ok(result) => {
                    for (m, c) in result {
                        let dest_dir = get_item_dir(&config.data_dir, &m, &c);
                        match module_handler.download(&m[..], &c[..], &dest_dir) {
                            Ok(()) => {}
                            Err(_e) => println!("Module {} unavailable", &m),
                        }
                    }
                }
                Err(e) => {
                    println!("Error pulling items: {}", e);
                }
            }
        }

        Some(("info", args)) => {
            cli_print_item(
                &shelf,
                args.value_of("module").unwrap(),
                args.value_of("code").unwrap(),
            );
        }

        Some(("edit", args)) => {
            shelf.edit_item(
                args.value_of("module"),
                args.value_of("code"),
                args.value_of("title"),
                args.value_of("authors"),
                args.value_of("genres"),
                args.is_present("favorite"),
            );
        }

        Some(("import", args)) => {
            import_shelf(&mut shelf, &PathBuf::from(args.value_of("file").unwrap()));
        }

        Some(("export", args)) => {
            export_shelf(&shelf, &PathBuf::from(args.value_of("file").unwrap()));
        }

        None => {
            // Start TUI if no argument is given
            let mut tui = TUI::new(&config, &mut shelf, &module_handler);
            match tui.start() {
                Ok(()) => {}
                Err(e) => {
                    println!("Error: {}", e)
                }
            }
        }

        _ => {
            println!("Invalid subcommand");
        }
    }

    /***** Save and exit *****/
    save_shelf(&shelf, &config.index_file);
}