embedded-td 0.1.2

Running tendermint as rust crate
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
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

use std::{env, fs, path::Path, process::Command};

use flate2::read::GzDecoder;
use tar::Archive;

fn check_run(p: &Path, version: Option<&str>) -> bool {
    let output = Command::new(p).arg("version").output().unwrap();

    if let Some(v) = version {
        let ver = String::from_utf8(output.stdout).unwrap();
        ver.trim() == v.trim()
    } else {
        output.status.success()
    }
}

// None: No this dir
// Some(true): tendermint right
// Some(false): tendermint error
fn check_has_tendermint(dir: &str, version: Option<&str>) -> Option<bool> {
    let out_path = Path::new(dir).join("tendermint");

    if out_path.exists() {
        Some(check_run(&out_path, version))
    } else {
        None
    }
}

fn download_unpack_tgz(url: &str, out_dir: &str) {
    let body = reqwest::blocking::get(url).unwrap();
    let decoder = GzDecoder::new(body);
    let mut archive = Archive::new(decoder);
    println!("ssss");
    archive.unpack(out_dir).unwrap();
}

fn main() {
    let arch = env::var("CARGO_CFG_TARGET_ARCH").unwrap();
    let sys = env::var("CARGO_CFG_TARGET_OS").unwrap();

    let platform = match (sys.as_str(), arch.as_str()) {
        ("linux", "aarch64") => "linux_arm64",
        ("linux", "arm") => "linux_armv6",
        ("linux", "x86_64") => "linux_amd64",
        ("android", "aarch64") => "linux_arm64",
        ("android", "arm") => "linux_armv6",
        ("android", "x86_64") => "linux_amd64",
        ("windows", "aarch64") => "windows_arm64",
        ("windows", "arm") => "windows_armv6",
        ("windows", "x86_64") => "windows_amd64",
        ("darwin", "aarch64") => "darwin_arm64",
        ("darwin", "x86_64") => "darwin_amd64",
        _ => {
            let target_triple = env::var("TARGET").unwrap();

            panic!("unsupport triple: {}", target_triple);
        }
    };

    let version = {
        if env::var("CARGO_FEATURE_TD_VER_0_34").unwrap() == "1" {
            "0.34.24"
        } else {
            panic!("must use special version of tendermint")
        }
    };

    let use_source_code = env::var("CARGO_FEATURE_USE_SOURCE_CODE");

    let upstream_url = env::var("EMBEDDED_TD_UPSTREAM_URL").ok();

    if use_source_code.is_err() {
        let dir = format!("{}/build", env::var("OUT_DIR").unwrap());

        if let Some(url) = upstream_url {
            if let Some(v) = check_has_tendermint(&dir, None) {
                if !v {
                    fs::remove_dir_all(&dir).unwrap();
                    download_unpack_tgz(&url, &dir);
                }
            } else {
                download_unpack_tgz(&url, &dir);
            }
        } else {
            let url = format!("https://github.com/tendermint/tendermint/releases/download/v{}/tendermint_{}_{}.tar.gz",version, version, platform);

            if let Some(v) = check_has_tendermint(&dir, Some(version)) {
                if !v {
                    fs::remove_dir_all(&dir).unwrap();
                    download_unpack_tgz(&url, &dir);
                }
            } else {
                download_unpack_tgz(&url, &dir);
            }
        }
    }

    println!("cargo:rerun-if-changed=build.rs");
    println!("cargo:rerun-if-env-changed=EMBEDDED_TD_UPSTREAM_URL");
}