use hpsvm::{HPSVM, types::FailedTransactionMetadata};
use smallvec::{SmallVec, smallvec};
use solana_address::Address;
use solana_keypair::Keypair;
use solana_signer::{Signer, signers::Signers};
use super::{
TOKEN_ID, get_multisig_signers,
spl_token::instruction::{AuthorityType, set_authority},
};
#[derive(Debug)]
pub struct SetAuthority<'a> {
svm: &'a mut HPSVM,
payer: &'a Keypair,
authority_type: AuthorityType,
account: &'a Address,
new_authority: Option<&'a Address>,
signers: SmallVec<[&'a Keypair; 1]>,
owner: Option<Address>,
token_program_id: Option<&'a Address>,
}
impl<'a> SetAuthority<'a> {
pub fn new(
svm: &'a mut HPSVM,
payer: &'a Keypair,
account: &'a Address,
authority_type: AuthorityType,
) -> Self {
SetAuthority {
svm,
payer,
owner: None,
authority_type,
account,
new_authority: None,
token_program_id: None,
signers: smallvec![payer],
}
}
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 token_program_id(mut self, program_id: &'a Address) -> Self {
self.token_program_id = Some(program_id);
self
}
pub fn new_authority(mut self, new_authority: &'a Address) -> Self {
self.new_authority = Some(new_authority);
self
}
pub fn send(self) -> Result<(), FailedTransactionMetadata> {
let token_program_id = self.token_program_id.unwrap_or(&TOKEN_ID);
let payer_pk = self.payer.pubkey();
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 = set_authority(
token_program_id,
self.account,
self.new_authority,
self.authority_type,
&authority,
&signer_keys,
)?;
super::sign_and_send(self.svm, self.payer, &self.signers, ix)
}
}