use std::collections::HashSet;
use std::fmt::{Display, Write};
use std::fs::File;
use std::hash::Hash;
use std::io::{BufReader, Lines};

#[derive(Debug, Eq, PartialOrd, Ord)]
pub struct Package {
    pub name: String,
    repo: Option<String>,
}

impl From<String> for Package {
    fn from(mut s: String) -> Self {
        s.remove_comment();
        s.remove_whitespace();
        let (name, repo) = Self::split_into_name_and_repo(s);
        Self { name, repo }
    }
}

impl Package {
    pub(crate) fn from_lines(lines: Lines<BufReader<File>>) -> HashSet<Self> {
        lines
            .into_iter()
            .map(|l| Package::from(l.unwrap()))
            .collect()
    }

    fn split_into_name_and_repo(mut s: String) -> (String, Option<String>) {
        match s.find('/') {
            None => (s, None),
            Some(pos) => {
                let mut name = s.split_off(pos);
                name = name.split_off(1);
                (name, Some(s))
            }
        }
    }
}

impl PartialEq for Package {
    fn eq(&self, other: &Self) -> bool {
        self.name == other.name
            && match &self.repo {
                None => true,
                Some(r) => match &other.repo {
                    None => true,
                    Some(r2) => r == r2,
                },
            }
    }
}

impl Hash for Package {
    fn hash<H: std::hash::Hasher>(&self, state: &mut H) {
        self.name.hash(state);
    }
}

trait Whitespace {
    fn remove_comment(&mut self) {}
    fn remove_whitespace(&mut self) {}
}

impl Whitespace for String {
    fn remove_comment(&mut self) {
        match self.find('#') {
            None => (),
            Some(idx) => self.truncate(idx),
        }
    }

    fn remove_whitespace(&mut self) {
        match self.find(char::is_whitespace) {
            None => (),
            Some(idx) => self.truncate(idx),
        }
    }
}

impl Display for Package {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match &self.repo {
            None => (),
            Some(repo) => {
                f.write_str(repo)?;
                f.write_char('/')?;
            }
        }
        f.write_str(&self.name)
    }
}

#[cfg(test)]
mod tests {
    use super::Package;

    #[test]
    fn split_into_name_and_repo() {
        let x = "repo/name".to_string();
        let (name, repo) = Package::split_into_name_and_repo(x);
        assert_eq!(name, "name");
        assert_eq!(repo, Some("repo".to_string()));

        let x = "something".to_string();
        let (name, repo) = super::Package::split_into_name_and_repo(x);
        assert_eq!(name, "something");
        assert_eq!(repo, None);
    }

    #[test]
    fn from() {
        let x = "myrepo/somepackage  #  ".to_string();
        let p = Package::from(x);
        assert_eq!(p.name, "somepackage");
        assert_eq!(p.repo, Some("myrepo".to_string()));
    }
}