ethrex-levm 17.0.0

use crate::{
    errors::{ContextResult, VMError},
    hooks::{L2Hook, backup_hook::BackupHook, default_hook::DefaultHook},
    vm::{VM, VMType},
};
use ethrex_common::types::fee_config::FeeConfig;
use std::{cell::RefCell, rc::Rc};

pub trait Hook {
    fn prepare_execution(&mut self, vm: &mut VM<'_>) -> Result<(), VMError>;

    fn finalize_execution(
        &mut self,
        vm: &mut VM<'_>,
        report: &mut ContextResult,
    ) -> Result<(), VMError>;

    /// True iff this hook reads the top-level call-frame backup (`db.tx_backup`) in
    /// `finalize_execution`. Drives `VM::preserve_top_level_backup`: when any installed hook
    /// returns true, the backup is deep-cloned (not moved out) on the revert / invalid-tx paths
    /// so the hook still sees it. Defaults to false; only `BackupHook` overrides it.
    fn reads_top_level_backup(&self) -> bool {
        false
    }
}

pub fn get_hooks(vm_type: &VMType) -> Vec<Rc<RefCell<dyn Hook + 'static>>> {
    match vm_type {
        VMType::L1 => l1_hooks(),
        VMType::L2(fee_config) => l2_hooks(*fee_config),
    }
}

pub fn l1_hooks() -> Vec<Rc<RefCell<dyn Hook + 'static>>> {
    vec![Rc::new(RefCell::new(DefaultHook))]
}

pub fn l2_hooks(fee_config: FeeConfig) -> Vec<Rc<RefCell<dyn Hook + 'static>>> {
    vec![
        Rc::new(RefCell::new(L2Hook::new(fee_config))),
        Rc::new(RefCell::new(BackupHook::default())),
    ]
}