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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

use {
    anchor_lang::{
        prelude::*,
        solana_program::system_program
    },
    crate::state,
    std::mem::size_of
};

#[derive(Accounts)]
#[instruction(
    config_transfer_fee_distributor: u64,
    config_transfer_fee_program: u64,
    authority_bump: u8,
    config_bump: u8,
    treasury_bump: u8,
)]
pub struct InitializeProgram<'info> {
    #[account(mut)]
    pub signer: Signer<'info>,

    #[account(
        init,
        seeds = [state::SEED_AUTHORITY],
        bump = authority_bump,
        payer = signer,
        space = 8 + size_of::<state::Authority>(),
    )]
    pub authority: Account<'info, state::Authority>,
    
    #[account(
        init, 
        seeds = [state::SEED_CONFIG], 
        bump = config_bump, 
        payer = signer, 
        space = 8 + size_of::<state::Config>()
    )]
    pub config: Account<'info, state::Config>,
    
    #[account(
        init, 
        seeds = [state::SEED_TREASURY], 
        bump = treasury_bump, 
        payer = signer, 
        space = 8 + size_of::<state::Treasury>()
    )]
    pub treasury: Account<'info, state::Treasury>,

    #[account(address = system_program::ID)]
    pub system_program: Program<'info, System>,
}

pub fn handler(
    ctx: Context<InitializeProgram>, 
    config_transfer_fee_distributor: u64,
    config_transfer_fee_program: u64,
    authority_bump: u8,
    config_bump: u8,
    treasury_bump: u8,
) -> ProgramResult {
    // Get accounts.
    let authority = &mut ctx.accounts.authority;
    let config = &mut ctx.accounts.config;
    let treasury = &mut ctx.accounts.treasury;

    // Initialize authority account.
    authority.bump = authority_bump;

    // Initialize config account.
    config.transfer_fee_distributor = config_transfer_fee_distributor;
    config.transfer_fee_program = config_transfer_fee_program;
    config.bump = config_bump;

    // Initialize treasury account.
    treasury.bump = treasury_bump;

    return Ok(());
}