1
  2
  3
  4
  5
  6
  7
  8
  9
 10
 11
 12
 13
 14
 15
 16
 17
 18
 19
 20
 21
 22
 23
 24
 25
 26
 27
 28
 29
 30
 31
 32
 33
 34
 35
 36
 37
 38
 39
 40
 41
 42
 43
 44
 45
 46
 47
 48
 49
 50
 51
 52
 53
 54
 55
 56
 57
 58
 59
 60
 61
 62
 63
 64
 65
 66
 67
 68
 69
 70
 71
 72
 73
 74
 75
 76
 77
 78
 79
 80
 81
 82
 83
 84
 85
 86
 87
 88
 89
 90
 91
 92
 93
 94
 95
 96
 97
 98
 99
100
101
102
103
104
105
106
107
108
109
110
111
112

use crate::intersect::{
    intersect_from_all, intersect_from_all_except_some, intersect_from_none, intersect_from_some,
};
use crate::remove::{
    remove_from_all, remove_from_all_except_some, remove_from_none, remove_from_some,
};
use crate::utils::clean_vec;
use serde::{Deserialize, Serialize};
use std::fmt::Debug;

mod intersect;
mod remove;
mod utils;

#[derive(Debug, Ord, PartialOrd, Eq, PartialEq, Hash, Serialize, Deserialize)]
pub enum KeySet<T>
where
    T: Ord + Debug + Clone,
{
    None,
    Some(Vec<T>),
    AllExceptSome(Vec<T>),
    All,
}

impl<T> KeySet<T>
where
    T: Ord + Debug + Clone,
{
    pub fn for_some(elements: &Vec<T>) -> KeySet<T> {
        let vec = clean_vec(elements.clone());
        utils::make_some_with(vec)
    }

    pub fn for_all_except_some(elements: &Vec<T>) -> KeySet<T> {
        let vec = clean_vec(elements.clone());
        utils::make_all_except_some_with(vec)
    }

    pub fn invert(&self) -> Self {
        match self {
            KeySet::All => KeySet::None,
            KeySet::None => KeySet::All,
            KeySet::Some(e) => KeySet::AllExceptSome(e.clone()),
            KeySet::AllExceptSome(e) => KeySet::Some(e.clone()),
        }
    }

    pub fn contains(&self, element: &T) -> bool {
        match self {
            KeySet::All => true,
            KeySet::None => false,
            KeySet::Some(e) => e.contains(&element),
            KeySet::AllExceptSome(e) => !e.contains(&element),
        }
    }

    pub fn remove(&self, other: &KeySet<T>) -> KeySet<T> {
        match self {
            KeySet::All => remove_from_all(other),
            KeySet::None => remove_from_none(other),
            KeySet::Some(e) => remove_from_some(e, other),
            KeySet::AllExceptSome(e) => remove_from_all_except_some(e, other),
        }
    }

    pub fn intersect(&self, other: &KeySet<T>) -> KeySet<T> {
        match self {
            KeySet::All => intersect_from_all(other),
            KeySet::None => intersect_from_none(other),
            KeySet::Some(e) => intersect_from_some(e, other),
            KeySet::AllExceptSome(e) => intersect_from_all_except_some(e, other),
        }
    }
}

impl<T> std::fmt::Display for KeySet<T>
where
    T: Ord + Debug + Clone,
{
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        match self {
            KeySet::All => write!(f, "KeySet::All"),
            KeySet::None => write!(f, "KeySet::None"),
            KeySet::Some(e) => write!(f, "KeySet::Some({:?})", e),
            KeySet::AllExceptSome(e) => write!(f, "KeySet::AllExceptSome({:?})", e),
        }
    }
}

impl<T> Clone for KeySet<T>
where
    T: Ord + Debug + Clone,
{
    fn clone(&self) -> Self {
        match self {
            KeySet::All => KeySet::All,
            KeySet::None => KeySet::None,
            KeySet::Some(e) => KeySet::Some(e.clone()),
            KeySet::AllExceptSome(e) => KeySet::AllExceptSome(e.clone()),
        }
    }
}

impl<T> Default for KeySet<T>
where
    T: Ord + Debug + Clone,
{
    fn default() -> Self {
        Self::All
    }
}