odra-modules 2.8.1

Collection of reusable Odra modules.
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96

#![allow(missing_docs)]

use odra::casper_types::U256;
use odra::named_keys::{
    base64_encoded_key_value_storage, compound_key_value_storage, single_value_storage
};
use odra::prelude::*;

const ALLOWANCES_KEY: &str = "allowances";
const BALANCES_KEY: &str = "balances";
const NAME_KEY: &str = "name";
const DECIMALS_KEY: &str = "decimals";
const SYMBOL_KEY: &str = "symbol";
const TOTAL_SUPPLY_KEY: &str = "total_supply";

single_value_storage!(
    Cep18NameStorage,
    String,
    NAME_KEY,
    ExecutionError::KeyNotFound
);
single_value_storage!(
    Cep18DecimalsStorage,
    u8,
    DECIMALS_KEY,
    ExecutionError::KeyNotFound
);
single_value_storage!(
    Cep18SymbolStorage,
    String,
    SYMBOL_KEY,
    ExecutionError::KeyNotFound
);
single_value_storage!(
    Cep18TotalSupplyStorage,
    U256,
    TOTAL_SUPPLY_KEY,
    ExecutionError::KeyNotFound
);
impl Cep18TotalSupplyStorage {
    /// Adds the given amount to the total supply of the token.
    pub fn add(&self, amount: U256) {
        let total_supply = self.get();
        let new_total_supply = total_supply
            .checked_add(amount)
            .unwrap_or_revert_with(&self.env(), ExecutionError::AdditionOverflow);
        self.set(new_total_supply);
    }

    /// Subtracts the given amount from the total supply of the token.
    pub fn subtract(&self, amount: U256) {
        let total_supply = self.get();
        let new_total_supply = total_supply
            .checked_sub(amount)
            .unwrap_or_revert_with(&self.env(), ExecutionError::SubtractionOverflow);
        self.set(new_total_supply);
    }
}
base64_encoded_key_value_storage!(Cep18BalancesStorage, BALANCES_KEY, Address, U256);
impl Cep18BalancesStorage {
    /// Adds the given amount to the balance of the given account.
    pub fn add(&self, account: &Address, amount: U256) {
        let balance = self.get(account).unwrap_or_default();
        let new_balance = balance.checked_add(amount).unwrap_or_revert(self);
        self.set(account, new_balance);
    }

    /// Subtracts the given amount from the balance of the given account.
    pub fn subtract(&self, account: &Address, amount: U256) {
        let balance = self.get(account).unwrap_or_default();
        let new_balance = balance
            .checked_sub(amount)
            .unwrap_or_revert_with(self, ExecutionError::SubtractionOverflow);
        self.set(account, new_balance);
    }
}
compound_key_value_storage!(Cep18AllowancesStorage, ALLOWANCES_KEY, Address, U256);
impl Cep18AllowancesStorage {
    /// Adds the given amount to the allowance of the given owner and spender.
    pub fn add(&self, owner: &Address, spender: &Address, amount: U256) {
        let allowance = self.get_or_default(owner, spender);
        let new_allowance = allowance
            .checked_add(amount)
            .unwrap_or_revert_with(&self.env(), ExecutionError::AdditionOverflow);
        self.set(owner, spender, new_allowance);
    }

    /// Subtracts the given amount from the allowance of the given owner and spender.
    pub fn subtract(&self, owner: &Address, spender: &Address, amount: U256) {
        let allowance = self.get_or_default(owner, spender);
        let new_allowance = allowance
            .checked_sub(amount)
            .unwrap_or_revert_with(&self.env(), ExecutionError::AdditionOverflow);
        self.set(owner, spender, new_allowance);
    }
}