// test_raw_input.cpp : Defines the entry point for the console application.
//

#include "stdafx.h"
#include <Windows.h>
#include <strsafe.h>
#include "resource.h"

INT_PTR CALLBACK DlgProc(HWND hwndDlg, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    switch (uMsg)
    {
    case WM_INITDIALOG:
        return TRUE;
    case WM_COMMAND:
        if (LOWORD(wParam) == IDCANCEL)
        {
            DestroyWindow(hwndDlg);
            PostQuitMessage(0);
            return TRUE;
        }
    case WM_INPUT:
        {
            UINT dwSize;

            GetRawInputData((HRAWINPUT)lParam, RID_INPUT, NULL, &dwSize,
                sizeof(RAWINPUTHEADER));
            LPBYTE lpb = new BYTE[dwSize];
            if (lpb == NULL)
            {
                return 0;
            }

            if (GetRawInputData((HRAWINPUT)lParam, RID_INPUT, lpb, &dwSize,
                sizeof(RAWINPUTHEADER)) != dwSize)
                wprintf(L"%s", TEXT("GetRawInputData does not return correct size !\n"));

            RAWINPUT* raw = (RAWINPUT*)lpb;

            WCHAR szTempOutput[1024];
            if (raw->header.dwType == RIM_TYPEKEYBOARD)
            {
                HRESULT hResult = StringCchPrintf(szTempOutput, 1024, TEXT(" Kbd: make=%04x Flags:%04x Reserved:%04x ExtraInformation:%08x, msg=%04x VK=%04x \n"),
                    raw->data.keyboard.MakeCode,
                    raw->data.keyboard.Flags,
                    raw->data.keyboard.Reserved,
                    raw->data.keyboard.ExtraInformation,
                    raw->data.keyboard.Message,
                    raw->data.keyboard.VKey);
                if (FAILED(hResult))
                {
                    printf("ERROR: StringCchPrintf failed-1\n");
                }
                wprintf(L"%s", szTempOutput);
            }
            else if (raw->header.dwType == RIM_TYPEMOUSE)
            {
                HRESULT hResult = StringCchPrintf(szTempOutput, 1024, TEXT("Mouse: usFlags=%04x ulButtons=%04x usButtonFlags=%04x usButtonData=%04x ulRawButtons=%04x lLastX=%04x lLastY=%04x ulExtraInformation=%04x\r\n"),
                    raw->data.mouse.usFlags,
                    raw->data.mouse.ulButtons,
                    raw->data.mouse.usButtonFlags,
                    raw->data.mouse.usButtonData,
                    raw->data.mouse.ulRawButtons,
                    raw->data.mouse.lLastX,
                    raw->data.mouse.lLastY,
                    raw->data.mouse.ulExtraInformation);

                if (FAILED(hResult))
                {
                    printf("ERROR: StringCchPrintf failed-2\n");
                }
                wprintf(L"%s", szTempOutput);
            }

            delete[] lpb;
            return 0;
        }
    }
    return FALSE;
}

int main()
{
    HWND hd = CreateDialog(NULL, MAKEINTRESOURCE(IDD_DIALOG1), NULL, DlgProc);
    ShowWindow(hd, SW_NORMAL);
    RAWINPUTDEVICE Rid[1];
    Rid[0].usUsagePage = 0x01;
    Rid[0].usUsage = 0x02;
    Rid[0].dwFlags = RIDEV_INPUTSINK;
    Rid[0].hwndTarget = hd;
    if (RegisterRawInputDevices(Rid, 1, sizeof(Rid[0])) == FALSE) {
        printf("RegisterRawInputDevices failed, code: %d\n", GetLastError());
        return 1;
    }

    printf("Cycle started.\n");
    BOOL bRet;
    MSG msg;
    while ((bRet = GetMessage(&msg, NULL, 0, 0)) != 0)
    {
        if (bRet == -1)
        {
            // Handle the error and possibly exit
        }
        else if (!IsWindow(hd) || !IsDialogMessage(hd, &msg))
        {
            TranslateMessage(&msg);
            DispatchMessage(&msg);
        }
    }

    return 0;
}