//! Copy trading contract

use std::{fmt::Display, num::ParseIntError, str::FromStr};

use super::market::{
    entry::{ExecuteMsg as MarketExecuteMsg, SlippageAssert, StopLoss},
    order::OrderId,
    position::PositionId,
};
use crate::{
    number::{Collateral, LpToken, NonZero, Usd},
    price::{PriceBaseInQuote, PricePoint, TakeProfitTrader},
    storage::{DirectionToBase, LeverageToBase, MarketId, RawAddr},
    time::Timestamp,
};
use anyhow::{anyhow, bail};
use cosmwasm_std::{Addr, Binary, Decimal256, StdError, StdResult, Uint128, Uint64};
use cw_storage_plus::{IntKey, Key, KeyDeserialize, Prefixer, PrimaryKey};
use thiserror::Error;

/// Message for instantiating a new copy trading contract.
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(rename_all = "snake_case")]
pub struct InstantiateMsg {
    /// Leader of the contract
    pub leader: RawAddr,
    /// Initial configuration values
    pub config: ConfigUpdate,
    /// Initial parameters
    pub parameters: FactoryConfigUpdate,
}

/// Full configuration
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(rename_all = "snake_case")]
/// Updates to configuration values.
pub struct Config {
    /// Factory we will allow trading on.
    pub factory: Addr,
    /// Administrator of the contract. Should be the factory contract
    /// which initializes this.
    pub admin: Addr,
    /// Pending administrator, ready to be accepted, if any.
    pub pending_admin: Option<Addr>,
    /// Leader of the contract
    pub leader: Addr,
    /// Name given to this copy_trading pool
    pub name: String,
    /// Description of the copy_trading pool. Not more than 128
    /// characters.
    pub description: String,
    /// Commission rate for the leader. Must be within 1-30%.
    pub commission_rate: Decimal256,
    /// Creation time of contract
    pub created_at: Timestamp,
    /// Allowed smart queries during Rebalance operation
    pub allowed_rebalance_queries: u32,
    /// Allowed smart queries during token value computation
    pub allowed_lp_token_queries: u32,
}

impl Config {
    /// Check validity of config values
    pub fn check(&self) -> anyhow::Result<()> {
        if self.name.len() > 128 {
            Err(anyhow!(
                "Description should not be more than 128 characters"
            ))
        } else if self.commission_rate < Decimal256::from_ratio(1u32, 100u32) {
            Err(anyhow!("Commission rate less than 1 percent"))
        } else if self.commission_rate > Decimal256::from_ratio(30u32, 100u32) {
            Err(anyhow!("Commission rate greater than 30 percent"))
        } else {
            Ok(())
        }
    }

    /// Check leader
    pub fn ensure_leader(&self, sender: &Addr) -> anyhow::Result<()> {
        if self.leader != sender {
            bail!("Unautorized access, only {} allowed", self.leader)
        }
        Ok(())
    }

    /// Ensure it is factory contract
    pub fn ensure_factory(&self, sender: &Addr) -> anyhow::Result<()> {
        if self.factory != sender {
            bail!("Unautorized access, only {} allowed", self.factory)
        }
        Ok(())
    }
}

/// Updates to configuration values.
///
/// See [Config] for field meanings.
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, Default)]
#[serde(rename_all = "snake_case")]
#[allow(missing_docs)]
pub struct ConfigUpdate {
    pub name: Option<String>,
    pub description: Option<String>,
    pub commission_rate: Option<Decimal256>,
}

/// Updates to configuration values.
///
/// See [Config] for field meanings.
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, Default)]
#[serde(rename_all = "snake_case")]
#[allow(missing_docs)]
pub struct FactoryConfigUpdate {
    pub allowed_rebalance_queries: Option<u32>,
    pub allowed_lp_token_queries: Option<u32>,
}

/// Executions available on the copy trading contract.
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(rename_all = "snake_case")]
pub enum ExecuteMsg {
    /// Cw20 interface
    Receive {
        /// Owner of funds sent to the contract
        sender: RawAddr,
        /// Amount of funds sent
        amount: Uint128,
        /// Must parse to a [ExecuteMsg]
        msg: Binary,
    },
    /// Deposit funds to the contract
    Deposit {},
    /// Withdraw funds from a given market
    Withdraw {
        /// The number of LP shares to remove
        shares: NonZero<LpToken>,
        /// Token type in which amount should be withdrawn
        token: Token,
    },
    /// Update configuration values that is allowed for leader.
    LeaderUpdateConfig(ConfigUpdate),
    /// Update configuration values that is allowed for facotr.
    FactoryUpdateConfig(FactoryConfigUpdate),
    /// Leader specific execute message for the market
    LeaderMsg {
        /// Market id that message is for
        market_id: MarketId,
        /// Message
        message: Box<MarketExecuteMsg>,
        /// Collateral to use for the action
        collateral: Option<NonZero<Collateral>>,
    },
    /// Leader withdrawal (for commission).
    LeaderWithdrawal {
        /// Fund to be withdrawn
        requested_funds: NonZero<Collateral>,
        /// Token type in which amount should be withdrawn
        token: Token,
    },
    /// Perform queue work
    DoWork {},
}

/// Queries that can be performed on the copy contract.
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(rename_all = "snake_case")]
pub enum QueryMsg {
    /// Get the current config
    ///
    /// Returns [Config]
    Config {},
    /// Get the queue status of a particular wallet
    ///
    /// Returns [QueueResp]
    QueueStatus {
        /// Address of the wallet
        address: RawAddr,
        /// Value from [QueueResp]
        start_after: Option<QueuePositionId>,
        /// How many values to return
        limit: Option<u32>,
    },
    /// Returns the share held by the wallet
    ///
    /// Returns [BalanceResp]
    Balance {
        /// Address of the token holder
        address: RawAddr,
        /// Value from [BalanceResp]
        start_after: Option<Token>,
        /// How many values to return
        limit: Option<u32>,
    },
    /// Check the status of the copy trading contract for all the
    /// markets that it's trading on
    ///
    /// Returns [LeaderStatusResp]
    LeaderStatus {
        /// Value from [BalanceResp::next_start_after]
        start_after: Option<Token>,
        /// How many values to return
        limit: Option<u32>,
    },
    /// Does it have any pending work ?
    ///
    /// Returns [WorkResp]
    HasWork {},
}

/// Individual response from [QueryMsg::QueueStatus]
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
pub struct QueueResp {
    /// Items in queue for the wallet
    pub items: Vec<QueueItemStatus>,
    /// Last processed [QueuePositionId]
    pub inc_processed_till: Option<IncQueuePositionId>,
    /// Last processed [QueuePositionId]
    pub dec_processed_till: Option<DecQueuePositionId>,
}

/// Queue item status
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq)]
pub struct QueueItemStatus {
    /// Queue item
    pub item: QueueItem,
    /// Status of processing
    pub status: ProcessingStatus,
}

/// Queue item Processing status
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq)]
pub enum ProcessingStatus {
    /// Not started processing yet
    NotProcessed,
    /// Successfully finished processing
    Finished,
    /// In progress
    InProgress,
    /// Failed during processing
    Failed(FailedReason),
}

impl ProcessingStatus {
    /// Did the processing fail ?
    pub fn failed(&self) -> bool {
        match self {
            ProcessingStatus::NotProcessed => false,
            ProcessingStatus::Finished => false,
            ProcessingStatus::Failed(_) => true,
            ProcessingStatus::InProgress => false,
        }
    }

    /// Is any status pending ?
    pub fn pending(&self) -> bool {
        match self {
            ProcessingStatus::NotProcessed => true,
            ProcessingStatus::Finished => false,
            ProcessingStatus::Failed(_) => false,
            ProcessingStatus::InProgress => true,
        }
    }

    /// Did the status finish ?
    pub fn finish(&self) -> bool {
        match self {
            ProcessingStatus::NotProcessed => false,
            ProcessingStatus::Finished => true,
            ProcessingStatus::Failed(_) => false,
            ProcessingStatus::InProgress => false,
        }
    }

    /// Is any status currently in progres ?
    pub fn in_progress(&self) -> bool {
        match self {
            ProcessingStatus::NotProcessed => false,
            ProcessingStatus::Finished => false,
            ProcessingStatus::Failed(_) => false,
            ProcessingStatus::InProgress => true,
        }
    }
}

/// Failure reason on why queue processing failed
#[derive(Error, Debug, serde::Serialize, serde::Deserialize, Clone, PartialEq)]
pub enum FailedReason {
    /// Not enough collateral available
    #[error("Collateral not available. Requested {requested}, but only available {available}")]
    NotEnoughCollateral {
        /// Available collateral
        available: Collateral,
        /// Requested collateral
        requested: NonZero<Collateral>,
    },
    /// Not enough crank fee
    #[error("Crank fee not available. Requested {requested}, but only available {available}")]
    NotEnoughCrankFee {
        /// Available collateral
        available: Collateral,
        /// Requested collateral
        requested: Collateral,
    },
    /// Fund less than chain's minimum representation
    #[error("Collateral amount {funds} is less than chain's minimum representation.not available")]
    FundLessThanMinChain {
        /// Requested collateral
        funds: NonZero<Collateral>,
    },
    /// Wallet does not have enough shares
    #[error("Shares not available. Requested {requested}, but only available {available}")]
    NotEnoughShares {
        /// Available shares
        available: LpToken,
        /// Requested shares
        requested: LpToken,
    },
    /// Received error from Market contract
    #[error("{market_id} result in error: {message}")]
    MarketError {
        /// Market ID which result in error
        market_id: MarketId,
        /// Error message
        message: String,
    },
    /// Deferred exec failure
    #[error("Deferred exec failure at {executed} because of: {reason}")]
    DeferredExecFailure {
        /// Reason it didn't apply successfully
        reason: String,
        /// Timestamp when it failed execution
        executed: Timestamp,
        /// Price point when it was cranked, if applicable
        crank_price: Option<PricePoint>,
    },
}

/// Queue Item
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq)]
pub enum QueueItem {
    /// Item that will lead to increase or no change of collateral
    IncCollaleteral {
        /// Item type
        item: IncQueueItem,
        /// Queue position id
        id: IncQueuePositionId,
    },
    /// Item that will lead to decrease of collateral
    DecCollateral {
        /// Item type
        item: Box<DecQueueItem>,
        /// Queue position id
        id: DecQueuePositionId,
    },
}

/// Queue item that needs to be processed
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq)]
pub enum IncQueueItem {
    /// Deposit the fund and get some [LpToken]
    Deposit {
        /// Funds to be deposited
        funds: NonZero<Collateral>,
        /// Token
        token: Token,
    },
    /// Market action items
    MarketItem {
        /// Market id
        id: MarketId,
        /// Market token
        token: Token,
        /// Market item
        item: Box<IncMarketItem>,
    },
}

/// Queue item that needs to be processed
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq)]
pub enum DecQueueItem {
    /// Withdraw via LpToken
    Withdrawal {
        /// Tokens to be withdrawn
        tokens: NonZero<LpToken>,
        /// Token type
        token: Token,
    },
    /// Market action items
    MarketItem {
        /// Market id
        id: MarketId,
        /// Market token
        token: Token,
        /// Market item
        item: Box<DecMarketItem>,
    },
}

/// Queue item that needs to be processed
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq)]
pub enum IncMarketItem {
    /// Remove collateral from a position, causing leverage to increase
    UpdatePositionRemoveCollateralImpactLeverage {
        /// ID of position to update
        id: PositionId,
        /// Amount of funds to remove from the position
        amount: NonZero<Collateral>,
    },
    /// Remove collateral from a position, causing leverage to increase
    UpdatePositionRemoveCollateralImpactSize {
        /// ID of position to update
        id: PositionId,
        /// Amount of funds to remove from the position
        amount: NonZero<Collateral>,
        /// Slippage alert
        slippage_assert: Option<SlippageAssert>,
    },
    /// Cancel an open limit order
    CancelLimitOrder {
        /// ID of the order
        order_id: OrderId,
    },
    /// Close a position
    ClosePosition {
        /// ID of position to close
        id: PositionId,
        /// Assertion that the price has not moved too far
        slippage_assert: Option<SlippageAssert>,
    },
}

/// Queue item that needs to be processed
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq)]
pub enum DecMarketItem {
    /// Open position
    OpenPosition {
        /// Collateral for the position
        collateral: NonZero<Collateral>,
        /// Assertion that the price has not moved too far
        slippage_assert: Option<SlippageAssert>,
        /// Leverage of new position
        leverage: LeverageToBase,
        /// Direction of new position
        direction: DirectionToBase,
        /// Stop loss price of new position
        stop_loss_override: Option<PriceBaseInQuote>,
        /// Take profit price of new position
        #[serde(alias = "take_profit_override")]
        take_profit: TakeProfitTrader,
    },
    /// Add collateral to position, causing leverage to decrease
    UpdatePositionAddCollateralImpactLeverage {
        /// Collateral that will be added
        collateral: NonZero<Collateral>,
        /// ID of position to update
        id: PositionId,
    },
    /// Add collateral to position, causing notional size to increase
    UpdatePositionAddCollateralImpactSize {
        /// Collateral that will be added
        collateral: NonZero<Collateral>,
        /// ID of position to update
        id: PositionId,
        /// Slippage assert
        slippage_assert: Option<SlippageAssert>,
    },
    /// Update position leverage
    UpdatePositionLeverage {
        /// ID of position to update
        id: PositionId,
        /// New leverage of the position
        leverage: LeverageToBase,
        /// Slippage assert
        slippage_assert: Option<SlippageAssert>,
    },
    /// Modify the take profit price of a position
    UpdatePositionTakeProfitPrice {
        /// ID of position to update
        id: PositionId,
        /// New take profit price of the position
        price: TakeProfitTrader,
    },
    /// Update the stop loss price of a position
    UpdatePositionStopLossPrice {
        /// ID of position to update
        id: PositionId,
        /// New stop loss price of the position, or remove
        stop_loss: StopLoss,
    },
    /// Set a limit order to open a position when the price of the asset hits
    /// the specified trigger price.
    PlaceLimitOrder {
        /// Collateral for the position
        collateral: NonZero<Collateral>,
        /// Price when the order should trigger
        trigger_price: PriceBaseInQuote,
        /// Leverage of new position
        leverage: LeverageToBase,
        /// Direction of new position
        direction: DirectionToBase,
        /// Stop loss price of new position
        stop_loss_override: Option<PriceBaseInQuote>,
        /// Take profit price of new position
        #[serde(alias = "take_profit_override")]
        take_profit: TakeProfitTrader,
    },
}

/// Token required for the queue item
pub enum RequiresToken {
    /// Token required for LP token value computation
    Token {
        /// Token
        token: Token,
    },
    /// Token not requird
    NoToken {},
}

impl IncQueueItem {
    /// Does this queue item require computation of LP token value
    pub fn requires_token(self) -> RequiresToken {
        match self {
            IncQueueItem::Deposit { token, .. } => RequiresToken::Token { token },
            IncQueueItem::MarketItem { .. } => RequiresToken::NoToken {},
        }
    }
}

impl DecQueueItem {
    /// Does this queue item require computation of LP token value
    pub fn requires_token(self) -> RequiresToken {
        match self {
            DecQueueItem::Withdrawal { token, .. } => RequiresToken::Token { token },
            // Market item does not require computation of lp token value
            DecQueueItem::MarketItem { .. } => RequiresToken::NoToken {},
        }
    }
}

/// Individual market response from [QueryMsg::Status]
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(rename_all = "snake_case")]
pub struct LeaderStatusResp {
    /// Tokens for the leader
    pub tokens: Vec<TokenStatus>,
}

/// Individual market response from [QueryMsg::Status]
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(rename_all = "snake_case")]
pub struct TokenStatus {
    /// Token
    pub token: Token,
    /// Available collateral for the leader
    pub collateral: Collateral,
    /// Total shares so far. Represents AUM.
    pub shares: LpToken,
    /// Unclaimed commission
    pub unclaimed_commission: Collateral,
    /// Claimed commission
    pub claimed_commission: Collateral,
}

/// Individual market response from [QueryMsg::Status]
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(rename_all = "snake_case")]
pub struct StatusResp {
    /// Market id
    pub market_id: MarketId,
    /// Sum of deposit collateral of all open positions
    pub tvl_open_positions_usd: Usd,
    /// Sum of deposit collateral of all closed positions
    pub tvl_closed_positions_usd: Usd,
    /// Total profit so far in the closed positions
    pub profit_in_usd: Usd,
    /// Total loss so far in the closed postions
    pub loss_in_usd: Usd,
}

/// Individual market response from [QueryMsg::Balance]
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(rename_all = "snake_case")]
pub struct BalanceResp {
    /// Shares of the pool held by the wallet
    pub balance: Vec<BalanceRespItem>,
    /// Start after that should be passed for next iteration
    pub start_after: Option<Token>,
}

/// Individual market response inside [BalanceResp]
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(rename_all = "snake_case")]
pub struct BalanceRespItem {
    /// Shares of the pool held by the wallet
    pub shares: NonZero<LpToken>,
    /// Token type
    pub token: Token,
}

/// Token accepted by the contract
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum Token {
    /// Native coin and its denom
    Native(String),
    /// CW20 contract and its address
    Cw20(Addr),
}

impl Token {
    /// Is it same as market token ?
    pub fn is_same(&self, token: &crate::token::Token) -> bool {
        match token {
            crate::token::Token::Cw20 { addr, .. } => match self {
                Token::Native(_) => false,
                Token::Cw20(cw20_addr) => {
                    let cw20_addr: &RawAddr = &cw20_addr.into();
                    cw20_addr == addr
                }
            },
            crate::token::Token::Native { denom, .. } => match self {
                Token::Native(native_denom) => *native_denom == *denom,
                Token::Cw20(_) => false,
            },
        }
    }
}

impl<'a> PrimaryKey<'a> for Token {
    type Prefix = ();
    type SubPrefix = ();
    type Suffix = Self;
    type SuperSuffix = Self;

    fn key(&self) -> Vec<Key> {
        let (token_type, bytes) = match self {
            Token::Native(native) => (0u8, native.as_bytes()),
            Token::Cw20(cw20) => (1u8, cw20.as_bytes()),
        };
        let token_type = Key::Val8([token_type]);
        let key = Key::Ref(bytes);

        vec![token_type, key]
    }
}

impl<'a> Prefixer<'a> for Token {
    fn prefix(&self) -> Vec<Key> {
        let (token_type, bytes) = match self {
            Token::Native(native) => (0u8, native.as_bytes()),
            Token::Cw20(cw20) => (1u8, cw20.as_bytes()),
        };
        let token_type = Key::Val8([token_type]);
        let key = Key::Ref(bytes);
        vec![token_type, key]
    }
}

impl KeyDeserialize for Token {
    type Output = Token;

    const KEY_ELEMS: u16 = 2;

    fn from_vec(value: Vec<u8>) -> StdResult<Self::Output> {
        let (token_type, token) = <(u8, Vec<u8>) as KeyDeserialize>::from_vec(value)?;
        let token = match token_type {
            0 => {
                let native_token = String::from_slice(&token)?;
                Token::Native(native_token)
            }
            1 => {
                let cw20_token = Addr::from_slice(&token)?;
                Token::Cw20(cw20_token)
            }
            _ => {
                return Err(StdError::serialize_err(
                    "Token",
                    "Invalid number in token_type",
                ))
            }
        };
        Ok(token)
    }
}

impl KeyDeserialize for &Token {
    type Output = Token;

    const KEY_ELEMS: u16 = 2;

    fn from_vec(value: Vec<u8>) -> StdResult<Self::Output> {
        let token = <Token as KeyDeserialize>::from_vec(value)?;
        Ok(token)
    }
}

impl Display for Token {
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        match self {
            Token::Native(denom) => f.write_str(denom),
            Token::Cw20(addr) => f.write_str(addr.as_str()),
        }
    }
}

impl Token {
    /// Ensure that the two versions of the token are compatible.
    pub fn ensure_matches(&self, token: &crate::token::Token) -> anyhow::Result<()> {
        match (self, token) {
            (Token::Native(_), crate::token::Token::Cw20 { addr, .. }) => {
                anyhow::bail!("Provided native funds, but market requires a CW20 (contract {addr})")
            }
            (
                Token::Native(denom1),
                crate::token::Token::Native {
                    denom: denom2,
                    decimal_places: _,
                },
            ) => {
                if denom1 == denom2 {
                    Ok(())
                } else {
                    Err(anyhow::anyhow!("Wrong denom provided. You sent {denom1}, but the contract expects {denom2}"))
                }
            }
            (
                Token::Cw20(addr1),
                crate::token::Token::Cw20 {
                    addr: addr2,
                    decimal_places: _,
                },
            ) => {
                if addr1.as_str() == addr2.as_str() {
                    Ok(())
                } else {
                    Err(anyhow::anyhow!(
                        "Wrong CW20 used. You used {addr1}, but the contract expects {addr2}"
                    ))
                }
            }
            (Token::Cw20(_), crate::token::Token::Native { denom, .. }) => {
                anyhow::bail!(
                    "Provided CW20 funds, but market requires native funds with denom {denom}"
                )
            }
        }
    }
}

/// Work response
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq)]
#[serde(rename_all = "snake_case")]
pub enum WorkResp {
    /// No work found
    NoWork,
    /// Has some work
    HasWork {
        /// Work description
        work_description: WorkDescription,
    },
}

impl WorkResp {
    /// Does it have work ?
    pub fn has_work(&self) -> bool {
        match self {
            WorkResp::NoWork => false,
            WorkResp::HasWork { .. } => true,
        }
    }

    /// Is it deferred work
    pub fn is_deferred_work(&self) -> bool {
        match self {
            WorkResp::NoWork => false,
            WorkResp::HasWork { work_description } => work_description.is_deferred_work(),
        }
    }

    /// Is it deferred work
    pub fn is_rebalance(&self) -> bool {
        match self {
            WorkResp::NoWork => false,
            WorkResp::HasWork { work_description } => work_description.is_rebalance(),
        }
    }

    /// Is it compute lp token work ?
    pub fn is_compute_lp_token(&self) -> bool {
        match self {
            WorkResp::NoWork => false,
            WorkResp::HasWork { work_description } => work_description.is_compute_lp_token(),
        }
    }

    /// Is it reset status work ?
    pub fn is_reset_status(&self) -> bool {
        match self {
            WorkResp::NoWork => false,
            WorkResp::HasWork { work_description } => work_description.is_reset_status(),
        }
    }
}

/// Work Description
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq)]
#[serde(rename_all = "snake_case")]
pub enum WorkDescription {
    /// Load Market
    LoadMarket {},
    /// Calculate LP token value
    ComputeLpTokenValue {
        /// Token
        token: Token,
        /// Start from specific market id in the processing phase
        process_start_from: Option<MarketId>,
        /// Start from specific market id in the validate phase
        validate_start_from: Option<MarketId>,
    },
    /// Process Queue item
    ProcessQueueItem {
        /// Id to process
        id: QueuePositionId,
    },
    /// Reset market specific statistics
    ResetStats {
        /// Token
        token: Token,
    },
    /// Handle deferred exec id
    HandleDeferredExecId {},
    /// Rebalance is done when contract balance is not same as the one
    /// internally tracked by it.
    Rebalance {
        /// Token
        token: Token,
        /// Amount that needs to be balanced
        amount: NonZero<Collateral>,
        /// Start from specific market id
        start_from: Option<MarketId>,
    },
}

impl WorkDescription {
    /// Is it the rebalance work ?
    pub fn is_rebalance(&self) -> bool {
        match self {
            WorkDescription::LoadMarket {} => false,
            WorkDescription::ComputeLpTokenValue { .. } => false,
            WorkDescription::ProcessQueueItem { .. } => false,
            WorkDescription::ResetStats { .. } => false,
            WorkDescription::HandleDeferredExecId {} => false,
            WorkDescription::Rebalance { .. } => true,
        }
    }

    /// Is it compute lp token work ?
    pub fn is_compute_lp_token(&self) -> bool {
        match self {
            WorkDescription::LoadMarket {} => false,
            WorkDescription::ComputeLpTokenValue { .. } => true,
            WorkDescription::ProcessQueueItem { .. } => false,
            WorkDescription::ResetStats { .. } => false,
            WorkDescription::HandleDeferredExecId {} => false,
            WorkDescription::Rebalance { .. } => false,
        }
    }

    /// Is it reset stats ?
    pub fn is_reset_status(&self) -> bool {
        match self {
            WorkDescription::LoadMarket {} => false,
            WorkDescription::ComputeLpTokenValue { .. } => false,
            WorkDescription::ProcessQueueItem { .. } => false,
            WorkDescription::ResetStats { .. } => true,
            WorkDescription::HandleDeferredExecId {} => false,
            WorkDescription::Rebalance { .. } => false,
        }
    }

    /// Is deferred work ?
    pub fn is_deferred_work(&self) -> bool {
        match self {
            WorkDescription::LoadMarket {} => false,
            WorkDescription::ComputeLpTokenValue { .. } => false,
            WorkDescription::ProcessQueueItem { .. } => false,
            WorkDescription::ResetStats { .. } => false,
            WorkDescription::HandleDeferredExecId {} => true,
            WorkDescription::Rebalance { .. } => false,
        }
    }
}

/// Queue position id that needs to be processed
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone, PartialEq)]
#[serde(rename_all = "snake_case")]
pub enum QueuePositionId {
    /// Queue position id corrsponding to the queue items that will
    /// increase or won't change the collateral
    IncQueuePositionId(IncQueuePositionId),
    /// Queue position id corresponding to the queue items that will
    /// decrease the collateral
    DecQueuePositionId(DecQueuePositionId),
}

impl<'a> PrimaryKey<'a> for QueuePositionId {
    type Prefix = ();
    type SubPrefix = ();
    type Suffix = Self;
    type SuperSuffix = Self;

    fn key(&self) -> Vec<Key> {
        let (queue_type, key) = match self {
            QueuePositionId::IncQueuePositionId(id) => (0u8, id.key()),
            QueuePositionId::DecQueuePositionId(id) => (1u8, id.key()),
        };
        let mut keys = vec![Key::Val8([queue_type])];
        keys.extend(key);
        keys
    }
}

impl KeyDeserialize for QueuePositionId {
    type Output = QueuePositionId;

    const KEY_ELEMS: u16 = 2;

    fn from_vec(value: Vec<u8>) -> StdResult<Self::Output> {
        let (queue_type, queue_id) = <(u8, u64) as KeyDeserialize>::from_vec(value)?;
        let position_id = match queue_type {
            0 => QueuePositionId::IncQueuePositionId(IncQueuePositionId(queue_id.into())),
            1 => QueuePositionId::DecQueuePositionId(DecQueuePositionId(queue_id.into())),
            _ => {
                return Err(StdError::serialize_err(
                    "QueuePositionId",
                    "Invalid number in queue_type",
                ))
            }
        };
        Ok(position_id)
    }
}

impl<'a> Prefixer<'a> for QueuePositionId {
    fn prefix(&self) -> Vec<Key> {
        match self {
            QueuePositionId::IncQueuePositionId(id) => {
                let mut keys = vec![Key::Val8([0u8])];
                keys.extend(id.key());
                keys
            }
            QueuePositionId::DecQueuePositionId(id) => {
                let mut keys = vec![Key::Val8([1u8])];
                keys.extend(id.key());
                keys
            }
        }
    }
}

/// Queue position number
#[derive(
    Copy, PartialOrd, Ord, Eq, Clone, PartialEq, serde::Serialize, serde::Deserialize, Debug,
)]
#[serde(rename_all = "snake_case")]
pub struct IncQueuePositionId(Uint64);

impl IncQueuePositionId {
    /// Construct a new value from a [u64].
    pub fn new(x: u64) -> Self {
        IncQueuePositionId(x.into())
    }

    /// The underlying `u64` representation.
    pub fn u64(self) -> u64 {
        self.0.u64()
    }

    /// Generate the next position ID
    ///
    /// Panics on overflow
    pub fn next(self) -> Self {
        IncQueuePositionId((self.u64() + 1).into())
    }
}

impl<'a> PrimaryKey<'a> for IncQueuePositionId {
    type Prefix = ();
    type SubPrefix = ();
    type Suffix = Self;
    type SuperSuffix = Self;

    fn key(&self) -> Vec<Key> {
        vec![Key::Val64(self.0.u64().to_cw_bytes())]
    }
}

impl<'a> Prefixer<'a> for IncQueuePositionId {
    fn prefix(&self) -> Vec<Key> {
        vec![Key::Val64(self.0.u64().to_cw_bytes())]
    }
}

impl KeyDeserialize for IncQueuePositionId {
    type Output = IncQueuePositionId;

    const KEY_ELEMS: u16 = 1;

    #[inline(always)]
    fn from_vec(value: Vec<u8>) -> StdResult<Self::Output> {
        u64::from_vec(value).map(|x| IncQueuePositionId(Uint64::new(x)))
    }
}

impl std::fmt::Display for IncQueuePositionId {
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        write!(f, "{}", self.0)
    }
}

impl FromStr for IncQueuePositionId {
    type Err = ParseIntError;
    fn from_str(src: &str) -> Result<Self, ParseIntError> {
        src.parse().map(|x| IncQueuePositionId(Uint64::new(x)))
    }
}

/// Queue position number
#[derive(
    Copy, PartialOrd, Ord, Eq, Clone, PartialEq, serde::Serialize, serde::Deserialize, Debug,
)]
#[serde(rename_all = "snake_case")]
pub struct DecQueuePositionId(Uint64);

impl DecQueuePositionId {
    /// Construct a new value from a [u64].
    pub fn new(x: u64) -> Self {
        DecQueuePositionId(x.into())
    }

    /// The underlying `u64` representation.
    pub fn u64(self) -> u64 {
        self.0.u64()
    }

    /// Generate the next position ID
    ///
    /// Panics on overflow
    pub fn next(self) -> Self {
        DecQueuePositionId((self.u64() + 1).into())
    }
}

impl<'a> PrimaryKey<'a> for DecQueuePositionId {
    type Prefix = ();
    type SubPrefix = ();
    type Suffix = Self;
    type SuperSuffix = Self;

    fn key(&self) -> Vec<Key> {
        vec![Key::Val64(self.0.u64().to_cw_bytes())]
    }
}

impl<'a> Prefixer<'a> for DecQueuePositionId {
    fn prefix(&self) -> Vec<Key> {
        vec![Key::Val64(self.0.u64().to_cw_bytes())]
    }
}

impl KeyDeserialize for DecQueuePositionId {
    type Output = DecQueuePositionId;

    const KEY_ELEMS: u16 = 1;

    #[inline(always)]
    fn from_vec(value: Vec<u8>) -> StdResult<Self::Output> {
        u64::from_vec(value).map(|x| DecQueuePositionId(Uint64::new(x)))
    }
}

impl std::fmt::Display for DecQueuePositionId {
    fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result {
        write!(f, "{}", self.0)
    }
}

impl FromStr for DecQueuePositionId {
    type Err = ParseIntError;
    fn from_str(src: &str) -> Result<Self, ParseIntError> {
        src.parse().map(|x| DecQueuePositionId(Uint64::new(x)))
    }
}

/// Migration message, currently no fields needed
#[derive(serde::Serialize, serde::Deserialize, Debug, Clone)]
#[serde(rename_all = "snake_case")]
pub struct MigrateMsg {}