bookshelf 1.1.4

use regex::Regex;
use serde::{Deserialize, Serialize};
use std::collections::{BTreeMap, BTreeSet};

/// Items to be stored in the Shelf
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
pub struct Item {
    title: String,
    authors: BTreeSet<String>,
    genres: BTreeSet<String>,
}

impl Item {
    pub fn new(title: String, authors: BTreeSet<String>, genres: BTreeSet<String>) -> Item {
        Item {
            title,
            authors,
            genres,
        }
    }

    /// Get the title, authors, and genres of the item
    pub fn export(&self) -> (&String, &BTreeSet<String>, &BTreeSet<String>) {
        return (&self.title, &self.authors, &self.genres);
    }
}

/// The shelf indexes all items and a list of favorites
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize)]
pub struct Shelf {
    index: BTreeMap<(String, String), Item>,
    favorites: BTreeSet<(String, String)>,
}

impl Shelf {
    pub fn new() -> Shelf {
        Shelf {
            index: BTreeMap::new(),
            favorites: BTreeSet::new(),
        }
    }

    /// Get the index
    pub fn get_index(&self) -> &BTreeMap<(String, String), Item> {
        &self.index
    }

    /// Check it an item is in the shelf
    pub fn has_item(&self, module: &str, code: &str) -> bool {
        self.index
            .contains_key(&(module.to_string(), code.to_string()))
    }

    // Add to index but not download
    pub fn add_item(
        &mut self,
        module: &str,
        code: &str,
        title: String,
        authors: BTreeSet<String>,
        genres: BTreeSet<String>,
    ) {
        self.index.insert(
            (module.to_string(), code.to_string()),
            Item::new(title, authors, genres),
        );
    }

    /// Search for items matching the provided parameters
    pub fn search_item(
        &self,
        module: Option<&str>,
        title_regex: Option<&str>,
        authors: Option<&str>,
        genres: Option<&str>,
        blacklist: Option<&str>,
        broad_search: bool,
        favorite: bool,
    ) -> Result<BTreeSet<(String, String)>, regex::Error> {
        // --favorite
        let mut result: BTreeSet<(String, String)> = if favorite {
            self.favorites.clone()
        } else {
            self.index.keys().cloned().collect()
        };

        // --modules
        if let Some(module) = module {
            let tmp = result.clone();
            for (m, c) in tmp {
                if m != module {
                    result.remove(&(m, c));
                }
            }
        }

        // --authors (match if any author matches)
        if let Some(authors) = authors {
            for key in result.clone() {
                for author in authors.split(",") {
                    if let Some(item) = self.index.get(&key) {
                        if !item.authors.contains(&author.to_string()) {
                            result.remove(&key);
                        }
                    } else {
                        result.remove(&key);
                    }
                }
            }
        }

        // --genres
        if let Some(genres) = genres {
            if broad_search {
                // broad search (match item if at least 1 genre matches)
                for key in result.clone() {
                    for genre in genres.split(",") {
                        if let Some(item) = self.index.get(&key) {
                            if !item.genres.contains(&genre.to_string()) {
                                result.remove(&key);
                            }
                        } else {
                            result.remove(&key);
                        }
                    }
                }
            } else {
                // normal search (match item if all genres match)
                for key in result.clone() {
                    let mut matches: bool = true;
                    for genre in genres.split(",") {
                        if let Some(item) = self.index.get(&key) {
                            if !item.genres.contains(&genre.to_string()) {
                                matches = false;
                            }
                        } else {
                            result.remove(&key);
                        }
                    }
                    if !matches {
                        result.remove(&key);
                    }
                }
            }
        }

        // --blacklist
        if let Some(blacklist) = blacklist {
            for key in result.clone() {
                for genre in blacklist.split(",") {
                    if let Some(item) = self.index.get(&key) {
                        if item.genres.contains(&genre.to_string()) {
                            result.remove(&key);
                        }
                    } else {
                        result.remove(&key);
                    }
                }
            }
        }

        // --title (match regex against title)
        if let Some(title_regex) = title_regex {
            match Regex::new(title_regex) {
                Ok(regex) => {
                    for key in result.clone() {
                        if let Some(item) = self.index.get(&key) {
                            if !&regex.is_match(&item.title) {
                                result.remove(&key);
                            }
                        } else {
                            result.remove(&key);
                        }
                    }
                }
                Err(e) => {
                    return Err(e);
                }
            }
        }
        Ok(result)
    }

    /// Remove item from index (and favorites)
    pub fn remove_item(&mut self, module: &str, code: &str) {
        let key: (String, String) = (module.to_string(), code.to_string());
        self.index.remove(&key);
        self.favorites.remove(&key);
    }

    /// Get the item corresponding to the module and code
    pub fn get_item(&self, module: &str, code: &str) -> Option<&Item> {
        self.index.get(&(module.to_string(), code.to_string()))
    }

    /// Edit item with provided parameters
    pub fn edit_item(
        &mut self,
        module: Option<&str>,
        code: Option<&str>,
        title: Option<&str>,
        authors: Option<&str>,
        genres: Option<&str>,
        favorite: bool,
    ) {
        // these 2 are required and can be safely unwrap'd
        let k = (module.unwrap().to_string(), code.unwrap().to_string());

        // update values
        if let Some(item) = self.index.get_mut(&k) {
            if let Some(t) = title {
                item.title = t.to_string();
            }
            if let Some(s) = authors {
                let mut authors: BTreeSet<String> = BTreeSet::new();
                for author in s.split(",") {
                    authors.insert(author.to_string());
                }
                item.authors = authors;
            }
            if let Some(t) = genres {
                let mut genres: BTreeSet<String> = BTreeSet::new();
                for genre in t.split(",") {
                    genres.insert(genre.to_string());
                }
                item.genres = genres;
            }
        }
        // insert if item wasn't in favorites
        if favorite {
            if !self.favorites.remove(&k) {
                self.favorites.insert(k);
            }
        }
    }

    /// Import a shelf into self, extending self's index and favorites
    pub fn import(&mut self, new_shelf: &Shelf) {
        // index
        for ((module, code), item) in new_shelf.index.iter() {
            self.index
                .insert((module.clone(), code.clone()), item.clone());
        }
        // favorites
        for (module, code) in new_shelf.favorites.iter() {
            self.favorites.insert((module.clone(), code.clone()));
        }
    }
}