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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172

//! # Account Proxy
//!
//! `abstract_std::proxy` hold all the assets associated with the Account instance. It accepts Cosmos messages from whitelisted addresses and executes them.
//!
//! ## Description
//! The proxy is part of the Core Account contracts along with the [`crate::manager`] contract.
//! This contract is responsible for executing Cosmos messages and calculating the value of its internal assets.
//!
//! ## Price Sources
//! [price sources](crate::objects::price_source) are what allow the proxy contract to provide value queries for its assets. It needs to be configured using the [`ExecuteMsg::UpdateAssets`] endpoint.
//! After configuring the price sources [`QueryMsg::TotalValue`] can be called to get the total holding value.

use cosmwasm_schema::QueryResponses;
use cosmwasm_std::{CosmosMsg, Empty, Uint128};
use cw_asset::{Asset, AssetInfo};

#[allow(unused_imports)]
use crate::{
    ibc_client::ExecuteMsg as IbcClientMsg,
    objects::{
        account::AccountId,
        oracle::{AccountValue, Complexity},
        price_source::{PriceSource, UncheckedPriceSource},
        AssetEntry,
    },
};

pub mod state {
    use cosmwasm_std::Addr;
    use cw_controllers::Admin;
    use cw_storage_plus::Item;

    pub use crate::objects::account::ACCOUNT_ID;
    use crate::objects::{ans_host::AnsHost, common_namespace::ADMIN_NAMESPACE};
    #[cosmwasm_schema::cw_serde]
    pub struct State {
        pub modules: Vec<Addr>,
    }
    pub const ANS_HOST: Item<AnsHost> = Item::new("\u{0}{6}ans_host");
    pub const STATE: Item<State> = Item::new("\u{0}{5}state");
    pub const ADMIN: Admin = Admin::new(ADMIN_NAMESPACE);
}

#[cosmwasm_schema::cw_serde]
pub struct InstantiateMsg {
    pub account_id: AccountId,
    pub ans_host_address: String,
    pub manager_addr: String,
    pub base_asset: Option<AssetEntry>,
}

#[cosmwasm_schema::cw_serde]
#[derive(cw_orch::ExecuteFns)]
pub enum ExecuteMsg {
    /// Sets the admin
    SetAdmin { admin: String },
    /// Executes the provided messages if sender is whitelisted
    ModuleAction { msgs: Vec<CosmosMsg<Empty>> },
    /// Execute a message and forward the Response data
    ModuleActionWithData { msg: CosmosMsg<Empty> },
    /// Execute IBC action on Client
    IbcAction { msg: IbcClientMsg },
    /// Adds the provided address to whitelisted dapps
    AddModules { modules: Vec<String> },
    /// Removes the provided address from the whitelisted dapps
    RemoveModule { module: String },
    /// Updates the VAULT_ASSETS map
    UpdateAssets {
        to_add: Vec<(AssetEntry, UncheckedPriceSource)>,
        to_remove: Vec<AssetEntry>,
    },
}
#[cosmwasm_schema::cw_serde]
pub struct MigrateMsg {}

#[cosmwasm_schema::cw_serde]
#[derive(QueryResponses, cw_orch::QueryFns)]
pub enum QueryMsg {
    /// Contains the enabled modules
    /// Returns [`ConfigResponse`]
    #[returns(ConfigResponse)]
    Config {},
    /// Returns the total value of the assets held by this account
    /// [`AccountValue`]
    #[returns(AccountValue)]
    TotalValue {},
    /// Returns the value of one token with an optional amount set.
    /// If amount is not set, the account's balance of the token is used.
    /// [`TokenValueResponse`]
    #[returns(TokenValueResponse)]
    TokenValue { identifier: AssetEntry },
    /// Returns the amount of specified tokens this contract holds
    /// [`HoldingAmountResponse`]
    #[returns(HoldingAmountResponse)]
    HoldingAmount { identifier: AssetEntry },
    /// Returns the oracle configuration value for the specified key
    /// [`AssetConfigResponse`]
    #[returns(AssetConfigResponse)]
    AssetConfig { identifier: AssetEntry },
    /// Returns [`AssetsConfigResponse`]
    /// Human readable
    #[returns(AssetsConfigResponse)]
    AssetsConfig {
        start_after: Option<AssetEntry>,
        limit: Option<u8>,
    },
    /// Returns [`AssetsInfoResponse`]
    /// Not human readable
    #[returns(AssetsInfoResponse)]
    AssetsInfo {
        start_after: Option<AssetInfo>,
        limit: Option<u8>,
    },
    /// Returns [`BaseAssetResponse`]
    #[returns(BaseAssetResponse)]
    BaseAsset {},
}

#[cosmwasm_schema::cw_serde]
pub struct ConfigResponse {
    pub modules: Vec<String>,
}

#[cosmwasm_schema::cw_serde]
pub struct TokenValueResponse {
    pub value: Uint128,
}

#[cosmwasm_schema::cw_serde]
pub struct BaseAssetResponse {
    pub base_asset: AssetInfo,
}

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

/// Human readable config for a single asset
#[cosmwasm_schema::cw_serde]
pub struct AssetConfigResponse {
    pub price_source: UncheckedPriceSource,
}

/// non-human readable asset configuration
#[cosmwasm_schema::cw_serde]
pub struct AssetsInfoResponse {
    pub assets: Vec<(AssetInfo, OracleAsset)>,
}

/// Human readable asset configuration
#[cosmwasm_schema::cw_serde]
pub struct AssetsConfigResponse {
    pub assets: Vec<(AssetEntry, UncheckedPriceSource)>,
}

#[cosmwasm_schema::cw_serde]
pub struct OracleAsset {
    pub price_source: PriceSource,
    pub complexity: Complexity,
}
/// Query message to external contract to get asset value
#[cosmwasm_schema::cw_serde]
pub struct ValueQueryMsg {
    pub asset: AssetInfo,
    pub amount: Uint128,
}
/// External contract value response
#[cosmwasm_schema::cw_serde]
pub struct ExternalValueResponse {
    pub value: Asset,
}