tintify 1.0.1

A lightweight library for terminal colors and styling
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

use std::env;
use std::ffi::OsString;
use std::process::Command;
use std::process::Output;
use std::str;
use std::str::Split;

fn main() -> () {
    println!("cargo:rustc-check-cfg=cfg(doc_cfg)");
    println!("cargo:rustc-check-cfg=cfg(const_mut_refs)");

    if let Some(compiler) = rustc_version() {
        if compiler.is_newer_than_minor(83) {
            println!("cargo:rustc-cfg=const_mut_refs");
        }
    }
}

struct Compiler {
    minor: u32,
    channel: ReleaseChannel
}

impl Compiler {
    fn is_newer_than_minor(&self, minor: u32) -> bool {
        match self.channel {
            ReleaseChannel::Stable | ReleaseChannel::Beta => self.minor >= minor,

            ReleaseChannel::Nightly                       => {
                self.minor > minor
            }
        }
    }
}

#[derive(Clone, Copy, Debug, Eq, PartialEq)]
enum ReleaseChannel {
    Stable,
    Beta,
    Nightly
}

fn rustc_version() -> Option<Compiler> {
    let rustc: OsString = env::var_os("RUSTC")?;
    let output: Output = Command::new(rustc).arg("--version").output().ok()?;
    let version: &str = str::from_utf8(&output.stdout).ok()?;
    let mut pieces: Split<char> = version.split('.');

    if pieces.next() != Some("rustc 1") {
        return None;
    }

    let minor: u32 = pieces.next()?.parse().ok()?;

    let channel: ReleaseChannel = if version.contains("nightly") {
        ReleaseChannel::Nightly
    } else if version.contains("beta") {
        ReleaseChannel::Beta
    } else {
        ReleaseChannel::Stable
    };

    Some(Compiler {
        minor,
        channel
    })
}