97 lines
3.5 KiB
Rust
97 lines
3.5 KiB
Rust
use editor::messages::message::Message;
|
|
use editor::utility_types::DebugMessageTree;
|
|
use std::io::Write;
|
|
|
|
fn main() {
|
|
let result = Message::message_tree();
|
|
std::fs::create_dir_all("../../website/generated").unwrap();
|
|
let mut file = std::fs::File::create("../../website/generated/hierarchical_message_system_tree.txt").unwrap();
|
|
file.write_all(format!("{} `{}#L{}`\n", result.name(), result.path(), result.line_number()).as_bytes()).unwrap();
|
|
if let Some(variants) = result.variants() {
|
|
for (i, variant) in variants.iter().enumerate() {
|
|
let is_last = i == variants.len() - 1;
|
|
print_tree_node(variant, "", is_last, &mut file);
|
|
}
|
|
}
|
|
}
|
|
|
|
fn print_tree_node(tree: &DebugMessageTree, prefix: &str, is_last: bool, file: &mut std::fs::File) {
|
|
// Print the current node
|
|
let (branch, child_prefix) = if tree.message_handler_data_fields().is_some() || tree.message_handler_fields().is_some() {
|
|
("├── ", format!("{prefix}│ "))
|
|
} else if is_last {
|
|
("└── ", format!("{prefix} "))
|
|
} else {
|
|
("├── ", format!("{prefix}│ "))
|
|
};
|
|
|
|
if tree.path().is_empty() {
|
|
file.write_all(format!("{}{}{}\n", prefix, branch, tree.name()).as_bytes()).unwrap();
|
|
} else {
|
|
file.write_all(format!("{}{}{} `{}#L{}`\n", prefix, branch, tree.name(), tree.path(), tree.line_number()).as_bytes())
|
|
.unwrap();
|
|
}
|
|
|
|
// Print children if any
|
|
if let Some(variants) = tree.variants() {
|
|
let len = variants.len();
|
|
for (i, variant) in variants.iter().enumerate() {
|
|
let is_last_child = i == len - 1;
|
|
print_tree_node(variant, &child_prefix, is_last_child, file);
|
|
}
|
|
}
|
|
|
|
// Print message field if any
|
|
if let Some(fields) = tree.fields() {
|
|
let len = fields.len();
|
|
for (i, field) in fields.iter().enumerate() {
|
|
let is_last_field = i == len - 1;
|
|
let branch = if is_last_field { "└── " } else { "├── " };
|
|
|
|
file.write_all(format!("{child_prefix}{branch}{field}\n").as_bytes()).unwrap();
|
|
}
|
|
}
|
|
|
|
// Print handler field if any
|
|
if let Some(data) = tree.message_handler_fields() {
|
|
let len = data.fields().len();
|
|
let (branch, child_prefix) = if tree.message_handler_data_fields().is_some() {
|
|
("├── ", format!("{prefix}│ "))
|
|
} else {
|
|
("└── ", format!("{prefix} "))
|
|
};
|
|
|
|
const FRONTEND_MESSAGE_STR: &str = "FrontendMessage";
|
|
if data.name().is_empty() && tree.name() != FRONTEND_MESSAGE_STR {
|
|
panic!("{}'s MessageHandler is missing #[message_handler_data]", tree.name());
|
|
} else if tree.name() != FRONTEND_MESSAGE_STR {
|
|
file.write_all(format!("{}{}{} `{}#L{}`\n", prefix, branch, data.name(), data.path(), data.line_number()).as_bytes())
|
|
.unwrap();
|
|
|
|
for (i, field) in data.fields().iter().enumerate() {
|
|
let is_last_field = i == len - 1;
|
|
let branch = if is_last_field { "└── " } else { "├── " };
|
|
|
|
file.write_all(format!("{}{}{}\n", child_prefix, branch, field.0).as_bytes()).unwrap();
|
|
}
|
|
}
|
|
}
|
|
|
|
// Print data field if any
|
|
if let Some(data) = tree.message_handler_data_fields() {
|
|
let len = data.fields().len();
|
|
if data.path().is_empty() {
|
|
file.write_all(format!("{}{}{}\n", prefix, "└── ", data.name()).as_bytes()).unwrap();
|
|
} else {
|
|
file.write_all(format!("{}{}{} `{}#L{}`\n", prefix, "└── ", data.name(), data.path(), data.line_number()).as_bytes())
|
|
.unwrap();
|
|
}
|
|
for (i, field) in data.fields().iter().enumerate() {
|
|
let is_last_field = i == len - 1;
|
|
let branch = if is_last_field { "└── " } else { "├── " };
|
|
let field = &field.0;
|
|
file.write_all(format!("{prefix} {branch}{field}\n").as_bytes()).unwrap();
|
|
}
|
|
}
|
|
}
|