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

use std::env::consts;
use std::fmt;

#[cfg(feature = "serialization")]
use serde::{Deserialize, Serialize};

/// Enum describing processor architectures.
#[derive(Clone, Debug, PartialEq)]
#[cfg_attr(feature = "serialization", derive(Deserialize, Serialize))]
#[cfg_attr(feature = "serialization", serde(rename_all = "lowercase"))]
pub enum Architecture {
    // TODO: Support more architectures.
    Amd64,
    ARM,

    Unknown,
}

impl Architecture {
    pub fn detect() -> Architecture {
        match consts::ARCH {
            "x86_64" => Architecture::Amd64,
            "arm" => Architecture::ARM,
            _ => Architecture::Unknown,
        }
    }

    pub fn value(&self) -> Vec<String> {
        match self {
            Architecture::Amd64 => vec![
                String::from("amd64"),
                String::from("x64"),
                String::from("x86_64"),
            ],
            Architecture::ARM => vec![String::from("arm")],
            _ => vec![String::from("unknown")],
        }
    }
}

impl fmt::Display for Architecture {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{}", self.value().first().unwrap())
    }
}

impl From<&str> for Architecture {
    fn from(v: &str) -> Architecture {
        match v {
            "amd64" => Architecture::Amd64,
            _ => Architecture::Unknown,
        }
    }
}