spl_token_2022/extension/memo_transfer/
mod.rs

1#[cfg(feature = "serde-traits")]
2use serde::{Deserialize, Serialize};
3use {
4    crate::{
5        error::TokenError,
6        extension::{BaseState, BaseStateWithExtensions, Extension, ExtensionType},
7    },
8    bytemuck::{Pod, Zeroable},
9    solana_program::{
10        instruction::get_processed_sibling_instruction, program_error::ProgramError, pubkey::Pubkey,
11    },
12    spl_pod::primitives::PodBool,
13};
14
15/// Memo Transfer extension instructions
16pub mod instruction;
17
18/// Memo Transfer extension processor
19pub mod processor;
20
21/// Memo Transfer extension for Accounts
22#[repr(C)]
23#[cfg_attr(feature = "serde-traits", derive(Serialize, Deserialize))]
24#[cfg_attr(feature = "serde-traits", serde(rename_all = "camelCase"))]
25#[derive(Clone, Copy, Debug, Default, PartialEq, Pod, Zeroable)]
26pub struct MemoTransfer {
27    /// Require transfers into this account to be accompanied by a memo
28    pub require_incoming_transfer_memos: PodBool,
29}
30impl Extension for MemoTransfer {
31    const TYPE: ExtensionType = ExtensionType::MemoTransfer;
32}
33
34/// Determine if a memo is required for transfers into this account
35pub fn memo_required<BSE: BaseStateWithExtensions<S>, S: BaseState>(account_state: &BSE) -> bool {
36    if let Ok(extension) = account_state.get_extension::<MemoTransfer>() {
37        return extension.require_incoming_transfer_memos.into();
38    }
39    false
40}
41
42/// Check if the previous sibling instruction is a memo
43pub fn check_previous_sibling_instruction_is_memo() -> Result<(), ProgramError> {
44    let is_memo_program = |program_id: &Pubkey| -> bool {
45        program_id == &spl_memo::id() || program_id == &spl_memo::v1::id()
46    };
47    let previous_instruction = get_processed_sibling_instruction(0);
48    match previous_instruction {
49        Some(instruction) if is_memo_program(&instruction.program_id) => {}
50        _ => {
51            return Err(TokenError::NoMemo.into());
52        }
53    }
54    Ok(())
55}
spl_token_2022/extension/memo_transfer/mod.rs

spl_token_2022/extension/memo_transfer/
mod.rs
