mirror of
https://github.com/Noratrieb/haesli.git
synced 2026-01-15 12:15:02 +01:00
split generated code so that methods are now in core
This commit is contained in:
parent
3b656b911a
commit
c333f20531
20 changed files with 1337 additions and 1206 deletions
|
|
@ -4,15 +4,15 @@ mod parser;
|
|||
mod random;
|
||||
mod write;
|
||||
|
||||
use anyhow::Context;
|
||||
use heck::ToUpperCamelCase;
|
||||
use parser::codegen_parser;
|
||||
use random::codegen_random;
|
||||
use std::fs;
|
||||
use std::fs::File;
|
||||
use std::io::Write;
|
||||
use std::iter::Peekable;
|
||||
use std::path::PathBuf;
|
||||
use std::str::FromStr;
|
||||
use strong_xml::XmlRead;
|
||||
use write::codegen_write;
|
||||
|
||||
#[derive(Debug, XmlRead)]
|
||||
#[xml(tag = "amqp")]
|
||||
|
|
@ -101,196 +101,235 @@ struct Doc {
|
|||
kind: Option<String>,
|
||||
}
|
||||
|
||||
pub fn main() {
|
||||
let this_file = PathBuf::from_str(file!()).unwrap();
|
||||
let expected_location = this_file
|
||||
.parent()
|
||||
.unwrap()
|
||||
.parent()
|
||||
.unwrap()
|
||||
.join("amqp0-9-1.xml");
|
||||
let content = fs::read_to_string(expected_location).unwrap();
|
||||
|
||||
let amqp = match Amqp::from_str(&content) {
|
||||
Ok(amqp) => amqp,
|
||||
Err(err) => {
|
||||
eprintln!("{err}");
|
||||
std::process::exit(1);
|
||||
}
|
||||
};
|
||||
codegen(&amqp);
|
||||
struct Codegen {
|
||||
output: Box<dyn Write>,
|
||||
}
|
||||
|
||||
fn codegen(amqp: &Amqp) {
|
||||
println!("#![allow(dead_code)]");
|
||||
println!("// This file has been generated by `xtask/src/codegen`. Do not edit it manually.\n");
|
||||
codegen_domain_defs(amqp);
|
||||
codegen_class_defs(amqp);
|
||||
codegen_parser(amqp);
|
||||
codegen_write(amqp);
|
||||
codegen_random(amqp);
|
||||
}
|
||||
pub fn main() -> anyhow::Result<()> {
|
||||
let this_file = PathBuf::from_str(file!()).context("own file path")?;
|
||||
let xtask_root = this_file
|
||||
.parent()
|
||||
.context("codegen directory path")?
|
||||
.parent()
|
||||
.context("src directory path")?
|
||||
.parent()
|
||||
.context("xtask root path")?;
|
||||
let amqp_spec = xtask_root.join("amqp0-9-1.xml");
|
||||
let project_root = xtask_root.parent().context("get project root parent")?;
|
||||
|
||||
fn codegen_domain_defs(amqp: &Amqp) {
|
||||
for domain in &amqp.domains {
|
||||
let invariants = invariants(domain.asserts.iter());
|
||||
let transport_generated_path = project_root.join("amqp_transport/src/methods/generated.rs");
|
||||
let core_generated_path = project_root.join("amqp_core/src/methods/generated.rs");
|
||||
|
||||
if let Some(label) = &domain.label {
|
||||
println!("/// {label}");
|
||||
}
|
||||
let content = fs::read_to_string(amqp_spec).context("read amqp spec file")?;
|
||||
|
||||
if !invariants.is_empty() {
|
||||
if domain.label.is_some() {
|
||||
println!("///");
|
||||
}
|
||||
println!("/// {invariants}");
|
||||
}
|
||||
if !domain.doc.is_empty() {
|
||||
println!("///");
|
||||
doc_comment(&domain.doc, 0);
|
||||
}
|
||||
println!(
|
||||
"pub type {} = {};\n",
|
||||
domain.name.to_upper_camel_case(),
|
||||
amqp_type_to_rust_type(&domain.kind),
|
||||
);
|
||||
let amqp = Amqp::from_str(&content).context("parse amqp spec file")?;
|
||||
|
||||
let transport_output =
|
||||
File::create(transport_generated_path).context("transport output file create")?;
|
||||
let core_output = File::create(core_generated_path).context("core output file create")?;
|
||||
|
||||
Codegen {
|
||||
output: Box::new(transport_output),
|
||||
}
|
||||
.transport_codegen(&amqp);
|
||||
|
||||
Codegen {
|
||||
output: Box::new(core_output),
|
||||
}
|
||||
.core_codegen(&amqp);
|
||||
|
||||
Ok(())
|
||||
}
|
||||
impl Codegen {
|
||||
fn transport_codegen(&mut self, amqp: &Amqp) {
|
||||
writeln!(self.output, "#![allow(dead_code)]").ok();
|
||||
writeln!(
|
||||
self.output,
|
||||
"// This file has been generated by `xtask/src/codegen`. Do not edit it manually.\n"
|
||||
)
|
||||
.ok();
|
||||
self.codegen_parser(amqp);
|
||||
self.codegen_write(amqp);
|
||||
self.codegen_random(amqp);
|
||||
}
|
||||
|
||||
fn codegen_class_defs(amqp: &Amqp) {
|
||||
println!("#[derive(Debug, Clone, PartialEq)]");
|
||||
println!("pub enum Method {{");
|
||||
fn core_codegen(&mut self, amqp: &Amqp) {
|
||||
writeln!(self.output, "#![allow(dead_code)]").ok();
|
||||
writeln!(
|
||||
self.output,
|
||||
"// This file has been generated by `xtask/src/codegen`. Do not edit it manually.\n"
|
||||
)
|
||||
.ok();
|
||||
self.codegen_domain_defs(amqp);
|
||||
self.codegen_class_defs(amqp);
|
||||
}
|
||||
|
||||
for class in &amqp.classes {
|
||||
let enum_name = class.name.to_upper_camel_case();
|
||||
for method in &class.methods {
|
||||
let method_name = method.name.to_upper_camel_case();
|
||||
doc_comment(&class.doc, 4);
|
||||
doc_comment(&method.doc, 4);
|
||||
print!(" {enum_name}{method_name}");
|
||||
if !method.fields.is_empty() {
|
||||
println!(" {{");
|
||||
for field in &method.fields {
|
||||
let field_name = snake_case(&field.name);
|
||||
let (field_type, field_docs) =
|
||||
get_invariants_with_type(field_type(field), field.asserts.as_ref());
|
||||
if !field_docs.is_empty() {
|
||||
println!(" /// {field_docs}");
|
||||
if !field.doc.is_empty() {
|
||||
println!(" ///");
|
||||
doc_comment(&field.doc, 8);
|
||||
}
|
||||
} else {
|
||||
doc_comment(&field.doc, 8);
|
||||
}
|
||||
println!(" {field_name}: {field_type},");
|
||||
fn codegen_domain_defs(&mut self, amqp: &Amqp) {
|
||||
for domain in &amqp.domains {
|
||||
let invariants = self.invariants(domain.asserts.iter());
|
||||
|
||||
if let Some(label) = &domain.label {
|
||||
writeln!(self.output, "/// {label}").ok();
|
||||
}
|
||||
|
||||
if !invariants.is_empty() {
|
||||
if domain.label.is_some() {
|
||||
writeln!(self.output, "///").ok();
|
||||
}
|
||||
println!(" }},");
|
||||
} else {
|
||||
println!(",");
|
||||
writeln!(self.output, "/// {invariants}").ok();
|
||||
}
|
||||
if !domain.doc.is_empty() {
|
||||
writeln!(self.output, "///").ok();
|
||||
self.doc_comment(&domain.doc, 0);
|
||||
}
|
||||
writeln!(
|
||||
self.output,
|
||||
"pub type {} = {};\n",
|
||||
domain.name.to_upper_camel_case(),
|
||||
self.amqp_type_to_rust_type(&domain.kind),
|
||||
)
|
||||
.ok();
|
||||
}
|
||||
}
|
||||
|
||||
println!("}}\n");
|
||||
}
|
||||
fn codegen_class_defs(&mut self, amqp: &Amqp) {
|
||||
writeln!(self.output, "#[derive(Debug, Clone, PartialEq)]").ok();
|
||||
writeln!(self.output, "pub enum Method {{").ok();
|
||||
|
||||
fn amqp_type_to_rust_type(amqp_type: &str) -> &'static str {
|
||||
match amqp_type {
|
||||
"octet" => "u8",
|
||||
"short" => "u16",
|
||||
"long" => "u32",
|
||||
"longlong" => "u64",
|
||||
"bit" => "bool",
|
||||
"shortstr" => "String",
|
||||
"longstr" => "Vec<u8>",
|
||||
"timestamp" => "u64",
|
||||
"table" => "super::Table",
|
||||
_ => unreachable!("invalid type {}", amqp_type),
|
||||
for class in &amqp.classes {
|
||||
let enum_name = class.name.to_upper_camel_case();
|
||||
for method in &class.methods {
|
||||
let method_name = method.name.to_upper_camel_case();
|
||||
self.doc_comment(&class.doc, 4);
|
||||
self.doc_comment(&method.doc, 4);
|
||||
write!(self.output, " {enum_name}{method_name}").ok();
|
||||
if !method.fields.is_empty() {
|
||||
writeln!(self.output, " {{").ok();
|
||||
for field in &method.fields {
|
||||
let field_name = self.snake_case(&field.name);
|
||||
let (field_type, field_docs) = self.get_invariants_with_type(
|
||||
self.field_type(field),
|
||||
field.asserts.as_ref(),
|
||||
);
|
||||
if !field_docs.is_empty() {
|
||||
writeln!(self.output, " /// {field_docs}").ok();
|
||||
if !field.doc.is_empty() {
|
||||
writeln!(self.output, " ///").ok();
|
||||
self.doc_comment(&field.doc, 8);
|
||||
}
|
||||
} else {
|
||||
self.doc_comment(&field.doc, 8);
|
||||
}
|
||||
writeln!(self.output, " {field_name}: {field_type},").ok();
|
||||
}
|
||||
writeln!(self.output, " }},").ok();
|
||||
} else {
|
||||
writeln!(self.output, ",").ok();
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
writeln!(self.output, "}}\n").ok();
|
||||
}
|
||||
}
|
||||
|
||||
fn field_type(field: &Field) -> &String {
|
||||
field.domain.as_ref().or(field.kind.as_ref()).unwrap()
|
||||
}
|
||||
|
||||
fn resolve_type_from_domain(amqp: &Amqp, domain: &str) -> String {
|
||||
amqp.domains
|
||||
.iter()
|
||||
.find(|d| d.name == domain)
|
||||
.map(|d| d.kind.clone())
|
||||
.unwrap()
|
||||
}
|
||||
|
||||
/// returns (type name, invariant docs)
|
||||
fn get_invariants_with_type(domain: &str, asserts: &[Assert]) -> (String, String) {
|
||||
let additional_docs = invariants(asserts.iter());
|
||||
|
||||
let type_name = domain.to_upper_camel_case();
|
||||
|
||||
(type_name, additional_docs)
|
||||
}
|
||||
|
||||
fn snake_case(ident: &str) -> String {
|
||||
use heck::ToSnakeCase;
|
||||
|
||||
if ident == "type" {
|
||||
"r#type".to_string()
|
||||
} else {
|
||||
ident.to_snake_case()
|
||||
fn amqp_type_to_rust_type(&self, amqp_type: &str) -> &'static str {
|
||||
match amqp_type {
|
||||
"octet" => "u8",
|
||||
"short" => "u16",
|
||||
"long" => "u32",
|
||||
"longlong" => "u64",
|
||||
"bit" => "bool",
|
||||
"shortstr" => "String",
|
||||
"longstr" => "Vec<u8>",
|
||||
"timestamp" => "u64",
|
||||
"table" => "super::Table",
|
||||
_ => unreachable!("invalid type {}", amqp_type),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn subsequent_bit_fields<'a>(
|
||||
amqp: &Amqp,
|
||||
bit_field: &'a Field,
|
||||
iter: &mut Peekable<impl Iterator<Item = &'a Field>>,
|
||||
) -> Vec<&'a Field> {
|
||||
let mut fields_with_bit = vec![bit_field];
|
||||
fn field_type<'a>(&self, field: &'a Field) -> &'a String {
|
||||
field.domain.as_ref().or(field.kind.as_ref()).unwrap()
|
||||
}
|
||||
|
||||
loop {
|
||||
if iter
|
||||
.peek()
|
||||
.map(|f| resolve_type_from_domain(amqp, field_type(f)) == "bit")
|
||||
.unwrap_or(false)
|
||||
{
|
||||
fields_with_bit.push(iter.next().unwrap());
|
||||
fn resolve_type_from_domain(&self, amqp: &Amqp, domain: &str) -> String {
|
||||
amqp.domains
|
||||
.iter()
|
||||
.find(|d| d.name == domain)
|
||||
.map(|d| d.kind.clone())
|
||||
.unwrap()
|
||||
}
|
||||
|
||||
/// returns (type name, invariant docs)
|
||||
fn get_invariants_with_type(&self, domain: &str, asserts: &[Assert]) -> (String, String) {
|
||||
let additional_docs = self.invariants(asserts.iter());
|
||||
|
||||
let type_name = domain.to_upper_camel_case();
|
||||
|
||||
(type_name, additional_docs)
|
||||
}
|
||||
|
||||
fn snake_case(&self, ident: &str) -> String {
|
||||
use heck::ToSnakeCase;
|
||||
|
||||
if ident == "type" {
|
||||
"r#type".to_string()
|
||||
} else {
|
||||
break;
|
||||
ident.to_snake_case()
|
||||
}
|
||||
}
|
||||
fields_with_bit
|
||||
}
|
||||
|
||||
fn invariants<'a>(asserts: impl Iterator<Item = &'a Assert>) -> String {
|
||||
asserts
|
||||
.map(|assert| match &*assert.check {
|
||||
"notnull" => "must not be null".to_string(),
|
||||
"length" => format!("must be shorter than {}", assert.value.as_ref().unwrap()),
|
||||
"regexp" => format!("must match `{}`", assert.value.as_ref().unwrap()),
|
||||
"le" => {
|
||||
format!(
|
||||
"must be less than the {} field of the method {}",
|
||||
assert.method.as_ref().unwrap(),
|
||||
assert.field.as_ref().unwrap()
|
||||
)
|
||||
fn subsequent_bit_fields<'a>(
|
||||
&self,
|
||||
bit_field: &'a Field,
|
||||
iter: &mut Peekable<impl Iterator<Item = &'a Field>>,
|
||||
amqp: &Amqp,
|
||||
) -> Vec<&'a Field> {
|
||||
let mut fields_with_bit = vec![bit_field];
|
||||
|
||||
loop {
|
||||
if iter
|
||||
.peek()
|
||||
.map(|f| self.resolve_type_from_domain(amqp, self.field_type(f)) == "bit")
|
||||
.unwrap_or(false)
|
||||
{
|
||||
fields_with_bit.push(iter.next().unwrap());
|
||||
} else {
|
||||
break;
|
||||
}
|
||||
_ => unimplemented!(),
|
||||
})
|
||||
.collect::<Vec<_>>()
|
||||
.join(", ")
|
||||
}
|
||||
|
||||
fn doc_comment(docs: &[Doc], indent: usize) {
|
||||
for doc in docs {
|
||||
if doc.kind == Some("grammar".to_string()) {
|
||||
continue;
|
||||
}
|
||||
for line in doc.text.lines() {
|
||||
let line = line.trim();
|
||||
if !line.is_empty() {
|
||||
let indent = " ".repeat(indent);
|
||||
println!("{indent}/// {line}");
|
||||
fields_with_bit
|
||||
}
|
||||
|
||||
fn invariants<'a>(&self, asserts: impl Iterator<Item = &'a Assert>) -> String {
|
||||
asserts
|
||||
.map(|assert| match &*assert.check {
|
||||
"notnull" => "must not be null".to_string(),
|
||||
"length" => format!("must be shorter than {}", assert.value.as_ref().unwrap()),
|
||||
"regexp" => format!("must match `{}`", assert.value.as_ref().unwrap()),
|
||||
"le" => {
|
||||
format!(
|
||||
"must be less than the {} field of the method {}",
|
||||
assert.method.as_ref().unwrap(),
|
||||
assert.field.as_ref().unwrap()
|
||||
)
|
||||
}
|
||||
_ => unimplemented!(),
|
||||
})
|
||||
.collect::<Vec<_>>()
|
||||
.join(", ")
|
||||
}
|
||||
|
||||
fn doc_comment(&mut self, docs: &[Doc], indent: usize) {
|
||||
for doc in docs {
|
||||
if doc.kind == Some("grammar".to_string()) {
|
||||
continue;
|
||||
}
|
||||
for line in doc.text.lines() {
|
||||
let line = line.trim();
|
||||
if !line.is_empty() {
|
||||
let indent = " ".repeat(indent);
|
||||
writeln!(self.output, "{indent}/// {line}").ok();
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue