aboutsummaryrefslogtreecommitdiff
path: root/build.rs
blob: c52bf3424993fe1ab587d136f910a74d75cdc580 (plain)
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
102
103
104
105
106
107
108
109
110
extern crate cc;

use std::env;
use std::io;
use std::io::{Read, Write};
use std::fs;
use std::path;

struct Version {
    major: String,
    minor: String,
    git: String,
}

fn stringify(err: env::VarError) -> String {
    format!("{}", err)
}

fn extract_git_version(pre: &str) -> Result<String, String> {
    // If a pre-release version is set, it is expected to have the format
    // pre.v<maj>.<min>.<n>.g<hash>, where <maj> and <min> are the last major and minor version,
    // <n> is the number of commits since this version and <hash> is the hash of the last commit.
    let parts: Vec<&str> = pre.split('.').collect();
    if parts.len() != 5 {
        return Err(format!("'{}' is not a valid pre-release version", pre));
    }
    Ok(format!("{}.{}-{}-{}", parts[1], parts[2], parts[3], parts[4]))
}

fn get_version() -> Result<Version, String> {
    let major = env::var("CARGO_PKG_VERSION_MAJOR").map_err(stringify)?;
    let minor = env::var("CARGO_PKG_VERSION_MINOR").map_err(stringify)?;
    let patch = env::var("CARGO_PKG_VERSION_PATCH").map_err(stringify)?;
    let pre = env::var("CARGO_PKG_VERSION_PRE").map_err(stringify)?;

    let git = match pre.is_empty() {
        true => match patch.is_empty() {
            true => format!("v{}.{}", major, minor),
            false => format!("v{}.{}.{}", major, minor, patch),
        },
        false => extract_git_version(&pre)?,
    };

    Ok(Version {
        major,
        minor,
        git,
    })
}

fn prepare_version_source(
    version: &Version,
    out_path: &path::Path,
    library_path: &path::Path
) -> io::Result<path::PathBuf> {
    let out = out_path.join("version.cc");
    let template = library_path.join("version.cc.in");

    let mut file = fs::File::open(template)?;
    let mut data = String::new();
    file.read_to_string(&mut data)?;
    drop(file);

    let data = data
        .replace("@PROJECT_VERSION_MAJOR@", &version.major)
        .replace("@PROJECT_VERSION_MINOR@", &version.minor)
        .replace("@PROJECT_VERSION_GIT@", &version.git);

    let mut file = fs::File::create(&out)?;
    file.write_all(data.as_bytes())?;

    Ok(out)
}

fn main() {
    let out_dir = env::var("OUT_DIR").expect("Environment variable OUT_DIR is not set");
    let out_path = path::PathBuf::from(out_dir);

    let version = get_version().expect("Could not extract library version");

    let sources = [
        "DeviceCommunicationExceptions.cpp",
        "NK_C_API.cc",
        "NitrokeyManager.cc",
        "command_id.cc",
        "device.cc",
        "log.cc",
        "misc.cc",
    ];
    let library_dir = format!("libnitrokey-{}", version.git);
    let library_path = path::Path::new(&library_dir);

    let version_source = prepare_version_source(&version, &out_path, &library_path)
        .expect("Could not prepare the version source file");

    cc::Build::new()
        .cpp(true)
        .flag("-std=c++14")
        .include(library_path.join("libnitrokey"))
        .files(sources.iter().map(|s| library_path.join(s)))
        .file(version_source)
        .compile("libnitrokey.a");

    let hidapi_library_name = if cfg!(target_os = "linux") {
        "hidapi-libusb"
    } else {
        "hidapi"
    };
    println!("cargo:rustc-link-lib={}", hidapi_library_name);
}