nitro-da-client 0.1.7

use anchor_lang::{prelude::Pubkey, InstructionData, ToAccountMetas};
use nitro_da_blober::instruction::InsertChunk;
use solana_sdk::instruction::Instruction;

use crate::tx::{MessageArguments, MessageBuilder};

impl MessageBuilder for InsertChunk {
    type Input = (Self, Pubkey);
    const COMPUTE_UNIT_LIMIT: u32 = 6_500;

    fn mutable_accounts(args: &MessageArguments<Self::Input>) -> Vec<Pubkey> {
        vec![args.input.1, args.payer]
    }

    fn generate_instructions(args: &MessageArguments<Self::Input>) -> Vec<Instruction> {
        let accounts = nitro_da_blober::accounts::InsertChunk {
            blob: args.input.1,
            blober: args.blober,
            payer: args.payer,
        };

        let data = Self {
            data: args.input.0.data.clone(),
            idx: args.input.0.idx,
        };

        vec![Instruction {
            program_id: args.program_id,
            accounts: accounts.to_account_metas(None),
            data: data.data(),
        }]
    }

    #[cfg(test)]
    fn generate_arbitrary_input(
        u: &mut arbitrary::Unstructured,
        payer: Pubkey,
        blober: Pubkey,
    ) -> arbitrary::Result<Self::Input> {
        let timestamp: u64 = u.arbitrary()?;
        let idx: u16 = u.arbitrary()?;
        let data: Vec<u8> = u.arbitrary()?;
        let blob = nitro_da_blober::find_blob_address(
            nitro_da_blober::id(),
            payer,
            blober,
            timestamp,
            data.len(),
        );

        Ok((InsertChunk { data, idx }, blob))
    }
}

#[cfg(test)]
mod tests {
    use nitro_da_blober::instruction::InsertChunk;

    use crate::tx::MessageBuilder;

    #[test]
    #[ignore]
    fn test_compute_unit_limit() {
        InsertChunk::test_compute_unit_limit();
    }
}