anchor_spl/token_2022_extensions/
transfer_hook.rs

1use anchor_lang::solana_program::account_info::AccountInfo;
2use anchor_lang::solana_program::pubkey::Pubkey;
3use anchor_lang::Result;
4use anchor_lang::{context::CpiContext, Accounts};
5
6pub fn transfer_hook_initialize<'info>(
7    ctx: CpiContext<'_, '_, '_, 'info, TransferHookInitialize<'info>>,
8    authority: Option<Pubkey>,
9    transfer_hook_program_id: Option<Pubkey>,
10) -> Result<()> {
11    let ix = spl_token_2022::extension::transfer_hook::instruction::initialize(
12        ctx.accounts.token_program_id.key,
13        ctx.accounts.mint.key,
14        authority,
15        transfer_hook_program_id,
16    )?;
17    anchor_lang::solana_program::program::invoke_signed(
18        &ix,
19        &[ctx.accounts.token_program_id, ctx.accounts.mint],
20        ctx.signer_seeds,
21    )
22    .map_err(Into::into)
23}
24
25#[derive(Accounts)]
26pub struct TransferHookInitialize<'info> {
27    pub token_program_id: AccountInfo<'info>,
28    pub mint: AccountInfo<'info>,
29}
30
31pub fn transfer_hook_update<'info>(
32    ctx: CpiContext<'_, '_, '_, 'info, TransferHookUpdate<'info>>,
33    transfer_hook_program_id: Option<Pubkey>,
34) -> Result<()> {
35    let ix = spl_token_2022::extension::transfer_hook::instruction::update(
36        ctx.accounts.token_program_id.key,
37        ctx.accounts.mint.key,
38        ctx.accounts.authority.key,
39        &[],
40        transfer_hook_program_id,
41    )?;
42    anchor_lang::solana_program::program::invoke_signed(
43        &ix,
44        &[
45            ctx.accounts.token_program_id,
46            ctx.accounts.mint,
47            ctx.accounts.authority,
48        ],
49        ctx.signer_seeds,
50    )
51    .map_err(Into::into)
52}
53
54#[derive(Accounts)]
55pub struct TransferHookUpdate<'info> {
56    pub token_program_id: AccountInfo<'info>,
57    pub mint: AccountInfo<'info>,
58    pub authority: AccountInfo<'info>,
59}