build.rs

use std::env;
use std::path::{Path, PathBuf};

const DEFAULT_ROUNDING_MODE: &str = "HalfUp";
const FMT_EXPONENTIAL_LOWER_THRESHOLD: &str = "5";
const FMT_EXPONENTIAL_UPPER_THRESHOLD: &str = "15";
const FMT_MAX_INTEGER_PADDING: &str = "1000";
const SERDE_DESERIALIZE_MODE: &str = "Strict";

fn main() {
    let out_dir: PathBuf = env::var_os("OUT_DIR").unwrap().into();
    write_default_rounding_mode(&out_dir);
    write_exponential_format_threshold(&out_dir);
    write_serde_deserialize_mode(&out_dir);
}

macro_rules! load_env {
    ($env:ident, $name:literal, $default:ident) => {{
        println!("cargo:rerun-if-env-changed={}", $name);
        $env::var($name).unwrap_or_else(|_| $default.to_owned())
    }};
}

fn write_default_rounding_mode(out_dir: &Path) {
    let rounding_mode_name = load_env!(
        env,
        "RUST_FASTNUM_DEFAULT_ROUNDING_MODE",
        DEFAULT_ROUNDING_MODE
    );

    let rust_file_path = out_dir.join("default_rounding_mode.rs");
    let rust_file_contents = format!(
        "const DEFAULT_ROUNDING_MODE: RoundingMode = RoundingMode::{};",
        rounding_mode_name
    );

    std::fs::write(rust_file_path, rust_file_contents).unwrap();
}

fn write_exponential_format_threshold(out_dir: &Path) {
    let low_value = load_env!(
        env,
        "RUST_FASTNUM_FMT_EXPONENTIAL_LOWER_THRESHOLD",
        FMT_EXPONENTIAL_LOWER_THRESHOLD
    );
    let high_value = load_env!(
        env,
        "RUST_FASTNUM_FMT_EXPONENTIAL_UPPER_THRESHOLD",
        FMT_EXPONENTIAL_UPPER_THRESHOLD
    );
    let max_padding = load_env!(
        env,
        "RUST_FASTNUM_FMT_MAX_INTEGER_PADDING",
        FMT_MAX_INTEGER_PADDING
    );

    let low_value: u32 = low_value
        .parse::<std::num::NonZeroU32>()
        .expect("$RUST_FASTNUM_FMT_EXPONENTIAL_LOWER_THRESHOLD must be an integer > 0")
        .into();

    let high_value: u32 = high_value
        .parse::<u32>()
        .expect("$RUST_FASTNUM_FMT_EXPONENTIAL_UPPER_THRESHOLD must be valid u32");

    let max_padding: u32 = max_padding
        .parse::<u32>()
        .expect("$RUST_FASTNUM_FMT_MAX_INTEGER_PADDING must be valid u32");

    let rust_file_path = out_dir.join("exponential_format_threshold.rs");

    let rust_file_contents = [
        format!(
            "const EXPONENTIAL_FORMAT_LEADING_ZERO_THRESHOLD: usize = {};",
            low_value
        ),
        format!(
            "const EXPONENTIAL_FORMAT_TRAILING_ZERO_THRESHOLD: usize = {};",
            high_value
        ),
        format!("const FMT_MAX_INTEGER_PADDING: usize = {};", max_padding),
    ];

    std::fs::write(rust_file_path, rust_file_contents.join("\n")).unwrap();
}

fn write_serde_deserialize_mode(out_dir: &Path) {
    let mode = load_env!(
        env,
        "RUST_FASTNUM_SERDE_DESERIALIZE_MODE",
        SERDE_DESERIALIZE_MODE
    );

    let rust_file_path = out_dir.join("serde_deserialize_mode.rs");
    let rust_file_contents = format!(
        "const SERDE_DESERIALIZE_MODE: DeserializeMode = DeserializeMode::{};",
        mode
    );

    std::fs::write(rust_file_path, rust_file_contents).unwrap();
}