rusteal-codegen 0.2.1

Code generator: UHT JSON to Rust bindings and C++ wrappers for rusteal
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

// Rust code generation orchestrator.

pub mod enums;
pub mod structs;
pub mod classes;
pub mod properties;
pub mod delegates;
pub mod module;
pub mod func_ids;
pub mod param_helpers;
pub mod cargo_toml;
pub mod manual;
pub mod prelude;

use std::path::Path;

use crate::context::CodegenContext;

/// Generate all Rust code into the output directory.
pub fn generate(ctx: &CodegenContext, out_dir: &Path) {
    // Ensure output directory exists
    std::fs::create_dir_all(out_dir).expect("Failed to create Rust output directory");

    // Generate per-module code
    for module_name in ctx.enabled_modules.iter() {
        let module_dir = out_dir.join(module_name);
        // Start from an empty module directory so types dropped by this run (blocklist,
        // engine upgrade) do not linger as stale files.
        if module_dir.exists() {
            std::fs::remove_dir_all(&module_dir).expect("Failed to clear module directory");
        }
        std::fs::create_dir_all(&module_dir).expect("Failed to create module directory");

        // Enums
        if let Some(module_enums) = ctx.module_enums.get(module_name) {
            for e in module_enums {
                let code = enums::generate_enum(e);
                let filename = crate::naming::to_snake_case(&e.name) + ".rs";
                std::fs::write(module_dir.join(&filename), code)
                    .unwrap_or_else(|err| panic!("Failed to write {filename}: {err}"));
            }
        }

        // Structs
        if let Some(module_structs) = ctx.module_structs.get(module_name) {
            for s in module_structs {
                let code = structs::generate_struct(s, ctx);
                let filename = crate::naming::to_snake_case(&s.name) + ".rs";
                std::fs::write(module_dir.join(&filename), code)
                    .unwrap_or_else(|err| panic!("Failed to write {filename}: {err}"));
            }
        }

        // Classes
        if let Some(module_classes) = ctx.module_classes.get(module_name) {
            for c in module_classes {
                let code = classes::generate_class(c, ctx);
                let filename = crate::naming::to_snake_case(&c.name) + ".rs";
                std::fs::write(module_dir.join(&filename), code)
                    .unwrap_or_else(|err| panic!("Failed to write {filename}: {err}"));
            }
        }

        // Module mod.rs
        let mod_code = module::generate_module_mod(
            module_name,
            ctx.module_enums.get(module_name).map(|v| v.as_slice()),
            ctx.module_structs.get(module_name).map(|v| v.as_slice()),
            ctx.module_classes.get(module_name).map(|v| v.as_slice()),
        );
        std::fs::write(module_dir.join("mod.rs"), mod_code)
            .expect("Failed to write module mod.rs");
    }

    // Generate func_ids.rs
    let func_ids_code = func_ids::generate_rust_func_ids(&ctx.func_table);
    std::fs::write(out_dir.join("func_ids.rs"), func_ids_code)
        .expect("Failed to write func_ids.rs");

    // Generate top-level lib.rs
    let lib_code = module::generate_lib_rs(ctx);
    std::fs::write(out_dir.join("lib.rs"), lib_code).expect("Failed to write lib.rs");
}