anchor_spl/token_2022_extensions/
memo_transfer.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 memo_transfer_initialize<'info>(
7    ctx: CpiContext<'_, '_, '_, 'info, MemoTransfer<'info>>,
8) -> Result<()> {
9    let ix = spl_token_2022::extension::memo_transfer::instruction::enable_required_transfer_memos(
10        ctx.accounts.token_program_id.key,
11        ctx.accounts.account.key,
12        ctx.accounts.owner.key,
13        &[],
14    )?;
15    anchor_lang::solana_program::program::invoke_signed(
16        &ix,
17        &[
18            ctx.accounts.token_program_id,
19            ctx.accounts.account,
20            ctx.accounts.owner,
21        ],
22        ctx.signer_seeds,
23    )
24    .map_err(Into::into)
25}
26
27pub fn memo_transfer_disable<'info>(
28    ctx: CpiContext<'_, '_, '_, 'info, MemoTransfer<'info>>,
29) -> Result<()> {
30    let ix =
31        spl_token_2022::extension::memo_transfer::instruction::disable_required_transfer_memos(
32            ctx.accounts.token_program_id.key,
33            ctx.accounts.account.key,
34            ctx.accounts.owner.key,
35            &[],
36        )?;
37    anchor_lang::solana_program::program::invoke_signed(
38        &ix,
39        &[
40            ctx.accounts.token_program_id,
41            ctx.accounts.account,
42            ctx.accounts.owner,
43        ],
44        ctx.signer_seeds,
45    )
46    .map_err(Into::into)
47}
48
49#[derive(Accounts)]
50pub struct MemoTransfer<'info> {
51    pub token_program_id: AccountInfo<'info>,
52    pub account: AccountInfo<'info>,
53    pub owner: AccountInfo<'info>,
54}