collectd-plugin 0.6.0

Provides ergonomic API ontop of collectd's C interface and macro for defining plugins easier
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
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

extern crate regex;

use regex::Regex;
use std::env;
use std::path::PathBuf;
use std::process::Command;

enum CollectdVersion {
    Collectd54,
    Collectd55,
    Collectd57,
}

fn main() {
    let collectd_version = detect_collectd_version();
    let out_path = PathBuf::from(env::var("OUT_DIR").unwrap());
    let version = match collectd_version.as_str() {
        "5.8" | "5.7" => {
            println!("cargo:rustc-cfg=collectd57");
            CollectdVersion::Collectd57
        }
        "5.6" | "5.5" => {
            println!("cargo:rustc-cfg=collectd55");
            CollectdVersion::Collectd55
        }
        "5.4" => {
            println!("cargo:rustc-cfg=collectd54");
            CollectdVersion::Collectd54
        }
        x => panic!("Unrecognized collectd version: {}", x),
    };

    bindings(out_path.join("bindings.rs"), version);
}

#[cfg(collectd_docs_rs)]
fn detect_collectd_version() -> String {
    String::from("5.5")
}

#[cfg(not(collectd_docs_rs))]
fn detect_collectd_version() -> String {
    let re = Regex::new(r"collectd (\d+\.\d+)\.\d+").expect("Valid collectd regex");
    println!("cargo:rerun-if-env-changed=COLLECTD_VERSION");

    env::var_os("COLLECTD_VERSION")
        .map(|x| {
            x.into_string()
                .expect("COLLECTD_VERSION to be a valid string")
        }).unwrap_or_else(|| {
            Command::new("collectd")
                .args(&["-h"])
                .output()
                .map(|x| String::from_utf8(x.stdout).expect("Collectd output to be utf8"))
                .map(|x| {
                    re.captures(&x)
                        .expect("Version info to be present in collectd")
                        .get(1)
                        .map(|x| String::from(x.as_str()))
                        .unwrap()
                }).expect("Collectd -h to execute successfully")
        })
}

#[cfg(feature = "bindgen")]
fn bindings(loc: PathBuf, version: CollectdVersion) {
    extern crate bindgen;
    let arg = match version {
        CollectdVersion::Collectd54 => "-DCOLLECTD_54",
        CollectdVersion::Collectd55 => "-DCOLLECTD_55",
        CollectdVersion::Collectd57 => "-DCOLLECTD_57",
    };

    bindgen::Builder::default()
        .header("wrapper.h")
        .clang_arg("-DHAVE_CONFIG_H")
        .clang_arg(arg)
        .rust_target(bindgen::RustTarget::Stable_1_21)
        .whitelist_type("cdtime_t")
        .whitelist_type("data_set_t")
        .whitelist_function("plugin_.*")
        .whitelist_var("OCONFIG_TYPE_.*")
        .whitelist_var("LOG_.*")
        .whitelist_var("DS_TYPE_.*")
        .whitelist_var("DATA_MAX_NAME_LEN")
        .generate()
        .expect("Unable to generate bindings")
        .write_to_file(loc)
        .expect("Couldn't write bindings!");
}

#[cfg(not(feature = "bindgen"))]
fn bindings(loc: PathBuf, version: CollectdVersion) {
    use std::fs;

    let path = match version {
        CollectdVersion::Collectd54 => "src/bindings-54.rs",
        CollectdVersion::Collectd55 => "src/bindings-55.rs",
        CollectdVersion::Collectd57 => "src/bindings-57.rs",
    };

    fs::copy(PathBuf::from(path), loc).expect("File to copy");
}