epoch

src/loaders.rs

@@ -1,5 +1,12 @@
 use solana_program::{
-    account_info::AccountInfo, program_error::ProgramError, pubkey::Pubkey, system_program,
+    account_info::AccountInfo, program_error::ProgramError, program_pack::Pack, pubkey::Pubkey,
+    system_program,
+};
+use spl_token::state::Mint;
+
+use crate::{
+    state::{Bus, Treasury},
+    BUS, MINT_ADDRESS, TREASURY,
 };
 
 pub fn load_signer<'a, 'info>(
@@ -42,6 +49,97 @@ pub fn load_uninitialized_pda<'a, 'info>(
     load_uninitialized_account(info)
 }
 
+pub fn load_bus<'a, 'info>(
+    info: &'a AccountInfo<'info>,
+) -> Result<&'a AccountInfo<'info>, ProgramError> {
+    if !info.owner.eq(&crate::id()) {
+        return Err(ProgramError::InvalidAccountOwner);
+    }
+    if info.data_is_empty() {
+        return Err(ProgramError::UninitializedAccount);
+    }
+
+    let bus_data = info.data.borrow();
+    let bus = bytemuck::try_from_bytes::<Bus>(&bus_data).unwrap();
+
+    let key =
+        Pubkey::create_program_address(&[BUS, &[bus.id as u8], &[bus.bump as u8]], &crate::id())?;
+    if !info.key.eq(&key) {
+        return Err(ProgramError::InvalidSeeds);
+    }
+
+    Ok(info)
+}
+
+pub fn load_treasury<'a, 'info>(
+    info: &'a AccountInfo<'info>,
+) -> Result<&'a AccountInfo<'info>, ProgramError> {
+    if !info.owner.eq(&crate::id()) {
+        return Err(ProgramError::InvalidAccountOwner);
+    }
+    if info.data_is_empty() {
+        return Err(ProgramError::UninitializedAccount);
+    }
+
+    let treasury_data = info.data.borrow();
+    let treasury = bytemuck::try_from_bytes::<Treasury>(&treasury_data).unwrap();
+
+    let key = Pubkey::create_program_address(&[TREASURY, &[treasury.bump as u8]], &crate::id())?;
+    if !info.key.eq(&key) {
+        return Err(ProgramError::InvalidSeeds);
+    }
+
+    Ok(info)
+}
+
+pub fn load_mint<'a, 'info>(
+    info: &'a AccountInfo<'info>,
+) -> Result<&'a AccountInfo<'info>, ProgramError> {
+    if !info.owner.eq(&spl_token::id()) {
+        return Err(ProgramError::InvalidAccountOwner);
+    }
+    if info.data_is_empty() {
+        return Err(ProgramError::UninitializedAccount);
+    }
+
+    let mint_data = info.data.borrow();
+    if Mint::unpack_unchecked(&mint_data).is_err() {
+        return Err(ProgramError::InvalidAccountData);
+    }
+
+    if !info.key.eq(&MINT_ADDRESS) {
+        return Err(ProgramError::InvalidAccountData);
+    }
+
+    Ok(info)
+}
+
+pub fn load_token_account<'a, 'info>(
+    info: &'a AccountInfo<'info>,
+    owner: &Pubkey,
+    mint: &Pubkey,
+) -> Result<&'a AccountInfo<'info>, ProgramError> {
+    if !info.owner.eq(&spl_token::id()) {
+        return Err(ProgramError::InvalidAccountOwner);
+    }
+    if info.data_is_empty() {
+        return Err(ProgramError::UninitializedAccount);
+    }
+
+    let account_data = info.data.borrow();
+    let account = spl_token::state::Account::unpack_unchecked(&account_data)
+        .or(Err(ProgramError::InvalidAccountData))?;
+
+    if !account.mint.eq(mint) {
+        return Err(ProgramError::InvalidAccountData);
+    }
+    if !account.owner.eq(owner) {
+        return Err(ProgramError::InvalidAccountData);
+    }
+
+    Ok(info)
+}
+
 pub fn load_uninitialized_account<'a, 'info>(
     info: &'a AccountInfo<'info>,
 ) -> Result<&'a AccountInfo<'info>, ProgramError> {