#pragma once
#include <functional>
#include <vmp_rtn.hpp>
#include <vmprofiler.hpp>

#include "llvm/IR/BasicBlock.h"
#include "llvm/IR/Constants.h"
#include "llvm/IR/DerivedTypes.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/IRPrintingPasses.h"
#include "llvm/IR/LLVMContext.h"
#include "llvm/IR/Module.h"
#include "llvm/IR/PassManager.h"
#include "llvm/IR/Type.h"
#include "llvm/IR/Verifier.h"
#include "llvm/Pass.h"

namespace vmp2::lifters
{
    using lifter_callback_t =
        std::function< void( vm::vmp_rtn_t *rtn, const vm::instrs::code_block_t &vm_code_block,
                             const vm::instrs::virt_instr_t &vinstr, llvm::IRBuilder<> *ir_builder ) >;

    extern lifter_callback_t lconstq;
    extern lifter_callback_t sregq;

    inline std::map< vm::handler::mnemonic_t, lifter_callback_t * > lifters = { { vm::handler::LCONSTQ, &lconstq },
                                                                                { vm::handler::SREGQ, &sregq } };

    inline bool lift( vm::vmp_rtn_t *rtn, const vm::instrs::code_block_t &vm_code_block,
                      const vm::instrs::virt_instr_t &vinstr, llvm::IRBuilder<> *ir_builder )
    {
        if ( vinstr.mnemonic_t == vm::handler::INVALID || lifters.find( vinstr.mnemonic_t ) == lifters.end() )
            return false;

        ( *( lifters[ vinstr.mnemonic_t ] ) )( rtn, vm_code_block, vinstr, ir_builder );
        return true;
    }
} // namespace vmp2::lifters