use defuse_core::{amounts::Amounts, token_id::TokenId};
use defuse_near_utils::NestPrefix;
use near_sdk::{BorshStorageKey, IntoStorageKey, near, store::IterableMap};

#[derive(Debug)]
#[near(serializers = [borsh])]
pub struct AccountState {
    pub token_balances: Amounts<IterableMap<TokenId, u128>>,
}

impl AccountState {
    pub fn new<S>(prefix: S) -> Self
    where
        S: IntoStorageKey,
    {
        let parent = prefix.into_storage_key();

        Self {
            token_balances: Amounts::new(IterableMap::new(
                parent.as_slice().nest(AccountStatePrefix::TokenBalances),
            )),
        }
    }
}

#[derive(BorshStorageKey)]
#[near(serializers = [borsh])]
enum AccountStatePrefix {
    TokenBalances,
}