use chrono::{DateTime, Utc};
use std::env;
use std::path::{Path, PathBuf};
use std::process::Command;
fn main() {
let builder = tonic_build::configure()
.type_attribute("netavark_proxy.Lease", "#[derive(serde::Serialize)]")
.type_attribute("netavark_proxy.DhcpV4Lease", "#[derive(serde::Serialize)]")
.type_attribute("netavark_proxy.DhcpV6Lease", "#[derive(serde::Serialize)]")
.type_attribute("netavark_proxy.IPResponse", "#[derive(serde::Serialize)]")
.type_attribute("netavark_proxy.MacAddress", "#[derive(serde::Serialize)]")
.type_attribute("netavark_proxy.NvIpv4Addr", "#[derive(serde::Serialize)]")
.type_attribute("netavark_proxy.Lease", "#[derive(serde::Deserialize)]")
.type_attribute(
"netavark_proxy.DhcpV4Lease",
"#[derive(serde::Deserialize)]",
)
.type_attribute(
"netavark_proxy.DhcpV6Lease",
"#[derive(serde::Deserialize)]",
)
.type_attribute("netavark_proxy.IPResponse", "#[derive(serde::Deserialize)]")
.type_attribute("netavark_proxy.MacAddress", "#[derive(serde::Deserialize)]")
.type_attribute("netavark_proxy.NvIpv4Addr", "#[derive(serde::Deserialize)]")
.type_attribute("netavark_proxy.MacAddress", "#[derive(Eq)]")
.type_attribute("netavark_proxy.MacAddress", "#[derive(Hash)]")
.type_attribute(
"netavark_proxy.NetworkConfig",
"#[derive(serde::Deserialize)]",
)
.type_attribute(
"netavark_proxy.NetworkConfig",
"#[derive(serde::Serialize)]",
)
.out_dir(PathBuf::from("src/proto-build"));
builder
.compile_protos(&[Path::new("src/proto/proxy.proto")], &[Path::new("proto")])
.unwrap_or_else(|e| panic!("Failed at builder: {:?}", e.to_string()));
println!("cargo:rerun-if-changed=build.rs");
let now = match env::var("SOURCE_DATE_EPOCH") {
Ok(val) => DateTime::from_timestamp(val.parse::<i64>().unwrap(), 0).unwrap(),
Err(_) => Utc::now(),
};
println!("cargo:rustc-env=BUILD_TIMESTAMP={}", now.to_rfc3339());
println!(
"cargo:rustc-env=BUILD_TARGET={}",
std::env::var("TARGET").unwrap()
);
let command = Command::new("git").args(["rev-parse", "HEAD"]).output();
let commit = match command {
Ok(output) => String::from_utf8(output.stdout).unwrap(),
Err(_) => "".to_string(),
};
println!("cargo:rustc-env=GIT_COMMIT={commit}");
let fwdriver = match env::var("NETAVARK_DEFAULT_FW")
.unwrap_or("iptables".to_string())
.as_str()
{
"nftables" => "nftables",
"iptables" => "iptables",
"none" => "none",
inv => panic!("Invalid default firewall driver {}", inv),
};
println!("cargo:rustc-check-cfg=cfg(default_fw, values(\"nftables\", \"iptables\", \"none\"))");
println!("cargo:rustc-cfg=default_fw=\"{}\"", fwdriver);
println!("cargo:rustc-env=DEFAULT_FW={fwdriver}");
}