triad_protocol/instructions/stake/
claim_stake.rs

1use crate::StakeV2;
2use crate::{ constraints::{ is_authority_for_stake, is_ttriad_mint }, state::StakeVault };
3use anchor_lang::prelude::*;
4use anchor_spl::token_2022::{ transfer_checked, Token2022, TransferChecked };
5use anchor_spl::{ associated_token::AssociatedToken, token_interface::{ Mint, TokenAccount } };
6
7#[derive(Accounts)]
8pub struct ClaimStake<'info> {
9    #[account(mut)]
10    pub signer: Signer<'info>,
11
12    #[account(mut)]
13    pub stake_vault: Box<Account<'info, StakeVault>>,
14
15    #[account(mut, constraint = is_authority_for_stake(&stake, &signer)?)]
16    pub stake: Box<Account<'info, StakeV2>>,
17
18    #[account(mut, constraint = is_ttriad_mint(&mint.key())?)]
19    pub mint: Box<InterfaceAccount<'info, Mint>>,
20
21    #[account(mut)]
22    pub from_ata: Box<InterfaceAccount<'info, TokenAccount>>,
23
24    #[account(
25        init_if_needed,
26        payer = signer,
27        constraint = to_ata.owner == *signer.key && to_ata.mint == mint.key(),
28        associated_token::mint = mint,
29        associated_token::authority = signer
30    )]
31    pub to_ata: Box<InterfaceAccount<'info, TokenAccount>>,
32
33    pub token_program: Program<'info, Token2022>,
34    pub associated_token_program: Program<'info, AssociatedToken>,
35    pub system_program: Program<'info, System>,
36}
37
38pub fn claim_stake(ctx: Context<ClaimStake>) -> Result<()> {
39    let stake_vault = &mut ctx.accounts.stake_vault;
40    let stake = &mut ctx.accounts.stake;
41
42    let signer: &[&[&[u8]]] = &[
43        &[b"stake_vault", stake_vault.name.as_bytes(), &[stake_vault.bump]],
44    ];
45
46    let cpi_context = CpiContext::new_with_signer(
47        ctx.accounts.token_program.to_account_info(),
48        TransferChecked {
49            from: ctx.accounts.from_ata.to_account_info(),
50            mint: ctx.accounts.mint.to_account_info(),
51            to: ctx.accounts.to_ata.to_account_info(),
52            authority: stake_vault.to_account_info(),
53        },
54        signer
55    );
56
57    let rewards = stake.available;
58
59    transfer_checked(cpi_context, rewards, ctx.accounts.mint.decimals)?;
60
61    stake_vault.amount -= rewards;
62    stake_vault.amount_paid += rewards;
63
64    stake.claimed += rewards;
65    stake.available = 0;
66
67    Ok(())
68}
triad_protocol/instructions/stake/claim_stake.rs

triad_protocol/instructions/stake/
claim_stake.rs
