syndicate-lang
/
syndicate-rs
1
0
Fork 0
Code Issues 2 Pull Requests Projects Releases Wiki Activity

Be more proc_macro2-centric

This commit is contained in:
Tony Garnock-Jones 2021-08-13 06:51:20 -04:00
parent bb519b625b
commit dc0a4cc1ab
1 changed files with 35 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

64
syndicate-macros/src/lib.rs
View File

@ -4,8 +4,9 @@ use syndicate::value::Value;
use syndicate::value::text::iovalue_from_str; use syndicate::value::text::iovalue_from_str;
extern crate proc_macro; extern crate proc_macro;
use proc_macro::Span;
use proc_macro::TokenStream; use proc_macro2::Span;
use proc_macro2::TokenStream;
use quote::ToTokens; use quote::ToTokens;
use quote::quote; use quote::quote;
@ -21,7 +22,17 @@ use syn::LitByteStr;
fn lit<T: ToTokens>(e: T) -> TokenStream { fn lit<T: ToTokens>(e: T) -> TokenStream {
quote!( quote!(
syndicate::schemas::dataspace_patterns::Pattern::DLit(Box::new( syndicate::schemas::dataspace_patterns::Pattern::DLit(Box::new(
syndicate::schemas::dataspace_patterns::DLit { value: #e }))).into() syndicate::schemas::dataspace_patterns::DLit { value: #e })))
}
fn compile_sequence_members(vs: &[IOValue]) -> Vec<TokenStream> {
let mut i = 0;
vs.iter().map(|f| {
let p = compile_pattern(f);
let result = quote!((#i .into(), #p));
i += 1;
result
}).collect::<Vec<_>>()
} }
fn compile_pattern(v: &IOValue) -> TokenStream { fn compile_pattern(v: &IOValue) -> TokenStream {
@ -46,16 +57,16 @@ fn compile_pattern(v: &IOValue) -> TokenStream {
} }
Value::String(s) => lit(quote!(#V_::Value::from(#s).wrap())), Value::String(s) => lit(quote!(#V_::Value::from(#s).wrap())),
Value::ByteString(bs) => { Value::ByteString(bs) => {
let bs = LitByteStr::new(bs, Span::call_site().into()); let bs = LitByteStr::new(bs, Span::call_site());
lit(quote!(#V_::Value::from(#bs).wrap())) lit(quote!(#V_::Value::from(#bs).wrap()))
} }
Value::Symbol(s) => match s.as_str() { Value::Symbol(s) => match s.as_str() {
"$" => quote!(#P_::Pattern::DBind(Box::new(#P_::DBind { "$" => quote!(#P_::Pattern::DBind(Box::new(#P_::DBind {
pattern: #P_::Pattern::DDiscard(Box::new(#P_::DDiscard)) pattern: #P_::Pattern::DDiscard(Box::new(#P_::DDiscard))
}))).into(), }))),
"_" => quote!(#P_::Pattern::DDiscard(Box::new(#P_::DDiscard))).into(), "_" => quote!(#P_::Pattern::DDiscard(Box::new(#P_::DDiscard))),
_ => if s.starts_with("=") { _ => if s.starts_with("=") {
lit(Ident::new(&s[1..], Span::call_site().into())) lit(Ident::new(&s[1..], Span::call_site()))
} else { } else {
lit(quote!(#V_::Value::symbol(#s).wrap())) lit(quote!(#V_::Value::symbol(#s).wrap()))
}, },
@ -66,33 +77,28 @@ fn compile_pattern(v: &IOValue) -> TokenStream {
None => panic!("Record labels in patterns must be symbols"), None => panic!("Record labels in patterns must be symbols"),
Some(label) => { Some(label) => {
let label_stx = if label.starts_with("=") { let label_stx = if label.starts_with("=") {
let id = Ident::new(&label[1..], Span::call_site().into()); let id = Ident::new(&label[1..], Span::call_site());
quote!(#id) quote!(#id)
} else { } else {
quote!(#V_::Value::symbol(#label).wrap()) quote!(#V_::Value::symbol(#label).wrap())
}; };
let mut i = 0; let members = compile_sequence_members(r.fields());
let members = r.fields().iter().map( quote!(#P_::Pattern::DCompound(Box::new(#P_::DCompound::Rec {
|f| { ctor: Box::new(#P_::CRec {
let p: proc_macro2::TokenStream = compile_pattern(f).into(); label: #label_stx,
let result = quote!((#i .into(), #p)); arity: #arity .into(),
i += 1; }),
result members: #V_::Map::from_iter(vec![#(#members),*])
}).collect::<Vec<_>>(); })))
quote!(
#P_::Pattern::DCompound(Box::new(
#P_::DCompound::Rec {
ctor: Box::new(#P_::CRec {
label: #label_stx,
arity: #arity .into(),
}),
members: #V_::Map::from_iter(vec![#(#members),*])
}))).into()
} }
} }
} }
// Value::Sequence(vs) => , // Value::Sequence(vs) => {
// Value::Set(_) => panic!("Cannot match sets in patterns"), // let arity = vs.len() as u128;
// let mut i = 0;
// let members = compile_sequence_members(vs);
// quote!(
Value::Set(_) => panic!("Cannot match sets in patterns"),
// Value::Dictionary(d) => , // Value::Dictionary(d) => ,
// Value::Embedded(e) => panic!("aiee"), // Value::Embedded(e) => panic!("aiee"),
_ => todo!(), _ => todo!(),
@ -100,13 +106,13 @@ fn compile_pattern(v: &IOValue) -> TokenStream {
} }
#[proc_macro] #[proc_macro]
pub fn pattern(src: TokenStream) -> TokenStream { pub fn pattern(src: proc_macro::TokenStream) -> proc_macro::TokenStream {
if let Lit::Str(s) = parse_macro_input!(src as ExprLit).lit { if let Lit::Str(s) = parse_macro_input!(src as ExprLit).lit {
match iovalue_from_str(&s.value()) { match iovalue_from_str(&s.value()) {
Ok(v) => { Ok(v) => {
let e = compile_pattern(&v); let e = compile_pattern(&v);
// println!("{:#}", &e); // println!("{:#}", &e);
return e; return e.into();
} }
Err(_) => (), Err(_) => (),
} }