build.rs

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

use fs_err as fs;
use fs_err::File;
use maplit::hashmap;
use memofs::VfsSnapshot;

fn snapshot_from_fs_path(path: &Path) -> io::Result<VfsSnapshot> {
    println!("cargo:rerun-if-changed={}", path.display());

    if path.is_dir() {
        let mut children = Vec::new();

        for entry in fs::read_dir(path)? {
            let entry = entry?;

            let file_name = entry.file_name().to_str().unwrap().to_owned();

            // We can skip any TestEZ test files since they aren't necessary for
            // the plugin to run.
            if file_name.ends_with(".spec.lua") || file_name.ends_with(".spec.luau") {
                continue;
            }

            let child_snapshot = snapshot_from_fs_path(&entry.path())?;
            children.push((file_name, child_snapshot));
        }

        Ok(VfsSnapshot::dir(children))
    } else {
        let content = fs::read_to_string(path)?;

        Ok(VfsSnapshot::file(content))
    }
}

fn update_plugin_version(plugin_root: &Path) -> Result<(), anyhow::Error> {
    let plugin_wally = fs::read_to_string(&plugin_root.join("wally.toml"))?;
    let mut plugin_wally: toml_edit::Document = plugin_wally.parse()?;

    plugin_wally["package"]["version"] = toml_edit::value(env!("CARGO_PKG_VERSION"));

    fs::write(&plugin_root.join("wally.toml"), plugin_wally.to_string())?;

    Ok(())
}

fn update_readme_version() -> Result<(), anyhow::Error> {
    let readme = fs::read_to_string("README.md")?;

    let regex = regex::Regex::new(r#"rojo = "[^/]+/rojo@[^"]+""#)?;
    let updated_readme = regex.replace_all(&readme, |_captures: &regex::Captures| {
        format!(r#"rojo = "UpliftGames/rojo@{}""#, env!("CARGO_PKG_VERSION"))
    });

    fs::write("README.md", updated_readme.as_bytes())?;

    Ok(())
}

fn main() -> Result<(), anyhow::Error> {
    let out_dir = env::var_os("OUT_DIR").unwrap();

    let root_dir = env::var_os("CARGO_MANIFEST_DIR").unwrap();
    let plugin_root = PathBuf::from(root_dir).join("plugin");

    update_plugin_version(&plugin_root)?;
    update_readme_version()?;

    let snapshot = VfsSnapshot::dir(hashmap! {
        "default.project.json" => snapshot_from_fs_path(&plugin_root.join("default.project.json"))?,
        "fmt" => snapshot_from_fs_path(&plugin_root.join("fmt"))?,
        "http" => snapshot_from_fs_path(&plugin_root.join("http"))?,
        "log" => snapshot_from_fs_path(&plugin_root.join("log"))?,
        "rbx_dom_lua" => snapshot_from_fs_path(&plugin_root.join("rbx_dom_lua"))?,
        "src" => snapshot_from_fs_path(&plugin_root.join("src"))?,
        "Packages" => snapshot_from_fs_path(&plugin_root.join("Packages"))?,
        "wally.toml" => snapshot_from_fs_path(&plugin_root.join("wally.toml"))?,
    });

    let out_path = Path::new(&out_dir).join("plugin.bincode");
    let out_file = File::create(&out_path)?;

    bincode::serialize_into(out_file, &snapshot)?;

    println!("cargo:rerun-if-changed=build/windows/rojo-manifest.rc");
    println!("cargo:rerun-if-changed=build/windows/rojo.manifest");
    embed_resource::compile("build/windows/rojo-manifest.rc");

    Ok(())
}