This commit is contained in:
Hardhat Chad
2025-06-25 14:33:58 -05:00
parent 11d5442dd2
commit d03acda291
6 changed files with 134 additions and 3 deletions

45
program/src/set_admin.rs Normal file
View File

@@ -0,0 +1,45 @@
use ore_api::prelude::*;
use steel::*;
/// Sets the admin.
pub fn process_set_admin(accounts: &[AccountInfo<'_>], data: &[u8]) -> ProgramResult {
// Parse data.
let args = SetAdmin::try_from_bytes(data)?;
let new_admin = Pubkey::new_from_array(args.admin);
// Load accounts.
let [signer_info, config_info, system_program] = accounts else {
return Err(ProgramError::NotEnoughAccountKeys);
};
signer_info.is_signer()?;
system_program.is_program(&system_program::ID)?;
// Load config account.
let config = if config_info.data_is_empty() {
// Assert signer is admin.
// signer_info.has_address(&ADMIN_ADDRESS)?;
// Create config account.
create_program_account::<Config>(
config_info,
system_program,
signer_info,
&ore_api::ID,
&[CONFIG],
)?;
let config = config_info.as_account_mut::<Config>(&ore_api::ID)?;
config.admin = *signer_info.key;
config.fee_collector = *signer_info.key;
config.fee_rate = 0;
config
} else {
config_info
.as_account_mut::<Config>(&ore_api::ID)?
.assert_mut(|c| c.admin == *signer_info.key)?
};
// Set admin.
config.admin = new_admin;
Ok(())
}