// Copyright (c) 2016-2017 Martijn Rijkeboer <[email protected]>
//
// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
// option. This file may not be copied, modified, or distributed
// except according to those terms.

use super::custom_icon_uuid::CustomIconUuid;
use super::entry::Entry;
use super::entry_uuid::EntryUuid;
use super::group_uuid::GroupUuid;
use super::icon::Icon;
use super::times::Times;
use chrono::{DateTime, Utc};
use std::collections::vec_deque::VecDeque;
use std::ptr;

/// A group in the database.
#[derive(Clone, Debug, PartialEq)]
pub struct Group {
    /// The date and time this group was created.
    pub creation_time: DateTime<Utc>,

    /// The identifier of this group's custom icon if any.
    pub custom_icon_uuid: Option<CustomIconUuid>,

    /// Default auto-type sequence.
    pub def_auto_type_sequence: String,

    /// Whether auto-type is enabled.
    pub enable_auto_type: Option<bool>,

    /// Whether searching is enabled.
    pub enable_searching: Option<bool>,

    /// Vector with entries that belong to this group.
    pub entries: Vec<Entry>,

    /// Whether this group expires.
    pub expires: bool,

    /// The date and time this group will expire if expires is true.
    pub expiry_time: DateTime<Utc>,

    /// Vector with subgroups of this group.
    pub groups: Vec<Group>,

    /// This group's icon.
    pub icon: Icon,

    /// Whether this group is expanded.
    pub is_expanded: bool,

    /// The date and time this group was last accessed.
    pub last_accessed: DateTime<Utc>,

    /// The date and time this group was last modified.
    pub last_modified: DateTime<Utc>,

    /// The identifier of the last top visible entry.
    pub last_top_visible_entry: EntryUuid,

    /// The date and time the location of this group was changed.
    pub location_changed: DateTime<Utc>,

    /// The name of this group.
    pub name: String,

    /// The notes of this group.
    pub notes: String,

    /// The usage count of this group.
    pub usage_count: i32,

    /// The identifier of this group.
    pub uuid: GroupUuid,

    /// The parent groups GroupUUID.
    pub parent: GroupUuid,
}

impl Group {
    /// Create a new group.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use kpdb::Group;
    ///
    /// let group = Group::new("Websites");
    /// ```
    pub fn new<S: Into<String>>(name: S) -> Group {
        let mut group = Group::default();
        group.name = name.into();
        group.uuid = GroupUuid::new_random();
        group
    }

    /// Add an entry to the current group.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use kpdb::{Entry, Group};
    ///
    /// let mut group = Group::new("group");
    /// let entry = Entry::new();
    ///
    /// assert_eq!(group.entries.len(), 0);
    /// group.add_entry(entry.clone());
    /// assert_eq!(group.entries.len(), 1);
    /// assert_eq!(group.entries[0], entry);
    /// ```
    pub fn add_entry(&mut self, entry: Entry) {
        self.entries.push(entry);
    }

    /// Add a sub group to the current group.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use kpdb::Group;
    ///
    /// let mut root = Group::new("root");
    /// let child = Group::new("child");
    ///
    /// assert_eq!(root.groups.len(), 0);
    /// root.add_group(child.clone());
    /// assert_eq!(root.groups.len(), 1);
    /// assert_eq!(root.groups[0], child);
    /// ```
    pub fn add_group(&mut self, group: Group) {
        self.groups.push(group);
    }

    /// Returns an iterator over the group and sub groups.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use kpdb::Group;
    ///
    /// let mut root = Group::new("root");
    /// let sub_1 = Group::new("sub_1");
    /// let sub_2 = Group::new("sub_2");
    /// root.add_group(sub_1.clone());
    /// root.add_group(sub_2.clone());
    ///
    /// let mut iterator = root.iter();
    /// assert_eq!(iterator.next(), Some(&root));
    /// assert_eq!(iterator.next(), Some(&sub_1));
    /// assert_eq!(iterator.next(), Some(&sub_2));
    /// assert_eq!(iterator.next(), None);
    /// ```
    pub fn iter(&self) -> Iter {
        Iter::new(self)
    }

    /// Returns an iterator that allows modifying each group.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use kpdb::Group;
    ///
    /// let mut root = Group::new("root");
    /// for group in root.iter_mut() {
    ///     group.name = String::from("name");
    /// }
    /// assert_eq!(root.name, "name");
    /// ```
    pub fn iter_mut(&mut self) -> IterMut {
        IterMut::new(self)
    }

    /// Remove an entry from the current group.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use kpdb::{Entry, Group};
    ///
    /// let mut group = Group::new("Sample");
    /// let entry = Entry::new();
    /// group.add_entry(entry.clone());
    /// assert_eq!(group.entries.len(), 1);
    /// assert_eq!(group.remove_entry(entry.uuid), Some(entry));
    /// assert_eq!(group.entries.len(), 0);
    /// ```
    pub fn remove_entry(&mut self, entry_uuid: EntryUuid) -> Option<Entry> {
        match self.entries.iter().position(|x| x.uuid == entry_uuid) {
            Some(x) => Some(self.entries.remove(x)),
            None => None,
        }
    }

    /// Remove a sub group from the current group.
    ///
    /// # Examples
    ///
    /// ```rust
    /// use kpdb::Group;
    ///
    /// let mut parent = Group::new("Parent");
    /// let child = Group::new("Child");
    /// parent.add_group(child.clone());
    /// assert_eq!(parent.groups.len(), 1);
    /// assert_eq!(parent.remove_group(child.uuid), Some(child));
    /// assert_eq!(parent.groups.len(), 0);
    /// ```
    pub fn remove_group(&mut self, group_uuid: GroupUuid) -> Option<Group> {
        match self.groups.iter().position(|x| x.uuid == group_uuid) {
            Some(x) => Some(self.groups.remove(x)),
            None => None,
        }
    }
}

impl Default for Group {
    fn default() -> Group {
        let now = Utc::now();
        Group {
            creation_time: now,
            custom_icon_uuid: None,
            def_auto_type_sequence: String::new(),
            enable_auto_type: None,
            enable_searching: None,
            entries: Vec::new(),
            expires: false,
            expiry_time: now,
            groups: Vec::new(),
            icon: Icon::Folder,
            is_expanded: true,
            last_accessed: now,
            last_modified: now,
            last_top_visible_entry: EntryUuid::nil(),
            location_changed: now,
            name: String::new(),
            notes: String::new(),
            usage_count: 0,
            uuid: GroupUuid::nil(),
            parent: GroupUuid::nil(),
        }
    }
}

impl Times for Group {
    fn creation_time(&self) -> DateTime<Utc> {
        self.creation_time
    }

    fn expires(&self) -> bool {
        self.expires
    }

    fn expiry_time(&self) -> DateTime<Utc> {
        self.expiry_time
    }

    fn last_accessed(&self) -> DateTime<Utc> {
        self.last_accessed
    }

    fn last_modified(&self) -> DateTime<Utc> {
        self.last_modified
    }

    fn location_changed(&self) -> DateTime<Utc> {
        self.location_changed
    }

    fn usage_count(&self) -> i32 {
        self.usage_count
    }

    fn set_creation_time(&mut self, val: DateTime<Utc>) {
        self.creation_time = val;
    }

    fn set_expires(&mut self, val: bool) {
        self.expires = val;
    }

    fn set_expiry_time(&mut self, val: DateTime<Utc>) {
        self.expiry_time = val;
    }

    fn set_last_accessed(&mut self, val: DateTime<Utc>) {
        self.last_accessed = val;
    }

    fn set_last_modified(&mut self, val: DateTime<Utc>) {
        self.last_modified = val;
    }

    fn set_location_changed(&mut self, val: DateTime<Utc>) {
        self.location_changed = val;
    }

    fn set_usage_count(&mut self, val: i32) {
        self.usage_count = val;
    }
}

/// Immutable group iterator.
pub struct Iter<'a> {
    curr: Option<&'a Group>,
    todo: VecDeque<&'a Group>,
}

impl<'a> Iter<'a> {
    fn new(group: &'a Group) -> Iter<'a> {
        let mut queue = VecDeque::new();
        queue.push_back(group);
        Iter {
            curr: None,
            todo: queue,
        }
    }
}

impl<'a> Iterator for Iter<'a> {
    type Item = &'a Group;

    fn next(&mut self) -> Option<&'a Group> {
        match self.curr.take() {
            Some(group) => {
                for sub in group.groups.iter() {
                    self.todo.push_back(sub);
                }
            }
            None => {}
        }
        self.curr = self.todo.pop_front();
        self.curr
    }
}

/// Mutable group iterator.
pub struct IterMut<'a> {
    curr: Option<&'a mut Group>,
    todo: VecDeque<&'a mut Group>,
}

impl<'a> IterMut<'a> {
    fn new(group: &'a mut Group) -> IterMut<'a> {
        let mut queue = VecDeque::new();
        queue.push_back(group);
        IterMut {
            curr: None,
            todo: queue,
        }
    }
}

impl<'a> Iterator for IterMut<'a> {
    type Item = &'a mut Group;

    fn next(&mut self) -> Option<&'a mut Group> {
        match self.curr.take() {
            Some(group) => {
                for sub in group.groups.iter_mut() {
                    self.todo.push_back(sub);
                }
            }
            None => {}
        }
        let curr = self.todo.pop_front();
        self.curr = unsafe { ptr::read(&curr) };
        curr
    }
}

#[cfg(test)]
mod tests {

    use super::*;
    use crate::types::EntryUuid;
    use crate::types::GroupUuid;
    use crate::types::Icon;
    use crate::utils::test::approx_equal_datetime;
    use chrono::Utc;

    #[test]
    fn test_new_returns_correct_instance() {
        let now = Utc::now();
        let name = "Root";
        let group = Group::new(name.clone());
        assert!(approx_equal_datetime(group.creation_time, now));
        assert_eq!(group.custom_icon_uuid, None);
        assert_eq!(group.def_auto_type_sequence, "");
        assert_eq!(group.enable_auto_type, None);
        assert_eq!(group.enable_searching, None);
        assert_eq!(group.entries, Vec::new());
        assert_eq!(group.expires, false);
        assert!(approx_equal_datetime(group.expiry_time, now));
        assert_eq!(group.groups, Vec::new());
        assert_eq!(group.icon, Icon::Folder);
        assert_eq!(group.is_expanded, true);
        assert!(approx_equal_datetime(group.last_accessed, now));
        assert!(approx_equal_datetime(group.last_modified, now));
        assert_eq!(group.last_top_visible_entry, EntryUuid::nil());
        assert!(approx_equal_datetime(group.location_changed, now));
        assert_eq!(group.name, name);
        assert_eq!(group.notes, "");
        assert_eq!(group.usage_count, 0);
        assert!(group.uuid != GroupUuid::nil());
    }

    #[test]
    fn test_add_entry_adds_entry() {
        let mut group = Group::new("group");
        let entry = Entry::new();

        assert_eq!(group.entries.len(), 0);
        group.add_entry(entry.clone());
        assert_eq!(group.entries.len(), 1);
        assert_eq!(group.entries[0], entry);
    }

    #[test]
    fn test_add_group_adds_group() {
        let mut root = Group::new("root");
        let child = Group::new("child");

        assert_eq!(root.groups.len(), 0);
        root.add_group(child.clone());
        assert_eq!(root.groups.len(), 1);
        assert_eq!(root.groups[0], child);
    }

    #[test]
    fn test_iter_returns_correct_iterator() {
        let mut root = Group::new("root");
        let mut sub_1 = Group::new("sub_1");
        let mut sub_2 = Group::new("sub_2");
        let sub_1_1 = Group::new("sub_1_1");
        let sub_1_2 = Group::new("sub_1_2");
        let sub_2_1 = Group::new("sub_2_1");
        let sub_2_2 = Group::new("sub_2_2");
        sub_1.add_group(sub_1_1.clone());
        sub_1.add_group(sub_1_2.clone());
        sub_2.add_group(sub_2_1.clone());
        sub_2.add_group(sub_2_2.clone());
        root.add_group(sub_1.clone());
        root.add_group(sub_2.clone());

        let mut iterator = root.iter();
        assert_eq!(iterator.next(), Some(&root));
        assert_eq!(iterator.next(), Some(&sub_1));
        assert_eq!(iterator.next(), Some(&sub_2));
        assert_eq!(iterator.next(), Some(&sub_1_1));
        assert_eq!(iterator.next(), Some(&sub_1_2));
        assert_eq!(iterator.next(), Some(&sub_2_1));
        assert_eq!(iterator.next(), Some(&sub_2_2));
        assert_eq!(iterator.next(), None);
    }

    #[test]
    fn test_iter_mut_returns_correct_iterator() {
        let mut root = Group::new("root");
        let sub_1 = Group::new("sub_1");
        let sub_2 = Group::new("sub_2");
        root.add_group(sub_1);
        root.add_group(sub_2);

        let mut num = 0;
        for group in root.iter_mut() {
            group.name = num.to_string();
            num += 1;
        }

        let mut num = 0;
        for group in root.iter() {
            assert_eq!(group.name, num.to_string());
            num += 1;
        }
    }

    #[test]
    fn test_remove_entry_removes_entry() {
        let mut group = Group::new("Sample");
        let entry = Entry::new();

        assert_eq!(group.entries.len(), 0);
        group.add_entry(entry.clone());
        assert_eq!(group.entries.len(), 1);
        assert_eq!(group.remove_entry(entry.uuid), Some(entry.clone()));
        assert_eq!(group.entries.len(), 0);
        assert_eq!(group.remove_entry(entry.uuid), None);
    }

    #[test]
    fn test_remove_group_removes_group() {
        let mut parent = Group::new("Parent");
        let child = Group::new("Child");

        assert_eq!(parent.groups.len(), 0);
        parent.add_group(child.clone());
        assert_eq!(parent.groups.len(), 1);
        assert_eq!(parent.remove_group(child.uuid), Some(child.clone()));
        assert_eq!(parent.groups.len(), 0);
        assert_eq!(parent.remove_group(child.uuid), None);
    }

    #[test]
    fn test_default_returns_correct_instance() {
        let now = Utc::now();
        let group = Group::default();
        assert!(approx_equal_datetime(group.creation_time, now));
        assert_eq!(group.custom_icon_uuid, None);
        assert_eq!(group.def_auto_type_sequence, "");
        assert_eq!(group.enable_auto_type, None);
        assert_eq!(group.enable_searching, None);
        assert_eq!(group.entries, Vec::new());
        assert_eq!(group.expires, false);
        assert!(approx_equal_datetime(group.expiry_time, now));
        assert_eq!(group.groups, Vec::new());
        assert_eq!(group.icon, Icon::Folder);
        assert_eq!(group.is_expanded, true);
        assert!(approx_equal_datetime(group.last_accessed, now));
        assert!(approx_equal_datetime(group.last_modified, now));
        assert_eq!(group.last_top_visible_entry, EntryUuid::nil());
        assert!(approx_equal_datetime(group.location_changed, now));
        assert_eq!(group.name, "");
        assert_eq!(group.notes, "");
        assert_eq!(group.usage_count, 0);
        assert_eq!(group.uuid, GroupUuid::nil());
    }
}