bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

Make #[custom_encodable] an attribute for newtype_index

Browse source 
Makes the syntax a little more rusty.
This commit is contained in:
Nilstrieb 2022-12-18 21:02:14 +01:00
parent b4d739ef12
commit 88d5f7f4ce
6 changed files with 19 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

22
compiler/rustc_macros/src/newtype.rs
View file

@ -6,7 +6,6 @@ use syn::*;
mod kw {
syn::custom_keyword!(DEBUG_FORMAT);
syn::custom_keyword!(MAX);
syn::custom_keyword!(ENCODABLE);
syn::custom_keyword!(custom);
syn::custom_keyword!(ORD_IMPL);
}
@ -27,7 +26,7 @@ struct Newtype(TokenStream);
impl Parse for Newtype {
fn parse(input: ParseStream<'_>) -> Result<Self> {
let attrs = input.call(Attribute::parse_outer)?;
let mut attrs = input.call(Attribute::parse_outer)?;
let vis: Visibility = input.parse()?;
input.parse::<Token![struct]>()?;
let name: Ident = input.parse()?;
@ -51,6 +50,17 @@ impl Parse for Newtype {
Ok(())
};
attrs.retain(|attr| match attr.path.get_ident() {
Some(ident) => match &*ident.to_string() {
"custom_encodable" => {
encodable = false;
false
}
_ => true,
},
_ => true,
});
if body.lookahead1().peek(Token![..]) {
body.parse::<Token![..]>()?;
} else {
@ -81,14 +91,6 @@ impl Parse for Newtype {
}
continue;
}
if body.lookahead1().peek(kw::ENCODABLE) {
body.parse::<kw::ENCODABLE>()?;
body.parse::<Token![=]>()?;
body.parse::<kw::custom>()?;
try_comma()?;
encodable = false;
continue;
}
if body.lookahead1().peek(kw::ORD_IMPL) {
body.parse::<kw::ORD_IMPL>()?;
body.parse::<Token![=]>()?;