ic-canister-kit 0.1.2

A ic canister package
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

use candid::CandidType;
use serde::Deserialize;

use crate::stable::Stable;
use crate::times::{now, Timestamp};

#[derive(CandidType, Deserialize, Debug, Default, Clone)]
pub struct LimitDuration {
    start: Timestamp,
    end: Timestamp,
}

impl LimitDuration {
    pub fn new(start: Timestamp, end: Timestamp) -> Self {
        LimitDuration { start, end }
    }
}

#[derive(CandidType, Deserialize, Debug, Default)]
pub struct NftLimit {
    limits: Vec<LimitDuration>, // 该时间段内,是限制时间
}

pub type NftLimitState = (Vec<LimitDuration>,);

impl Stable<NftLimitState, NftLimitState> for NftLimit {
    fn save(&mut self) -> NftLimitState {
        let limits = std::mem::take(&mut self.limits);
        (limits,)
    }

    fn restore(&mut self, state: NftLimitState) {
        let _ = std::mem::replace(&mut self.limits, state.0);
    }
}

impl NftLimit {
    pub fn is_limit(&self) -> bool {
        let now = now();
        for LimitDuration { start, end } in self.limits.iter() {
            if start <= &now && &now < end {
                return false;
            }
        }
        true
    }

    pub fn set_limits(&mut self, limits: Vec<LimitDuration>) {
        self.limits = limits;
    }

    pub fn get_limits(&self) -> Vec<LimitDuration> {
        self.limits.clone()
    }
}