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

use once_cell::sync::Lazy;

use crate::PColor;

/// Options to configure behaviours.
#[derive(Clone, Debug, Eq, PartialEq)]
pub struct Options {
    /// Sorting mode.
    pub mode: Mode,
}

impl Default for Options {
    fn default() -> Self {
        Self {
            mode: Default::default(),
        }
    }
}

/// Sorting modes.
#[derive(Clone, Debug, Eq, PartialEq)]
pub enum Mode {
    /// Black mode with a threshold color.
    Black(PColor),

    /// Brightness mode with a threshold value `0` to `255`.
    Brightness(u8),

    /// White mode with a threshold color.
    White(PColor),
}

impl Default for Mode {
    fn default() -> Self {
        Self::brightness()
    }
}

pub static DEFAULT_BLACK: Lazy<PColor> = Lazy::new(|| PColor::new(11, 220, 0));
pub static DEFAULT_BRIGHTNESS: u8 = 60;
pub static DEFAULT_WHITE: Lazy<PColor> = Lazy::new(|| PColor::new(57, 162, 192));

impl Mode {
    /// Black mode with a default threshold.
    pub fn black() -> Self {
        Self::Black(DEFAULT_BLACK.clone())
    }

    /// Brightness mode with a default threshold.
    pub fn brightness() -> Self {
        Self::Brightness(DEFAULT_BRIGHTNESS)
    }

    /// White mode with a default threshold.
    pub fn white() -> Self {
        Self::White(DEFAULT_WHITE.clone())
    }
}

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

    #[test]
    fn test_options_default() {
        let expected = Options {
            mode: Mode::Brightness(60),
        };
        assert_eq!(expected, Options::default());
    }

    #[test]
    fn test_mode_default() {
        let expected = Mode::Brightness(60);
        assert_eq!(expected, Mode::default());
    }

    #[test]
    fn test_mode_black() {
        let color = PColor::new(11, 220, 0);
        let expected = Mode::Black(color);
        assert_eq!(expected, Mode::black());
    }

    #[test]
    fn test_mode_brightness() {
        let value = 60;
        let expected = Mode::Brightness(value);
        assert_eq!(expected, Mode::brightness());
    }

    #[test]
    fn test_mode_white() {
        let color = PColor::new(57, 162, 192);
        let expected = Mode::White(color);
        assert_eq!(expected, Mode::white());
    }
}