solarti-program-error-derive 0.3.11

Proc-Macro Library for Solarti Program error attributes and derive macro
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
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

//! Token parsing

use {
    proc_macro2::Ident,
    syn::{
        parse::{Parse, ParseStream},
        token::Comma,
        LitInt, Token,
    },
};

/// Possible arguments to the `#[spl_program_error]` attribute
pub struct SplProgramErrorArgs {
    /// Whether to hash the error codes using `miraland_program::hash`
    /// or to use the default error code assigned by `num_traits`.
    pub hash_error_code_start: Option<u32>,
}

impl Parse for SplProgramErrorArgs {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        if input.is_empty() {
            return Ok(Self {
                hash_error_code_start: None,
            });
        }
        match SplProgramErrorArgParser::parse(input)? {
            SplProgramErrorArgParser::HashErrorCodes { value, .. } => Ok(Self {
                hash_error_code_start: Some(value.base10_parse::<u32>()?),
            }),
        }
    }
}

/// Parser for args to the `#[spl_program_error]` attribute
/// ie. `#[spl_program_error(hash_error_code_start = 1275525928)]`
enum SplProgramErrorArgParser {
    HashErrorCodes {
        _ident: Ident,
        _equals_sign: Token![=],
        value: LitInt,
        _comma: Option<Comma>,
    },
}

impl Parse for SplProgramErrorArgParser {
    fn parse(input: ParseStream) -> syn::Result<Self> {
        let _ident = {
            let ident = input.parse::<Ident>()?;
            if ident != "hash_error_code_start" {
                return Err(input.error("Expected argument 'hash_error_code_start'"));
            }
            ident
        };
        let _equals_sign = input.parse::<Token![=]>()?;
        let value = input.parse::<LitInt>()?;
        let _comma: Option<Comma> = input.parse().unwrap_or(None);
        Ok(Self::HashErrorCodes {
            _ident,
            _equals_sign,
            value,
            _comma,
        })
    }
}