rialo-sol-attribute-program 0.11.0

Sol attribute macro for defining a program
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41

// Copyright (c) Subzero Labs, Inc.
// SPDX-License-Identifier: Apache-2.0

use quote::{format_ident, quote, ToTokens};
use rialo_sol_lang_idl::types::{Idl, IdlType};

use super::common::{convert_idl_type_to_syn_type, gen_docs};

pub fn gen_constants_mod(idl: &Idl) -> proc_macro2::TokenStream {
    let constants = idl.constants.iter().map(|c| {
        let name = format_ident!("{}", c.name);
        let docs = gen_docs(&c.docs);
        let val = syn::parse_str::<syn::Expr>(&c.value)
            .unwrap()
            .to_token_stream();
        let (ty, val) = match &c.ty {
            IdlType::Bytes => (quote!(&[u8]), quote! { &#val }),
            IdlType::String => (quote!(&str), val),
            IdlType::Pubkey => {
                let pubkey_str = c.value.to_string();
                (
                    quote!(rialo_sol_lang::prelude::Pubkey),
                    quote!(rialo_sol_lang::prelude::Pubkey::from_str_const(#pubkey_str)),
                )
            }
            _ => (convert_idl_type_to_syn_type(&c.ty).to_token_stream(), val),
        };

        quote! {
            #docs
            pub const #name: #ty = #val;
        }
    });

    quote! {
        /// Program constants.
        pub mod constants {
            #(#constants)*
        }
    }
}