use solana_bpf_loader_program::{
    create_vm,
    serialization::{deserialize_parameters, serialize_parameters},
};
use solana_rbpf::vm::{EbpfVm, InstructionMeter};
use solana_sdk::{
    account::{Account as SolanaAccount, KeyedAccount},
    bpf_loader,
    entrypoint::SUCCESS,
    entrypoint_native::{
        ComputeBudget, ComputeMeter, Executor, InvokeContext, Logger, ProcessInstruction,
    },
    instruction::{CompiledInstruction, InstructionError},
    message::Message,
    program_option::COption,
    program_pack::Pack,
    pubkey::Pubkey,
    sysvar::rent::{self, Rent},
};
use spl_token::{
    instruction::TokenInstruction,
    state::{Account, Mint},
};
use std::{cell::RefCell, fs::File, io::Read, path::PathBuf, rc::Rc, sync::Arc};

fn load_program(name: &str) -> Vec<u8> {
    let mut path = PathBuf::new();
    path.push("../../target/bpfel-unknown-unknown/release");
    path.push(name);
    path.set_extension("so");
    let mut file = File::open(path).unwrap();

    let mut program = Vec::new();
    file.read_to_end(&mut program).unwrap();
    program
}

fn run_program(
    program_id: &Pubkey,
    parameter_accounts: &[KeyedAccount],
    instruction_data: &[u8],
) -> Result<u64, InstructionError> {
    let mut program_account = SolanaAccount::default();
    program_account.data = load_program("spl_token");
    let loader_id = bpf_loader::id();
    let mut invoke_context = MockInvokeContext::default();
    let executable = EbpfVm::<solana_bpf_loader_program::BPFError>::create_executable_from_elf(
        &&program_account.data,
        None,
    )
    .unwrap();
    let (mut vm, heap_region) = create_vm(
        &loader_id,
        executable.as_ref(),
        parameter_accounts,
        &mut invoke_context,
    )
    .unwrap();
    let mut parameter_bytes = serialize_parameters(
        &loader_id,
        program_id,
        parameter_accounts,
        &instruction_data,
    )
    .unwrap();
    assert_eq!(
        SUCCESS,
        vm.execute_program(parameter_bytes.as_mut_slice(), &[], &[heap_region])
            .unwrap()
    );
    deserialize_parameters(&loader_id, parameter_accounts, &parameter_bytes).unwrap();
    Ok(vm.get_total_instruction_count())
}

#[test]
fn assert_instruction_count() {
    let program_id = Pubkey::new_rand();
    let source_key = Pubkey::new_rand();
    let source_account = SolanaAccount::new_ref(u64::MAX, Account::get_packed_len(), &program_id);
    let destination_key = Pubkey::new_rand();
    let destination_account =
        SolanaAccount::new_ref(u64::MAX, Account::get_packed_len(), &program_id);
    let owner_key = Pubkey::new_rand();
    let owner_account = RefCell::new(SolanaAccount::default());
    let mint_key = Pubkey::new_rand();
    let mint_account = SolanaAccount::new_ref(0, Mint::get_packed_len(), &program_id);
    let rent_key = rent::id();
    let rent_account = RefCell::new(rent::create_account(42, &Rent::default()));

    // Create new mint
    let instruction_data = TokenInstruction::InitializeMint {
        decimals: 9,
        mint_authority: owner_key,
        freeze_authority: COption::None,
    }
    .pack();
    let parameter_accounts = vec![
        KeyedAccount::new(&mint_key, false, &mint_account),
        KeyedAccount::new(&source_key, false, &source_account),
    ];
    let initialize_mint_count =
        run_program(&program_id, &parameter_accounts[..], &instruction_data).unwrap();

    // Create source account
    let instruction_data = TokenInstruction::InitializeAccount.pack();
    let parameter_accounts = vec![
        KeyedAccount::new(&source_key, false, &source_account),
        KeyedAccount::new(&mint_key, false, &mint_account),
        KeyedAccount::new(&owner_key, false, &owner_account),
        KeyedAccount::new(&rent_key, false, &rent_account),
    ];
    let mintto_count =
        run_program(&program_id, &parameter_accounts[..], &instruction_data).unwrap();

    // Create destination account
    let instruction_data = TokenInstruction::InitializeAccount.pack();
    let parameter_accounts = vec![
        KeyedAccount::new(&destination_key, false, &destination_account),
        KeyedAccount::new(&mint_key, false, &mint_account),
        KeyedAccount::new(&owner_key, false, &owner_account),
        KeyedAccount::new(&rent_key, false, &rent_account),
    ];
    let _ = run_program(&program_id, &parameter_accounts[..], &instruction_data).unwrap();

    // MintTo source account
    let instruction_data = TokenInstruction::MintTo { amount: 100 }.pack();
    let parameter_accounts = vec![
        KeyedAccount::new(&mint_key, false, &mint_account),
        KeyedAccount::new(&source_key, false, &source_account),
        KeyedAccount::new(&owner_key, true, &owner_account),
    ];
    let initialize_account_count =
        run_program(&program_id, &parameter_accounts[..], &instruction_data).unwrap();

    // Transfer from source to destination
    let instruction = TokenInstruction::Transfer { amount: 100 };
    let instruction_data = instruction.pack();
    let parameter_accounts = vec![
        KeyedAccount::new(&source_key, false, &source_account),
        KeyedAccount::new(&destination_key, false, &destination_account),
        KeyedAccount::new(&owner_key, true, &owner_account),
    ];
    let transfer_count =
        run_program(&program_id, &parameter_accounts[..], &instruction_data).unwrap();

    const BASELINE_NEW_MINT_COUNT: u64 = 4000; // last known 3802
    const BASELINE_INITIALIZE_ACCOUNT_COUNT: u64 = 6500; // last known 6445
    const BASELINE_MINTTO_COUNT: u64 = 6500; // last known 6194
    const BASELINE_TRANSFER_COUNT: u64 = 8000; // last known 7609

    println!("BPF instructions executed");
    println!(
        "  InitializeMint   : {:?} ({:?})",
        initialize_mint_count, BASELINE_NEW_MINT_COUNT
    );
    println!(
        "  InitializeAccount: {:?} ({:?})",
        initialize_account_count, BASELINE_INITIALIZE_ACCOUNT_COUNT
    );
    println!(
        "  MintTo           : {:?} ({:?})",
        mintto_count, BASELINE_MINTTO_COUNT
    );
    println!(
        "  Transfer         : {:?} ({:?})",
        transfer_count, BASELINE_TRANSFER_COUNT,
    );

    assert!(initialize_account_count <= BASELINE_INITIALIZE_ACCOUNT_COUNT);
    assert!(initialize_mint_count <= BASELINE_NEW_MINT_COUNT);
    assert!(transfer_count <= BASELINE_TRANSFER_COUNT);
}

// Mock InvokeContext

#[derive(Debug, Default)]
struct MockInvokeContext {
    pub key: Pubkey,
    pub logger: MockLogger,
    pub compute_meter: MockComputeMeter,
}
impl InvokeContext for MockInvokeContext {
    fn push(&mut self, _key: &Pubkey) -> Result<(), InstructionError> {
        Ok(())
    }
    fn pop(&mut self) {}
    fn verify_and_update(
        &mut self,
        _message: &Message,
        _instruction: &CompiledInstruction,
        _accounts: &[Rc<RefCell<SolanaAccount>>],
    ) -> Result<(), InstructionError> {
        Ok(())
    }
    fn get_caller(&self) -> Result<&Pubkey, InstructionError> {
        Ok(&self.key)
    }
    fn get_programs(&self) -> &[(Pubkey, ProcessInstruction)] {
        &[]
    }
    fn get_logger(&self) -> Rc<RefCell<dyn Logger>> {
        Rc::new(RefCell::new(self.logger.clone()))
    }
    fn is_cross_program_supported(&self) -> bool {
        true
    }
    fn get_compute_budget(&self) -> ComputeBudget {
        ComputeBudget::default()
    }
    fn get_compute_meter(&self) -> Rc<RefCell<dyn ComputeMeter>> {
        Rc::new(RefCell::new(self.compute_meter.clone()))
    }
    fn add_executor(&mut self, _pubkey: &Pubkey, _executor: Arc<dyn Executor>) {}
    fn get_executor(&mut self, _pubkey: &Pubkey) -> Option<Arc<dyn Executor>> {
        None
    }
}

#[derive(Debug, Default, Clone)]
struct MockComputeMeter {}
impl ComputeMeter for MockComputeMeter {
    fn consume(&mut self, _amount: u64) -> Result<(), InstructionError> {
        Ok(())
    }
    fn get_remaining(&self) -> u64 {
        u64::MAX
    }
}
#[derive(Debug, Default, Clone)]
struct MockLogger {}
impl Logger for MockLogger {
    fn log_enabled(&self) -> bool {
        true
    }
    fn log(&mut self, message: &str) {
        println!("{}", message);
    }
}

struct TestInstructionMeter {}
impl InstructionMeter for TestInstructionMeter {
    fn consume(&mut self, _amount: u64) {}
    fn get_remaining(&self) -> u64 {
        u64::MAX
    }
}