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
//! The `TriState` crate provides a three-valued type equivalent to `Option<bool>`.
//! It also provides several convenience methods for testing tri-state values.

use std::convert::From;

/// A three-valued type equivalent to `Option<bool>`.
#[derive(Clone, Copy, Eq, PartialEq, Ord, PartialOrd)]
pub enum TriState {
    /// The tri-state value signifying definitive truth.
    Yes,

    /// The tri-state value signifying definitive falsity.
    No,

    /// The tri-state value signifying an unknown truth value.
    Unknown
}

impl TriState {
    /// Is `self` equal to `Yes`?
    pub fn yes(self) -> bool { self == TriState::Yes }

    /// Is `self` equal to `No`?
    pub fn no(self) -> bool { self == TriState::No }

    /// Is `self` equal to `Unknown`?
    pub fn unknown(self) -> bool { self == TriState::Unknown }

    /// A mnemonic for `yes()`.
    pub fn definitely(self) -> bool { self.yes() }

    /// A mnemonic for `no()`.
    pub fn definitely_not(self) -> bool { self.no() }

    /// Converts `self` to an `Option<bool>`. Equivalent to `Option::<bool>::from(self)`.
    pub fn to_bool(self) -> Option<bool> {
        match self {
            TriState::Yes => Some(true),
            TriState::No => Some(false),
            TriState::Unknown => None
        }
    }
}

impl From<TriState> for Option<bool> {
    fn from(t: TriState) -> Option<bool> { t.to_bool() }
}

impl From<Option<bool>> for TriState {
    fn from(b: Option<bool>) -> TriState {
        match b {
            Some(true) => TriState::Yes,
            Some(false) => TriState::No,
            None => TriState::Unknown
        }
    }
}

impl From<bool> for TriState {
    fn from(b: bool) -> TriState {
        match b {
            true => TriState::Yes,
            false => TriState::No
        }
    }
}