globalkeyinterceptor.cpp

#include "globalkeyinterceptor.h"
#include "mainwindow.h"

#include <QMetaObject>

bool GlobalKeyInterceptor::capturing = false;
HHOOK GlobalKeyInterceptor::hookHandle = nullptr;
int GlobalKeyInterceptor::modState = 0;
int GlobalKeyInterceptor::captureTarget = TARGET_NONE;

bool GlobalKeyInterceptor::isModifierKey(int vk) {
    switch (vk) {
        case VK_CONTROL:
        case VK_LCONTROL:
        case VK_RCONTROL:
        case VK_MENU:
        case VK_LMENU:
        case VK_RMENU:
        case VK_SHIFT:
        case VK_LSHIFT:
        case VK_RSHIFT:
        case VK_LWIN:
        case VK_RWIN:
            return true;
        default:
            return false;
    }
}

int GlobalKeyInterceptor::modBitForKey(int vk) {
    switch (vk) {
        case VK_CONTROL:
        case VK_LCONTROL:
        case VK_RCONTROL:
            return MB_CTRL;
        case VK_MENU:
        case VK_LMENU:
        case VK_RMENU:
            return MB_ALT;
        case VK_SHIFT:
        case VK_LSHIFT:
        case VK_RSHIFT:
            return MB_SHIFT;
        case VK_LWIN:
        case VK_RWIN:
            return MB_WIN;
        default:
            return 0;
    }
}

QString GlobalKeyInterceptor::buildSequenceString(int vk) {
    QString seq;
    if (modState & MB_CTRL) seq += "Ctrl+";
    if (modState & MB_ALT) seq += "Alt+";
    if (modState & MB_SHIFT) seq += "Shift+";
    if (modState & MB_WIN) seq += "Win+";

    wchar_t name[64] = {0};
    if (GetKeyNameTextW(MapVirtualKeyW(vk, MAPVK_VK_TO_VSC) << 16, name, 64) > 0) {
        seq += QString::fromWCharArray(name);
    } else {
        seq += QString::number(vk);
    }
    return seq;
}

void GlobalKeyInterceptor::handleKeyDown(int vk) {
    if (isModifierKey(vk)) {
        modState |= modBitForKey(vk);
        return;
    }

    const QString seq = buildSequenceString(vk);
    if (MainWindow::instance) {
        const char *method = (captureTarget == TARGET_FINALIZATION_HOTKEY)
                                 ? "setFinalizationHotkeyText"
                                 : "setHotkeyText";
        QMetaObject::invokeMethod(MainWindow::instance, method, Qt::QueuedConnection,
                                  Q_ARG(QString, seq));
    }
    modState = 0;
}

void GlobalKeyInterceptor::handleKeyUp(int vk) {
    if (isModifierKey(vk)) {
        modState &= ~modBitForKey(vk);
    }
}

LRESULT CALLBACK GlobalKeyInterceptor::hookProc(int nCode, WPARAM wParam, LPARAM lParam) {
    if (nCode < 0 || !capturing)
        return CallNextHookEx(hookHandle, nCode, wParam, lParam);

    auto *kb = reinterpret_cast<KBDLLHOOKSTRUCT *>(lParam);

    if (wParam == WM_KEYDOWN) {
        handleKeyDown(static_cast<int>(kb->vkCode));
        return 1;
    }

    if (wParam == WM_KEYUP) {
        handleKeyUp(static_cast<int>(kb->vkCode));
    }

    return CallNextHookEx(hookHandle, nCode, wParam, lParam);
}

void GlobalKeyInterceptor::start() {
    if (!hookHandle) {
        hookHandle = SetWindowsHookExW(WH_KEYBOARD_LL, hookProc, GetModuleHandleW(nullptr), 0);
    }
}

void GlobalKeyInterceptor::stop() {
    if (hookHandle) {
        UnhookWindowsHookEx(hookHandle);
        hookHandle = nullptr;
    }
}