shank_render 0.4.8

Renders implementaions derived from shank macros
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

use proc_macro2::TokenStream;
use shank_macro_impl::{
    account::extract_account_structs,
    parsed_struct::{ProcessedSeed, StructAttrs},
    syn::{self, Ident, ItemStruct, Result as ParseResult},
};
use shank_render::pda::try_process_seeds;

fn parse_struct(code: TokenStream) -> ItemStruct {
    syn::parse2::<ItemStruct>(code).expect("Should parse successfully")
}

pub fn parse_struct_attrs(code: TokenStream) -> (Ident, StructAttrs) {
    let account_struct = parse_struct(code);
    let all_structs = vec![&account_struct].into_iter();
    let parsed_structs = extract_account_structs(all_structs)
        .expect("Should parse struct without error");

    (
        account_struct.ident,
        parsed_structs.first().unwrap().struct_attrs.clone(),
    )
}

pub fn process_seeds(code: TokenStream) -> ParseResult<Vec<ProcessedSeed>> {
    let (_, struct_attrs) = parse_struct_attrs(code);
    try_process_seeds(&struct_attrs)
}

pub fn pretty_print(code: TokenStream) -> String {
    let syn_tree = syn::parse_file(code.to_string().as_str()).unwrap();
    prettyplease::unparse(&syn_tree)
}