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

use crate::*;
use vipers::{assert_keys_eq, Validate};

#[derive(Accounts)]
pub struct CreateUserInfo<'info> {
    /// This is a vault info account.
    #[account(
        init,
        seeds = [
            b"raydium_vault_user_info".as_ref(),
            vault.to_account_info().key.as_ref(),
            user_account.key.as_ref(),
        ],
        bump,
        payer = payer
    )]
    pub user_info_account: Box<Account<'info, UserInfo>>,

    pub vault: Box<Account<'info, Vault>>,

    /// Base of the [User] address.
    pub user_account: Signer<'info>,

    /// CHECK: Raydium will check it.
    pub farm_program: UncheckedAccount<'info>,

    /// Fee payer.
    #[account(mut)]
    pub payer: Signer<'info>,

    /// [System] program.
    pub system_program: Program<'info, System>,
}

/// Creates a new [UserInfo].
pub fn handler(ctx: Context<CreateUserInfo>, user_account: Pubkey) -> Result<()> {
    // Creating the vault info account.
    let user_info_account = &mut ctx.accounts.user_info_account;
    user_info_account.vault = ctx.accounts.vault.key();
    user_info_account.bump = *ctx.bumps.get("user_info_account").unwrap();

    user_info_account.user_account = user_account;

    // Setting vault authority
    user_info_account.manager = ctx.accounts.vault.manager;

    user_info_account.last_deposit_at = 0;
    user_info_account.last_withdraw_at = 0;
    user_info_account.token_a_added = 0;
    user_info_account.token_a_removed = 0;
    user_info_account.token_b_added = 0;
    user_info_account.token_b_removed = 0;
    user_info_account.lp_tokens_debt = 0;

    Ok(())
}

impl<'info> Validate<'info> for CreateUserInfo<'info> {
    fn validate(&self) -> Result<()> {
        assert_keys_eq!(self.farm_program, self.vault.farm_program_id);
        Ok(())
    }
}