#include "core/JIT/jit.h"
#include <sys/mman.h>
#include <cstring>
#include <cstdio>
#include <vector>

using JitFunc = void(*)();

void JIT::translate_and_run(CPU& cpu) {
    uint8_t* code = (uint8_t*)mmap(nullptr, 64, PROT_READ | PROT_WRITE | PROT_EXEC,
                                   MAP_PRIVATE | MAP_ANON, -1, 0);
    size_t offset = 0;

    // Decode mock instructions from cpu.memory
    if (cpu.memory[0] == 0x05) { // MOVZ placeholder
        code[offset++] = 0x48; // mov rax, imm64
        code[offset++] = 0xB8;
        uint64_t imm = 5;
        std::memcpy(&code[offset], &imm, sizeof(imm));
        offset += 8;
    }

    if (cpu.memory[4] == 0x03) { // ADD placeholder
        code[offset++] = 0x48; // add rax, imm32
        code[offset++] = 0x05;
        uint32_t addval = 3;
        std::memcpy(&code[offset], &addval, sizeof(addval));
        offset += 4;
    }

    code[offset++] = 0xC3; // ret

    JitFunc fn = reinterpret_cast<JitFunc>(code);
    uint64_t result;
    asm volatile (
        "call *%1\n"
        "mov %%rax, %0\n"
        : "=r"(result)
        : "r"(fn)
        : "%rax"
    );

    cpu.regs[0] = result;
}