Skip to main content

factorio_api_gen/
lib.rs

1#![allow(
2    clippy::missing_errors_doc,
3    clippy::missing_panics_doc,
4    clippy::module_name_repetitions,
5    clippy::must_use_candidate,
6    clippy::too_many_lines,
7    clippy::cognitive_complexity,
8    clippy::doc_markdown,
9    clippy::unnested_or_patterns,
10    clippy::option_if_let_else,
11    clippy::map_unwrap_or,
12    clippy::needless_pass_by_value,
13    clippy::use_self,
14    clippy::similar_names,
15    clippy::single_match_else,
16    clippy::match_same_arms,
17    clippy::items_after_statements,
18    clippy::struct_field_names,
19    clippy::if_not_else,
20    clippy::from_iter_instead_of_collect,
21    clippy::manual_string_new,
22    clippy::wildcard_imports,
23    clippy::enum_glob_use,
24    clippy::unused_self,
25    clippy::trivially_copy_pass_by_ref,
26    clippy::as_conversions,
27    clippy::indexing_slicing,
28    clippy::integer_division,
29    clippy::arithmetic_side_effects,
30    clippy::missing_const_for_fn,
31    clippy::implicit_hasher,
32    clippy::semicolon_if_nothing_returned,
33    clippy::redundant_closure_for_method_calls,
34    clippy::unnecessary_wraps,
35    clippy::return_self_not_must_use,
36    clippy::pub_use,
37    clippy::module_inception,
38    clippy::redundant_pub_crate
39)]
40
41mod generate;
42mod schema;
43
44use std::path::Path;
45
46pub use schema::RuntimeApi;
47
48pub struct GeneratedApi {
49    pub application_version: String,
50    pub api_version: u32,
51    pub events: String,
52    pub event_map: String,
53    pub event_lookup: String,
54    pub event_filter_lookup: String,
55    pub event_module_lookup: String,
56    pub event_filters: String,
57    pub event_data: String,
58    pub defines: String,
59    pub classes: String,
60    pub globals: String,
61    pub concepts: String,
62    pub unions: String,
63    pub debug_types: String,
64}
65
66pub fn parse_runtime_api(json: &str) -> Result<RuntimeApi, serde_json::Error> {
67    serde_json::from_str(json)
68}
69
70pub fn generate_runtime_api(api: &RuntimeApi) -> GeneratedApi {
71    let mappings = generate::collect_event_mappings(api);
72    let class_names = generate::class_names(api);
73    let filter_concept_names = generate::event_filter_concept_names(api);
74    let identification_names = generate::identification_concept_names(api, &filter_concept_names);
75    let identification_signatures = generate::identification_signatures(api, &identification_names);
76    let mut concept_names = generate::generatable_concept_names(api, &filter_concept_names);
77    concept_names.extend(identification_names.iter().cloned());
78    let union_registry = generate::collect_literal_unions(api);
79    let known = generate::KnownTypes {
80        classes: &class_names,
81        concepts: &concept_names,
82        identifications: &identification_names,
83        identification_signatures: &identification_signatures,
84        unions: union_registry.names(),
85        union_registry: &union_registry,
86    };
87    let event_filters = generate::generate_event_filters(api);
88    let concepts = {
89        let mut structs = generate::generate_concepts(api, &known, &filter_concept_names);
90        let ids = generate::generate_identifications(api, &known).to_string();
91        structs.push_str(&ids);
92        structs
93    };
94
95    GeneratedApi {
96        application_version: api.application_version.clone(),
97        api_version: api.api_version,
98        events: generate::generate_events(api),
99        event_map: generate::generate_event_map(&mappings),
100        event_lookup: generate::generate_event_lookup(&mappings),
101        event_filter_lookup: generate::generate_event_filter_lookup(&mappings),
102        event_module_lookup: generate::generate_event_module_lookup(&mappings),
103        event_filters,
104        event_data: generate::generate_event_data(api, &known),
105        defines: generate::generate_defines(&api.defines),
106        classes: generate::generate_classes(api, &known),
107        globals: generate::generate_globals(api, &known),
108        concepts,
109        unions: generate::generate_unions(&union_registry),
110        debug_types: generate::generate_debug_types(api, &known),
111    }
112}
113
114pub fn generate_from_json(json: &str) -> Result<GeneratedApi, serde_json::Error> {
115    let api = parse_runtime_api(json)?;
116    Ok(generate_runtime_api(&api))
117}
118
119pub fn bundled_runtime_api_json() -> &'static str {
120    include_str!("../api/runtime-api.json")
121}
122
123pub fn generate_from_bundled_api() -> Result<GeneratedApi, serde_json::Error> {
124    generate_from_json(bundled_runtime_api_json())
125}
126
127pub fn write_generated_api(output_dir: &Path, generated: &GeneratedApi) -> std::io::Result<()> {
128    std::fs::create_dir_all(output_dir)?;
129
130    write_module(
131        output_dir,
132        "mod.rs",
133        &format!(
134            "// Generated from Factorio runtime API v{} (format v{}).\n\
135             #[allow(unused, clippy::all, clippy::pedantic, clippy::nursery)]\n\n\
136             pub mod classes;\n\
137             pub mod concepts;\n\
138             pub mod debug_types;\n\
139             pub mod defines;\n\
140             pub mod event_data;\n\
141             pub mod event_filters;\n\
142             pub mod events;\n\
143             pub mod globals;\n\
144             pub mod map;\n\
145             pub mod unions;\n",
146            generated.application_version, generated.api_version
147        ),
148    )?;
149    write_module(output_dir, "events.rs", &generated.events)?;
150    write_module(output_dir, "map.rs", &generated.event_map)?;
151    write_module(output_dir, "event_filters.rs", &generated.event_filters)?;
152    write_module(output_dir, "event_data.rs", &generated.event_data)?;
153    write_module(output_dir, "defines.rs", &generated.defines)?;
154    write_module(output_dir, "classes.rs", &generated.classes)?;
155    write_module(output_dir, "globals.rs", &generated.globals)?;
156    write_module(output_dir, "concepts.rs", &generated.concepts)?;
157    write_module(output_dir, "unions.rs", &generated.unions)?;
158    write_module(output_dir, "debug_types.rs", &generated.debug_types)?;
159
160    Ok(())
161}
162
163pub fn write_macro_event_lookup(
164    output_dir: &Path,
165    generated: &GeneratedApi,
166) -> std::io::Result<()> {
167    std::fs::create_dir_all(output_dir)?;
168    write_module(output_dir, "event_lookup.rs", &generated.event_lookup)?;
169    write_module(
170        output_dir,
171        "event_filter_lookup.rs",
172        &generated.event_filter_lookup,
173    )?;
174    write_module(
175        output_dir,
176        "event_module_lookup.rs",
177        &generated.event_module_lookup,
178    )
179}
180
181fn write_module(output_dir: &Path, file_name: &str, contents: &str) -> std::io::Result<()> {
182    let path = output_dir.join(file_name);
183    std::fs::write(path, contents)
184}