drt-sc-derive 0.0.2

Dharitri smart contract API procedural 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
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
65
66
67
68
69

use crate::parse::attributes::util::{clean_string, is_first_char_numeric};

use super::attr_names::*;

pub struct PayableAttribute {
    pub identifier: Option<String>,
}

impl PayableAttribute {
    pub fn parse(attr: &syn::Attribute) -> Option<PayableAttribute> {
        if let Some(first_seg) = attr.path().segments.first() {
            if first_seg.ident == ATTR_PAYABLE {
                Some(PayableAttribute {
                    identifier: extract_token_identifier(attr),
                })
            } else {
                None
            }
        } else {
            None
        }
    }
}

/// Current implementation only works with 1 token name.
/// Might be extended in the future.
fn extract_token_identifier(attr: &syn::Attribute) -> Option<String> {
    match attr.meta.clone() {
        syn::Meta::Path(_) => {
            panic!("attribute needs 1 string argument: Replace with #[payable(\"*\")] or #[payable(\"REWA\")]")
        },
        syn::Meta::List(list) => {
            let mut iter = list.tokens.into_iter();
            let ticker = match iter.next() {
                Some(proc_macro2::TokenTree::Literal(literal)) => {
                    let clean = clean_string(literal.to_string());
                    assert!(
                        !clean.is_empty(),
                        "ticker can not be empty. attribute needs 1 string argument: Replace with #[payable(\"*\")] or #[payable(\"REWA\")"
                    );

                    assert!(!is_first_char_numeric(&clean), "argument can not be a number");

                    if clean
                    .chars()
                    .next()
                    .is_some_and(|s|
                        s == '*'
                    ) {
                        assert!(clean.len() == 1usize, "attribute needs 1 string argument: \"*\", \"REWA\" or token identifier");
                    }

                    clean
                },
                Some(_) => panic!("expected a string as argument"),
                None => panic!("argument can not be empty. attribute needs 1 string argument: Replace with #[payable(\"*\")] or #[payable(\"REWA\")"),
            };

            assert!(
                iter.next().is_none(),
                "too many tokens in attribute argument"
            );
            Some(ticker)
        },
        syn::Meta::NameValue(_) => panic!(
            "attribute can not be name value. attribute needs 1 string argument: \"*\" or \"REWA\""
        ),
    }
}