mirror of
https://github.com/github/codeql.git
synced 2025-12-17 17:23:36 +01:00
13 lines
418 B
Rust
13 lines
418 B
Rust
use proc_macro::TokenStream;
|
|
use quote::quote;
|
|
|
|
#[proc_macro_attribute]
|
|
pub fn add_suffix(attr: TokenStream, item: TokenStream) -> TokenStream {
|
|
let suff = syn::parse_macro_input!(attr as syn::LitStr).value();
|
|
let mut ast = syn::parse_macro_input!(item as syn::ItemFn);
|
|
ast.sig.ident = syn::Ident::new(&format!("{}_{}", ast.sig.ident, suff), ast.sig.ident.span());
|
|
quote! {
|
|
#ast
|
|
}.into()
|
|
}
|