aboutsummaryrefslogtreecommitdiff
path: root/src/playerclass.rs
blob: 247e8b1752a1f50303068744dc6cdeeddb27108f (plain)
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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
use std::{fmt, str::FromStr};

use evtclib::{EliteSpec, Profession};

use thiserror::Error;

/// An enum containing either a profession or an elite spec.
///
/// This enum provides us with a variety of things:
///
/// Game mechanic wise, a Dragonhunter is also a Guardian, because Dragonhunter is only the elite
/// specialization. However, when outputting that to the user, we usually only write Dragonhunter,
/// and not Guardian, as that is implied. Same when filtering, when we filter for Guardian, we
/// probably don't want any Dragonhunters.
///
/// So this enum unifies the handling between core specs and elite specs, and provides them with a
/// convenient [`Display`][Display] implementation as well.
#[derive(Debug, Clone, Copy, Hash, PartialEq, Eq)]
pub enum PlayerClass {
    Profession(Profession),
    EliteSpec(EliteSpec),
}

#[derive(Debug, Clone, Hash, PartialEq, Eq, Error)]
#[error("could not parse the class: {0}")]
pub struct ParsePlayerClassError(String);

impl From<Profession> for PlayerClass {
    fn from(p: Profession) -> Self {
        PlayerClass::Profession(p)
    }
}

impl From<EliteSpec> for PlayerClass {
    fn from(e: EliteSpec) -> Self {
        PlayerClass::EliteSpec(e)
    }
}

impl FromStr for PlayerClass {
    type Err = ParsePlayerClassError;

    fn from_str(s: &str) -> Result<Self, Self::Err> {
        let err = ParsePlayerClassError(s.to_owned());
        Profession::from_str(s)
            .map(Into::into)
            .map_err(|_| err.clone())
            .or_else(|_| EliteSpec::from_str(s).map(Into::into).map_err(|_| err))
    }
}

impl From<(Profession, Option<EliteSpec>)> for PlayerClass {
    fn from((profession, elite_spec): (Profession, Option<EliteSpec>)) -> Self {
        if let Some(spec) = elite_spec {
            spec.into()
        } else {
            profession.into()
        }
    }
}

impl From<&evtclib::Player> for PlayerClass {
    fn from(player: &evtclib::Player) -> Self {
        (player.profession(), player.elite()).into()
    }
}

impl fmt::Display for PlayerClass {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        use EliteSpec::*;
        use Profession::*;

        let name = match *self {
            PlayerClass::EliteSpec(elite) => match elite {
                Dragonhunter => "Dragonhunter",
                Firebrand => "Firebrand",
                Berserker => "Berserker",
                Spellbreaker => "Spellbreaker",
                Herald => "Herald",
                Renegade => "Renegade",
                Scrapper => "Scrapper",
                Holosmith => "Holosmith",
                Druid => "Druid",
                Soulbeast => "Soulbeast",
                Daredevil => "Daredevil",
                Deadeye => "Deadeye",
                Tempest => "Tempest",
                Weaver => "Weaver",
                Chronomancer => "Chronomancer",
                Mirage => "Mirage",
                Reaper => "Reaper",
                Scourge => "Scourge",
            },
            PlayerClass::Profession(prof) => match prof {
                Guardian => "Guardian",
                Warrior => "Warrior",
                Revenant => "Revenant",
                Engineer => "Engineer",
                Ranger => "Ranger",
                Thief => "Thief",
                Elementalist => "Elementalist",
                Mesmer => "Mesmer",
                Necromancer => "Necromancer",
            },
        };
        write!(f, "{}", name)
    }
}

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

    #[test]
    fn test_player_class_from() {
        let tests: &[(Profession, Option<EliteSpec>, PlayerClass)] = &[
            (
                Profession::Guardian,
                None,
                PlayerClass::Profession(Profession::Guardian),
            ),
            (
                Profession::Guardian,
                Some(EliteSpec::Dragonhunter),
                PlayerClass::EliteSpec(EliteSpec::Dragonhunter),
            ),
        ];

        for (prof, elite_spec, expected) in tests {
            assert_eq!(PlayerClass::from((*prof, *elite_spec)), *expected);
        }
    }

    #[test]
    fn test_parse_player_class() {
        let tests: &[(&'static str, PlayerClass)] = &[
            ("guardian", Profession::Guardian.into()),
            ("dragonhunter", EliteSpec::Dragonhunter.into()),
            ("warrior", Profession::Warrior.into()),
            ("scourge", EliteSpec::Scourge.into()),
        ];

        for (input, expected) in tests {
            assert_eq!(input.parse(), Ok(*expected));
        }
    }
}