feat: withdraw and call spl (#77)

Author: skosito

Anchor.toml

@@ -6,6 +6,7 @@ skip-lint = false
 
 [programs.localnet]
 connected = "4xEw862A2SEwMjofPkUyd4NEekmVJKJsdHkK3UkAtDrc"
+connected_spl = "8iUjRRhUCn8BjrvsWPfj8mguTe9L81ES4oAUApiF8JFC"
 gateway = "ZETAjseVjuFsxdRxo6MmTCvqFwb3ZHUx56Co3vCmGis"
 
 [registry]

Cargo.lock

@@ -31,6 +31,12 @@ pub mod connected {
         pda.sub_lamports(amount / 2)?;
         ctx.accounts.random_wallet.add_lamports(amount / 2)?;
 
+        // Check if the message is "revert" and return an error if so
+        if pda.last_message == "revert" {
+            msg!("Reverting transaction due to 'revert' message.");
+            return Err(ErrorCode::RevertMessage.into());
+        }
+
         msg!(
             "On call executed with amount {}, sender {:?} and message {}",
             amount,
@@ -75,4 +81,7 @@ pub struct Pda {
 pub enum ErrorCode {
     #[msg("The data provided could not be converted to a valid UTF-8 string.")]
     InvalidDataFormat,
+
+    #[msg("Revert message detected. Transaction execution halted.")]
+    RevertMessage,
 }

go-idl/generated/gateway.go

@@ -0,0 +1,23 @@
+[package]
+name = "connectedSPL"
+version = "0.1.0"
+description = "Test program used for testing withdraw and call SPL feature"
+edition = "2021"
+
+[lib]
+crate-type = ["cdylib", "lib"]
+name = "connected_spl"
+
+[features]
+default = []
+cpi = ["no-entrypoint"]
+no-entrypoint = []
+no-idl = []
+no-log-ix-name = []
+idl-build = ["anchor-lang/idl-build"]
+
+[dependencies]
+anchor-lang = { version = "=0.30.0" }
+anchor-spl = { version = "=0.30.0", features = ["idl-build"] }
+spl-associated-token-account = "3.0.2"
+solana-program = "=1.18.15"

programs/connected/src/lib.rs

@@ -0,0 +1,2 @@
+[target.bpfel-unknown-unknown.dependencies.std]
+features = []

programs/connectedSPL/Cargo.toml

@@ -0,0 +1,111 @@
+use anchor_lang::prelude::*;
+use anchor_spl::token::{transfer_checked, Mint, Token, TokenAccount};
+use std::mem::size_of;
+
+declare_id!("8iUjRRhUCn8BjrvsWPfj8mguTe9L81ES4oAUApiF8JFC");
+
+// NOTE: this is just example contract that can be called from gateway in execute_spl_token function for testing withdraw and call spl
+#[program]
+pub mod connected_spl {
+    use super::*;
+
+    pub fn initialize(ctx: Context<Initialize>) -> Result<()> {
+        Ok(())
+    }
+
+    pub fn on_call(
+        ctx: Context<OnCall>,
+        amount: u64,
+        sender: [u8; 20],
+        data: Vec<u8>,
+    ) -> Result<()> {
+        let pda = &mut ctx.accounts.pda;
+
+        // Store the sender's public key
+        pda.last_sender = sender;
+
+        // Convert data to a string and store it
+        let message = String::from_utf8(data).map_err(|_| ErrorCode::InvalidDataFormat)?;
+        pda.last_message = message;
+
+        // Transfer some portion of tokens transferred from gateway to another account
+        let token = &ctx.accounts.token_program;
+        let signer_seeds: &[&[&[u8]]] = &[&[b"connectedSPL", &[ctx.bumps.pda]]];
+
+        let xfer_ctx = CpiContext::new_with_signer(
+            token.to_account_info(),
+            anchor_spl::token::TransferChecked {
+                from: ctx.accounts.pda_ata.to_account_info(),
+                mint: ctx.accounts.mint_account.to_account_info(),
+                to: ctx.accounts.random_wallet_ata.to_account_info(),
+                authority: pda.to_account_info(),
+            },
+            signer_seeds,
+        );
+
+        transfer_checked(xfer_ctx, amount / 2, 6)?;
+
+        // Check if the message is "revert" and return an error if so
+        if pda.last_message == "revert" {
+            msg!("Reverting transaction due to 'revert' message.");
+            return Err(ErrorCode::RevertMessage.into());
+        }
+
+        msg!(
+            "On call executed with amount {}, sender {:?} and message {}",
+            amount,
+            pda.last_sender,
+            pda.last_message
+        );
+
+        Ok(())
+    }
+}
+
+#[derive(Accounts)]
+pub struct Initialize<'info> {
+    #[account(mut)]
+    pub signer: Signer<'info>,
+
+    #[account(init, payer = signer, space = size_of::<Pda>() + 32, seeds = [b"connectedSPL"], bump)]
+    pub pda: Account<'info, Pda>,
+
+    pub system_program: Program<'info, System>,
+}
+
+#[derive(Accounts)]
+pub struct OnCall<'info> {
+    #[account(mut, seeds = [b"connectedSPL"], bump)]
+    pub pda: Account<'info, Pda>,
+
+    #[account(mut)]
+    pub pda_ata: Account<'info, TokenAccount>,
+
+    pub mint_account: Account<'info, Mint>,
+
+    pub gateway_pda: UncheckedAccount<'info>,
+
+    pub random_wallet: UncheckedAccount<'info>,
+
+    #[account(mut)]
+    pub random_wallet_ata: AccountInfo<'info>,
+
+    pub token_program: Program<'info, Token>,
+
+    pub system_program: Program<'info, System>,
+}
+
+#[account]
+pub struct Pda {
+    pub last_sender: [u8; 20],
+    pub last_message: String,
+}
+
+#[error_code]
+pub enum ErrorCode {
+    #[msg("The data provided could not be converted to a valid UTF-8 string.")]
+    InvalidDataFormat,
+
+    #[msg("Revert message detected. Transaction execution halted.")]
+    RevertMessage,
+}