WindowManager.cpp

#include "WindowManager.h"
#include "DeviceMonitor.h"
#include <stdexcept>
#include <wchar.h>

WindowManager::WindowManager(DeviceMonitor* diskMonitor)
    : m_hwnd(NULL), m_hDeviceNotify(NULL), m_diskMonitor(diskMonitor) {
}

WindowManager::~WindowManager() {
    UnregisterDeviceNotification();
    if (m_hwnd) {
        DestroyWindow(m_hwnd);
        m_hwnd = NULL;
    }
}

void WindowManager::InitDeviceNotification() {
    // 注册窗口类
    WNDCLASSEX wc = { 0 };
    wc.cbSize = sizeof(WNDCLASSEX);
    wc.lpfnWndProc = WindowProc;
    wc.hInstance = GetModuleHandle(NULL);
    wc.lpszClassName = L"DiskMonitorWindowClass";
    if (!RegisterClassEx(&wc)) {
        throw std::runtime_error("注册窗口类失败");
    }

    // 创建设备通知窗口
    m_hwnd = CreateWindowEx(0, L"DiskMonitorWindowClass", L"Disk Monitor Hidden Window",
        0, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT, CW_USEDEFAULT,
        HWND_MESSAGE, NULL, GetModuleHandle(NULL), m_diskMonitor);
    if (!m_hwnd) {
        throw std::runtime_error("创建设备通知窗口失败");
    }

    // 注册设备通知
    DEV_BROADCAST_DEVICEINTERFACE filter = { 0 };
    filter.dbcc_size = sizeof(DEV_BROADCAST_DEVICEINTERFACE);
    filter.dbcc_devicetype = DBT_DEVTYP_DEVICEINTERFACE;
    filter.dbcc_classguid = GUID_DEVINTERFACE_VOLUME;

    m_hDeviceNotify = RegisterDeviceNotification(m_hwnd, &filter, DEVICE_NOTIFY_WINDOW_HANDLE);
    if (!m_hDeviceNotify) {
        throw std::runtime_error("注册设备通知失败");
    }
}

void WindowManager::UnregisterDeviceNotification() {
    if (m_hDeviceNotify) {
        ::UnregisterDeviceNotification(m_hDeviceNotify);
        m_hDeviceNotify = NULL;
    }
}

LRESULT CALLBACK WindowManager::WindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam) {
    if (uMsg == WM_CREATE) {
        LPCREATESTRUCT pCreateStruct = (LPCREATESTRUCT)lParam;
        SetWindowLongPtr(hwnd, GWLP_USERDATA, (LONG_PTR)pCreateStruct->lpCreateParams);
        return 0;
    }

    DeviceMonitor* pThis = (DeviceMonitor*)GetWindowLongPtr(hwnd, GWLP_USERDATA);
    if (pThis && uMsg == WM_DEVICECHANGE) {
        return pThis->HandleDeviceChange(static_cast<UINT>(wParam), lParam);
    }

    return DefWindowProc(hwnd, uMsg, wParam, lParam);
}