hpsvm 0.1.20

use solana_account::{Account, AccountSharedData};
use solana_address::Address;
use solana_instruction::{Instruction, account_meta::AccountMeta};
use solana_message::inner_instruction::InnerInstructionsList;
use solana_program_error::ProgramError;
use solana_signature::Signature;
use solana_transaction::InstructionError;
use solana_transaction_context::transaction::TransactionReturnData;
use solana_transaction_error::{TransactionError, TransactionResult as Result};

use crate::{error::HPSVMError, format_logs::format_logs};

/// Transaction metadata captured during execution.
#[derive(Debug, Default, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct TransactionMetadata {
    /// The transaction signature.
    pub signature: Signature,
    /// Transaction log messages.
    pub logs: Vec<String>,
    /// Inner (CPI) instructions executed during the transaction.
    pub inner_instructions: InnerInstructionsList,
    /// Compute units consumed by the transaction.
    pub compute_units_consumed: u64,
    /// Data returned by programs during execution.
    pub return_data: TransactionReturnData,
    /// Transaction fee in lamports.
    pub fee: u64,
    /// Execution diagnostics including balance diffs, token balances, and trace.
    #[cfg_attr(feature = "serde", serde(default))]
    pub diagnostics: ExecutionDiagnostics,
}

impl TransactionMetadata {
    /// Returns the transaction logs formatted with ANSI color codes.
    pub fn pretty_logs(&self) -> String {
        format_logs(&self.logs)
    }
}

/// Structured execution details captured alongside transaction metadata.
#[derive(Debug, Default, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[non_exhaustive]
pub struct ExecutionDiagnostics {
    /// Account lamport balances before execution.
    pub pre_balances: Vec<u64>,
    /// Account lamport balances after execution.
    pub post_balances: Vec<u64>,
    /// Per-account state diffs for writable accounts.
    pub account_diffs: Vec<AccountDiff>,
    /// Account source failures observed during execution preparation.
    pub account_source_failures: Vec<AccountSourceFailure>,
    /// SPL token balances before execution.
    pub pre_token_balances: Vec<TokenBalance>,
    /// SPL token balances after execution.
    pub post_token_balances: Vec<TokenBalance>,
    /// Execution trace of CPI instruction frames.
    pub execution_trace: ExecutionTrace,
}

/// External account source failures observed while preparing execution.
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[non_exhaustive]
pub struct AccountSourceFailure {
    /// The account address that failed.
    pub pubkey: Address,
    /// Description of the failure.
    pub error: String,
}

/// Pre/post account state for a writable account touched by execution.
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[non_exhaustive]
pub struct AccountDiff {
    /// The account address.
    pub address: Address,
    /// Account state before execution, if the account existed.
    pub pre: Option<Account>,
    /// Account state after execution.
    pub post: Option<Account>,
}

/// SPL token balance metadata for a token account present in execution diagnostics.
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[non_exhaustive]
pub struct TokenBalance {
    /// Index of the token account in the transaction.
    pub account_index: usize,
    /// Address of the token account.
    pub address: Address,
    /// Address of the token mint.
    pub mint: Address,
    /// Address of the token account owner.
    pub owner: Address,
    /// Token balance amount.
    pub amount: u64,
    /// Token decimals, if known.
    pub decimals: Option<u8>,
}

/// Instruction trace frames captured directly from the Solana transaction context.
#[derive(Debug, Default, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[non_exhaustive]
pub struct ExecutionTrace {
    /// The executed instructions in order.
    pub instructions: Vec<ExecutedInstruction>,
}

/// One executed top-level or CPI instruction.
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
#[non_exhaustive]
pub struct ExecutedInstruction {
    /// Call stack depth (0 for top-level instructions).
    pub stack_height: u8,
    /// Address of the program that was invoked.
    pub program_id: Address,
    /// Account metas passed to the instruction.
    pub accounts: Vec<AccountMeta>,
    /// Raw instruction data.
    pub data: Vec<u8>,
}

impl ExecutedInstruction {
    /// Returns this trace frame as a normal Solana instruction.
    #[must_use]
    pub fn instruction(&self) -> Instruction {
        Instruction {
            program_id: self.program_id,
            accounts: self.accounts.clone(),
            data: self.data.clone(),
        }
    }
}

/// Simulated transaction information including metadata and post-execution accounts.
#[derive(Debug, Default, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct SimulatedTransactionInfo {
    /// Transaction metadata.
    pub meta: TransactionMetadata,
    /// Post-execution writable account snapshot.
    pub post_accounts: Vec<(Address, AccountSharedData)>,
}

/// Metadata for a failed transaction including the error and logs.
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(serde::Serialize, serde::Deserialize))]
pub struct FailedTransactionMetadata {
    /// The transaction error.
    pub err: TransactionError,
    /// Transaction metadata captured before failure.
    pub meta: TransactionMetadata,
}

impl From<ProgramError> for FailedTransactionMetadata {
    fn from(value: ProgramError) -> Self {
        Self {
            err: TransactionError::InstructionError(
                0,
                InstructionError::Custom(u64::from(value) as u32),
            ),
            meta: Default::default(),
        }
    }
}

/// A result type that either holds successful transaction metadata or failure metadata.
pub type TransactionResult = std::result::Result<TransactionMetadata, FailedTransactionMetadata>;

#[derive(Debug)]
pub(crate) struct ExecutionResult {
    pub(crate) post_accounts: Vec<(Address, AccountSharedData)>,
    pub(crate) tx_result: Result<()>,
    pub(crate) signature: Signature,
    pub(crate) compute_units_consumed: u64,
    pub(crate) inner_instructions: InnerInstructionsList,
    pub(crate) return_data: TransactionReturnData,
    pub(crate) execution_trace: ExecutionTrace,
    /// Whether the transaction can be included in a block
    pub(crate) included: bool,
    pub(crate) fee: u64,
    pub(crate) fee_payer: Option<Address>,
    pub(crate) account_source_failures: Vec<AccountSourceFailure>,
    pub(crate) fatal_error: Option<HPSVMError>,
    /// Whether the installed inspector accepted this execution, so staged
    /// commit paths know without consulting it a second time.
    ///
    /// A plain `bool`, not the `InspectMode` const the execution ran under:
    /// the const is gone once execution returns, and re-consulting at commit
    /// would be wrong (the inspector could have changed between `transact`
    /// and `commit`). Set from `MODE.is_live()` at the mode-selected entry.
    pub(crate) inspector_live: bool,
}

impl Default for ExecutionResult {
    fn default() -> Self {
        Self {
            post_accounts: Default::default(),
            tx_result: Err(TransactionError::UnsupportedVersion),
            signature: Default::default(),
            compute_units_consumed: Default::default(),
            inner_instructions: Default::default(),
            return_data: Default::default(),
            execution_trace: Default::default(),
            included: false,
            fee: 0,
            fee_payer: None,
            account_source_failures: Default::default(),
            fatal_error: None,
            inspector_live: false,
        }
    }
}

#[cfg(test)]
mod tests {
    use proptest::prelude::*;
    use solana_instruction::{Instruction, account_meta::AccountMeta};

    use super::*;

    fn trace_frame(stack_height: u8, program_id: Address) -> ExecutedInstruction {
        ExecutedInstruction {
            stack_height,
            program_id,
            accounts: vec![AccountMeta::new_readonly(program_id, false)],
            data: vec![1, 2, 3],
        }
    }

    #[test]
    fn pretty_logs_matches_format_logs() {
        let meta = TransactionMetadata {
            logs: vec![String::from("Program log: hello"), String::new()],
            ..Default::default()
        };
        assert_eq!(meta.pretty_logs(), format_logs(&meta.logs));
        assert!(meta.pretty_logs().contains("hello"));
    }

    #[test]
    fn pretty_logs_of_metadata_without_logs_is_empty() {
        assert_eq!(TransactionMetadata::default().pretty_logs(), "");
    }

    #[test]
    fn executed_instruction_round_trips_back_into_an_instruction() {
        let program_id = Address::new_unique();
        let other = Address::new_unique();
        let frame = ExecutedInstruction {
            stack_height: 2,
            program_id,
            accounts: vec![
                AccountMeta::new(program_id, true),
                AccountMeta::new_readonly(other, false),
            ],
            data: vec![9, 8, 7],
        };

        let instruction = frame.instruction();

        assert_eq!(instruction.program_id, program_id);
        assert_eq!(instruction.accounts, frame.accounts);
        assert_eq!(instruction.data, frame.data);
        assert_eq!(instruction, frame.instruction());
    }

    #[test]
    fn executed_instruction_of_an_empty_trace_frame_is_empty() {
        let frame = ExecutedInstruction {
            stack_height: 0,
            program_id: Address::default(),
            accounts: Vec::new(),
            data: Vec::new(),
        };
        assert_eq!(
            frame.instruction(),
            Instruction { program_id: Address::default(), accounts: Vec::new(), data: Vec::new() }
        );
    }

    /// `ProgramError` carries a u64 but the wire format is a `u32`, so the
    /// conversion truncates. Every representable custom code must survive
    /// intact, which is what the runtime relies on.
    #[test]
    fn program_error_conversion_wraps_a_custom_instruction_error() {
        let metadata = FailedTransactionMetadata::from(ProgramError::Custom(4242));
        match metadata.err {
            TransactionError::InstructionError(0, InstructionError::Custom(code)) => {
                assert_eq!(code, 4242);
            }
            other => panic!("expected InstructionError(0, Custom(4242)), got {other:?}"),
        }
        assert_eq!(metadata.meta, TransactionMetadata::default());
    }

    #[test]
    fn program_error_conversion_preserves_the_widest_custom_code() {
        // `ProgramError::Custom` holds a `u32`, so the widest representable code
        // is the boundary that must survive the conversion intact.
        let widest = u32::MAX;
        let metadata = FailedTransactionMetadata::from(ProgramError::Custom(widest));
        match metadata.err {
            TransactionError::InstructionError(0, InstructionError::Custom(code)) => {
                assert_eq!(code, widest);
            }
            other => panic!("expected InstructionError(0, Custom(_)), got {other:?}"),
        }
    }

    #[test]
    fn execution_result_default_is_a_non_included_unsupported_version_failure() {
        let result = ExecutionResult::default();
        assert!(matches!(result.tx_result, Err(TransactionError::UnsupportedVersion)));
        assert!(!result.included);
        assert_eq!(result.fee, 0);
        assert_eq!(result.compute_units_consumed, 0);
        assert_eq!(result.signature, Signature::default());
        assert_eq!(result.fee_payer, None);
        assert!(result.post_accounts.is_empty());
        assert!(result.account_source_failures.is_empty());
        assert!(result.fatal_error.is_none());
    }

    // Property: every `ProgramError` becomes a `Custom` code equal to
    // `u64::from(error) as u32` at instruction index 0.
    proptest! {
        #[test]
        fn program_error_conversion_is_a_truncating_custom_code(code in any::<u32>()) {
            let metadata = FailedTransactionMetadata::from(ProgramError::Custom(code));
            match metadata.err {
                TransactionError::InstructionError(0, InstructionError::Custom(actual)) => {
                    prop_assert_eq!(actual, code);
                }
                other => prop_assert!(false, "unexpected error {other:?}"),
            }
            prop_assert_eq!(metadata.meta, TransactionMetadata::default());
        }
    }

    // Property: `ExecutedInstruction::instruction()` is an exact round trip of
    // the frame's program id, accounts, and data for any frame.
    proptest! {
        #[test]
        fn executed_instruction_round_trips_for_any_frame(
            stack_height in any::<u8>(),
            account_count in 0usize..6,
            data in prop::collection::vec(any::<u8>(), 0..32),
        ) {
            let program_id = Address::new_unique();
            let accounts: Vec<AccountMeta> = (0..account_count)
                .map(|i| {
                    let pubkey = Address::new_unique();
                    if i % 2 == 0 {
                        AccountMeta::new(pubkey, true)
                    } else {
                        AccountMeta::new_readonly(pubkey, false)
                    }
                })
                .collect();
            let expected_accounts = accounts.clone();
            let expected_data = data.clone();
            let frame = ExecutedInstruction { stack_height, program_id, accounts, data };

            let instruction = frame.instruction();

            prop_assert_eq!(instruction.program_id, program_id);
            prop_assert_eq!(&instruction.accounts, &expected_accounts);
            prop_assert_eq!(&instruction.data, &expected_data);
            // The frame itself is untouched: `instruction()` clones rather than moves.
            prop_assert_eq!(frame.accounts, expected_accounts);
            prop_assert_eq!(frame.data, expected_data);
        }
    }

    // Property: `pretty_logs` never panics and emits exactly one line per
    // non-empty input, whatever the log slice contains.
    proptest! {
        #[test]
        fn pretty_logs_never_panics(
            logs in prop::collection::vec("[^\r\n]*", 0..8),
        ) {
            let meta = TransactionMetadata { logs, ..Default::default() };
            let pretty = meta.pretty_logs();
            let expected = meta.logs.iter().filter(|line| !line.is_empty()).count();
            // Bound first: `prop_assert!` treats its single expression as a format
            // string, so literal braces must not appear inline.
            let reset = "\u{1b}[0m";

            prop_assert_eq!(pretty.lines().count(), expected);
            prop_assert_eq!(pretty.matches(reset).count(), expected);
        }
    }
}