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
113
114
115
116
117
118

use strum_macros::{Display, EnumString};
use num_enum::{TryFromPrimitive, IntoPrimitive};
use serde_repr::{Serialize_repr, Deserialize_repr};

#[derive(Serialize_repr, Deserialize_repr, Debug, Hash, Eq, PartialEq, Display, EnumString, TryFromPrimitive, IntoPrimitive, Clone)]
#[repr(u8)]
pub enum Quality {
    Normal = 0,
    Genuine = 1,
    #[strum(serialize = "rarity2")]
    Rarity2 = 2,
    Vintage = 3,
    #[strum(serialize = "rarity3")]
    Rarity3 = 4,
    Unusual = 5,
    Unique = 6,
    Community = 7,
    Valve = 8,
    #[strum(serialize = "Self-Made")]
    SelfMade = 9,
    Customized = 10,
    Strange = 11,
    Completed = 12,
    Haunted = 13,
    #[strum(serialize = "Collector's")]
    Collectors = 14,
    #[strum(serialize = "Decorated Weapon")]
    DecoratedWeapon = 15,
}

impl Quality {
    
    pub fn from_color(color: u32) -> Option<Self> {
        match color {
            0xB2B2B2 => Some(Quality::Normal),
            0x4D7455 => Some(Quality::Genuine),
            0x476291 => Some(Quality::Vintage),
            0x8650AC => Some(Quality::Unusual),
            0xFFD700 => Some(Quality::Unique),
            0x56083F => Some(Quality::Valve),
            0x70B04A => Some(Quality::SelfMade),
            0xCF6A32 => Some(Quality::Strange),
            0x38F3AB => Some(Quality::Haunted),
            0xAA0000 => Some(Quality::Collectors),
            0xFAFAFA => Some(Quality::DecoratedWeapon),
            _ => None,
        }
    }
    
    pub fn from_color_str(color: &str) -> Option<Self> {
        match color {
            "B2B2B2" => Some(Quality::Normal),
            "4D7455" => Some(Quality::Genuine),
            "476291" => Some(Quality::Vintage),
            "8650AC" => Some(Quality::Unusual),
            "FFD700" => Some(Quality::Unique),
            "56083F" => Some(Quality::Valve),
            "70B04A" => Some(Quality::SelfMade),
            "CF6A32" => Some(Quality::Strange),
            "38F3AB" => Some(Quality::Haunted),
            "AA0000" => Some(Quality::Collectors),
            "FAFAFA" => Some(Quality::DecoratedWeapon),
            _ => None,
        }
    }
    
    pub fn color(&self) -> u32 {
        match self {
            Quality::Normal => 0xB2B2B2,
            Quality::Genuine => 0x4D7455,
            Quality::Rarity2 => 0xFFFFFF,
            Quality::Vintage => 0x476291,
            Quality::Rarity3 => 0xFFFFFF,
            Quality::Unusual => 0x8650AC,
            Quality::Unique => 0xFFD700,
            Quality::Community => 0x70B04A,
            Quality::Valve => 0x56083F,
            Quality::SelfMade => 0x70B04A,
            Quality::Customized => 0xFFFFFF,
            Quality::Strange => 0xCF6A32,
            Quality::Completed => 0xFFFFFF,
            Quality::Haunted => 0x38F3AB,
            Quality::Collectors => 0xAA0000,
            Quality::DecoratedWeapon => 0xFAFAFA,
        }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use std::str::FromStr;
    
    #[test]
    fn from_color() {
        assert_eq!(Some(Quality::Strange), Quality::from_color(0xCF6A32));
    }
    
    #[test]
    fn to_color() {
        assert_eq!(Quality::Strange.color(), 0xCF6A32);
    }

    #[test]
    fn converts_to_primitive() {
        assert_eq!(11 as u8, Quality::Strange.into());
    }
    
    #[test]
    fn converts_string_to_quality() {
        assert_eq!(Quality::DecoratedWeapon, Quality::from_str("Decorated Weapon").unwrap());
    }
    
    #[test]
    fn displays_as_string() {
        assert_eq!("Collector's", &format!("{}", Quality::Collectors));
    }
}