unspeaker/tengri
1
0
Fork 0
mirror of https://codeberg.org/unspeaker/tengri.git synced 2025-12-06 03:36:42 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

proc: working command, expose

Browse source
This commit is contained in:
🪞👃🪞 2025-05-08 13:46:29 +03:00
parent 751e01a41e
commit 046be9a9e1
2 changed files with 24 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

5
proc/src/proc_command.rs
View file

@ -62,6 +62,7 @@ impl ToTokens for CommandDef {
let implementations = exposed.values().map(CommandArm::to_implementation); let implementations = exposed.values().map(CommandArm::to_implementation);
write_quote_to(out, quote! { write_quote_to(out, quote! {
/// Generated by [tengri_proc]. /// Generated by [tengri_proc].
#[derive(Clone, Debug)]
pub enum #enumeration { pub enum #enumeration {
#(#variants)* #(#variants)*
} }
@ -123,9 +124,11 @@ impl CommandArm {
if with_values { if with_values {
let take_err = LitStr::new(&format!("{}: missing argument \"{}\" ({})", let take_err = LitStr::new(&format!("{}: missing argument \"{}\" ({})",
quote!{#ident}, quote!{#pat}, quote!{#ty}), Span::call_site()); quote!{#ident}, quote!{#pat}, quote!{#ty}), Span::call_site());
let give_err = LitStr::new(&format!("{}: missing value \"{}\" ({})",
quote!{#ident}, quote!{#pat}, quote!{#ty}), Span::call_site());
write_quote_to(&mut out, quote! { write_quote_to(&mut out, quote! {
: Context::get(state, &iter.next().expect(#take_err).value) : Context::get(state, &iter.next().expect(#take_err).value)
}); .expect(#give_err) });
} }
} else { } else {
unreachable!("only typed args should be present at this position") unreachable!("only typed args should be present at this position")

32
proc/src/proc_expose.rs
View file

@ -70,23 +70,31 @@ impl ToTokens for ExposeImpl {
out.append(token); out.append(token);
} }
for (t, variants) in exposed.iter() { for (t, variants) in exposed.iter() {
let predefined = match format!("{}", quote! { #t }).as_str() { let formatted_type = format!("{}", quote! { #t });
"bool" => vec![ let predefined = match formatted_type.as_str() {
quote! { ::tengri::dsl::Value::Sym(":true") => true }, "bool" => quote! {
quote! { ::tengri::dsl::Value::Sym(":false") => false }, ::tengri::dsl::Value::Sym(":true") => true,
], ::tengri::dsl::Value::Sym(":false") => false,
},
"u8" | "u16" | "u32" | "u64" | "usize" | "u8" | "u16" | "u32" | "u64" | "usize" |
"i8" | "i16" | "i32" | "i64" | "isize" => vec![ "i8" | "i16" | "i32" | "i64" | "isize" => {
quote! { ::tengri::dsl::Value::Num(n) => *n }, let num_err = LitStr::new(
], &format!("{{n}}: failed to convert to {formatted_type}"),
_ => vec![], Span::call_site()
);
quote! {
::tengri::dsl::Value::Num(n) => TryInto::<#t>::try_into(*n)
.unwrap_or_else(|_|panic!(#num_err)),
}
},
_ => quote! {},
}; };
let values = variants.iter().map(|(k, v)|ExposeArm(k.clone(), v.clone())); let values = variants.iter().map(|(k, v)|ExposeArm(k.clone(), v.clone()));
let trait_impl = quote! { let trait_impl = quote! {
impl ::tengri::dsl::Context<#t> for #target { impl ::tengri::dsl::Context<#t> for #target {
fn get (&self, dsl: &::tengri::dsl::Value) -> Option<#t> { fn get (&self, dsl: &::tengri::dsl::Value) -> Option<#t> {
Some(match dsl { Some(match dsl {
#(#predefined,)* #predefined
#(#values,)* #(#values,)*
_ => return None _ => return None
}) })
@ -97,9 +105,9 @@ impl ToTokens for ExposeImpl {
out.append(token); out.append(token);
} }
} }
//if exposed.len() > 0 { if exposed.len() > 0 {
//panic!("{}", quote! {#out}); //panic!("{}", quote! {#out});
//} }
} }
} }