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

/// Deposits collateral.
pub fn process_deposit(accounts: &[AccountInfo<'_>], data: &[u8]) -> ProgramResult {
    // Parse data.
    let args = Deposit::try_from_bytes(data)?;
    let amount = u64::from_le_bytes(args.amount);

    // Load accounts.
    let clock = Clock::get()?;
    let [signer_info, block_info, collateral_info, mint_ore_info, sender_info, stake_info, system_program, token_program] =
        accounts
    else {
        return Err(ProgramError::NotEnoughAccountKeys);
    };
    signer_info.is_signer()?;
    let block = block_info
        .as_account_mut::<Block>(&ore_api::ID)?
        .assert_mut(|b| clock.slot < b.start_slot)?;
    collateral_info
        .is_writable()?
        .as_associated_token_account(block_info.key, mint_ore_info.key)?;
    mint_ore_info.has_address(&MINT_ADDRESS)?.as_mint()?;
    sender_info
        .is_writable()?
        .as_associated_token_account(signer_info.key, &mint_ore_info.key)?
        .assert(|t| t.amount() >= amount)?;
    system_program.is_program(&system_program::ID)?;
    token_program.is_program(&spl_token::ID)?;

    // Load stake account.
    let stake = if stake_info.data_is_empty() {
        create_program_account::<Stake>(
            stake_info,
            system_program,
            signer_info,
            &ore_api::ID,
            &[STAKE, &signer_info.key.to_bytes(), &block.id.to_le_bytes()],
        )?;
        let stake = stake_info.as_account_mut::<Stake>(&ore_api::ID)?;
        stake.authority = *signer_info.key;
        stake.block_id = block.id;
        stake.collateral = 0;
        stake.spend = 0;
        stake
    } else {
        stake_info
            .as_account_mut::<Stake>(&ore_api::ID)?
            .assert_mut(|p| p.authority == *signer_info.key)?
            .assert_mut(|p| p.block_id == block.id)?
    };

    // Update stake state.
    stake.collateral += amount;

    // Transfer collateral.
    transfer(
        signer_info,
        sender_info,
        collateral_info,
        token_program,
        amount,
    )?;

    // Emit event.
    DepositEvent {
        disc: OreEvent::Deposit as u64,
        authority: *signer_info.key,
        block_id: block.id,
        amount,
        collateral: stake.collateral,
        ts: clock.unix_timestamp,
    }
    .log_return();

    Ok(())
}