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

use std::{
    borrow::Cow,
    env::{self, VarError},
    fmt,
};

/// `target_family`: A a more generic description of a target, such as the family of
/// the operating systems or architectures that the target generally falls into.
#[derive(Clone, Debug, Eq, Hash, PartialEq, PartialOrd, Ord)]
#[non_exhaustive]
pub enum Family<'a> {
    /// `unix`: Unix based operating systems
    Unix,

    /// `windows`: Microsoft's Windows operating system
    Windows,

    /// `wasm`: WebAssembly
    Wasm,

    /// Operating system family we don't know about
    Other(Cow<'a, str>),
}

impl<'a> Family<'a> {
    /// String representing this target family which matches `#[cfg(target_family)]`
    pub fn as_str(&self) -> &str {
        match self {
            Family::Unix => "unix",
            Family::Windows => "windows",
            Family::Wasm => "wasm",
            Family::Other(s) => s,
        }
    }

    /// Create a new [`Arch`] from the given string.
    fn from_str(os_name: impl Into<Cow<'a, str>>) -> Self {
        let os_name = os_name.into();
        match os_name.as_ref() {
            "unix" => Family::Unix,
            "windows" => Family::Windows,
            "wasm" => Family::Wasm,
            _ => Family::Other(os_name),
        }
    }

    pub fn target() -> Result<Self, VarError> {
        env::var("CARGO_CFG_TARGET_FAMILY").map(Self::from_str)
    }
}

impl<'a> fmt::Display for Family<'a> {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.write_str(self.as_str())
    }
}