triad_protocol/instructions/stake/
deposit_stake_rewards.rs

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

triad_protocol/instructions/stake/
deposit_stake_rewards.rs
