Cross Program Invocation

Обновление инструкции Update

Сначала программе нужен простой механизм "pay-to-update" — для этого измените структуру Update и функцию update.

Начните с обновления файла lib.rs, чтобы подключить элементы из модуля system_program.

use anchor_lang::system_program::{transfer, Transfer};

Далее обновите структуру Update, чтобы добавить новый аккаунт с именем vault_account. Этот аккаунт, управляемый программой, получает SOL от пользователя при обновлении его аккаунта сообщения.

#[account(
    mut,
    seeds = [b"vault", user.key().as_ref()],
    bump,
)]
pub vault_account: SystemAccount<'info>,

Далее добавьте логику CPI в инструкцию update для перевода 0,001 SOL с аккаунта пользователя на аккаунт хранилища.

let transfer_accounts = Transfer {
    from: ctx.accounts.user.to_account_info(),
    to: ctx.accounts.vault_account.to_account_info(),
};

let cpi_context = CpiContext::new(
    ctx.accounts.system_program.to_account_info(),
    transfer_accounts,
);

transfer(cpi_context, 1_000_000)?;

Перестройте программу.

$ 
buildCopy

Обновите инструкцию удаления

Теперь добавьте механизм "возврата при удалении", изменив структуру Delete и функцию delete.

Сначала обновите структуру Delete и добавьте vault_account. Это позволит переводить все SOL из хранилища обратно пользователю при закрытии его аккаунта сообщений.

#[account(
    mut,
    seeds = [b"vault", user.key().as_ref()],
    bump,
)]
pub vault_account: SystemAccount<'info>,

Также добавьте system_program, так как для CPI-перевода требуется вызов System Program.

pub system_program: Program<'info, System>,

Далее добавьте логику CPI в инструкцию delete для перевода SOL с аккаунта хранилища обратно на аккаунт пользователя.

let user_key = ctx.accounts.user.key();
let signer_seeds: &[&[&[u8]]] =
    &[&[b"vault", user_key.as_ref(), &[ctx.bumps.vault_account]]];

let transfer_accounts = Transfer {
    from: ctx.accounts.vault_account.to_account_info(),
    to: ctx.accounts.user.to_account_info(),
};
let cpi_context = CpiContext::new(
    ctx.accounts.system_program.to_account_info(),
    transfer_accounts,
).with_signer(signer_seeds);
transfer(cpi_context, ctx.accounts.vault_account.lamports())?;

Обратите внимание, что _ctx: Context<Delete> меняется на ctx: Context<Delete> для использования контекста в теле функции.

Перестройте программу.

$ 
buildCopy

Повторное развертывание программы

После внесения этих изменений повторно разверните обновленную программу. Это гарантирует, что модифицированная программа станет доступной для тестирования. В Solana обновление программы требует просто развертывания программы с тем же идентификатором программы.

$ 
deploy
Copy
[show output]

Обновление тестового файла

Далее обновите файл anchor.test.ts, чтобы добавить новый аккаунт хранилища в инструкции. Для этого нужно получить vault PDA и включить его в вызовы инструкций update и delete.

Получение PDA хранилища

Сначала добавьте получение PDA хранилища:

const [vaultPda, vaultBump] = PublicKey.findProgramAddressSync(
  [Buffer.from("vault"), wallet.publicKey.toBuffer()],
  program.programId
);

Изменение теста обновления

Затем обновите инструкцию update, чтобы включить vaultAccount

const transactionSignature = await program.methods
  .update(message)
  .accounts({
    messageAccount: messagePda,
    vaultAccount: vaultPda
  })
  .rpc({ commitment: "confirmed" });

Изменение теста удаления

Затем обновите инструкцию delete, чтобы включить vaultAccount

const transactionSignature = await program.methods
  .delete()
  .accounts({
    messageAccount: messagePda,
    vaultAccount: vaultPda
  })
  .rpc({ commitment: "confirmed" });

Повторный запуск теста

После внесения этих изменений запустите тесты, чтобы убедиться, что всё работает как ожидается:

$ 
test
Copy
[show output]

Затем вы можете проверить ссылки SolanaFM, чтобы просмотреть детали транзакции, где вы найдете CPI для инструкций передачи в рамках инструкций обновления и удаления.

Обновление CPI

Удаление CPI

Если вы столкнетесь с какими-либо ошибками, вы можете обратиться к финальному коду.