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

/* data_category.rs
 *
 * Copyright 2021-2022 Aman Kumar <amankrx.com>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

use gtk::glib;

/// All supported [DataCategory]s are listed in this enum.
#[derive(
    Debug,
    num_derive::FromPrimitive,
    num_derive::ToPrimitive,
    Clone,
    Copy,
    Hash,
    PartialEq,
    Eq,
    strum::EnumString,
    strum::IntoStaticStr,
    strum::AsRefStr,
)]
#[strum(serialize_all = "snake_case")]
pub enum DataCategory {
    Activity,
    Weight,
}

impl Default for DataCategory {
    fn default() -> Self {
        Self::Activity
    }
}

impl glib::ToValue for DataCategory {
    fn to_value(&self) -> glib::Value {
        self.as_ref().to_value()
    }

    fn value_type(&self) -> glib::Type {
        <String as glib::StaticType>::static_type()
    }
}

#[cfg(test)]
mod test {
    use super::DataCategory;
    use std::str::FromStr;

    #[test]
    fn deserialize_data_category() {
        assert_eq!(
            DataCategory::from_str("activity"),
            Ok(DataCategory::Activity)
        );
        assert_eq!(DataCategory::from_str("weight"), Ok(DataCategory::Weight),);
        assert!(DataCategory::from_str("unknown").is_err());
    }

    #[test]
    fn serialize_data_category() {
        let a: &str = DataCategory::Activity.into();
        assert_eq!(a, "activity");
        let a: &str = DataCategory::Weight.into();
        assert_eq!(a, "weight");
    }
}