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
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

//! # Staking Api
//!
//! `4t2::cw-staking`

use crate::api;
use crate::objects::{AnsAsset, AssetEntry};
use cosmwasm_schema::QueryResponses;
use cosmwasm_std::{Addr, Empty, Uint128};
use cw20::Expiration;
use cw_asset::AssetInfo;
use cw_utils::Duration;

pub type ProviderName = String;

/// The callback id for staking over ibc
pub const IBC_STAKING_PROVIDER_ID: u32 = 22335;

pub const CW_STAKING: &str = "4t2:cw-staking";

pub type ExecuteMsg = api::ExecuteMsg<CwStakingExecuteMsg>;
pub type InstantiateMsg = api::InstantiateMsg<Empty>;
pub type QueryMsg = api::QueryMsg<CwStakingQueryMsg>;

impl api::ApiExecuteMsg for CwStakingExecuteMsg {}

impl api::ApiQueryMsg for CwStakingQueryMsg {}

/// A request message that's sent to this staking api
#[cosmwasm_schema::cw_serde]
pub struct CwStakingExecuteMsg {
    /// The name of the staking provider
    pub provider: ProviderName,
    pub action: CwStakingAction,
}

#[cosmwasm_schema::cw_serde]
/// Possible actions to perform on the staking contract
/// All provide the staking token information
pub enum CwStakingAction {
    /// Stakes/bonds a given token
    Stake {
        staking_token: AnsAsset,
        unbonding_period: Option<Duration>,
    },
    /// Unstake a given token
    Unstake {
        staking_token: AnsAsset,
        unbonding_period: Option<Duration>,
    },
    /// Claim rewards for a given token
    ClaimRewards { staking_token: AssetEntry },
}

#[cosmwasm_schema::cw_serde]
#[derive(QueryResponses)]
#[cfg_attr(feature = "boot", derive(boot_core::QueryFns))]
#[cfg_attr(feature = "boot", impl_into(QueryMsg))]
pub enum CwStakingQueryMsg {
    #[returns(StakingInfoResponse)]
    Info {
        provider: ProviderName,
        staking_token: AssetEntry,
    },
    #[returns(StakeResponse)]
    Staked {
        provider: ProviderName,
        staking_token: AssetEntry,
        staker_address: String,
        unbonding_period: Option<Duration>,
    },
    #[returns(UnbondingResponse)]
    Unbonding {
        provider: ProviderName,
        staking_token: AssetEntry,
        staker_address: String,
    },
    #[returns(RewardTokensResponse)]
    RewardTokens {
        provider: ProviderName,
        staking_token: AssetEntry,
    },
}

#[cosmwasm_schema::cw_serde]
pub struct StakingInfoResponse {
    pub staking_contract_address: Addr,
    pub staking_token: AssetInfo,
    pub unbonding_periods: Option<Vec<Duration>>,
    pub max_claims: Option<u32>,
}

#[cosmwasm_schema::cw_serde]
pub struct StakeResponse {
    pub amount: Uint128,
}

#[cosmwasm_schema::cw_serde]
pub struct RewardTokensResponse {
    pub tokens: Vec<AssetInfo>,
}

#[cosmwasm_schema::cw_serde]
pub struct UnbondingResponse {
    pub claims: Vec<Claim>,
}

#[cosmwasm_schema::cw_serde]
pub struct Claim {
    pub amount: Uint128,
    pub claimable_at: Expiration,
}