MainFrm.cpp

/*----------------------------------------------------------
    MainFrmクラス
                                    2002/11/16 (c)Keizi
----------------------------------------------------------*/

#include "stdafx.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

#include <locale.h>

#include "MainFrm.h"
#include "Charu3.h"
#include "log.h"
#include "resource.h"

//---------------------------------------------------
// CMainFrame
//---------------------------------------------------

IMPLEMENT_DYNAMIC(CMainFrame, CFrameWnd)

BEGIN_MESSAGE_MAP(CMainFrame, CFrameWnd)
    //{{AFX_MSG_MAP(CMainFrame)
    ON_WM_CREATE()
    ON_WM_SETFOCUS()
    ON_COMMAND(IDM_EXIT, OnExit)
    ON_WM_TIMER()
    ON_WM_QUERYENDSESSION()
    //}}AFX_MSG_MAP
END_MESSAGE_MAP()


//---------------------------------------------------
//関数名	CMainFrame
//機能		コンストラクタ
//---------------------------------------------------
CMainFrame::CMainFrame() : m_hIcon(nullptr), m_hStopIcon(nullptr), m_nIcon(), m_hActive(nullptr)
{
    setlocale(LC_ALL, "");

    //メニューを原則有効化
    CFrameWnd::m_bAutoMenuEnable = FALSE;

    m_PopupMenu.LoadMenu(MAKEINTRESOURCE(IDR_MAINFRAME));
}

//---------------------------------------------------
//関数名	~CMainFrame
//機能		デストラクタ
//---------------------------------------------------
CMainFrame::~CMainFrame()
{
    Shell_NotifyIcon(NIM_DELETE, &m_nIcon);
}

//---------------------------------------------------
//関数名	OnCreate(LPCREATESTRUCT lpCreateStruct)
//機能		初期化
//---------------------------------------------------
int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
{
    if (CFrameWnd::OnCreate(lpCreateStruct) == -1)
        return -1;
    // フレームのクライアント領域全体を占めるビューを作成します。
    if (!m_wndView.Create(NULL, NULL, AFX_WS_DEFAULT_VIEW, CRect(0, 0, 0, 0), this, AFX_IDW_PANE_FIRST, NULL)) {
        TRACE0("Failed to create view window\n");
        return -1;
    }
    //アイコン処理
#if true
    m_hIcon = (HICON)LoadImage(theApp.m_hInstance, MAKEINTRESOURCE(IDI_RUN256), IMAGE_ICON, 16, 16, 0);//スモールアイコン
    m_hStopIcon = (HICON)LoadImage(theApp.m_hInstance, MAKEINTRESOURCE(IDI_STOP256), IMAGE_ICON, 16, 16, 0);//スモールアイコン
#else
    m_hIcon = (HICON)LoadImage(theApp.m_hInstance, MAKEINTRESOURCE(IDI_RUN), IMAGE_ICON, 16, 16, 0);//スモールアイコン
    m_hStopIcon = (HICON)LoadImage(theApp.m_hInstance, MAKEINTRESOURCE(IDI_STOP), IMAGE_ICON, 16, 16, 0);//スモールアイコン
#endif
    // 通知領域アイコンを設定する
    m_nIcon.cbSize = sizeof(NOTIFYICONDATA);
    m_nIcon.uID = 4;
    m_nIcon.hWnd = m_hWnd;
    m_nIcon.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
    m_nIcon.hIcon = m_hStopIcon;
    m_nIcon.uCallbackMessage = WM_TASKTRAY;

    _tcscpy_s(m_nIcon.szTip, _T("Charu3"));
    Shell_NotifyIcon(NIM_ADD, &m_nIcon);

    return 0;
}

//---------------------------------------------------
//関数名	CheckTrayPos()
//機能		カーソルがタスクトレイに重なってるか判定
//---------------------------------------------------
bool CMainFrame::CheckTrayPos()
{
    HWND hTaskBarWnd;
    HWND hTrayNotifyWnd;
    RECT rtTrayNotify;
    POINT ptCursolPos;
    bool isRet = false;

    GetCursorPos(&ptCursolPos);

    // タスクバーのインジケータ領域の矩形を得る
    hTaskBarWnd = ::FindWindow(_T("Shell_TrayWnd"), NULL);
    hTrayNotifyWnd = FindWindowEx(hTaskBarWnd, NULL, _T("TrayNotifyWnd"), NULL)->GetSafeHwnd();
    if (hTrayNotifyWnd == NULL)	return FALSE;

    ZeroMemory(&rtTrayNotify, sizeof(rtTrayNotify));
    ::GetWindowRect(hTrayNotifyWnd, &rtTrayNotify);

    if (ptCursolPos.x > rtTrayNotify.left && ptCursolPos.x < rtTrayNotify.right
        && ptCursolPos.y > rtTrayNotify.top && ptCursolPos.y < rtTrayNotify.bottom)
        isRet = true;

    return isRet;
}

//---------------------------------------------------
//関数名	PreCreateWindow(CREATESTRUCT& cs)
//機能		初期化前処理
//---------------------------------------------------
BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs)
{
    if (!CFrameWnd::PreCreateWindow(cs))	return FALSE;

    cs.cx = 185;
    cs.cy = 256;
    cs.x = -500;
    cs.y = -500;

    cs.style = WS_TILED | WS_MINIMIZEBOX | WS_SYSMENU;
    cs.dwExStyle &= ~WS_EX_CLIENTEDGE;
    cs.lpszClass = AfxRegisterWndClass(0);

    return TRUE;
}

//---------------------------------------------------
//関数名	OnClipboardChanged()
//機能		クリップボードの変更処理
//---------------------------------------------------
void CMainFrame::OnClipboardChanged()
{
    // Check the program that owns the currently active window if stealth program is registered.
    bool isStealth = false;
    if (theApp.m_ini.m_stealth.size() > 0) {
        const HWND hActiveWindow = ::GetForegroundWindow();
        DWORD actorBufSize = 32768;
        LPWSTR actorPathnameBuf = new WCHAR[actorBufSize];
        DWORD dwProcessId;
        ::GetWindowThreadProcessId(hActiveWindow, &dwProcessId);
        const HANDLE hProcess = ::OpenProcess(PROCESS_QUERY_LIMITED_INFORMATION, FALSE, dwProcessId);
        if (::QueryFullProcessImageName(hProcess, 0, actorPathnameBuf, &actorBufSize)) {
            CString csActorPath = CString(actorPathnameBuf);
            LPWSTR p = _tcsrchr(actorPathnameBuf, _T('\\'));
            CString csActorName = p ? CString(p + 1) : csActorPath;
            for (const CString& s : theApp.m_ini.m_stealth) {
                CString csStealth = s;
                csStealth.Replace(_T('/'), _T('\\'));
                if (csStealth.Find(_T('\\')) < 0) {
                    isStealth = csActorName.CompareNoCase(csStealth) == 0;
                }
                else {
                    isStealth = csActorPath.CompareNoCase(csStealth) == 0;
                }
                if (isStealth) {
                    break;
                }
            }
        }
        delete[] actorPathnameBuf;
    }

    CString strClipboard;
    if (!isStealth) {
        if (theApp.m_ini.m_nClipboardOpenDelay > 0) {
            Sleep(theApp.m_ini.m_nClipboardOpenDelay);
        }
        theApp.GetClipboardText(strClipboard);
        theApp.Record(strClipboard);
    }

    if (theApp.m_ini.m_bShowClipboardInTooltipOfNofifyIcon) {
        if (isStealth) {
            _tcscpy_s(m_nIcon.szTip, _T("**** Stealthed ****"));
        }
        else {
            strClipboard = strClipboard.Left(1024);
            strClipboard.Replace(_T('\t'), _T(' ')); // Replace tab with space.
            _tcscpy_s(m_nIcon.szTip, strClipboard.Left(63));
        }
        if (!Shell_NotifyIcon(NIM_MODIFY, &m_nIcon)) {
            Shell_NotifyIcon(NIM_ADD, &m_nIcon);
        }
    }

}

//---------------------------------------------------
//関数名	SwitchNotificationAreaIcon(bool stockmode)
//機能		トレイアイコン切り替え
//---------------------------------------------------
void CMainFrame::SwitchNotificationAreaIcon(bool stockmode)
{
    m_nIcon.hIcon = stockmode ? m_hIcon : m_hStopIcon;
    if (!Shell_NotifyIcon(NIM_MODIFY, &m_nIcon)) {
        Shell_NotifyIcon(NIM_ADD, &m_nIcon);
    }
}

//---------------------------------------------------
// CMainFrame クラスの診断
//---------------------------------------------------

#ifdef _DEBUG
void CMainFrame::AssertValid() const
{
    CFrameWnd::AssertValid();
}

void CMainFrame::Dump(CDumpContext& dc) const
{
    CFrameWnd::Dump(dc);
}

#endif //_DEBUG

//---------------------------------------------------
//関数名	OnSetFocus(CWnd* pOldWnd)
//機能		フォーカスを与える
//---------------------------------------------------
void CMainFrame::OnSetFocus(CWnd* pOldWnd)
{
    // ビュー ウィンドウにフォーカスを与えます。
    m_wndView.SetFocus();
}

//---------------------------------------------------
//関数名	OnCmdMsg(UINT nID, int nCode, void* pExtra, AFX_CMDHANDLERINFO* pHandlerInfo)
//機能		コマンド処理
//---------------------------------------------------
BOOL CMainFrame::OnCmdMsg(UINT nID, int nCode, void* pExtra, AFX_CMDHANDLERINFO* pHandlerInfo)
{
    // ビューに最初にコマンドを処理する機会を与えます。
    if (m_wndView.OnCmdMsg(nID, nCode, pExtra, pHandlerInfo))
        return TRUE;

    // 処理されなかった場合にはデフォルトの処理を行います。
    return CFrameWnd::OnCmdMsg(nID, nCode, pExtra, pHandlerInfo);
}


//---------------------------------------------------
//関数名	WindowProc(UINT message, WPARAM wParam, LPARAM lParam)
//機能		メッセージ処理
//---------------------------------------------------
LRESULT CMainFrame::WindowProc(UINT message, WPARAM wParam, LPARAM lParam)
{
    // tasktray messages
    if (WM_TASKTRAY == message && theApp.GetPhase() == PHASE_IDOL) {
        if (WM_RBUTTONUP == lParam) {
            // menu
            POINT point;
            GetCursorPos(&point);
            this->SetForegroundWindow();
            m_PopupMenu.GetSubMenu(0)->TrackPopupMenu(TPM_BOTTOMALIGN | TPM_LEFTALIGN | TPM_LEFTBUTTON, point.x, point.y, this);
            PostMessage(0, 0, 0);
            return FALSE; // to suppress notification area menu
        }
        else if (WM_LBUTTONDOWN == lParam) {
            theApp.OnClick(m_hActive);
        }
        else if (CheckTrayPos()) {
            m_hActive = ::GetForegroundWindow();
            theApp.CheckFocusInfo(m_hActive);
        }
    }
    // clipboard update notification
    if (WM_CLIPBOARDUPDATE == message) {
        if (theApp.m_ini.m_bDebug) {
            LOG(_T("WM_CLIPBOARDUPDATE wParam:%x lParam:%x"), wParam, lParam);
        }
        OnClipboardChanged();
    }

    return CFrameWnd::WindowProc(message, wParam, lParam);
}

//---------------------------------------------------
//関数名	OnExit()
//機能		終了処理
//---------------------------------------------------
void CMainFrame::OnExit()
{
    theApp.OnExit();
    Shell_NotifyIcon(NIM_DELETE, &m_nIcon);
    CFrameWnd::DestroyWindow();
}

BOOL CMainFrame::OnQueryEndSession()
{
    if (!CFrameWnd::OnQueryEndSession()) {
        return FALSE;
    }
    theApp.OnExit();
    Shell_NotifyIcon(NIM_DELETE, &m_nIcon);

    return TRUE;
}