1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use primitive_types::H160;
use proc_macro::TokenStream;
use std::str::FromStr;
use syn::{parse_macro_input, LitStr};

#[proc_macro]
pub fn addr(input: TokenStream) -> TokenStream {
    let input = parse_macro_input!(input as LitStr);
    let address =
        H160::from_str(input.value().as_str()).expect("the string can't be parsed as Address");
    let bytes: Vec<String> = address
        .as_fixed_bytes()
        .into_iter()
        .map(|b| format!("{}u8", b))
        .collect();
    format!("primitive_types::H160([{}])", bytes.join(","))
        .parse()
        .unwrap()
}