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