openethereum/util/version/build.rs

// Copyright 2015-2019 Parity Technologies (UK) Ltd.
// This file is part of Parity Ethereum.

// Parity Ethereum is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.

// Parity Ethereum is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU General Public License for more details.

// You should have received a copy of the GNU General Public License
// along with Parity Ethereum.  If not, see <http://www.gnu.org/licenses/>.

extern crate rustc_version;
extern crate toml;
extern crate vergen;

use std::{env, fs::File, io::Write, path::Path};
use vergen::{vergen, OutputFns};

const ERROR_MSG: &'static str = "Failed to generate metadata files";

fn main() {
    vergen(OutputFns::all()).expect(ERROR_MSG);

    let version = rustc_version::version().expect(ERROR_MSG);

    let cargo: toml::Value = toml::from_str(include_str!("./Cargo.toml")).expect(ERROR_MSG);
    let track = cargo["package"]["metadata"]["track"]
        .as_str()
        .expect("'track' has to be a string!");

    create_file(
        "meta.rs",
        format!(
            "
			/// This versions track.
			#[allow(unused)]
			pub const TRACK: &str = {track:?};

			/// Returns compiler version.
			pub fn rustc_version() -> &'static str {{
				\"{version}\"
			}}
		",
            track = track,
            version = version,
        ),
    );
}

fn create_file(filename: &str, data: String) {
    let out_dir = env::var("OUT_DIR").expect(ERROR_MSG);
    let dest_path = Path::new(&out_dir).join(filename);
    let mut f = File::create(&dest_path).expect(ERROR_MSG);
    f.write_all(data.as_bytes()).expect(ERROR_MSG);
}