macroquest-build-config 0.1.3

use std::env;
use std::ffi::CStr;
use std::fs;
use std::os::raw::c_char;
use std::path::{Path, PathBuf};

use serde::Serialize;

// NOTE: this has to be kept in sync with the BuildConfig located in srcs/lib.rs
#[derive(Debug, Serialize)]
struct BuildConfig {
    eq_version: String,
    profile: String,
    root_dir: PathBuf,
    bin_dir: PathBuf,
}

fn eq_version(dir: &Path) -> Result<String, Box<dyn std::error::Error>> {
    let current_path = env::var_os("PATH").unwrap();
    let mut new_path = current_path.clone();
    new_path.push(";");
    new_path.push(dir.join("release"));
    env::set_var("PATH", new_path);

    Ok(unsafe {
        let lib = libloading::Library::new(dir.join("release/MQ2Main.dll"))?;

        let version_ptr: libloading::Symbol<*const c_char> = lib.get(b"gszVersion\0")?;
        let version = CStr::from_ptr(*version_ptr).to_str()?;

        let time_ptr: libloading::Symbol<*const c_char> = lib.get(b"gszTime\0")?;
        let time = CStr::from_ptr(*time_ptr).to_str()?;

        format!("{} {}", version, time)
    })
}

fn main() {
    // We need to rerun if a number of things change, so mark them all.
    println!("cargo:rerun-if-env-changed=MACROQUEST_BUILD_PROFILE");
    println!("cargo:rerun-if-env-changed=MACROQUEST_DIR");
    println!("cargo:rerun-if-env-changed=MACROQUEST_BUILD_BIN_DIR");
    println!("cargo:rerun-if-changed=build.rs");

    // If we're building on docs.rs, then we synthesize a build configuration
    let config = if std::env::var("DOCS_RS").is_ok() {
        BuildConfig {
            eq_version: String::from("docs build"),
            root_dir: PathBuf::from("docs build"),
            profile: String::from("docs build"),
            bin_dir: PathBuf::from("docs build"),
        }
    } else {
        // Compute our Build Configuration
        let mq_profile = env::var("MACROQUEST_BUILD_PROFILE").unwrap_or_else(|_| "release".into());
        let mq_root_dir = PathBuf::from(
            env::var_os("MACROQUEST_DIR")
                .expect("Must set MACROQUEST_DIR to the root of a MacroQuest checkout"),
        );
        let mq_bin_dir = env::var_os("MACROQUEST_BUILD_BIN_DIR")
            .map(PathBuf::from)
            .unwrap_or_else(|| mq_root_dir.join("build/bin/"));

        // Determine what version of EverQuest we're building against
        let eq_version = eq_version(mq_bin_dir.as_path()).unwrap();

        BuildConfig {
            eq_version,
            root_dir: mq_root_dir,
            profile: mq_profile,
            bin_dir: mq_bin_dir,
        }
    };

    // Actually write out our configuration file so that our crate can read it
    // at compile time.
    let out_dir = env::var_os("OUT_DIR").unwrap();
    let dest_path = Path::new(&out_dir).join("config.toml");
    fs::write(dest_path, toml::to_string(&config).unwrap()).unwrap();
}