#include <jni.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <android/log.h>
#include <dlfcn.h>
#include <unistd.h>
#include <pthread.h>
#include <vector>
#include <string>

// 核心库包含
#include "dobby.h"
#include "imgui.h"
#include "imgui_impl_android.h"
#include "imgui_impl_opengl3.h"
#include <GLES3/gl3.h>
#include <EGL/egl.h>

#define LOG_TAG "JKMenu"
#define LOGI(...) __android_log_print(ANDROID_LOG_INFO, LOG_TAG, __VA_ARGS__)
#define LOGE(...) __android_log_print(ANDROID_LOG_ERROR, LOG_TAG, __VA_ARGS__)

// --- 原始函数指针存储 ---
typedef EGLBoolean (*eglSwapBuffers_t)(EGLDisplay dpy, EGLSurface surface);
eglSwapBuffers_t orig_eglSwapBuffers = nullptr;

typedef void* (*vkQueuePresentKHR_t)(void* queue, const void* pPresentInfo);
vkQueuePresentKHR_t orig_vkQueuePresentKHR = nullptr;

// --- 全局状态 ---
bool g_Initialized = false;
bool g_ShowMenu = true;
int g_Width = 0;
int g_Height = 0;
bool g_IsVulkanMode = false;

// 菜单渲染逻辑 (仅限 GLES 环境)
void DrawImGuiMenu() {
    if (!g_ShowMenu) return;

    ImGui::SetNextWindowSize(ImVec2(400, 350), ImGuiCond_FirstUseEver);
    ImGui::Begin("JKMenu - Renderer Check", &g_ShowMenu);

    ImGui::Text("Device: OnePlus Pad Pro 2");
    ImGui::Text("Target: com.tencent.jkchess");
    ImGui::Text("Mode: %s", g_IsVulkanMode ? "Vulkan (Unsupported)" : "GLES3 (Active)");
    ImGui::Text("Screen: %dx%d", g_Width, g_Height);
    ImGui::Separator();

    if (g_IsVulkanMode) {
        ImGui::TextColored(ImVec4(1, 0, 0, 1), "ERROR: Cannot draw in Vulkan mode!");
        ImGui::Text("Please disable High Quality in settings.");
    } else {
        static float f = 0.5f;
        ImGui::SliderFloat("Scale", &f, 0.1f, 2.0f);
    }

    if (ImGui::Button("Hide Menu")) {
        g_ShowMenu = false;
    }

    ImGui::End();
}

// OpenGL 渲染执行器
void PerformRenderGLES(EGLDisplay dpy, EGLSurface surface) {
    if (!g_Initialized) {
        LOGI("SUCCESS! OpenGL (GLES) Render triggered.");
        eglQuerySurface(dpy, surface, EGL_WIDTH, &g_Width);
        eglQuerySurface(dpy, surface, EGL_HEIGHT, &g_Height);
        
        if (g_Width <= 0 || g_Height <= 0) return;

        ImGui::CreateContext();
        ImGuiIO& io = ImGui::GetIO();
        io.DisplaySize = ImVec2((float)g_Width, (float)g_Height);
        
        if (ImGui_ImplOpenGL3_Init("#version 300 es")) {
            g_Initialized = true;
            LOGI("ImGui GLES Initialized: %dx%d", g_Width, g_Height);
        }
    }

    if (g_Initialized) {
        ImGui_ImplOpenGL3_NewFrame();
        ImGui::NewFrame();
        DrawImGuiMenu();
        ImGui::Render();
        ImGui_ImplOpenGL3_RenderDrawData(ImGui::GetDrawData());
    }
}

// Hook 回调: OpenGL
EGLBoolean hooked_eglSwapBuffers(EGLDisplay dpy, EGLSurface surface) {
    g_IsVulkanMode = false;
    PerformRenderGLES(dpy, surface);
    return orig_eglSwapBuffers(dpy, surface);
}

// Hook 回调: Vulkan
void* hooked_vkQueuePresentKHR(void* queue, const void* pPresentInfo) {
    static int vk_warn_count = 0;
    g_IsVulkanMode = true;
    if (vk_warn_count % 300 == 0) {
        LOGE("Vulkan Present triggered! Please switch game to GLES mode to see menu.");
    }
    vk_warn_count++;
    return orig_vkQueuePresentKHR(queue, pPresentInfo);
}

// 动态搜索并 Hook
void* hack_thread(void*) {
    LOGI("SO Loaded! Starting Multi-Renderer Scanner...");
    sleep(10);

    // 1. 尝试 Hook OpenGL (GLES)
    const char* gles_syms[] = {"eglSwapBuffers", "nw_main_swap_buffers"};
    for (const char* sym : gles_syms) {
        void* target = DobbySymbolResolver("libEGL.so", sym);
        if (!target) target = dlsym(RTLD_DEFAULT, sym);
        if (target) {
            LOGI("Found GLES Hook Point: %s at %p", sym, target);
            DobbyHook(target, (dobby_dummy_func_t)hooked_eglSwapBuffers, (dobby_dummy_func_t*)&orig_eglSwapBuffers);
        }
    }

    // 2. 尝试 Hook Vulkan (仅用于识别状态)
    void* vk_target = DobbySymbolResolver("libvulkan.so", "vkQueuePresentKHR");
    if (vk_target) {
        LOGI("Found Vulkan Hook Point: vkQueuePresentKHR at %p", vk_target);
        DobbyHook(vk_target, (dobby_dummy_func_t)hooked_vkQueuePresentKHR, (dobby_dummy_func_t*)&orig_vkQueuePresentKHR);
    }

    if (!orig_eglSwapBuffers && !orig_vkQueuePresentKHR) {
        LOGE("Could not find any render functions!");
    }

    return nullptr;
}

void __attribute__((constructor)) init() {
    pthread_t t;
    pthread_create(&t, nullptr, hack_thread, nullptr);
}