Invocation inter-programmes

Mettre à jour l'instruction de mise à jour

Tout d'abord, le programme nécessite un mécanisme simple de "paiement pour mise à jour" en modifiant la structure Update et la fonction update.

Commencez par mettre à jour le fichier lib.rs pour importer les éléments du module system_program.

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

Ensuite, mettez à jour la structure Update pour inclure un nouveau compte appelé vault_account. Ce compte, contrôlé par le programme, reçoit du SOL de la part d'un utilisateur lorsqu'il met à jour son compte de message.

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

Ensuite, ajoutez la logique CPI dans l'instruction update pour transférer 0,001 SOL du compte de l'utilisateur vers le compte coffre-fort.

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)?;

Reconstruisez le programme.

$ 
buildCopy

Mettre à jour l'instruction de suppression

Ajoutez maintenant un mécanisme de "remboursement lors de la suppression" en modifiant la structure Delete et la fonction delete.

Tout d'abord, mettez à jour la structure Delete pour inclure le vault_account. Cela permet le transfert de tout SOL dans le vault vers l'utilisateur lorsqu'ils ferment leur compte de message.

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

Ajoutez également le system_program car le CPI pour le transfert nécessite d'invoquer le System Program.

pub system_program: Program<'info, System>,

Ensuite, ajoutez la logique CPI dans l'instruction delete pour transférer le SOL du compte coffre-fort vers le compte de l'utilisateur.

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())?;

Notez que _ctx: Context<Delete> devient ctx: Context<Delete> pour utiliser le contexte dans le corps de la fonction.

Reconstruisez le programme.

$ 
buildCopy

Redéployer le programme

Après avoir effectué ces modifications, redéployez le programme mis à jour. Cela garantit que le programme modifié devient disponible pour les tests. Sur Solana, la mise à jour d'un programme nécessite simplement de déployer le programme avec le même ID de programme.

$ 
deploy
Copy
[show output]

Mettre à jour le fichier de test

Ensuite, mettez à jour le fichier anchor.test.ts pour inclure le nouveau compte vault dans les instructions. Cela nécessite de dériver le PDA du vault et de l'inclure dans les appels d'instruction de mise à jour et de suppression.

Dériver le PDA du coffre-fort

Tout d'abord, ajoutez la dérivation du PDA du coffre-fort :

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

Modifier le test de mise à jour

Ensuite, mettez à jour l'instruction de mise à jour pour inclure le vaultAccount

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

Modifier le test de suppression

Ensuite, mettez à jour l'instruction de suppression pour inclure le vaultAccount

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

Relancer le test

Après avoir effectué ces modifications, exécutez les tests pour vous assurer que tout fonctionne comme prévu :

$ 
test
Copy
[show output]

Vous pouvez ensuite examiner les liens SolanaFM pour visualiser les détails de la transaction, où vous trouverez les CPI pour les instructions de transfert au sein des instructions de mise à jour et de suppression.

CPI de mise à jour

CPI de suppression

Si vous rencontrez des erreurs, vous pouvez consulter le code final.