lssa/test_program_methods/guest/src/bin/simple_balance_transfer.rs
2026-04-03 00:58:11 +02:00

40 lines
1.1 KiB
Rust

use nssa_core::program::{AccountPostState, ProgramInput, ProgramOutput, read_nssa_inputs};
type Instruction = u128;
fn main() {
let (
ProgramInput { self_program_id, caller_program_id: _,
pre_states,
instruction: balance,
},
instruction_words,
) = read_nssa_inputs::<Instruction>();
let Ok([sender_pre, receiver_pre]) = <[_; 2]>::try_from(pre_states) else {
return;
};
let mut sender_post = sender_pre.account.clone();
let mut receiver_post = receiver_pre.account.clone();
sender_post.balance = sender_post
.balance
.checked_sub(balance)
.expect("Not enough balance to transfer");
receiver_post.balance = receiver_post
.balance
.checked_add(balance)
.expect("Overflow when adding balance");
ProgramOutput::new(
self_program_id,
instruction_words,
vec![sender_pre, receiver_pre],
vec![
AccountPostState::new(sender_post),
AccountPostState::new(receiver_post),
],
)
.write();
}