forked from 1b-agent/opencadstudio-upstream
first commit
This commit is contained in:
1 parent
d52c6440ff
commit
d2fdac7f1e
274 files changed
+89330
No files matched your search
@@ -0,0 +1,91 @@
|
||||
// build.rs — auto-generates src/modules/registry.rs
|
||||
//
|
||||
// Scans src/modules/ for subdirectories containing a mod.rs file.
|
||||
// Each subdirectory name like "home" maps to a struct "HomeModule"
|
||||
// defined in src/modules/home/mod.rs.
|
||||
//
|
||||
// Rules for a directory to be picked up:
|
||||
// • Located directly in src/modules/
|
||||
// • Contains a mod.rs file
|
||||
// • Defines a pub struct {PascalCase}Module implementing CadModule
|
||||
// (unit struct — no ::new() needed)
|
||||
//
|
||||
// Tab display order: the PRIORITY list below controls order.
|
||||
// Unlisted modules appear after priority modules, sorted alphabetically.
|
||||
//
|
||||
// Adding a new module:
|
||||
// 1. Create src/modules/my_name/mod.rs (implement CadModule as MyNameModule)
|
||||
// 2. Add `pub mod my_name;` to src/modules/mod.rs
|
||||
// 3. cargo build → module appears in the ribbon
|
||||
|
||||
use std::fs;
|
||||
use std::path::Path;
|
||||
|
||||
// Display order for known modules; unknown modules follow alphabetically.
|
||||
const PRIORITY: &[&str] = &["home", "annotate", "insert", "view", "manage"];
|
||||
|
||||
fn main() {
|
||||
let mods_dir = Path::new("src/modules");
|
||||
println!("cargo:rerun-if-changed=src/modules");
|
||||
|
||||
// Collect subdirectories that contain mod.rs.
|
||||
let mut modules: Vec<String> = fs::read_dir(mods_dir)
|
||||
.expect("src/modules/ not found")
|
||||
.filter_map(|e| {
|
||||
let entry = e.ok()?;
|
||||
if !entry.file_type().ok()?.is_dir() {
|
||||
return None;
|
||||
}
|
||||
let name = entry.file_name().into_string().ok()?;
|
||||
let mod_file = mods_dir.join(&name).join("mod.rs");
|
||||
if mod_file.exists() {
|
||||
Some(name)
|
||||
} else {
|
||||
None
|
||||
}
|
||||
})
|
||||
.collect();
|
||||
|
||||
// Sort: priority order first, then alphabetical for the rest.
|
||||
modules.sort_by_key(|name| {
|
||||
let pri = PRIORITY.iter().position(|p| *p == name.as_str());
|
||||
(pri.map(|i| i as u32).unwrap_or(u32::MAX), name.clone())
|
||||
});
|
||||
|
||||
// snake_case → PascalCase + "Module"
|
||||
let to_struct = |s: &str| -> String {
|
||||
s.split('_')
|
||||
.map(|w| {
|
||||
let mut c = w.chars();
|
||||
match c.next() {
|
||||
None => String::new(),
|
||||
Some(f) => f.to_uppercase().collect::<String>() + c.as_str(),
|
||||
}
|
||||
})
|
||||
.collect::<String>()
|
||||
+ "Module"
|
||||
};
|
||||
|
||||
let mut out = String::from(
|
||||
"// AUTO-GENERATED by build.rs — do not edit by hand.\n\
|
||||
// To add a module: create src/modules/my_name/mod.rs + add `pub mod my_name;` to modules/mod.rs.\n\n\
|
||||
use crate::modules::CadModule;\n\n\
|
||||
/// Returns one boxed instance of every registered CAD module.\n\
|
||||
/// Called once at startup by `Ribbon::new()`.\n\
|
||||
pub fn all_modules() -> Vec<Box<dyn CadModule>> {\n\
|
||||
\tvec![\n",
|
||||
);
|
||||
|
||||
for module in &modules {
|
||||
let s = to_struct(module);
|
||||
out.push_str(&format!("\t\tBox::new(super::{module}::{s}),\n"));
|
||||
}
|
||||
|
||||
out.push_str("\t]\n}\n");
|
||||
|
||||
let out_path = mods_dir.join("registry.rs");
|
||||
let current = fs::read_to_string(&out_path).unwrap_or_default();
|
||||
if current != out {
|
||||
fs::write(&out_path, &out).expect("failed to write registry.rs");
|
||||
}
|
||||
}
|
||||
Reference in new issue
Block a user