clawback_now.

Author: RustNinja

programs/merkle-distributor/src/instructions/clawback_now.rs

@@ -0,0 +1,73 @@
+// Instruction to clawback funds once they have expired
+
+use anchor_lang::{context::Context, prelude::*, Accounts, Key, Result};
+use anchor_spl::{
+    token,
+    token::{Token, TokenAccount},
+};
+
+use crate::{error::ErrorCode, state::merkle_distributor::MerkleDistributor};
+
+/// [merkle_distributor::clawback] accounts.
+#[derive(Accounts)]
+pub struct ClawbackNow<'info> {
+    /// The [MerkleDistributor].
+    #[account(mut)]
+    pub distributor: Account<'info, MerkleDistributor>,
+
+    /// Distributor ATA containing the tokens to distribute.
+    #[account(
+        mut,
+        associated_token::mint = distributor.mint,
+        associated_token::authority = distributor.key(),
+        address = distributor.token_vault
+    )]
+    pub from: Account<'info, TokenAccount>,
+
+    /// The Clawback token account.
+    #[account(mut, address = distributor.clawback_receiver)]
+    pub to: Account<'info, TokenAccount>,
+
+    /// Claimant account
+    /// Anyone can claw back the funds
+    pub claimant: Signer<'info>,
+
+    /// The [System] program.
+    pub system_program: Program<'info, System>,
+
+    /// SPL [Token] program.
+    pub token_program: Program<'info, Token>,
+}
+
+/// Claws back unclaimed tokens by:
+/// 1. Checking that the lockup has expired
+/// 2. Transferring remaining funds from the vault to the clawback receiver
+/// 3. Marking the distributor as clawed back
+/// CHECK:
+///     1. The distributor has not already been clawed back
+#[allow(clippy::result_large_err)]
+pub fn handle_clawback_now(ctx: Context<ClawbackNow>, amount: u64) -> Result<()> {
+    let distributor = &ctx.accounts.distributor;
+
+    let seeds = [
+        b"MerkleDistributor".as_ref(),
+        &distributor.mint.to_bytes(),
+        &distributor.version.to_le_bytes(),
+        &[ctx.accounts.distributor.bump],
+    ];
+
+    token::transfer(
+        CpiContext::new(
+            ctx.accounts.token_program.to_account_info(),
+            token::Transfer {
+                from: ctx.accounts.from.to_account_info(),
+                to: ctx.accounts.to.to_account_info(),
+                authority: ctx.accounts.distributor.to_account_info(),
+            },
+        )
+        .with_signer(&[&seeds[..]]),
+        amount,
+    )?;
+
+    Ok(())
+}

programs/merkle-distributor/src/instructions/mod.rs

@@ -8,6 +8,7 @@ pub use remove_new_claim::*;
 pub use admin_new_claim::*;
 pub use claim_new_claim::*;
 pub use remove_claim::*;
+pub use clawback_now::*;
 pub mod claim_locked;
 pub mod clawback;
 pub mod new_claim;
@@ -16,6 +17,7 @@ pub mod admin_new_claim;
 pub mod claim_new_claim;
 pub mod remove_new_claim;
 pub mod remove_claim;
+pub mod clawback_now;
 
 pub mod set_admin;
 pub mod set_clawback_receiver;