zhipuzi_pay_plugin/zhipuzi_pay_plugin/wnd/CValueWnd.cpp

#include "../pch/pch.h"
#include "CValueWnd.h"

#include "../tool/CComHelper.h"

#include "../tool/CSerialPort.h"

void CValueWnd::Notify(TNotifyUI& msg)
{
    if(msg.sType == _T("click"))
    {
        if(msg.pSender->GetName() == _T("closebtn"))
        {
            PostQuitMessage(0);
            return;
        }
    }
}

void CValueWnd::Init()
{
    SetWindowPos(m_hWnd, HWND_TOPMOST, 0, 0, 0, 0, SWP_NOMOVE | SWP_NOSIZE);

    BOOL isAdmin = CSystem::IsRunasAdmin();

    if(isAdmin)
    {
        //MessageBox(m_hWnd, L"管理员", L"111", MB_OK);
    }
    else
    {
        //MessageBox(m_hWnd, L"非管理员", L"111", MB_OK);
    }

    SetHwnd(m_hWnd);

    BOOL ret = InstallHook();

    LONG id = GetWindowLongW((HWND)0x00211108, GWL_ID);
    LONG b = id;

    TCHAR szTitle[MAX_PATH];
    LPWSTR lpWindowName = szTitle;

    ::SendMessageW((HWND)0x00081342, WM_GETTEXT, (WPARAM)(MAX_PATH), (LPARAM)lpWindowName);
    wstring a = lpWindowName;

    std::thread(&CValueWnd::ReadValue, this).detach();

    std::thread(&CValueWnd::WatchWnd, this).detach();
}

void CValueWnd::ReadValue()
{
    //读取串口的模式
    //CComHelper helper;
    //std::vector<std::wstring> comVector = helper.getComPort();

    //for (std::vector<std::wstring>::iterator it = comVector.begin(); it != comVector.end(); it++)
    //{
    //	std::wstring comName = *it;
    //	LOG_INFO("comName:" << comName.c_str());
    //}

    //int nReadType = 1;

    //if (nReadType == 1)
    //{
    //	std::wstring com_name = L"COM2";
    //	CSerialPort serial(com_name.c_str(), 2400);

    //	bool ret = serial.openComm();
    //	if (!ret)
    //	{
    //		LOG_INFO("open com failed!");
    //		return;
    //	}

    //	while (true)
    //	{
    //		//开始读取串口的数据
    //		char a[1024] = { 0 };
    //		serial.readFromComm(a, 1024);

    //		LOG_INFO("a:" << a);

    //		Sleep(200);
    //	}

    //	serial.closeComm();
    //}


    FindValueWnd(L"");
}

void CValueWnd::WatchWnd()
{
    while(true)
    {
        if(m_watchWnd != NULL)
        {
            char szTitle[MAX_PATH] = { 0 };

            LPSTR lpWindowName = szTitle;

            std::string value;

            int ret = ::SendMessageA(m_watchWnd, WM_GETTEXT, (WPARAM)(MAX_PATH), (LPARAM)lpWindowName);

            if(ret == 0)
            {
                //说明m_watchWnd已经失效了
                value = "0.00";

                m_watchWnd = NULL;
            }
            else
            {
                value = lpWindowName;
            }

            if(value != m_watchValue)
            {
                m_watchValue = value;

                SendMessage(WM_WATCH_WND_UPDATE, NULL, NULL);
            }
        }
        else
        {
            //根据上次保存的记录，来查找符合条件的窗口
            if(g_valuewnd_vector.size() > 0)
            {
                ValueWnd lastWnd = g_valuewnd_vector[0];

                //获得进程名字
                std::wstring processName = lastWnd.processName;

                //根据进程名字，找到对应的主窗口
                HWND mainWnd = GetWndHwnd(processName.c_str(), lastWnd.wndTitle);

                if(mainWnd ==  NULL)
                {
                    //说明这个进程还没有启动
                    m_watchWnd = NULL;
                }
                else
                {
                    //根据Z序查找目标窗口
                    char szTitle[MAX_PATH] = { 0 };
                    int nMaxCount = MAX_PATH;

                    LPSTR lpWindowName = szTitle;

                    GetWindowTextA(mainWnd, lpWindowName, nMaxCount);

					if (string(lpWindowName) != lastWnd.wndTitle)
					{
						//这次根据进程名字找到的主窗口，窗口名字和上次的不一样，无效
						m_watchWnd = NULL;
					}
					else
					{
						ValueWnd newWnd;
						newWnd.wndTitle = lpWindowName;
						newWnd.TopWnd = mainWnd;
						newWnd.z_order = lastWnd.z_order;
						newWnd.processName = lastWnd.processName;
						newWnd.selfWnd = NULL;

						g_zorder = 0;

						EnumChildWindows(mainWnd, EnumMainwndChildProc, (LPARAM)(&newWnd));

						if (newWnd.selfWnd != NULL)
						{
							//找到了
							m_watchWnd = newWnd.selfWnd;
						}
					}
                }
            }
        }

        Sleep(200);
    }
}

LRESULT CValueWnd::OnCreate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    LONG styleValue = ::GetWindowLong(*this, GWL_STYLE);
    styleValue &= ~WS_CAPTION;
    ::SetWindowLong(*this, GWL_STYLE, styleValue | WS_CLIPSIBLINGS | WS_CLIPCHILDREN);

    // 把自己的窗口句柄与窗口绘制管理器挂接在一起
    m_pm.Init(m_hWnd);

    m_pm.AddPreMessageFilter(this);

    CDialogBuilder builder;

    CControlUI* pRoot = builder.Create(_T("value.xml"), (UINT)0, NULL, &m_pm);
    ASSERT(pRoot && "Failed to parse XML");

    // 把这些控件绘制到本窗口上
    m_pm.AttachDialog(pRoot);

    // 把自己加入到CPaintManagerUI的m_aNotifiers数组中，用于处理Notify函数
    m_pm.AddNotifier(this);

    Init();

    return 0;
}

LRESULT CValueWnd::OnClose(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    bHandled = FALSE;
    return 0;
}

/*
 *这个是窗口被销毁的时候调用的
 **/
LRESULT CValueWnd::OnDestroy(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    bHandled = FALSE;
    return 0;
}

LRESULT CValueWnd::OnNcActivate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    if(::IsIconic(*this))
    {
        bHandled = FALSE;
    }

    return (wParam == 0) ? TRUE : FALSE;
}

LRESULT CValueWnd::OnNcCalcSize(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    return 0;
}

LRESULT CValueWnd::OnNcPaint(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    //在这里设置焦点才有用
    CEditUI* pAccountEdit = static_cast<CEditUI*>(m_pm.FindControl(_T("accountedit")));

    if(pAccountEdit)
    {
        pAccountEdit->SetFocus();
    }

    return 0;
}

LRESULT CValueWnd::OnNcHitTest(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    POINT pt;
    pt.x = GET_X_LPARAM(lParam);
    pt.y = GET_Y_LPARAM(lParam);
    ::ScreenToClient(*this, &pt);

    RECT rcClient;
    ::GetClientRect(*this, &rcClient);

    RECT rcCaption = m_pm.GetCaptionRect();

    if(pt.x >= rcClient.left + rcCaption.left && pt.x < rcClient.right - rcCaption.right \
            && pt.y >= rcCaption.top && pt.y < rcCaption.bottom)
    {
        CControlUI* pControl = static_cast<CControlUI*>(m_pm.FindControl(pt));

        if(pControl && _tcscmp(pControl->GetClass(), DUI_CTR_BUTTON) != 0)
        {
            return HTCAPTION;
        }
    }

    return HTCLIENT;
}

LRESULT CValueWnd::OnKexianUpdate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    //收到了客显数据消息，读取客显数据
    char data[MAX_DATA_LENGTH] = { 0 };
    int data_length = 0;

    HookGetData(data, &data_length);

    if(data_length == 0)
    {
        return true;
    }

    UpdateKexian(data, data_length);

    return true;
}

LRESULT CValueWnd::OnWatchWndUpdate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    CLabelUI* valueLabel = static_cast<CLabelUI*>(m_pm.FindControl(_T("value")));

    std::string money_show = CLewaimaiString::DoubleToString(atof(m_watchValue.c_str()), 2);
    valueLabel->SetText(CLewaimaiString::UTF8ToUnicode(money_show).c_str());

    return TRUE;
}

LRESULT CValueWnd::HandleMessage(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
    LRESULT lRes = 0;
    BOOL bHandled = TRUE;

    switch(uMsg)
    {
    case WM_CREATE:
        lRes = OnCreate(uMsg, wParam, lParam, bHandled);
        break;

    case WM_CLOSE:
        lRes = OnClose(uMsg, wParam, lParam, bHandled);
        break;

    case WM_DESTROY:
        lRes = OnDestroy(uMsg, wParam, lParam, bHandled);
        break;

    case WM_NCACTIVATE:
        lRes = OnNcActivate(uMsg, wParam, lParam, bHandled);
        break;

    case WM_NCCALCSIZE:
        lRes = OnNcCalcSize(uMsg, wParam, lParam, bHandled);
        break;

    case WM_NCPAINT:
        lRes = OnNcPaint(uMsg, wParam, lParam, bHandled);
        break;

    case WM_NCHITTEST:
        lRes = OnNcHitTest(uMsg, wParam, lParam, bHandled);
        break;

    case WM_HOOK_MESSAGE:
        lRes = OnKexianUpdate(uMsg, wParam, lParam, bHandled);
        break;

    case WM_WATCH_WND_UPDATE:
        lRes = OnWatchWndUpdate(uMsg, wParam, lParam, bHandled);
        break;

    default:
        bHandled = FALSE;
    }

    if(bHandled)
    {
        return lRes;
    }

    if(m_pm.MessageHandler(uMsg, wParam, lParam, lRes))
    {
        return lRes;
    }

    return CWindowWnd::HandleMessage(uMsg, wParam, lParam);
}

LRESULT CValueWnd::MessageHandler(UINT uMsg, WPARAM wParam, LPARAM lParam, bool& bHandled)
{
    return false;
}

void CValueWnd::UpdateKexian(const char* data, int length)
{
    std::string new_data = data;

    std::string show_command = "\x1bQA";
    std::string type_command = "\x1bs";

    std::string money;

    if(new_data.find(show_command) == new_data.npos)
    {
        return;
    }

    int nPos = new_data.find(show_command);

    int nTypePos = new_data.find(type_command, nPos + 1);

    if(nTypePos == new_data.npos)
    {
        //没有类型信息，有些收银机（比如思迅天店）就是不传类型信息，这种只有传什么显示什么了
        int nTabPos = new_data.find("\r", nPos + 1);

        if(nTabPos == new_data.npos)
        {
            //制表符也没有
            return;
        }

        money = new_data.substr(nPos + 3, nTabPos - nPos - 3);

        CLabelUI* valueLabel = static_cast<CLabelUI*>(m_pm.FindControl(_T("value")));

        std::string money_show = CLewaimaiString::DoubleToString(atof(money.c_str()), 2);
        valueLabel->SetText(CLewaimaiString::UTF8ToUnicode(money_show).c_str());
    }
    else
    {
        std::string value_type = new_data.substr(nTypePos + 2, 1);
        money = new_data.substr(nPos + 3, nTypePos - nPos - 3 - 1);

        if(value_type == "2")
        {
            //这个表示是总计的，那么就要刷新金额显示
            CLabelUI* valueLabel = static_cast<CLabelUI*>(m_pm.FindControl(_T("value")));

            std::string money_show = CLewaimaiString::DoubleToString(atof(money.c_str()), 2);
            valueLabel->SetText(CLewaimaiString::UTF8ToUnicode(money_show).c_str());
        }
    }

    return;
}

void CValueWnd::FindValueWnd(std::wstring value)
{
    //先查找子窗口
    EnumWindows(EnumWindowsProc, NULL);

    //展示查找结果
    int nCount = (int)g_valuewnd_vector.size();

    if(nCount > 0)
    {
        //开始监听这个窗口的内容变化
        ValueWnd selectWnd = g_valuewnd_vector[0];

        m_watchWnd = selectWnd.selfWnd;
    }
}