Створення
Інструкція make виконує три завдання:
Ініціалізує запис Escrow та зберігає всі умови угоди.
Створює Сховище (ATA для
mint_a, що належитьescrow).Переміщує токени A творця до цього сховища за допомогою CPI до програми SPL-Token.
Required Accounts
Нижче наведено облікові записи, які потрібні для контексту:
maker: творець escrow. Має бути підписантом і змінюваним
escrow: обліковий запис escrow, який ми ініціалізуємо. Має бути змінюваним
mint_a: токен, який ми вносимо в escrow
mint_b: токен, який ми хочемо отримати
maker_ata_a: пов'язаний токен-акаунт, що належить творцю. Має бути змінюваним
vault: пов'язаний токен-акаунт, що належить escrow. Має бути змінюваним
system_program: системна програма. Має бути виконуваною
token_program: токен-програма. Має бути виконуваною
Примітка: Ми будемо використовувати типи, представлені у Вступі до Pinocchio.
У коді це виглядає так:
pub struct MakeAccounts<'a> {
pub maker: &'a AccountInfo,
pub escrow: &'a AccountInfo,
pub mint_a: &'a AccountInfo,
pub mint_b: &'a AccountInfo,
pub maker_ata_a: &'a AccountInfo,
pub vault: &'a AccountInfo,
pub system_program: &'a AccountInfo,
pub token_program: &'a AccountInfo,
}
impl<'a> TryFrom<&'a [AccountInfo]> for MakeAccounts<'a> {
type Error = ProgramError;
fn try_from(accounts: &'a [AccountInfo]) -> Result<Self, Self::Error> {
let [maker, escrow, mint_a, mint_b, maker_ata_a, vault, system_program, token_program, _] = accounts else {
return Err(ProgramError::NotEnoughAccountKeys);
};
// Basic Accounts Checks
SignerAccount::check(maker)?;
MintInterface::check(mint_a)?;
MintInterface::check(mint_b)?;
AssociatedTokenAccount::check(maker_ata_a, maker, mint_a, token_program)?;
// Return the accounts
Ok(Self {
maker,
escrow,
mint_a,
mint_b,
maker_ata_a,
vault,
system_program,
token_program,
})
}
}Instruction Data
Ось дані інструкції, які нам потрібно передати:
seed: випадкове число, що використовується під час виведення сіда. Має бути u64
receive: кількість, яку творець хоче отримати. Має бути u64
amount: кількість, яку творець хоче внести. Має бути u64
Ми перевіримо, щоб amount не був нульовим, оскільки це не мало б сенсу для escrow.
Ось як це виглядає в коді:
pub struct MakeInstructionData {
pub seed: u64,
pub receive: u64,
pub amount: u64,
}
impl<'a> TryFrom<&'a [u8]> for MakeInstructionData {
type Error = ProgramError;
fn try_from(data: &'a [u8]) -> Result<Self, Self::Error> {
if data.len() != size_of::<u64>() * 3 {
return Err(ProgramError::InvalidInstructionData);
}
let seed = u64::from_le_bytes(data[0..8].try_into().unwrap());
let receive = u64::from_le_bytes(data[8..16].try_into().unwrap());
let amount = u64::from_le_bytes(data[16..24].try_into().unwrap());
// Instruction Checks
if amount == 0 {
return Err(ProgramError::InvalidInstructionData);
}
Ok(Self {
seed,
receive,
amount,
})
}
}Instruction Logic
Ми починаємо з ініціалізації необхідних облікових записів у реалізації TryFrom, після того як ми десеріалізували як instruction_data, так і accounts.
Для цього кроку ми створюємо обліковий запис Escrow за допомогою трейту ProgramAccount::init::<Escrow> з допоміжних функцій, представлених у Вступі до Pinocchio. Аналогічно ми ініціалізуємо обліковий запис Vault, оскільки він має бути створений з нуля:
pub struct Make<'a> {
pub accounts: MakeAccounts<'a>,
pub instruction_data: MakeInstructionData,
pub bump: u8,
}
impl<'a> TryFrom<(&'a [u8], &'a [AccountInfo])> for Make<'a> {
type Error = ProgramError;
fn try_from((data, accounts): (&'a [u8], &'a [AccountInfo])) -> Result<Self, Self::Error> {
let accounts = MakeAccounts::try_from(accounts)?;
let instruction_data = MakeInstructionData::try_from(data)?;
// Initialize the Accounts needed
let (_, bump) = find_program_address(&[b"escrow", accounts.maker.key(), &instruction_data.seed.to_le_bytes()], &crate::ID);
let seed_binding = instruction_data.seed.to_le_bytes();
let bump_binding = [bump];
let escrow_seeds = [
Seed::from(b"escrow"),
Seed::from(accounts.maker.key().as_ref()),
Seed::from(&seed_binding),
Seed::from(&bump_binding),
];
ProgramAccount::init::<Escrow>(
accounts.maker,
accounts.escrow,
&escrow_seeds,
Escrow::LEN
)?;
// Initialize the vault
AssociatedTokenAccount::init(
accounts.vault,
accounts.mint_a,
accounts.maker,
accounts.escrow,
accounts.system_program,
accounts.token_program,
)?;
Ok(Self {
accounts,
instruction_data,
bump,
})
}
}Тепер ми можемо зосередитися на самій логіці, яка полягатиме у заповненні облікового запису escrow, а потім переказі токенів до сховища.
impl<'a> Make<'a> {
pub const DISCRIMINATOR: &'a u8 = &0;
pub fn process(&mut self) -> ProgramResult {
// Populate the escrow account
let mut data = self.accounts.escrow.try_borrow_mut_data()?;
let escrow = Escrow::load_mut(data.as_mut())?;
escrow.set_inner(
self.instruction_data.seed,
*self.accounts.maker.key(),
*self.accounts.mint_a.key(),
*self.accounts.mint_b.key(),
self.instruction_data.receive,
[self.bump],
);
// Transfer tokens to vault
Transfer {
from: self.accounts.maker_ata_a,
to: self.accounts.vault,
authority: self.accounts.maker,
amount: self.instruction_data.amount
}.invoke()?;
Ok(())
}
}