use borsh::{BorshDeserialize, BorshSerialize};
use mpl_utils::{assert_derivation, assert_signer, resize_or_reallocate_account_raw};
use solana_program::{
account_info::{next_account_info, AccountInfo},
entrypoint::ProgramResult,
program_memory::sol_memcpy,
system_program,
};
use crate::{
error::OnchainMetadataError, instruction::AddAuthorityArgs, pda::PREFIX, state::JsonMetadata,
};
pub(crate) fn process_add_authority(
accounts: &[AccountInfo],
args: AddAuthorityArgs,
) -> ProgramResult {
let account_info_iter = &mut accounts.iter();
let json_account = next_account_info(account_info_iter)?;
if (json_account.owner != &crate::ID) || json_account.data_is_empty() {
return Err(OnchainMetadataError::NotInitialized.into());
}
let json_metadata_account = next_account_info(account_info_iter)?;
if (json_metadata_account.owner != &crate::ID) || json_metadata_account.data_is_empty() {
return Err(OnchainMetadataError::NotInitialized.into());
}
let mut json_metadata = JsonMetadata::try_from_slice(&json_metadata_account.data.borrow())?;
let bump = assert_derivation(
&crate::ID,
json_metadata_account,
&[
PREFIX.as_bytes(),
crate::ID.as_ref(),
json_account.key.as_ref(),
],
OnchainMetadataError::MetadataDerivedKeyInvalid,
)?;
if bump != json_metadata.bump {
return Err(OnchainMetadataError::MetadataDerivedKeyInvalid.into());
}
let payer = next_account_info(account_info_iter)?;
assert_signer(payer)?;
if !json_metadata.authorities.contains(payer.key) {
return Err(OnchainMetadataError::InvalidAuthority.into());
}
let system_program = next_account_info(account_info_iter)?;
if system_program.key != &system_program::ID {
return Err(OnchainMetadataError::InvalidSystemProgram.into());
}
json_metadata.authorities.push(args.new_authority);
let serialized_data = json_metadata.try_to_vec()?;
resize_or_reallocate_account_raw(
json_metadata_account,
payer,
system_program,
serialized_data.len(),
)?;
sol_memcpy(
&mut json_metadata_account.try_borrow_mut_data()?,
&serialized_data,
serialized_data.len(),
);
Ok(())
}