main.cpp

#include "ezui.hpp"

using DrawCommand = DX11Renderer::DrawCommand;
using Color = DX11Renderer::Color;

class WindowHijacker {
public:
    HWND findWindow(const std::string& partialName);
    HWND createWindow(const std::string& windowName, int width, int height);

private:
    struct SearchData {
        std::string name;
        HWND hwnd;
    };
    static BOOL CALLBACK enumWindowsProc(HWND hwnd, LPARAM lParam);
    std::string toLower(const std::string& str);
};

HWND WindowHijacker::findWindow(const std::string& partialName) {
    SearchData searchData;
    searchData.name = toLower(partialName);
    searchData.hwnd = nullptr;

    EnumWindows(enumWindowsProc, reinterpret_cast<LPARAM>(&searchData));
    return searchData.hwnd;
}

BOOL CALLBACK WindowHijacker::enumWindowsProc(HWND hwnd, LPARAM lParam) {
    char windowTitle[256];
    GetWindowTextA(hwnd, windowTitle, sizeof(windowTitle));

    std::string titleStr = windowTitle;
    std::transform(titleStr.begin(), titleStr.end(), titleStr.begin(), ::tolower);

    SearchData* searchData = reinterpret_cast<SearchData*>(lParam);
    if (titleStr.find(searchData->name) != std::string::npos) {
        searchData->hwnd = hwnd;
        return FALSE;
    }
    return TRUE;
}

std::string WindowHijacker::toLower(const std::string& str) {
    std::string result = str;
    std::transform(result.begin(), result.end(), result.begin(), ::tolower);
    return result;
}

HWND WindowHijacker::createWindow(const std::string& windowName, int width, int height) {
    WNDCLASS wc = { 0 };
    wc.lpfnWndProc = DefWindowProc;
    wc.hInstance = GetModuleHandle(nullptr);
    wc.lpszClassName = L"ezUIwindowClass";
    wc.hCursor = LoadCursor(nullptr, IDC_ARROW);

    if (!RegisterClass(&wc)) {
        std::cerr << "Failed to register window class! Error: " << GetLastError() << std::endl;
        return nullptr;
    }

    std::wstring wWindowName(windowName.begin(), windowName.end());

    HWND hwnd = CreateWindowEx(
        WS_EX_TOPMOST | WS_EX_LAYERED | WS_EX_TRANSPARENT,
        wc.lpszClassName,
        wWindowName.c_str(),
        WS_POPUP | WS_VISIBLE,
        CW_USEDEFAULT, CW_USEDEFAULT, width, height,
        nullptr, nullptr, wc.hInstance, nullptr
    );

    if (!hwnd) {
        std::cerr << "Failed to create window! Error: " << GetLastError() << std::endl;
        return nullptr;
    }

    MARGINS margins = { -1 };
    DwmExtendFrameIntoClientArea(hwnd, &margins);

    SetLayeredWindowAttributes(hwnd, 0, 255, LWA_ALPHA);

    ShowWindow(hwnd, SW_SHOW);
    UpdateWindow(hwnd);

    return hwnd;
}

int main() {
    WindowHijacker hijacker;
    HWND hwnd = hijacker.findWindow("yourwindowtohijack");

    int screenWidth = GetSystemMetrics(SM_CXSCREEN);
    int screenHeight = GetSystemMetrics(SM_CYSCREEN);

    if (!hwnd) {
        std::cout << "Window not found, creating a new window.\n";
        hwnd = hijacker.createWindow("My ezUI Window", screenWidth, screenHeight);
    }

    DX11Renderer renderer(hwnd);
    renderer.initD3D11();


    ezUI ui(renderer);


    ui.addContainer("A", 100.0f, 100.0f, 250.0f, 350.0f);
    ui.toggleVisibility("A");

    ui.addButton("A", "TestButton", { 210.0f, 330.0f, 40.0f, 20.0f, 0.0f, DX11Renderer::Color(.45f, 0.45f, 0.45f, 1.0f) },
        [](ezUI::Button& button) {
        PostQuitMessage(0);
    },
        [](ezUI::Button& button) {
        button.bounds.color = DX11Renderer::Color(.7f, 0.7f, 0.7f, 1.0f);
    },  [](ezUI::Button& button) {
        button.bounds.color = DX11Renderer::Color(.45f, 0.45f, 0.45f, 1.0f);
    }
    );


    ui.addHotkey("A", VK_END, []() {
        PostQuitMessage(0);
    });

    ui.addHotkey("A", VK_HOME, [&ui]() {
        ui.masterToggle();
    });

    MSG msg = {};
    while (msg.message != WM_QUIT) {
        if (PeekMessage(&msg, nullptr, 0, 0, PM_REMOVE)) {
            TranslateMessage(&msg);
            DispatchMessage(&msg);
        }

        ui.handleInput();
        ui.drawAllElements();
    }

    return 0;
}