ci_env 0.4.1

Detect CI information from the environment.
Documentation 
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

use crate::api::{opt_var, var, CiEnvironment, CiOutput, CiProvider};
use std::{fs, path::PathBuf};

pub const TEAMCITY_OUTPUT: CiOutput = CiOutput {
    close_log_group: "##teamcity[blockClosed name='{name}']",
    open_log_group: "##teamcity[blockOpened name='{name}']",
};

// https://www.jetbrains.com/help/teamcity/predefined-build-parameters.html
pub fn create_environment() -> CiEnvironment {
    let mut env = CiEnvironment {
        base_branch: None,
        base_revision: None,
        branch: "".into(),
        env_prefix: None,
        head_revision: None,
        id: var("BUILD_NUMBER"),
        provider: CiProvider::TeamCity,
        request_id: None,
        request_url: None,
        revision: var("BUILD_VCS_NUMBER"),
        url: None,
    };

    if let Some(file) = opt_var("TEAMCITY_BUILD_PROPERTIES_FILE") {
        parse_build_file(PathBuf::from(file), &mut env);
    }

    env
}

fn parse_build_file(path: PathBuf, env: &mut CiEnvironment) {
    if !path.exists() {
        return;
    }

    if let Ok(contents) = fs::read_to_string(path) {
        for line in contents.lines() {
            if let Some(value) = line.strip_prefix("teamcity.configuration.properties.file=") {
                parse_config_file(PathBuf::from(value), env);
            }
        }
    }
}

fn parse_config_file(path: PathBuf, env: &mut CiEnvironment) {
    if !path.exists() {
        return;
    }

    if let Ok(contents) = fs::read_to_string(path) {
        for line in contents.lines() {
            if let Some(value) = line.strip_prefix("teamcity.build.branch=") {
                env.branch = value.to_owned();
            }
        }
    }
}