~/hooky

hooky/victim/build.rs -rw-r--r-- 640 B
9537b5f6 — Arthur Melton read file to run from a config! 2 years ago
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
use std::io::prelude::*;
use std::fs::File;
use serde_derive::Deserialize;

#[derive(Deserialize)]
struct Config {
    payload: Option<String>,
}

fn main() {
    println!("cargo:rerun-if-changed=config.toml");
    let mut file = File::open("config.toml").expect("Unable to open the file");
    let mut contents = String::new();
    file.read_to_string(&mut contents).expect("Unable to read the file");
    let config: Config = toml::from_str(&contents).expect("Cant convert to toml");
    if let Some(payload) = config.payload {
        println!("cargo:rustc-cfg=payload");
        println!("cargo:rustc-env=payload={}", payload);
    }
}