use std::collections::BTreeMap;

use anyhow;
use cosmwasm_std::{Addr, HexBinary, Uint128};
use serde::{Deserialize, Serialize};

#[derive(Clone, Debug)]
pub struct State {
    pub state: BTreeMap<Addr, Uint128>,
}

#[derive(Clone, Debug, PartialEq, Eq, Deserialize, Serialize)]
pub struct RawState {
    pub state: BTreeMap<Addr, Uint128>,
}

#[derive(Clone, Debug, Serialize, Deserialize)]
pub struct RawEncryptedState {
    pub ciphertext: HexBinary,
}

impl From<State> for RawState {
    fn from(o: State) -> Self {
        Self { state: o.state }
    }
}

impl TryFrom<RawState> for State {
    type Error = anyhow::Error;

    fn try_from(o: RawState) -> Result<Self, anyhow::Error> {
        Ok(Self { state: o.state })
    }
}

#[derive(Clone, Debug)]
pub struct Balance {
    pub balance: Uint128,
}

#[derive(Clone, Debug, PartialEq, Eq, Deserialize, Serialize)]
pub struct RawBalance {
    pub balance: Uint128,
}

impl From<Balance> for RawBalance {
    fn from(o: Balance) -> Self {
        Self { balance: o.balance }
    }
}

impl TryFrom<RawBalance> for Balance {
    type Error = anyhow::Error;

    fn try_from(o: RawBalance) -> Result<Self, anyhow::Error> {
        Ok(Self { balance: o.balance })
    }
}