yewprint/build.rs

//! This build script parses the svg paths located in iconSvgPaths.js
//! to generate the IconName enum and all the icons.
//!
//! If the js file need an update, you can download a npm archive at this url:
//! https://registry.npmjs.org/@blueprintjs/icons/-/icons-3.19.0.tgz
//!
//! After that you can extract the file following this path:
//! package/lib/esnext/generated/iconSvgPaths.js

use heck::ToUpperCamelCase;
use regex::Regex;
use std::collections::HashSet;
use std::env;
use std::fs;
use std::path::Path;

fn main() {
    let icon_svg_paths = fs::read_to_string("iconSvgPaths.js").expect("cannot read file");
    let out_dir = env::var_os("OUT_DIR").unwrap();
    let dest_path = Path::new(&out_dir).join("icon_svg_paths.rs");

    let mut src = String::new();
    let re_map = Regex::new(r"export const IconSvgPaths([0-9]+) = \{([^}]+)\};").unwrap();
    let re_item = Regex::new(r#"(?m)(?s)"([^"]+)": (\[.*?\]),\s*$"#).unwrap();
    let mut keys = HashSet::new();

    for map in re_map.captures_iter(icon_svg_paths.as_str()) {
        src.push_str("fn icon_svg_paths_");
        src.push_str(&map[1]);
        src.push_str("(icon: IconName) -> &'static [&'static str] { match icon {\n");
        for item in re_item.captures_iter(&map[2]) {
            let key = item[1].to_upper_camel_case();
            src.push_str("IconName::");
            src.push_str(&key);
            src.push_str(" => &");
            src.push_str(&item[2]);
            src.push_str(",\n");
            keys.insert(key);
        }
        src.push_str(" }}\n\n");
    }

    if keys.is_empty() {
        panic!("failed parse icons");
    }

    let mut keys: Vec<_> = keys.iter().collect();
    keys.sort();
    src.push_str(
        "#[derive(Debug, Copy, Clone, PartialEq, Eq, Hash)]\n\
         pub enum IconName {\n",
    );
    for icon in &keys {
        src.push_str(icon);
        src.push_str(",\n");
    }
    src.push_str("}\n\n");

    src.push_str("impl IconName {\n");
    src.push_str("pub const ALL: &[IconName] = &[\n");
    for icon in keys {
        src.push_str("IconName::");
        src.push_str(icon);
        src.push_str(",\n");
    }
    src.push_str("];\n");
    src.push('}');

    fs::write(&dest_path, src).unwrap();
    println!("cargo:rerun-if-changed=build.rs");
}