ore/program/src/automate.rs

use ore_api::prelude::*;
use steel::*;

/// Sets the executor.
pub fn process_automate(accounts: &[AccountInfo<'_>], data: &[u8]) -> ProgramResult {
    // Parse data.
    let args = Automate::try_from_bytes(data)?;
    let amount = u64::from_le_bytes(args.amount);
    let deposit = u64::from_le_bytes(args.deposit);
    let fee = u64::from_le_bytes(args.fee);
    let mask = u64::from_le_bytes(args.mask);
    let strategy = AutomationStrategy::from_u64(args.strategy as u64);

    // Load accounts.
    let [signer_info, automation_info, executor_info, miner_info, system_program] = accounts else {
        return Err(ProgramError::NotEnoughAccountKeys);
    };
    signer_info.is_signer()?;
    automation_info.is_writable()?;
    miner_info
        .as_account_mut::<Miner>(&ore_api::ID)?
        .assert_mut_err(
            |m| m.authority == *signer_info.key,
            OreError::NotAuthorized.into(),
        )?;
    system_program.is_program(&system_program::ID)?;

    // // Check whitelist
    // if !AUTHORIZED_ACCOUNTS.contains(&signer_info.key) {
    //     return Err(trace("Not authorized", OreError::NotAuthorized.into()));
    // }

    // Close account if executor is Pubkey::default().
    if *executor_info.key == Pubkey::default() {
        automation_info
            .as_account_mut::<Automation>(&ore_api::ID)?
            .assert_mut_err(
                |a| a.authority == *signer_info.key,
                OreError::NotAuthorized.into(),
            )?;
        automation_info.close(signer_info)?;
        return Ok(());
    }

    // Create automation.
    let automation = if automation_info.data_is_empty() {
        create_program_account::<Automation>(
            automation_info,
            system_program,
            signer_info,
            &ore_api::ID,
            &[AUTOMATION, &signer_info.key.to_bytes()],
        )?;
        let automation = automation_info.as_account_mut::<Automation>(&ore_api::ID)?;
        automation.balance = 0;
        automation.authority = *signer_info.key;
        automation
    } else {
        automation_info
            .as_account_mut::<Automation>(&ore_api::ID)?
            .assert_mut_err(
                |a| a.authority == *signer_info.key,
                OreError::NotAuthorized.into(),
            )?
    };

    // Set strategy and mask.
    automation.amount = amount;
    automation.balance += deposit;
    automation.executor = *executor_info.key;
    automation.fee = fee;
    automation.mask = mask;
    automation.strategy = strategy as u64;

    // Transfer balance to executor.
    automation_info.collect(deposit, signer_info)?;

    Ok(())
}