logos-execution-zone/test_program_methods/guest/src/bin/data_changer.rs

use nssa_core::program::{AccountPostState, ProgramInput, ProgramOutput, read_nssa_inputs};

type Instruction = Vec<u8>;

/// A program that modifies the account data by setting bytes sent in instruction.
fn main() {
    let (
        ProgramInput {
            pre_states,
            instruction: data,
        },
        instruction_words,
    ) = read_nssa_inputs::<Instruction>();

    let Ok([pre]) = <[_; 1]>::try_from(pre_states) else {
        return;
    };

    let account_pre = &pre.account;
    let mut account_post = account_pre.clone();
    account_post.data = data
        .try_into()
        .expect("provided data should fit into data limit");

    ProgramOutput::new(
        instruction_words,
        vec![pre],
        vec![AccountPostState::new_claimed(account_post)],
    )
    .write();
}