105 lines
3.9 KiB
Rust
105 lines
3.9 KiB
Rust
|
use crate::parse_instruction::{
|
||
|
check_num_accounts, ParsableProgram, ParseInstructionError, ParsedInstructionEnum,
|
||
|
};
|
||
|
use serde_json::json;
|
||
|
use solana_sdk::{instruction::CompiledInstruction, pubkey::Pubkey};
|
||
|
use std::str::FromStr;
|
||
|
|
||
|
// A helper function to convert spl_associated_token_account_v1_0::id() as spl_sdk::pubkey::Pubkey
|
||
|
// to solana_sdk::pubkey::Pubkey
|
||
|
pub fn spl_associated_token_id_v1_0() -> Pubkey {
|
||
|
Pubkey::from_str(&spl_associated_token_account_v1_0::id().to_string()).unwrap()
|
||
|
}
|
||
|
|
||
|
pub fn parse_associated_token(
|
||
|
instruction: &CompiledInstruction,
|
||
|
account_keys: &[Pubkey],
|
||
|
) -> Result<ParsedInstructionEnum, ParseInstructionError> {
|
||
|
match instruction.accounts.iter().max() {
|
||
|
Some(index) if (*index as usize) < account_keys.len() => {}
|
||
|
_ => {
|
||
|
// Runtime should prevent this from ever happening
|
||
|
return Err(ParseInstructionError::InstructionKeyMismatch(
|
||
|
ParsableProgram::SplAssociatedTokenAccount,
|
||
|
));
|
||
|
}
|
||
|
}
|
||
|
check_num_associated_token_accounts(&instruction.accounts, 7)?;
|
||
|
Ok(ParsedInstructionEnum {
|
||
|
instruction_type: "create".to_string(),
|
||
|
info: json!({
|
||
|
"source": account_keys[instruction.accounts[0] as usize].to_string(),
|
||
|
"account": account_keys[instruction.accounts[1] as usize].to_string(),
|
||
|
"wallet": account_keys[instruction.accounts[2] as usize].to_string(),
|
||
|
"mint": account_keys[instruction.accounts[3] as usize].to_string(),
|
||
|
"systemProgram": account_keys[instruction.accounts[4] as usize].to_string(),
|
||
|
"tokenProgram": account_keys[instruction.accounts[5] as usize].to_string(),
|
||
|
"rentSysvar": account_keys[instruction.accounts[6] as usize].to_string(),
|
||
|
}),
|
||
|
})
|
||
|
}
|
||
|
|
||
|
fn check_num_associated_token_accounts(
|
||
|
accounts: &[u8],
|
||
|
num: usize,
|
||
|
) -> Result<(), ParseInstructionError> {
|
||
|
check_num_accounts(accounts, num, ParsableProgram::SplAssociatedTokenAccount)
|
||
|
}
|
||
|
|
||
|
#[cfg(test)]
|
||
|
mod test {
|
||
|
use super::*;
|
||
|
use spl_associated_token_account_v1_0::{
|
||
|
create_associated_token_account,
|
||
|
solana_program::{
|
||
|
instruction::CompiledInstruction as SplAssociatedTokenCompiledInstruction,
|
||
|
message::Message, pubkey::Pubkey as SplAssociatedTokenPubkey,
|
||
|
},
|
||
|
};
|
||
|
|
||
|
fn convert_pubkey(pubkey: Pubkey) -> SplAssociatedTokenPubkey {
|
||
|
SplAssociatedTokenPubkey::from_str(&pubkey.to_string()).unwrap()
|
||
|
}
|
||
|
|
||
|
fn convert_compiled_instruction(
|
||
|
instruction: &SplAssociatedTokenCompiledInstruction,
|
||
|
) -> CompiledInstruction {
|
||
|
CompiledInstruction {
|
||
|
program_id_index: instruction.program_id_index,
|
||
|
accounts: instruction.accounts.clone(),
|
||
|
data: instruction.data.clone(),
|
||
|
}
|
||
|
}
|
||
|
|
||
|
#[test]
|
||
|
fn test_parse_associated_token() {
|
||
|
let mut keys: Vec<Pubkey> = vec![];
|
||
|
for _ in 0..7 {
|
||
|
keys.push(solana_sdk::pubkey::new_rand());
|
||
|
}
|
||
|
|
||
|
let create_ix = create_associated_token_account(
|
||
|
&convert_pubkey(keys[0]),
|
||
|
&convert_pubkey(keys[1]),
|
||
|
&convert_pubkey(keys[2]),
|
||
|
);
|
||
|
let message = Message::new(&[create_ix], None);
|
||
|
let compiled_instruction = convert_compiled_instruction(&message.instructions[0]);
|
||
|
assert_eq!(
|
||
|
parse_associated_token(&compiled_instruction, &keys).unwrap(),
|
||
|
ParsedInstructionEnum {
|
||
|
instruction_type: "create".to_string(),
|
||
|
info: json!({
|
||
|
"source": keys[0].to_string(),
|
||
|
"account": keys[1].to_string(),
|
||
|
"wallet": keys[2].to_string(),
|
||
|
"mint": keys[3].to_string(),
|
||
|
"systemProgram": keys[4].to_string(),
|
||
|
"tokenProgram": keys[5].to_string(),
|
||
|
"rentSysvar": keys[6].to_string(),
|
||
|
})
|
||
|
}
|
||
|
);
|
||
|
}
|
||
|
}
|