SearchWindow.cpp

/*
    查找窗口模块
    这个模块可以让用户根据窗口类名和标题搜索窗口。
*/

#include <windows.h>
#include <CommCtrl.h>
#include <string>
#include <thread>

#include "framework_base.hpp"
#include "SearchWindowIDs.hpp"
#include "PropertyWindow.hpp"

#include "toolset.h"
#include "config.h"

extern HINSTANCE g_hInstance;
extern HWND hTreeView;
extern bool gb_quitEvent;

bool SearchWindowRunning = false;
HWND s_searchWindowHWND = nullptr;

LRESULT CALLBACK SearchWindowProc(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{

    enum {
        sw_lbClass, sw_lbTitle, sw_lbHwnd,
        sw_titleEdit, sw_classEdit,
        sw_cbAutoUpdateOnEdit,
        sw_SearchBtn, sw_ClearBtn,
        sw_hwndEdit,
        sw_treeBtn, sw_propertyBtn,
        SWH_HwndID_MAXCOUNT
    };

    using Framework::ChildWindow;
    thread_local static ChildWindow ctrls[SWH_HwndID_MAXCOUNT];
    thread_local static bool AutoUpdateOnEdit;

    // thread_local: 每个线程一份、跨消息存活；线程退出时析构（自动 DeleteObject）
    thread_local static Framework::Font font;

    thread_local static HWND targetHwnd;

    if(gb_quitEvent) DestroyWindow(hwnd);
    switch(uMsg) {
    case WM_CREATE:
    {
        FRAMEWORK_INIT_WM_CREATE

        // Labels
        ctrls[sw_lbClass] = ChildWindow::createStatic("类名", { 15, 15, 40, 24 }, hwnd, 0);
        ctrls[sw_lbTitle] = ChildWindow::createStatic("标题", { 15, 45, 40, 24 }, hwnd, 0);
        ctrls[sw_lbHwnd] = ChildWindow::createStatic("hwnd", { 15, 113, 40, 24 }, hwnd, 0);

        ctrls[sw_classEdit] = ChildWindow::createEdit("", { 55, 15, 255, 24 }, hwnd, SWH_EDIT_CLASS, false, ES_LEFT);
        ctrls[sw_titleEdit] = ChildWindow::createEdit("", { 55, 45, 255, 24 }, hwnd, SWH_EDIT_TITLE, false, ES_LEFT);

        ctrls[sw_cbAutoUpdateOnEdit] = ChildWindow::createCheckBox("自动更新", { 15, 79, 90, 25 }, hwnd, SWH_CB_AUTOUPDATEONEDIT);

        ctrls[sw_SearchBtn] = ChildWindow::createButton("搜索", { 185, 79, 90, 25 }, hwnd, SWH_ID_SEARCHBTN);
        ctrls[sw_SearchBtn].setEnabled(false);
        ctrls[sw_ClearBtn] = ChildWindow::createButton("清空", { 285, 79, 90, 25 }, hwnd, SWH_ID_CLEARBTN);
        ctrls[sw_ClearBtn].setEnabled(false);

        ctrls[sw_hwndEdit] = ChildWindow::createEdit("", { 55, 113, 90, 24 }, hwnd, 0, true, ES_LEFT);

        ctrls[sw_treeBtn] = ChildWindow::createButton("查看列表", { 185, 230, 90, 25 }, hwnd, SWH_ID_TREEBTN);
        ctrls[sw_treeBtn].setEnabled(false);
        ctrls[sw_propertyBtn] = ChildWindow::createButton("查看属性", { 285, 230, 90, 25 }, hwnd, SWH_ID_PROPERTYBTN);
        ctrls[sw_propertyBtn].setEnabled(false);

        // Apply high dpi scaling factor and font
        Framework::batchApplyScale(ctrls, SWH_HwndID_MAXCOUNT, m_dpi);
        font.recreateFont(m_dpi);
        Framework::batchApplyFont(ctrls, SWH_HwndID_MAXCOUNT, font);

        AutoUpdateOnEdit = Config.SearchWindow.AutoUpdateOnEdit;
        ctrls[sw_cbAutoUpdateOnEdit].setChecked(AutoUpdateOnEdit);

        break;
    }
    case WM_COMMAND:
    {
        switch(HIWORD(wParam)) {
        case BN_CLICKED: { // 按下按钮
            switch(LOWORD(wParam)) {
            case SWH_ID_TREEBTN: {
                HTREEITEM item = FindTreeViewItemByHwnd(hTreeView, targetHwnd);
                if(item) {
                    FocusControlAcrossThreads(hTreeView);
                    TreeView_SelectItem(hTreeView, item);
                }
                    else MessageBox(NULL, "Failed", "ERROR", 0);
                break;
            }
            case SWH_ID_PROPERTYBTN: {
                StartPropertyWindow(hwnd, targetHwnd);
                break;
            }
            case SWH_ID_SEARCHBTN: {
                std::string className = ctrls[sw_classEdit].getText();
                std::string wndTitle = ctrls[sw_titleEdit].getText();

                targetHwnd = FindWindow(className.c_str(), wndTitle.c_str());
                if(targetHwnd != NULL) {
                    std::stringstream tsstr; std::string tstr;
                    tsstr << std::hex << targetHwnd; tstr = tsstr.str(); tsstr.str("");
                    tstr = tstr.substr(tstr.length()-6); tsstr << "0x" << tstr;
                    ctrls[sw_hwndEdit].setText(tsstr.str()); // 显示窗口句柄（16进制）
                    tsstr.str("");

                    ctrls[sw_treeBtn].setEnabled(true);
                    ctrls[sw_propertyBtn].setEnabled(true);
                } else {
                    ctrls[sw_hwndEdit].setText("");

                    ctrls[sw_treeBtn].setEnabled(false);
                    ctrls[sw_propertyBtn].setEnabled(false);
                }
                break;
            }
            case SWH_ID_CLEARBTN: {
                ctrls[sw_classEdit].setText("");
                ctrls[sw_titleEdit].setText("");
                
                break;
            }
            case SWH_CB_AUTOUPDATEONEDIT: {
                AutoUpdateOnEdit = ctrls[sw_cbAutoUpdateOnEdit].isChecked();
                Config.SearchWindow.AutoUpdateOnEdit = AutoUpdateOnEdit;
                break;
            }}
            break;
        }
        case EN_CHANGE: { // 编辑框修改
            WORD control = LOWORD(wParam);
            // Search boxes
            if(control == SWH_EDIT_TITLE || control == SWH_EDIT_CLASS) { // 修改的是搜索框（而不是结果显示）
                std::string className = ctrls[sw_classEdit].getText();
                std::string wndTitle = ctrls[sw_titleEdit].getText();

                if(className.empty() && wndTitle.empty()) {
                    ctrls[sw_SearchBtn].setEnabled(false);
                    ctrls[sw_ClearBtn].setEnabled(false);

                    ctrls[sw_hwndEdit].setText("");
                } else { // 不同时为空
                    ctrls[sw_SearchBtn].setEnabled(true);
                    ctrls[sw_ClearBtn].setEnabled(true);

                    if(AutoUpdateOnEdit) { // 自动搜索
                        SendMessage(hwnd, WM_COMMAND, MAKEWPARAM(SWH_ID_SEARCHBTN, BN_CLICKED), (LPARAM)ctrls[sw_SearchBtn].hwnd());
                    }
                }
            }
            
            break;
        }}
        break;
    }
    case WM_DPICHANGED:
    {
        RECT* prc = (RECT*)lParam;
        SetWindowPos(hwnd, NULL, prc->left, prc->top, prc->right-prc->left, prc->bottom-prc->top, SWP_NOZORDER | SWP_NOACTIVATE);

        UINT m_dpi = LOWORD(wParam);
        // Rearrange child controls
        Framework::batchApplyScale(ctrls, SWH_HwndID_MAXCOUNT, m_dpi);
        font.recreateFont(m_dpi);
        Framework::batchApplyFont(ctrls, SWH_HwndID_MAXCOUNT, font);

        break;
    }
    case WM_CLOSE:
        DestroyWindow(hwnd);
        break;
    case WM_DESTROY:
        PostQuitMessage(0);
        break;
    default:
        return DefWindowProc(hwnd, uMsg, wParam, lParam);
    }
    return 0;
}

int SearchWindow_WorkerThread() {
    HWND searchWindow;

    SearchWindowRunning = true;

    WNDCLASSEX wcex = { 0 };
    wcex.cbSize = sizeof(WNDCLASSEX);
    wcex.style = 0;
    wcex.cbClsExtra = 0;
	wcex.cbWndExtra = 0;
    wcex.lpfnWndProc = SearchWindowProc;
    wcex.hInstance = g_hInstance;
    wcex.hCursor = LoadCursor(NULL, IDC_ARROW);
    wcex.lpszClassName = TEXT("WindowContolExClass_SearchWindow{}");
    wcex.hbrBackground = HBRUSH(COLOR_WINDOW);

    RegisterClassExA(&wcex);

    searchWindow = CreateWindowEx(
        /*WS_EX_NOREDIRECTIONBITMAP,*/ 0,
        TEXT("WindowContolExClass_SearchWindow{}"), TEXT("搜索窗口"),
        WS_OVERLAPPEDWINDOW &~ WS_SIZEBOX &~ WS_MAXIMIZEBOX,
        CW_USEDEFAULT, CW_USEDEFAULT, 400, 300,
        NULL, NULL, g_hInstance, NULL
    );
    s_searchWindowHWND = searchWindow;

    subWindows.push_back(searchWindow);

    ShowWindow(searchWindow, SW_SHOW);

    MSG msg;
    while (GetMessage(&msg, NULL, 0, 0))
    {
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }

    SearchWindowRunning = false;
    s_searchWindowHWND = nullptr;

    return (int)msg.wParam;
}


int StartSearchWindow() {
    if(!SearchWindowRunning) {
        std::thread* thrd = new std::thread(SearchWindow_WorkerThread);
        subThreadItem item { thrd, ST_SearchWindow, nullptr };
        subThreads.push_back(item);
    } else SetForegroundWindow(s_searchWindowHWND);
    return 0;
}