1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use {
    crate::state::*,
    anchor_lang::prelude::*,
};

#[derive(Accounts)]
#[instruction()]
pub struct FeeCollect<'info> {
    #[account(
        mut,
        seeds = [
            SEED_FEE, 
            fee.daemon.as_ref()
        ],
        bump = fee.bump,
    )]
    pub fee: Account<'info, Fee>,

    #[account(mut)]
    pub signer: Signer<'info>,

    #[account(
        mut,
        seeds = [SEED_TREASURY],
        bump = treasury.bump,
    )]
    pub treasury: Account<'info, Treasury>,
}

pub fn handler(ctx: Context<FeeCollect>) -> ProgramResult {
    // Get accounts.
    let fee = &mut ctx.accounts.fee;
    let treasury = &mut ctx.accounts.treasury;

    // Collect lamports fee account to treasury.
    **fee.to_account_info().try_borrow_mut_lamports()? -= fee.balance;
    **treasury.to_account_info().try_borrow_mut_lamports()? += fee.balance;

    // Null out collectable fee balance.
    fee.balance = 0;
    
    Ok(())
}