use hpsvm::{HPSVM, types::FailedTransactionMetadata};
use smallvec::{SmallVec, smallvec};
use solana_address::Address;
use solana_keypair::Keypair;
use solana_signer::{Signer, signers::Signers};
use solana_transaction::Transaction;
use super::{TOKEN_ID, get_multisig_signers, spl_token::instruction::burn};
#[derive(Debug)]
pub struct Burn<'a> {
svm: &'a mut HPSVM,
payer: &'a Keypair,
mint: &'a Address,
account: &'a Address,
token_program_id: Option<&'a Address>,
amount: u64,
signers: SmallVec<[&'a Keypair; 1]>,
owner: Option<Address>,
}
impl<'a> Burn<'a> {
pub fn new(
svm: &'a mut HPSVM,
payer: &'a Keypair,
mint: &'a Address,
account: &'a Address,
amount: u64,
) -> Self {
Burn {
svm,
payer,
mint,
account,
token_program_id: None,
amount,
owner: None,
signers: smallvec![payer],
}
}
pub fn token_program_id(mut self, program_id: &'a Address) -> Self {
self.token_program_id = Some(program_id);
self
}
pub fn owner(mut self, owner: &'a Keypair) -> Self {
self.owner = Some(owner.pubkey());
self.signers = smallvec![owner];
self
}
pub fn multisig(mut self, multisig: &'a Address, signers: &'a [&'a Keypair]) -> Self {
self.owner = Some(*multisig);
self.signers = SmallVec::from(signers);
self
}
pub fn send(self) -> Result<(), FailedTransactionMetadata> {
let payer_pk = self.payer.pubkey();
let token_program_id = self.token_program_id.unwrap_or(&TOKEN_ID);
let authority = self.owner.unwrap_or(payer_pk);
let signing_keys = self.signers.pubkeys();
let signer_keys = get_multisig_signers(&authority, &signing_keys);
let ix =
burn(token_program_id, self.account, self.mint, &authority, &signer_keys, self.amount)?;
let block_hash = self.svm.latest_blockhash();
let mut tx = Transaction::new_with_payer(&[ix], Some(&payer_pk));
tx.partial_sign(&[self.payer], block_hash);
tx.partial_sign(self.signers.as_ref(), block_hash);
self.svm.send_transaction(tx)?;
Ok(())
}
}