use std::convert::TryFrom;
use std::collections::HashSet;
use std::iter::FromIterator;
use std::hash::{Hash, Hasher};
use std::cmp::{Ord, Ordering, Eq};

use std::fs::remove_dir_all;

use serde::{Serialize, Deserialize};

use json_surf::prelude::*;


// Main struct
#[derive(Serialize, Debug, Deserialize, PartialEq, PartialOrd, Clone)]
struct UserInfo {
    first: String,
    last: String,
    age: u8,
}

impl UserInfo {
    pub fn new(first: String, last: String, age: u8) -> Self {
        Self {
            first,
            last,
            age,
        }
    }
}

impl Default for UserInfo {
    fn default() -> Self {
        let first = "".to_string();
        let last = "".to_string();
        let age = 0u8;
        UserInfo::new(first, last, age)
    }
}


fn main() {
    // Specify home location for indexes
    let home = ".store".to_string();
    // Specify index name
    let index_name = "test_user_info".to_string();

    // Prepare builder
    let mut builder = SurferBuilder::default();
    builder.set_home(&home);

    let data = UserInfo::default();
    builder.add_struct(index_name.clone(), &data);

    // Prepare Surfer
    let mut surfer = Surfer::try_from(builder).unwrap();

    // Prepare data to insert & search

    // User 1: John Doe
    let first = "John".to_string();
    let last = "Doe".to_string();
    let age = 20u8;
    let john_doe = UserInfo::new(first, last, age);

    // User 2: Jane Doe
    let first = "Jane".to_string();
    let last = "Doe".to_string();
    let age = 18u8;
    let jane_doe = UserInfo::new(first, last, age);

    // User 3: Jonny Doe
    let first = "Jonny".to_string();
    let last = "Doe".to_string();
    let age = 10u8;
    let jonny_doe = UserInfo::new(first, last, age);

    // User 4: Jinny Doe
    let first = "Jinny".to_string();
    let last = "Doe".to_string();
    let age = 10u8;
    let jinny_doe = UserInfo::new(first, last, age);

    // Writing structs

    // Option 1: One struct at a time
    let _ = surfer.insert_struct(&index_name, &john_doe).unwrap();
    let _ = surfer.insert_struct(&index_name, &jane_doe).unwrap();

    // Option 2: Write all structs together
    let users = vec![jonny_doe.clone(), jinny_doe.clone()];
    let _ = surfer.insert_structs(&index_name, &users).unwrap();

    block_thread(1);

    // Reading structs

    // Option 1: Full text search
    let expected = vec![john_doe.clone()];
    let computed = surfer.read_all_structs::<UserInfo>(&index_name, "John").unwrap().unwrap();
    assert_eq!(expected, computed);

    let mut expected = vec![john_doe.clone(), jane_doe.clone(), jonny_doe.clone(), jinny_doe.clone()];
    expected.sort();
    let mut computed = surfer.read_all_structs::<UserInfo>(&index_name, "doe").unwrap().unwrap();
    computed.sort();
    assert_eq!(expected, computed);

    // Option 2: Term search
    let mut expected = vec![jonny_doe.clone(), jinny_doe.clone()];
    expected.sort();
    let mut computed = surfer.read_all_structs_by_field::<UserInfo>(&index_name, "age", "10").unwrap().unwrap();
    computed.sort();
    assert_eq!(expected, computed);

    // Delete structs

    // Option 1: Delete based on all text fields
    // Before delete
    let before = surfer.read_all_structs::<UserInfo>(&index_name, "doe").unwrap().unwrap();
    let before: HashSet<UserInfo> = HashSet::from_iter(before.into_iter());

    // Delete any occurrence of John (Actual call to delete)
    surfer.delete_structs(&index_name, "john").unwrap();

    // After delete
    let after = surfer.read_all_structs::<UserInfo>(&index_name, "doe").unwrap().unwrap();
    let after: HashSet<UserInfo> = HashSet::from_iter(after.into_iter());
    // Check difference
    let computed: Vec<UserInfo> = before.difference(&after).map(|e| e.clone()).collect();
    // Only John should be deleted
    let expected = vec![john_doe];
    assert_eq!(expected, computed);

    // Option 2: Delete based on a specific field
    // Before delete
    let before = surfer.read_all_structs_by_field::<UserInfo>(&index_name, "age", "10").unwrap().unwrap();
    let before: HashSet<UserInfo> = HashSet::from_iter(before.into_iter());

    // Delete any occurrence where age = 10 (Actual call to delete)
    surfer.delete_structs_by_field(&index_name, "age", "10").unwrap();

    // After delete
    let after = surfer.read_all_structs_by_field::<UserInfo>(&index_name, "age", "10").unwrap().unwrap();
    let after: HashSet<UserInfo> = HashSet::from_iter(after.into_iter());
    // Check difference
    let mut computed: Vec<UserInfo> = before.difference(&after).map(|e| e.clone()).collect();
    computed.sort();
    // Both Jonny & Jinny should be deleted
    let mut expected = vec![jonny_doe, jinny_doe];
    expected.sort();
    assert_eq!(expected, computed);


    // Clean-up
    let path = surfer.which_index(&index_name).unwrap();
    let _ = remove_dir_all(&path);
    let _ = remove_dir_all(&home);
}

/// Convenience method for sorting & likely not required in user code
impl Ord for UserInfo {
    fn cmp(&self, other: &Self) -> Ordering {
        if self.first == other.first && self.last == other.last {
            return Ordering::Equal;
        };
        if self.first == other.first {
            if self.last > other.last {
                Ordering::Greater
            } else {
                Ordering::Less
            }
        } else {
            if self.first > other.first {
                Ordering::Greater
            } else {
                Ordering::Less
            }
        }
    }
}

/// Convenience method for sorting & likely not required in user code
impl Eq for UserInfo {}

/// Convenience method for sorting & likely not required in user code
impl Hash for UserInfo {
    fn hash<H: Hasher>(&self, state: &mut H) {
        for i in self.first.as_bytes() {
            state.write_u8(*i);
        }
        for i in self.last.as_bytes() {
            state.write_u8(*i);
        }
        state.write_u8(self.age);
        state.finish();
    }
}