#include <pthread.h>
#include <unistd.h>
#include <android/log.h>
#include <EGL/egl.h>
#include <GLES3/gl3.h>
#include <dlfcn.h>
#include <android/input.h>
#include <stdio.h>
#include <algorithm>
#include "dobby.h"
#include "imgui.h"
#include "imgui_impl_android.h"
#include "imgui_impl_opengl3.h"

#define TAG "ZeroConfig_Drive"

static bool g_Initialized = false;

struct {
    float x = 0.0f;
    float y = 0.0f;
    bool down = false;
    
    float renderW = 0.0f; 
    float renderH = 0.0f;

    float physW = 3392.0f;
    float physH = 2400.0f;
    bool isSizeAutoDetected = false;
} g_Touch;

// 高度缩放基准
static float g_DynamicScale = 1.0f;
static float g_BaseWindowHeight = 0.0f; 

typedef EGLBoolean (*p_eglSwapBuffers)(EGLDisplay dpy, EGLSurface surface);
static p_eglSwapBuffers old_eglSwapBuffers = nullptr;

typedef int (*p_InputConsumer_consume)(void* instance, void* factory, bool consumeBatches, int64_t frameTime, uint32_t* outSeq, void** outEvent);
static p_InputConsumer_consume old_consume = nullptr;

void AutoFetchDeviceResolution() {
    FILE* pipe = popen("wm size", "r");
    if (!pipe) return;
    
    char buffer[128];
    while (fgets(buffer, sizeof(buffer), pipe) != NULL) {
        int w = 0, h = 0;
        if (sscanf(buffer, "Physical size: %dx%d", &w, &h) == 2 ||
            sscanf(buffer, "Override size: %dx%d", &w, &h) == 2) {
            g_Touch.physW = (float)std::max(w, h);
            g_Touch.physH = (float)std::min(w, h);
            g_Touch.isSizeAutoDetected = true;
        }
    }
    pclose(pipe);
}

void handle_android_event(AInputEvent* event) {
    if (AInputEvent_getType(event) == AINPUT_EVENT_TYPE_MOTION) {
        int32_t action = AMotionEvent_getAction(event) & AMOTION_EVENT_ACTION_MASK;
        
        float rawX = AMotionEvent_getX(event, 0);
        float rawY = AMotionEvent_getY(event, 0);
        
        float autoScaleX = g_Touch.renderW / g_Touch.physW;
        float autoScaleY = g_Touch.renderH / g_Touch.physH;

        g_Touch.x = rawX * autoScaleX;
        g_Touch.y = rawY * autoScaleY;

        if (action == AMOTION_EVENT_ACTION_DOWN) {
            g_Touch.down = true;
        } else if (action == AMOTION_EVENT_ACTION_UP || action == AMOTION_EVENT_ACTION_CANCEL) {
            g_Touch.down = false;
        }
    }
}

// =================================================================
// 核心照抄区 1：完美触摸不穿透
// =================================================================
int hook_consume(void* instance, void* factory, bool consumeBatches, int64_t frameTime, uint32_t* outSeq, void** outEvent) {
    // 1. 先让系统原函数去读取事件
    int res = old_consume(instance, factory, consumeBatches, frameTime, outSeq, outEvent);
    
    if (res == 0 && outEvent && *outEvent) {
        // 2. 我们先拿到事件并转换坐标给 ImGui
        AInputEvent* event = (AInputEvent*)(*outEvent);
        handle_android_event(event);
        
        // 3. 【防穿透绝杀】判断当前手指是否点在菜单上
        if (g_Initialized && ImGui::GetCurrentContext() != nullptr) {
            ImGuiIO& io = ImGui::GetIO();
            if (io.WantCaptureMouse) {
                // 如果点在菜单上，直接把底层传给游戏引擎的事件指针“没收”
                *outEvent = nullptr; 
                // 返回 -EWOULDBLOCK (-11)，欺骗安卓系统，让它以为当前队列为空！
                return -11; 
            }
        }
    }
    return res;
}

EGLBoolean hook_eglSwapBuffers(EGLDisplay dpy, EGLSurface surface) {
    EGLint w, h;
    eglQuerySurface(dpy, surface, EGL_WIDTH, &w);
    eglQuerySurface(dpy, surface, EGL_HEIGHT, &h);
    
    if (w <= 0 || h <= 0) return old_eglSwapBuffers(dpy, surface);

    g_Touch.renderW = (float)w;
    g_Touch.renderH = (float)h;

    if (!g_Initialized) {
        ImGui::CreateContext();
        ImGui_ImplOpenGL3_Init("#version 300 es");
        g_Initialized = true;
    }

    ImGuiIO& io = ImGui::GetIO();
    io.DisplaySize = ImVec2(g_Touch.renderW, g_Touch.renderH);
    io.MousePos = ImVec2(g_Touch.x, g_Touch.y);
    io.MouseDown[0] = g_Touch.down;

    float deviceScale = (float)h / 1000.0f;
    if (deviceScale < 1.0f) deviceScale = 1.0f;
    float finalScale = deviceScale * g_DynamicScale; 

    static ImGuiStyle default_style;
    static bool style_backed_up = false;
    if (!style_backed_up) {
        default_style = ImGui::GetStyle();
        style_backed_up = true;
    }

    ImGuiStyle& style = ImGui::GetStyle();
    style = default_style;
    style.ScaleAllSizes(3.0f * finalScale);
    style.WindowRounding = 12.0f;
    style.WindowMinSize = ImVec2(100.0f, 100.0f); 
    io.FontGlobalScale = 3.0f * finalScale;

    ImGui_ImplOpenGL3_NewFrame();
    ImGui::NewFrame();

    if (g_Touch.down) {
        ImGui::GetForegroundDrawList()->AddCircleFilled(io.MousePos, 20.0f, IM_COL32(0, 255, 0, 200));
    }

    if (g_BaseWindowHeight == 0.0f) {
        g_BaseWindowHeight = h * 0.45f; 
    }

    ImGui::SetNextWindowPos(ImVec2(30, 30), ImGuiCond_FirstUseEver);
    ImGui::SetNextWindowSize(ImVec2(w * 0.7f, g_BaseWindowHeight), ImGuiCond_FirstUseEver);
    
    if (ImGui::Begin("Zero-Config Engine")) {
        
        float currentHeight = ImGui::GetWindowHeight();
        g_DynamicScale = currentHeight / g_BaseWindowHeight;
        
        if (g_DynamicScale < 0.4f) g_DynamicScale = 0.4f;
        if (g_DynamicScale > 3.0f) g_DynamicScale = 3.0f;

        ImGui::TextColored(ImVec4(0, 1, 1, 1), "[*] Magic Resize Active");
        ImGui::Separator();
        
        ImGui::Text("Render Surface: %.0f x %.0f", g_Touch.renderW, g_Touch.renderH);
        
        if (g_Touch.isSizeAutoDetected) {
            ImGui::TextColored(ImVec4(0, 1, 0, 1), "Physical Screen: %.0f x %.0f (Auto)", g_Touch.physW, g_Touch.physH);
        } else {
            ImGui::TextColored(ImVec4(1, 1, 0, 1), "Physical Screen: %.0f x %.0f (Fallback)", g_Touch.physW, g_Touch.physH);
        }
        
        ImGui::Separator();
        ImGui::TextColored(ImVec4(1, 1, 0, 1), "-> Drag RIGHT edge to show more text.");
        ImGui::TextColored(ImVec4(1, 1, 0, 1), "-> Drag BOTTOM edge to ZOOM IN/OUT.");
        ImGui::Separator();

        static bool esp_enabled = true;
        ImGui::Checkbox("Enable ESP (Demo)", &esp_enabled);
        
        static float test_slider = 50.0f;
        ImGui::SliderFloat("FOV Size", &test_slider, 0.0f, 100.0f);
    }
    ImGui::End();

    ImGui::Render();
    
    // =================================================================
    // 核心照抄区 2：Viewport 渲染防污染机制
    // =================================================================
    // 备份游戏本身的 Viewport 和 深度测试状态
    GLint last_viewport[4];
    glGetIntegerv(GL_VIEWPORT, last_viewport);
    GLint last_depth;
    glGetIntegerv(GL_DEPTH_TEST, &last_depth);
    
    // 关闭深度测试，防止 UI 被游戏的 3D 模型遮挡
    glDisable(GL_DEPTH_TEST);
    
    // 强制把渲染视口设置到与我们的 DisplaySize 一致
    // 这行就是 2.txt 里的：glViewport(0, 0, g_gl_width, g_gl_height);
    glViewport(0, 0, w, h); 
    
    // 绘制 ImGui 数据
    ImGui_ImplOpenGL3_RenderDrawData(ImGui::GetDrawData());
    
    // 完美复原游戏环境，游戏引擎就不会崩溃或画面扭曲
    glViewport(last_viewport[0], last_viewport[1], last_viewport[2], last_viewport[3]);
    if (last_depth) glEnable(GL_DEPTH_TEST);

    return old_eglSwapBuffers(dpy, surface);
}

void* init_thread(void*) {
    AutoFetchDeviceResolution();
    
    sleep(6); 
    void* sym_egl = dlsym(RTLD_DEFAULT, "eglSwapBuffers");
    if (sym_egl) DobbyHook(sym_egl, (void*)hook_eglSwapBuffers, (void**)&old_eglSwapBuffers);
    
    void* consume_addr = DobbySymbolResolver("libinput.so", "_ZN7android13InputConsumer7consumeEPNS_26InputEventFactoryInterfaceEblPjPPNS_10InputEventE");
    if (consume_addr) DobbyHook(consume_addr, (void*)hook_consume, (void**)&old_consume);
    
    return nullptr;
}

__attribute__((constructor)) void init_plugin() {
    pthread_t t;
    pthread_create(&t, nullptr, init_thread, nullptr);
}