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

use crate::whitelist::AUTHORIZED_ACCOUNTS;

/// Claims a block reward.
pub fn process_prospect(accounts: &[AccountInfo<'_>], data: &[u8]) -> ProgramResult {
    // Parse data.
    let args = Prospect::try_from_bytes(data)?;
    let amount = u64::from_le_bytes(args.amount);
    let square_id = usize::from_le_bytes(args.square_id);

    // Load accounts.
    let clock = Clock::get()?;
    let [signer_info, board_info, config_info, fee_collector_info, miner_info, square_info, system_program] =
        accounts
    else {
        return Err(ProgramError::NotEnoughAccountKeys);
    };
    signer_info.is_signer()?;
    let board = board_info
        .as_account_mut::<Board>(&ore_api::ID)?
        .assert_mut(|b| {
            (clock.slot >= b.start_slot && clock.slot < b.end_slot && b.slot_hash == [0; 32])
                || (clock.slot >= b.end_slot + 25 && b.slot_hash != [0; 32])
        })?;
    let config = config_info.as_account::<Config>(&ore_api::ID)?;
    fee_collector_info
        .has_address(&config.fee_collector)?
        .is_writable()?;
    miner_info.is_writable()?;
    let square = square_info.as_account_mut::<Square>(&ore_api::ID)?;
    system_program.is_program(&system_program::ID)?;

    // Check whitelist
    if !AUTHORIZED_ACCOUNTS.contains(&signer_info.key) {
        return Err(ProgramError::InvalidAccountData);
    }

    // Check minimum amount.
    if amount < config.min_prospect_amount {
        return Err(ProgramError::InvalidAccountData);
    }

    // Create miner.
    let miner = if miner_info.data_is_empty() {
        create_program_account::<Miner>(
            miner_info,
            system_program,
            signer_info,
            &ore_api::ID,
            &[MINER, &signer_info.key.to_bytes()],
        )?;
        let miner = miner_info.as_account_mut::<Miner>(&ore_api::ID)?;
        miner.authority = *signer_info.key;
        miner.prospects = [0; 25];
        miner.rewards_sol = 0;
        miner.rewards_ore = 0;
        miner.round_id = board.id;
        miner.lifetime_rewards_sol = 0;
        miner.lifetime_rewards_ore = 0;
        miner
    } else {
        miner_info
            .as_account_mut::<Miner>(&ore_api::ID)?
            .assert_mut(|m| m.authority == *signer_info.key)?
    };

    // Reset board.
    if board.slot_hash != [0; 32] {
        // Reset board
        board.prospects = [0; 25];
        board.id += 1;
        board.slot_hash = [0; 32];
        board.start_slot = clock.slot;
        board.end_slot = clock.slot + 150; // one minute
        board.top_miner = Pubkey::default();
        board.total_prospects = 0;
        board.total_vaulted = 0;
        board.total_winnings = 0;

        // Reset squares
        square.count = [0; 25];
        square.miners = [[Pubkey::default(); 16]; 25];
    }

    // Reset miner
    if miner.round_id != board.id {
        miner.prospects = [0; 25];
        miner.round_id = board.id;
    }

    // Normalize amount.
    let fee = amount / 100;
    let amount = amount - fee;

    // Update miner
    let is_first_move = miner.prospects[square_id] == 0;
    miner.prospects[square_id] += amount;

    // Update square
    if is_first_move {
        square.miners[square_id][square.count[square_id] as usize] = *signer_info.key;
        square.count[square_id] += 1;
    }

    // Update board
    board.prospects[square_id] += amount;
    board.total_prospects += amount;

    // Transfer prospects.
    board_info.collect(amount, &signer_info)?;
    fee_collector_info.collect(fee, &signer_info)?;

    Ok(())
}