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

use {anchor_lang::prelude::*, hpl_utils::traits::*};

/// The staking_pool multiplier state account
#[account]
pub struct Multipliers {
    pub bump: u8,

    /// The staking_pool this multiplier is associated with
    pub staking_pool: Pubkey,

    /// The decimals for multipliers
    pub decimals: u8,

    /// The duration multipliers for the staking_pool
    pub duration_multipliers: Vec<Multiplier>,

    /// The duration multipliers for the staking_pool
    pub count_multipliers: Vec<Multiplier>,

    /// The duration multipliers for the staking_pool
    pub creator_multipliers: Vec<Multiplier>,

    /// The duration multipliers for the staking_pool
    pub collection_multipliers: Vec<Multiplier>,
}
impl Default for Multipliers {
    const LEN: usize = 8 + 136;

    fn set_defaults(&mut self) {
        self.bump = 0;
        self.staking_pool = Pubkey::default();
        self.decimals = 0;
        self.duration_multipliers = vec![];
        self.count_multipliers = vec![];
        self.creator_multipliers = vec![];
        self.collection_multipliers = vec![];
    }
}

#[derive(AnchorSerialize, AnchorDeserialize, Clone, Debug, PartialEq)]
pub struct Multiplier {
    pub value: u64,
    pub multiplier_type: MultiplierType,
}
impl Default for Multiplier {
    const LEN: usize = 48;

    fn set_defaults(&mut self) {
        self.value = 0;
        self.multiplier_type = MultiplierType::StakeDuration { min_duration: 0 };
    }
}

#[derive(AnchorSerialize, AnchorDeserialize, Clone, Debug, PartialEq)]
pub enum MultiplierType {
    /// The multiplier is applied to the stake duration
    StakeDuration { min_duration: u64 },

    /// The multiplier is applied to the number of NFTs staked
    NFTCount { min_count: u64 },

    /// The multiplier is applied based on creator
    Creator { creator: Pubkey },

    /// The multiplier is applied based on collection
    Collection { collection: Pubkey },
}