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

//! KYC info module.
use odra::{types::Address, Mapping, UnwrapOrRevert};

use crate::{utils::Error, voting::types::VotingId};

use super::refs::ContractRefs;

/// A utility module that provides information about the current status of the KYC process.
#[odra::module]
pub struct KycInfo {
    refs: ContractRefs,
    votings: Mapping<Address, Option<VotingId>>,
    addresses: Mapping<VotingId, Address>,
}

impl KycInfo {
    /// Returns true if the `address` has a non-zero balance of kyc token, false otherwise.
    pub fn is_kycd(&self, address: &Address) -> bool {
        !self.refs.kyc_token().balance_of(address).is_zero()
    }

    /// Sets a flag indicating there is ongoing voting for the given `address`.
    pub fn set_voting(&mut self, address: Address, voting_id: VotingId) {
        self.votings.set(&address, Some(voting_id));
        self.addresses.set(&voting_id, address);
    }

    /// Clears the flag indicating there is ongoing voting for the given `address`.
    pub fn clear_voting(&mut self, address: &Address) {
        self.votings.set(address, None);
    }

    /// Indicates whether there is ongoing voting for the given `address`.
    pub fn exists_ongoing_voting(&self, address: &Address) -> bool {
        self.votings.get(address).is_some()
    }

    /// Gets the address of the voting subject.
    pub fn get_voting_subject(&self, voting_id: VotingId) -> Address {
        self.addresses
            .get(&voting_id)
            .unwrap_or_revert_with(Error::VotingAddressNotFound)
    }
}