use solana_pubkey::Pubkey;
use borsh::BorshSerialize;
use borsh::BorshDeserialize;
#[derive(BorshSerialize, BorshDeserialize, Clone, Debug, Eq, PartialEq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct Treasury {
pub discriminator: [u8; 8],
pub version: u16,
pub bump: [u8; 1],
#[cfg_attr(feature = "serde", serde(with = "serde_with::As::<serde_with::DisplayFromStr>"))]
pub staked_token_mint: Pubkey,
#[cfg_attr(feature = "serde", serde(with = "serde_with::As::<serde_with::DisplayFromStr>"))]
pub reward_token_mint: Pubkey,
#[cfg_attr(feature = "serde", serde(with = "serde_with::As::<serde_with::DisplayFromStr>"))]
pub admin_authority: Pubkey,
#[cfg_attr(feature = "serde", serde(with = "serde_with::As::<serde_with::DisplayFromStr>"))]
pub reward_authority: Pubkey,
pub flags: u32,
pub cooldown_interval: u64,
pub total_staked_shares: u64,
pub acc_reward_per_share: u128,
pub total_reward: u64,
pub total_unclaimed_reward: u64,
pub total_unstaked_shares: u64,
#[cfg_attr(feature = "serde", serde(with = "serde_with::As::<serde_with::DisplayFromStr>"))]
pub owner_authority: Pubkey,
#[cfg_attr(feature = "serde", serde(with = "serde_with::As::<serde_with::Bytes>"))]
pub reserved: [u8; 80],
}
impl Treasury {
pub const LEN: usize = 311;
#[inline(always)]
pub fn from_bytes(data: &[u8]) -> Result<Self, std::io::Error> {
let mut data = data;
Self::deserialize(&mut data)
}
}
impl<'a> TryFrom<&solana_account_info::AccountInfo<'a>> for Treasury {
type Error = std::io::Error;
fn try_from(account_info: &solana_account_info::AccountInfo<'a>) -> Result<Self, Self::Error> {
let mut data: &[u8] = &(*account_info.data).borrow();
Self::deserialize(&mut data)
}
}
#[cfg(feature = "fetch")]
pub fn fetch_treasury(
rpc: &solana_client::rpc_client::RpcClient,
address: &solana_pubkey::Pubkey,
) -> Result<crate::shared::DecodedAccount<Treasury>, std::io::Error> {
let accounts = fetch_all_treasury(rpc, &[*address])?;
Ok(accounts[0].clone())
}
#[cfg(feature = "fetch")]
pub fn fetch_all_treasury(
rpc: &solana_client::rpc_client::RpcClient,
addresses: &[solana_pubkey::Pubkey],
) -> Result<Vec<crate::shared::DecodedAccount<Treasury>>, std::io::Error> {
let accounts = rpc.get_multiple_accounts(addresses)
.map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e.to_string()))?;
let mut decoded_accounts: Vec<crate::shared::DecodedAccount<Treasury>> = Vec::new();
for i in 0..addresses.len() {
let address = addresses[i];
let account = accounts[i].as_ref()
.ok_or(std::io::Error::new(std::io::ErrorKind::Other, format!("Account not found: {}", address)))?;
let data = Treasury::from_bytes(&account.data)?;
decoded_accounts.push(crate::shared::DecodedAccount { address, account: account.clone(), data });
}
Ok(decoded_accounts)
}
#[cfg(feature = "fetch")]
pub fn fetch_maybe_treasury(
rpc: &solana_client::rpc_client::RpcClient,
address: &solana_pubkey::Pubkey,
) -> Result<crate::shared::MaybeAccount<Treasury>, std::io::Error> {
let accounts = fetch_all_maybe_treasury(rpc, &[*address])?;
Ok(accounts[0].clone())
}
#[cfg(feature = "fetch")]
pub fn fetch_all_maybe_treasury(
rpc: &solana_client::rpc_client::RpcClient,
addresses: &[solana_pubkey::Pubkey],
) -> Result<Vec<crate::shared::MaybeAccount<Treasury>>, std::io::Error> {
let accounts = rpc.get_multiple_accounts(addresses)
.map_err(|e| std::io::Error::new(std::io::ErrorKind::Other, e.to_string()))?;
let mut decoded_accounts: Vec<crate::shared::MaybeAccount<Treasury>> = Vec::new();
for i in 0..addresses.len() {
let address = addresses[i];
if let Some(account) = accounts[i].as_ref() {
let data = Treasury::from_bytes(&account.data)?;
decoded_accounts.push(crate::shared::MaybeAccount::Exists(crate::shared::DecodedAccount { address, account: account.clone(), data }));
} else {
decoded_accounts.push(crate::shared::MaybeAccount::NotFound(address));
}
}
Ok(decoded_accounts)
}
#[cfg(feature = "anchor")]
impl anchor_lang::AccountDeserialize for Treasury {
fn try_deserialize_unchecked(buf: &mut &[u8]) -> anchor_lang::Result<Self> {
Ok(Self::deserialize(buf)?)
}
}
#[cfg(feature = "anchor")]
impl anchor_lang::AccountSerialize for Treasury {}
#[cfg(feature = "anchor")]
impl anchor_lang::Owner for Treasury {
fn owner() -> Pubkey {
crate::TUNA_STAKING_ID
}
}
#[cfg(feature = "anchor-idl-build")]
impl anchor_lang::IdlBuild for Treasury {}
#[cfg(feature = "anchor-idl-build")]
impl anchor_lang::Discriminator for Treasury {
const DISCRIMINATOR: [u8; 8] = [0; 8];
}