mirror of
https://codeberg.org/unspeaker/tengri.git
synced 2025-12-06 11:46:42 +01:00
This commit is contained in:
parent
b543c43e68
commit
cba23a005c
3 changed files with 417 additions and 34 deletions
392
proc/src/proc_expose.rs
Normal file
392
proc/src/proc_expose.rs
Normal file
|
|
@ -0,0 +1,392 @@
|
|||
use crate::*;
|
||||
use syn::parse::discouraged::Speculative;
|
||||
|
||||
pub(crate) fn expose_impl (meta: TokenStream, data: TokenStream) -> TokenStream {
|
||||
let mut out = TokenStream2::new();
|
||||
ExposeDefinition {
|
||||
meta: parse_macro_input!(meta as ExposeMeta),
|
||||
data: parse_macro_input!(data as ExposeImpl),
|
||||
}.to_tokens(&mut out);
|
||||
out.into()
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
struct ExposeDefinition {
|
||||
meta: ExposeMeta,
|
||||
data: ExposeImpl,
|
||||
}
|
||||
|
||||
impl Parse for ExposeDefinition {
|
||||
fn parse (input: ParseStream) -> Result<Self> {
|
||||
Ok(Self {
|
||||
meta: input.parse::<ExposeMeta>()?,
|
||||
data: input.parse::<ExposeImpl>()?,
|
||||
})
|
||||
}
|
||||
}
|
||||
|
||||
impl ToTokens for ExposeDefinition {
|
||||
fn to_tokens (&self, out: &mut TokenStream2) {
|
||||
let Self { meta, data } = self;
|
||||
for token in quote! { #data } {
|
||||
out.append(token)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
struct ExposeMeta {}
|
||||
|
||||
impl Parse for ExposeMeta {
|
||||
fn parse (input: ParseStream) -> Result<Self> {
|
||||
Ok(Self {})
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
struct ExposeImpl {
|
||||
target: Ident,
|
||||
items: Vec<ImplItem>,
|
||||
types: BTreeMap<ExposeType, BTreeMap<ExposeSym, ExposeArm>>,
|
||||
}
|
||||
|
||||
impl Parse for ExposeImpl {
|
||||
fn parse (input: ParseStream) -> Result<Self> {
|
||||
let _impl = input.parse::<Token![impl]>()?;
|
||||
let target = input.parse::<Ident>()?;
|
||||
let group;
|
||||
let brace = braced!(group in input);
|
||||
let mut items = vec![];
|
||||
let mut types: BTreeMap<ExposeType, BTreeMap<ExposeSym, ExposeArm>> = Default::default();
|
||||
while !group.is_empty() {
|
||||
let fork = group.fork();
|
||||
if let Ok(block) = fork.parse::<ExposeBlock>() {
|
||||
let t = block.type_.into();
|
||||
if let Some(values) = types.get_mut(&t) {
|
||||
for (key, value) in block.values.into_iter() {
|
||||
if values.contains_key(&key) {
|
||||
return Err(input.error(format!("{key:?} ({t:?}): already exists")))
|
||||
} else {
|
||||
values.insert(key, value);
|
||||
}
|
||||
}
|
||||
} else {
|
||||
types.insert(t, block.values);
|
||||
}
|
||||
group.advance_to(&fork);
|
||||
continue
|
||||
}
|
||||
let fork = group.fork();
|
||||
if let Ok(item) = fork.parse::<ImplItem>() {
|
||||
items.push(item);
|
||||
group.advance_to(&fork);
|
||||
continue
|
||||
}
|
||||
return Err(input.error(
|
||||
"expected either item or #[tengri::expose(type)] { \":key\" => value }"
|
||||
));
|
||||
}
|
||||
Ok(Self { target, items, types })
|
||||
}
|
||||
}
|
||||
|
||||
impl ToTokens for ExposeImpl {
|
||||
fn to_tokens (&self, out: &mut TokenStream2) {
|
||||
let Self { target, items, types } = self;
|
||||
for token in quote! { impl #target { #(#items)* } } {
|
||||
out.append(token);
|
||||
}
|
||||
for (t, variants) in types.iter() {
|
||||
let predef = match format!("{}", quote! { #t }).as_str() {
|
||||
"bool" => vec![
|
||||
quote! { ::tengri::dsl::Value::Sym(":true") => true },
|
||||
quote! { ::tengri::dsl::Value::Sym(":false") => false },
|
||||
],
|
||||
"u8" | "u16" | "u32" | "u64" | "usize" |
|
||||
"i8" | "i16" | "i32" | "i64" | "isize" => vec![
|
||||
quote! { ::tengri::dsl::Value::Num(n) => *n },
|
||||
],
|
||||
_ => vec![],
|
||||
};
|
||||
let values = variants.values();
|
||||
let trait_impl = quote! {
|
||||
impl ::tengri::dsl::Context<#t> for #target {
|
||||
fn get (&self, dsl: &::tengri::dsl::Value) -> Option<#t> {
|
||||
Some(match dsl {
|
||||
#(#predef,)*
|
||||
#(#values,)*
|
||||
_ => return None
|
||||
})
|
||||
}
|
||||
}
|
||||
};
|
||||
for token in trait_impl {
|
||||
out.append(token);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
struct ExposeBlock {
|
||||
type_: Type,
|
||||
values: BTreeMap<ExposeSym, ExposeArm>,
|
||||
}
|
||||
|
||||
impl Parse for ExposeBlock {
|
||||
fn parse (input: ParseStream) -> Result<Self> {
|
||||
let _ = input.parse::<Token![#]>()?;
|
||||
|
||||
let group;
|
||||
let bracket = bracketed!(group in input);
|
||||
let path = group.parse::<Path>()?;
|
||||
let type_ = if
|
||||
path.segments.get(0).map(|x|x.ident.to_string()) == Some("tengri".to_string()) &&
|
||||
path.segments.get(1).map(|x|x.ident.to_string()) == Some("expose".to_string())
|
||||
{
|
||||
let token;
|
||||
let paren = parenthesized!(token in group);
|
||||
token.parse::<Type>()?
|
||||
} else {
|
||||
return Err(input.error("expected #[tengri::expose(type)]"))
|
||||
};
|
||||
|
||||
let group;
|
||||
let brace = braced!(group in input);
|
||||
let mut values = BTreeMap::new();
|
||||
while !group.is_empty() {
|
||||
let arm = group.parse::<ExposeArm>()?;
|
||||
values.insert(arm.key.clone(), arm);
|
||||
let _ = group.parse::<Token![,]>()?;
|
||||
}
|
||||
Ok(Self { type_, values })
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
struct ExposeArm {
|
||||
key: ExposeSym,
|
||||
value: Expr
|
||||
}
|
||||
|
||||
impl Parse for ExposeArm {
|
||||
fn parse (input: ParseStream) -> Result<Self> {
|
||||
let key = input.parse::<LitStr>()?.into();
|
||||
let _ = input.parse::<Token![=]>()?;
|
||||
let _ = input.parse::<Token![>]>()?;
|
||||
let value = input.parse::<Expr>()?;
|
||||
Ok(Self { key, value })
|
||||
}
|
||||
}
|
||||
|
||||
impl ToTokens for ExposeArm {
|
||||
fn to_tokens (&self, out: &mut TokenStream2) {
|
||||
let Self { key, value } = self;
|
||||
out.append(Punct::new(':', Joint));
|
||||
out.append(Punct::new(':', Alone));
|
||||
out.append(Ident::new("tengri", Span::call_site()));
|
||||
out.append(Punct::new(':', Joint));
|
||||
out.append(Punct::new(':', Alone));
|
||||
out.append(Ident::new("dsl", Span::call_site()));
|
||||
out.append(Punct::new(':', Joint));
|
||||
out.append(Punct::new(':', Alone));
|
||||
out.append(Ident::new("Value", Span::call_site()));
|
||||
out.append(Punct::new(':', Joint));
|
||||
out.append(Punct::new(':', Alone));
|
||||
out.append(Ident::new("Sym", Span::call_site()));
|
||||
out.append(Group::new(Delimiter::Parenthesis, {
|
||||
let mut out = TokenStream2::new();
|
||||
out.append(key.0.token());
|
||||
out
|
||||
}));
|
||||
out.append(Punct::new('=', Joint));
|
||||
out.append(Punct::new('>', Alone));
|
||||
for token in quote! { #value } {
|
||||
out.append(token);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
struct ExposeSym(LitStr);
|
||||
|
||||
impl From<LitStr> for ExposeSym {
|
||||
fn from (this: LitStr) -> Self {
|
||||
Self(this)
|
||||
}
|
||||
}
|
||||
|
||||
impl PartialOrd for ExposeSym {
|
||||
fn partial_cmp (&self, other: &Self) -> Option<Ordering> {
|
||||
let this = &self.0;
|
||||
let that = &other.0;
|
||||
Some(format!("{}", quote! { #this }).cmp(&format!("{}", quote! { #that })))
|
||||
}
|
||||
}
|
||||
|
||||
impl Ord for ExposeSym {
|
||||
fn cmp (&self, other: &Self) -> Ordering {
|
||||
let this = &self.0;
|
||||
let that = &other.0;
|
||||
format!("{}", quote! { #this }).cmp(&format!("{}", quote! { #that }))
|
||||
}
|
||||
}
|
||||
|
||||
impl PartialEq for ExposeSym {
|
||||
fn eq (&self, other: &Self) -> bool {
|
||||
let this = &self.0;
|
||||
let that = &other.0;
|
||||
format!("{}", quote! { #this }) == format!("{}", quote! { #that })
|
||||
}
|
||||
}
|
||||
|
||||
impl Eq for ExposeSym {}
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
struct ExposeType(Type);
|
||||
|
||||
impl From<Type> for ExposeType {
|
||||
fn from (this: Type) -> Self {
|
||||
Self(this)
|
||||
}
|
||||
}
|
||||
|
||||
impl PartialOrd for ExposeType {
|
||||
fn partial_cmp (&self, other: &Self) -> Option<Ordering> {
|
||||
let this = &self.0;
|
||||
let that = &other.0;
|
||||
Some(format!("{}", quote! { #this }).cmp(&format!("{}", quote! { #that })))
|
||||
}
|
||||
}
|
||||
|
||||
impl Ord for ExposeType {
|
||||
fn cmp (&self, other: &Self) -> Ordering {
|
||||
let this = &self.0;
|
||||
let that = &other.0;
|
||||
format!("{}", quote! { #this }).cmp(&format!("{}", quote! { #that }))
|
||||
}
|
||||
}
|
||||
|
||||
impl PartialEq for ExposeType {
|
||||
fn eq (&self, other: &Self) -> bool {
|
||||
let this = &self.0;
|
||||
let that = &other.0;
|
||||
format!("{}", quote! { #this }) == format!("{}", quote! { #that })
|
||||
}
|
||||
}
|
||||
|
||||
impl Eq for ExposeType {}
|
||||
|
||||
impl ToTokens for ExposeType {
|
||||
fn to_tokens (&self, out: &mut TokenStream2) {
|
||||
self.0.to_tokens(out)
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)] #[test] fn test_expose_definition () {
|
||||
let parsed: ExposeImpl = pq! {
|
||||
//#[tengri_proc::expose]
|
||||
impl Something {
|
||||
#[tengri::expose(bool)] {
|
||||
":bool1" => true || false,
|
||||
}
|
||||
fn something () {}
|
||||
}
|
||||
};
|
||||
// FIXME:
|
||||
//assert_eq!(
|
||||
//format!("{}", quote! { #parsed }),
|
||||
//format!("{}", quote! {
|
||||
//impl Something {
|
||||
//fn something () {}
|
||||
//}
|
||||
//impl ::tengri::Context<bool> for Something {
|
||||
//fn get (&self, dsl: &::tengri::Value) -> Option<bool> {
|
||||
//Some(match dsl {
|
||||
//::tengri::Value::Sym(":true") => true,
|
||||
//::tengri::Value::Sym(":false") => false,
|
||||
//::tengri::Value::Sym(":bool1") => true || false,
|
||||
//_ => return None
|
||||
//})
|
||||
//}
|
||||
//}
|
||||
//})
|
||||
//);
|
||||
|
||||
let parsed: ExposeImpl = pq! {
|
||||
//#[tengri_proc::expose]
|
||||
impl Something {
|
||||
#[tengri::expose(bool)] {
|
||||
":bool1" => true || false,
|
||||
}
|
||||
#[tengri::expose(u16)] {
|
||||
":u161" => 0 + 1,
|
||||
}
|
||||
#[tengri::expose(usize)] {
|
||||
":usize1" => 1 + 2,
|
||||
}
|
||||
#[tengri::expose(Arc<str>)] {
|
||||
":arcstr1" => "foo".into(),
|
||||
}
|
||||
#[tengri::expose(Option<Arc<str>>)] {
|
||||
":optarcstr1" => Some("bar".into()),
|
||||
":optarcstr2" => Some("baz".into()),
|
||||
}
|
||||
fn something () {}
|
||||
}
|
||||
};
|
||||
// FIXME:
|
||||
//assert_eq!(
|
||||
//format!("{}", quote! { #parsed }),
|
||||
//format!("{}", quote! {
|
||||
//impl Something {
|
||||
//fn something () {}
|
||||
//}
|
||||
//impl ::tengri::Context<Arc<str>> for Something {
|
||||
//fn get (&self, dsl: &::tengri::Value) -> Option<Arc<str>> {
|
||||
//Some(match dsl {
|
||||
//::tengri::Value::Sym(":arcstr1") => "foo".into(),
|
||||
//_ => return None
|
||||
//})
|
||||
//}
|
||||
//}
|
||||
//impl ::tengri::Context<Option<Arc<str>>> for Something {
|
||||
//fn get (&self, dsl: &::tengri::Value) -> Option<Option<Arc<str>>> {
|
||||
//Some(match dsl {
|
||||
//::tengri::Value::Sym(":optarcstr1") => Some("bar".into()),
|
||||
//::tengri::Value::Sym(":optarcstr2") => Some("baz".into()),
|
||||
//_ => return None
|
||||
//})
|
||||
//}
|
||||
//}
|
||||
//impl ::tengri::Context<bool> for Something {
|
||||
//fn get (&self, dsl: &::tengri::Value) -> Option<bool> {
|
||||
//Some(match dsl {
|
||||
//::tengri::Value::Sym(":true") => true,
|
||||
//::tengri::Value::Sym(":false") => false,
|
||||
//::tengri::Value::Sym(":bool1") => true || false,
|
||||
//_ => return None
|
||||
//})
|
||||
//}
|
||||
//}
|
||||
//impl ::tengri::Context<u16> for Something {
|
||||
//fn get (&self, dsl: &::tengri::Value) -> Option<u16> {
|
||||
//Some(match dsl {
|
||||
//::tengri::Value::Num(n) => *n as u16,
|
||||
//::tengri::Value::Sym(":u161") => 0 + 1,
|
||||
//_ => return None
|
||||
//})
|
||||
//}
|
||||
//}
|
||||
//impl ::tengri::Context<usize> for Something {
|
||||
//fn get (&self, dsl: &::tengri::Value) -> Option<usize> {
|
||||
//Some(match dsl {
|
||||
//::tengri::Value::Num(n) => *n as usize,
|
||||
//::tengri::Value::Sym(":usize1") => 1 + 2,
|
||||
//_ => return None
|
||||
//})
|
||||
//}
|
||||
//}
|
||||
//})
|
||||
//)
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue