Noratrieb/haesli
1
0
Fork 0
mirror of https://github.com/Noratrieb/haesli.git synced 2026-01-17 05:05:03 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

split generated code so that methods are now in core

Browse source
This commit is contained in:
nora 2022-02-20 21:22:30 +01:00
parent 3b656b911a
commit c333f20531
20 changed files with 1337 additions and 1206 deletions
Download patch file Download diff file Expand all files Collapse all files

85
xtask/src/codegen/random.rs
View file

@ -1,59 +1,78 @@
use super::{snake_case, Amqp};
use crate::codegen::{Amqp, Codegen};
use heck::ToUpperCamelCase;
pub(super) fn codegen_random(amqp: &Amqp) {
println!(
"
impl Codegen {
pub fn codegen_random(&mut self, amqp: &Amqp) {
writeln!(
self.output,
"
mod random {{
use rand::Rng;
use amqp_core::methods::*;
use crate::methods::RandomMethod;
use super::*;
"
);
)
.ok();
writeln!(
self.output,
"impl<R: Rng> RandomMethod<R> for Method {{
#[allow(unused_variables)]
fn random(rng: &mut R) -> Self {{"
)
.ok();
impl_random("Method", || {
let class_lens = amqp.classes.len();
println!(" match rng.gen_range(0u32..{class_lens}) {{");
writeln!(
self.output,
" match rng.gen_range(0u32..{class_lens}) {{"
)
.ok();
for (i, class) in amqp.classes.iter().enumerate() {
let class_name = class.name.to_upper_camel_case();
println!(" {i} => {{");
writeln!(self.output, " {i} => {{").ok();
let method_len = class.methods.len();
println!(" match rng.gen_range(0u32..{method_len}) {{");
writeln!(
self.output,
" match rng.gen_range(0u32..{method_len}) {{"
)
.ok();
for (i, method) in class.methods.iter().enumerate() {
let method_name = method.name.to_upper_camel_case();
println!(" {i} => Method::{class_name}{method_name} {{");
writeln!(
self.output,
" {i} => Method::{class_name}{method_name} {{"
)
.ok();
for field in &method.fields {
let field_name = snake_case(&field.name);
println!(" {field_name}: RandomMethod::random(rng),");
let field_name = self.snake_case(&field.name);
writeln!(
self.output,
" {field_name}: RandomMethod::random(rng),"
)
.ok();
}
println!(" }},");
writeln!(self.output, " }},").ok();
}
println!(
writeln!(
self.output,
" _ => unreachable!(),
}}"
);
)
.ok();
println!(" }}");
writeln!(self.output, " }}").ok();
}
println!(
writeln!(
self.output,
" _ => unreachable!(),
}}"
);
});
)
.ok();
writeln!(self.output, " }}\n}}").ok();
println!("}}");
}
fn impl_random(name: &str, body: impl FnOnce()) {
println!(
"impl<R: Rng> RandomMethod<R> for {name} {{
#[allow(unused_variables)]
fn random(rng: &mut R) -> Self {{"
);
body();
println!(" }}\n}}");
writeln!(self.output, "}}").ok();
}
}