1
// This file is part of hnefatafl-copenhagen.
2
//
3
// hnefatafl-copenhagen is free software: you can redistribute it and/or modify
4
// it under the terms of the GNU Affero General Public License as published by
5
// the Free Software Foundation, either version 3 of the License, or
6
// (at your option) any later version.
7
//
8
// hnefatafl-copenhagen is distributed in the hope that it will be useful,
9
// but WITHOUT ANY WARRANTY; without even the implied warranty of
10
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
11
// GNU Affero General Public License for more details.
12
//
13
// You should have received a copy of the GNU Affero General Public License
14
// along with this program.  If not, see <https://www.gnu.org/licenses/>.
15
//
16
// SPDX-License-Identifier: AGPL-3.0-or-later
17
// SPDX-FileCopyrightText: 2025 David Campbell <david@hnefatafl.org>
18

            
19
use std::{fmt, str::FromStr};
20

            
21
use serde::{Deserialize, Serialize};
22

            
23
#[derive(
24
    Clone, Copy, Debug, Default, Deserialize, Eq, Hash, Ord, PartialEq, PartialOrd, Serialize,
25
)]
26
pub enum Role {
27
    #[default]
28
    Attacker,
29
    Defender,
30
    Roleless,
31
}
32

            
33
impl Role {
34
    #[must_use]
35
20784632
    pub fn opposite(&self) -> Self {
36
20784632
        match self {
37
14247803
            Self::Attacker => Self::Defender,
38
6536829
            Self::Defender => Self::Attacker,
39
            Self::Roleless => Self::Roleless,
40
        }
41
20784632
    }
42
}
43

            
44
impl fmt::Display for Role {
45
24
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
46
24
        match self {
47
12
            Role::Attacker => write!(f, "attacker"),
48
12
            Role::Defender => write!(f, "defender"),
49
            Role::Roleless => write!(f, "roleless"),
50
        }
51
24
    }
52
}
53

            
54
impl FromStr for Role {
55
    type Err = anyhow::Error;
56

            
57
438
    fn from_str(string: &str) -> anyhow::Result<Self> {
58
438
        let string = string.to_lowercase();
59

            
60
438
        match string.as_str() {
61
438
            "a" | "attacker" => Ok(Self::Attacker),
62
240
            "d" | "defender" => Ok(Self::Defender),
63
            _ => Err(anyhow::Error::msg(format!(
64
                "Error trying to convert '{string}' to a Role!"
65
            ))),
66
        }
67
438
    }
68
}