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

// Copyright 2021 Axiom-Team
//
// This file is part of Duniter-v2S.
//
// Duniter-v2S is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, version 3 of the License.
//
// Duniter-v2S is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with Duniter-v2S. If not, see <https://www.gnu.org/licenses/>.

use codec::{Decode, Encode, MaxEncodedLen};
use frame_support::pallet_prelude::*;
use scale_info::TypeInfo;
use sp_runtime::traits::Zero;

/// Account data structure.
///
/// For details, refer to `struct AccountData` in Substrate code.
#[derive(Clone, Decode, Encode, Eq, MaxEncodedLen, PartialEq, RuntimeDebug, TypeInfo)] // Default,
pub struct AccountData<Balance, IdtyId> {
    /// Free balance of the account.
    pub(super) free: Balance,
    /// Reserved balance of the account.
    pub(super) reserved: Balance,
    /// Frozen fee balance of the account.
    fee_frozen: Balance,
    /// Optional pointer to an identity used to determine if this account is linked to a member and in the quota system for fee refunds.
    pub linked_idty: Option<IdtyId>,
}

// explicit implementation of default trait (can not be derived)
impl<Balance: Zero, IdtyId> Default for AccountData<Balance, IdtyId> {
    fn default() -> Self {
        Self {
            linked_idty: None,
            free: Balance::zero(),
            reserved: Balance::zero(),
            fee_frozen: Balance::zero(),
        }
    }
}

impl<Balance: Zero, IdtyId> AccountData<Balance, IdtyId> {
    pub fn set_balances(&mut self, new_balances: pallet_balances::AccountData<Balance>) {
        self.free = new_balances.free;
        self.reserved = new_balances.reserved;
        self.fee_frozen = new_balances.frozen;
    }
}

// convert Duniter AccountData to Balances AccountData
// needed for trait implementation
impl<Balance: Zero, IdtyId> From<AccountData<Balance, IdtyId>>
    for pallet_balances::AccountData<Balance>
{
    fn from(account_data: AccountData<Balance, IdtyId>) -> Self {
        Self {
            free: account_data.free,
            reserved: account_data.reserved,
            frozen: account_data.fee_frozen,
            flags: Default::default(), // default flags since not used
        }
    }
}

#[derive(
    Clone,
    Decode,
    Default,
    Encode,
    Eq,
    MaxEncodedLen,
    PartialEq,
    RuntimeDebug,
    TypeInfo,
    serde::Serialize,
    serde::Deserialize,
)]
#[serde(deny_unknown_fields)]
pub struct GenesisAccountData<Balance, IdtyId> {
    pub balance: Balance,
    pub idty_id: Option<IdtyId>,
}