#pragma execution_character_set("utf-8")
#include "mousekeytest.h"
#include "ui_mousekeytest.h"

#include <qdebug.h>

#ifdef Q_OS_WIN
#include <windows.h>
#include "dbt.h"
#include "initguid.h"
#include "usbiodef.h"
#include "hidclass.h"
#include "ntddkbd.h"
#include "ntddmou.h"
#pragma comment(lib, "user32.lib")
#endif


MouseKeyTest::MouseKeyTest(QWidget *parent) :
    QWidget(parent),
    ui(new Ui::MouseKeyTest)
{
    ui->setupUi(this);

    this->setWindowTitle("鼠标、键盘热插拔监测。");
    registerGUID();
}

MouseKeyTest::~MouseKeyTest()
{
    delete ui;
}

/**
 * @brief  给设备注册通知
 */
void MouseKeyTest::registerGUID()
{
#ifdef Q_OS_WIN
    DEV_BROADCAST_DEVICEINTERFACE mouseInterface;
    ZeroMemory(&mouseInterface, sizeof(DEV_BROADCAST_DEVICEINTERFACE));
    mouseInterface.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
    mouseInterface.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
    mouseInterface.dbcc_classguid = GUID_DEVINTERFACE_MOUSE ;
    if(!RegisterDeviceNotificationW((HANDLE)this->winId(), &mouseInterface, DEVICE_NOTIFY_WINDOW_HANDLE))
    {
        qDebug() << "鼠标GUID注册失败";
    }

    DEV_BROADCAST_DEVICEINTERFACE keyboardInterface;
    ZeroMemory(&keyboardInterface, sizeof(DEV_BROADCAST_DEVICEINTERFACE));
    keyboardInterface.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
    keyboardInterface.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
    keyboardInterface.dbcc_classguid = GUID_DEVINTERFACE_KEYBOARD;
    if(!RegisterDeviceNotificationW((HANDLE)this->winId(), &keyboardInterface, DEVICE_NOTIFY_WINDOW_HANDLE))
    {
        qDebug() << "键盘GUID注册失败";
    }
#endif
}

/**
 * @brief
 * @param eventType
 * @param message
 * @param result
 * @return
 */
bool MouseKeyTest::nativeEvent(const QByteArray &eventType, void *message, long *result)
{
#ifdef Q_OS_WIN
    if(eventType == "windows_generic_MSG")
    {
        MSG* msg = reinterpret_cast<MSG*>(message);
        if(msg->message == WM_DEVICECHANGE)                // 通知应用程序设备或计算机的硬件配置发生更改。
        {
            PDEV_BROADCAST_HDR lpdb = (PDEV_BROADCAST_HDR)msg->lParam;
            PDEV_BROADCAST_DEVICEINTERFACE lpdbv = (PDEV_BROADCAST_DEVICEINTERFACE)lpdb;
            switch (msg->wParam)
            {
            case DBT_DEVICEARRIVAL:             // 插入
            {
                if(lpdb->dbch_devicetype == DBT_DEVTYP_DEVICEINTERFACE)
                {
                    if(lpdbv->dbcc_classguid == GUID_DEVINTERFACE_KEYBOARD)
                    {
                        ui->textEdit->append("键盘插入：" + QString::fromWCharArray(lpdbv->dbcc_name));
                    }
                    else if(lpdbv->dbcc_classguid == GUID_DEVINTERFACE_MOUSE)
                    {
                        ui->textEdit->append("鼠标插入：" + QString::fromWCharArray(lpdbv->dbcc_name));
                    }
                }
                break;
            }
            case DBT_DEVICEREMOVECOMPLETE:      // 拔出
            {
                if(lpdb->dbch_devicetype == DBT_DEVTYP_DEVICEINTERFACE)
                {
                    if(lpdbv->dbcc_classguid == GUID_DEVINTERFACE_KEYBOARD)
                    {
                        ui->textEdit->append("键盘移除：" + QString::fromWCharArray(lpdbv->dbcc_name));
                    }
                    else if(lpdbv->dbcc_classguid == GUID_DEVINTERFACE_MOUSE)
                    {
                        ui->textEdit->append("鼠标移除：" + QString::fromWCharArray(lpdbv->dbcc_name));
                    }
                }
                break;
            }
            default:
                break;
            }
        }
    }

#endif
    return false;
}