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()
{
	//初始化窗口位置
    InitWndPos();

	//添加托盘图标
	AddTrayIcon();

	//初始化菜单窗口
	pMenu = new CMenuWnd();
	pMenu->Create(m_hWnd, L"智铺子收银插件菜单", WS_POPUP, WS_EX_TOOLWINDOW);
	pMenu->ShowWindow(false);

	//先暂停Hook工作（因为不一定是串口模式）
	::SetHookStatus(false);

	//再安装钩子
    BOOL ret = InstallHook();

	//开始启动监听
	RestartWatch();

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

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

    //开始测试ocr
    tesseract::TessBaseAPI tess;

    if(tess.Init("./tessdata", "eng"))
    {
        std::cout << "OCRTesseract: Could not initialize tesseract." << std::endl;
        return ;
    }

    // setup
    tess.SetPageSegMode(tesseract::PageSegMode::PSM_AUTO);
    tess.SetVariable("save_best_choices", "T");

    // read image
    auto pixs = pixRead("D:\\3.png");
    if(!pixs)
    {
        std::cout << "Cannot open input file: " << "D:\1.png" << std::endl;
        return ;
    }

    // recognize
    tess.SetImage(pixs);
    tess.Recognize(0);

    // get result and delete[] returned char* string
    std::string aaa = tess.GetUTF8Text();

    // cleanup
    tess.Clear();
    pixDestroy(&pixs);*/
}

void CValueWnd::RestartWatch()
{
	std::string system_setting_jinezhuaqu_setting_type = CSetting::GetParam("system_setting_jinezhuaqu_setting_type");

	m_nWatchType = atoi(system_setting_jinezhuaqu_setting_type.c_str());

	if (m_nWatchType == 1)
	{
		//更新监听的端口号
		std::string system_setting_jinezhuaqu_setting_chuankou_num = CSetting::GetParam("system_setting_jinezhuaqu_setting_chuankou_num");
		if (system_setting_jinezhuaqu_setting_chuankou_num == "")
		{
			//参数错误
			return;
		}

		std::wstring ws_system_setting_jinezhuaqu_setting_chuankou_num = CLewaimaiString::UTF8ToUnicode(system_setting_jinezhuaqu_setting_chuankou_num);

		//设置窗口句柄
		::SetHwnd(m_hWnd);

		//设置串口端口号
		//统一成大写
		std::transform(system_setting_jinezhuaqu_setting_chuankou_num.begin(), system_setting_jinezhuaqu_setting_chuankou_num.end(), system_setting_jinezhuaqu_setting_chuankou_num.begin(), ::toupper);
		std::transform(ws_system_setting_jinezhuaqu_setting_chuankou_num.begin(), ws_system_setting_jinezhuaqu_setting_chuankou_num.end(), ws_system_setting_jinezhuaqu_setting_chuankou_num.begin(), ::toupper);

		::SetComNum(system_setting_jinezhuaqu_setting_chuankou_num.c_str(), ws_system_setting_jinezhuaqu_setting_chuankou_num.c_str(), system_setting_jinezhuaqu_setting_chuankou_num.size());

		//开始监听工作
		::SetHookStatus(true);

		//暂停其他监听的
	}
}

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::OnLButtonDown(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
/*
	//获取鼠标点击位置
	int xPos = LOWORD(lParam);
	int yPos = HIWORD(lParam);

	//將位置转换为屏幕坐标
	POINT pt = { xPos,yPos };
	ClientToScreen(m_hWnd, &pt);

	RECT menuWndRect = { 0 };
	::GetWindowRect(pMenu->GetHWND(), &menuWndRect);

	if (pt.x <= menuWndRect.left || pt.x >= menuWndRect.right
		|| pt.y <= menuWndRect.top || pt.y >= menuWndRect.bottom)
	{
		pMenu->ShowWindow(false);
	}

	bHandled = FALSE;
	return 0;*/

	bHandled = FALSE;
	return 0;
}

LRESULT CValueWnd::OnRButtonDown(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
/*
	int xPos = LOWORD(lParam);
	int yPos = HIWORD(lParam);

	POINT pt = { xPos,yPos };

	//可能是因为整个窗口都是标题区域，这里不需要做转换
	//ClientToScreen(m_hWnd, &pt);

	pMenu->ShowWindow(true);

	::SetWindowPos(pMenu->GetHWND(), NULL, pt.x, pt.y, 0, 0, SWP_NOSIZE | SWP_NOACTIVATE);

	bHandled = FALSE;
	return 0;*/

	ShowMenu();

	bHandled = true;
	return 0;
}

LRESULT CValueWnd::OnKexianUpdate(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
	//如果当前不是串口模式，直接忽略
	if (m_nWatchType != 1)
	{
		bHandled = TRUE;
		return true;
	}

    //收到了客显数据消息，读取客显数据
    char data[MAX_DATA_LENGTH] = { 0 };
    int data_length = 0;

    HookGetData(data, &data_length);

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

    UpdateKexian(data, data_length);

	bHandled = TRUE;
    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_NCLBUTTONDOWN:
		lRes = OnLButtonDown(uMsg, wParam, lParam, bHandled);
		break;
	case WM_NCRBUTTONDOWN:
		lRes = OnRButtonDown(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_SHOWTASK:
        lRes = OnTrayIcon(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::AddTrayIcon()
{
    memset(&m_trayIcon, 0, sizeof(NOTIFYICONDATA));

    m_trayIcon.cbSize = sizeof(NOTIFYICONDATA);
    m_trayIcon.hIcon = ::LoadIcon(GetModuleHandle(NULL), MAKEINTRESOURCE(IDI_ICON_DUILIB));
    m_trayIcon.hWnd = m_hWnd;
    lstrcpy(m_trayIcon.szTip, _T("智铺子收银插件"));
    m_trayIcon.uCallbackMessage = WM_SHOWTASK;
    m_trayIcon.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;

    Shell_NotifyIcon(NIM_ADD, &m_trayIcon);
}

LRESULT CValueWnd::OnTrayIcon(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
{
    //如果在图标中单击左键则还原
    if(lParam == WM_LBUTTONDOWN)
    {

    }

    //如果在图标中单击右键则弹出声明式菜单
    if(lParam == WM_RBUTTONDOWN)
    {
		ShowMenu();
    }

    bHandled = true;
    return 0;
}

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

	//清屏命令
	std::string clear_command = "\f";

	//4个信号灯全灭
	std::string quanmie_command = "\x1bs0";


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

	if (new_data == clear_command || new_data == quanmie_command)
	{
		//清屏命令，或者是熄灭灯的命令
		return;
	}

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

	std::string money;

    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());
        }
		else
		{
			//这种情况传输的是一些找零、单价之类的信息，不更新客显金额
		}
    }

    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;
    }
}

void CValueWnd::InitWndPos()
{
    ASSERT(::IsWindow(m_hWnd));
    ASSERT((GetWindowStyle(m_hWnd)&WS_CHILD) == 0);
    RECT rcDlg = { 0 };
    ::GetWindowRect(m_hWnd, &rcDlg);
    RECT rcArea = { 0 };
    RECT rcCenter = { 0 };
    HWND hWnd = *this;
    HWND hWndParent = ::GetParent(m_hWnd);
    HWND hWndCenter = ::GetWindowOwner(m_hWnd);
    if(hWndCenter != NULL)
    {
        hWnd = hWndCenter;
    }

    // 处理多显示器模式下屏幕居中
    MONITORINFO oMonitor = {};
    oMonitor.cbSize = sizeof(oMonitor);
    ::GetMonitorInfo(::MonitorFromWindow(hWnd, MONITOR_DEFAULTTONEAREST), &oMonitor);
    rcArea = oMonitor.rcWork;

    if(hWndCenter == NULL)
    {
        rcCenter = rcArea;
    }
    else
    {
        ::GetWindowRect(hWndCenter, &rcCenter);
    }

    int DlgWidth = rcDlg.right - rcDlg.left;
    int DlgHeight = rcDlg.bottom - rcDlg.top;

    // Find dialog's upper left based on rcCenter
    int xLeft = rcCenter.right - DlgWidth - 60;
    int yTop = 100;

    // The dialog is outside the screen, move it inside
    if(xLeft < rcArea.left)
    {
        xLeft = rcArea.left;
    }
    else if(xLeft + DlgWidth > rcArea.right)
    {
        xLeft = rcArea.right - DlgWidth;
    }
    if(yTop < rcArea.top)
    {
        yTop = rcArea.top;
    }
    else if(yTop + DlgHeight > rcArea.bottom)
    {
        yTop = rcArea.bottom - DlgHeight;
    }

    ::SetWindowPos(m_hWnd, HWND_TOPMOST, xLeft, yTop, 0, 0, SWP_NOSIZE | SWP_NOACTIVATE);
}

void CValueWnd::ShowMenu()
{
	//获取鼠标坐标
	POINT pt;
	GetCursorPos(&pt);

	//右击后点别地可以清除“右击出来的菜单”
	SetForegroundWindow(m_hWnd);

	//托盘菜单,win32程序使用的是HMENU，如果是MFC程序可以使用CMenu
	HMENU hMenu;

	//生成托盘菜单
	hMenu = CreatePopupMenu();

	//添加菜单,关键在于设置的一个标识符  WM_ONCLOSE 点击后会用到
	AppendMenu(hMenu, MF_STRING, WM_MENU_SYSTEM_SETTING, _T("系统设置"));
	AppendMenu(hMenu, MF_STRING, WM_ONCLOSE, _T("退出"));

	//弹出菜单,并把用户所选菜单项的标识符返回
	int cmd = TrackPopupMenu(hMenu, TPM_RETURNCMD, pt.x, pt.y, NULL, m_hWnd, NULL);

	//如果标识符是WM_ONCLOSE则关闭
	if (cmd == WM_ONCLOSE)
	{
		//退出程序
		::PostQuitMessage(0);
	}
	if (cmd == WM_MENU_SYSTEM_SETTING)
	{
		if (m_is_show_setting == true)
		{
			return;
		}

		CSystemSettingWnd* pSystemSettingFrame = new CSystemSettingWnd(this);

		if (pSystemSettingFrame != NULL)
		{
			m_is_show_setting = true;

			pSystemSettingFrame->Create(NULL, _T(""), UI_WNDSTYLE_DIALOG, WS_EX_TOOLWINDOW);
			pSystemSettingFrame->SetIcon(IDI_ICON_DUILIB);
			pSystemSettingFrame->CenterWindow();
			UINT ret = pSystemSettingFrame->ShowModal();

			m_is_show_setting = false;

			if (ret == IDOK)
			{
				
			}
			else if (ret == IDCANCEL)
			{

			}
		}
	}
}