#pragma once
#include <vector>

struct TaskInfo
{
    // 程序名稱
    CString strProgramName;

    // 窗口句柄
    HWND hWnd;
};

class TaskbarInfo
{
public:
    // 回調函數
    static BOOL CALLBACK EnumWindowProc(HWND hWnd, LPARAM lParam);
public:

    // 記錄在任務欄打開的程序
    std::vector<TaskInfo> m_TaskbarInfoArr;

    // 頂層窗口句柄
    HWND m_hWnd;

public:
    TaskbarInfo();
    ~TaskbarInfo();

    // 獲得在任務欄打開程序信息
    void GetTaskbarInof (void);

    // 獲得頂層窗口句柄，失敗返回NULL
    HWND GetTopWnd (void);
};

#include "stdafx.h"        // 如果編譯出錯，請刪除這句
#include "OpenProgramOnTaskbar.h"

#include "Psapi.h"
#pragma comment(lib, "Psapi.lib") 

TaskbarInfo::TaskbarInfo()
{

}

TaskbarInfo::~TaskbarInfo()
{

}

void TaskbarInfo::GetTaskbarInof (void)
{
    EnumWindows(EnumWindowProc, (LPARAM)this);
}

BOOL CALLBACK TaskbarInfo::EnumWindowProc(HWND hWnd, LPARAM lParam)
{
    TaskbarInfo * pTaskbarInfo = (TaskbarInfo*)lParam;

    TCHAR szWindow[256]={0};
    TaskInfo taskInfo;

    ::GetWindowText(hWnd, szWindow, 255); //獲取窗口標題
    if ( ::IsWindow(hWnd) &&
        ::IsWindowVisible(hWnd) &&
        (::GetWindowLong(hWnd, GWL_EXSTYLE)&WS_EX_TOOLWINDOW)!=WS_EX_TOOLWINDOW &&
        ::GetWindowLong(hWnd, GWL_HWNDPARENT)==0 )
    {
        DWORD dwPID;  //保存進程標識符

        GetWindowThreadProcessId(hWnd, &dwPID);  //接受一個窗口句柄。dwPID保存窗口的創建者的進程標識符，GetWindowThreadProcessId返回值是該創建者的線程標識符
        HANDLE hBrowser = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwPID);  //打開一個已存在的進程對象,并返回進程的句柄，這就是我們要的進程句柄了

        WCHAR exePath[256];
        memset(exePath, 0, 256);
        //獲取程序的path，并保存到exePath
        GetModuleFileNameEx(hBrowser, NULL, exePath, MAX_PATH);
        CString strTemp = exePath;
        strTemp.MakeLower();

        // 過濾在桌面打開的文件夾等
        if (!strTemp.IsEmpty() && strTemp.Find(_T("explorer.exe")) == -1)
        {
            taskInfo.strProgramName = szWindow;
            taskInfo.hWnd = hWnd;
            pTaskbarInfo->m_TaskbarInfoArr.push_back(taskInfo);
            memset(szWindow, 0, 256);
        }
    }
    return TRUE;
}

HWND TaskbarInfo::GetTopWnd (void)
{
    DWORD dwPID;  //保存進程標識符
    HWND hWnd = NULL;
    hWnd = ::GetForegroundWindow();

    GetWindowThreadProcessId(hWnd, &dwPID);  //接受一個窗口句柄。dwPID保存窗口的創建者的進程標識符，GetWindowThreadProcessId返回值是該創建者的線程標識符
    HANDLE hBrowser = OpenProcess(PROCESS_ALL_ACCESS, FALSE, dwPID);  //打開一個已存在的進程對象,并返回進程的句柄，這就是我們要的進程句柄了

    WCHAR exePath[256];
    memset(exePath, 0, 256);
    //獲取程序的path，并保存到exePath
    GetModuleFileNameEx(hBrowser, NULL, exePath, MAX_PATH);
    CString strTemp = exePath;
    strTemp.MakeLower();

    if (!strTemp.IsEmpty() && strTemp.Find(_T("explorer.exe")) == -1)
    {
        m_hWnd = hWnd;
        if (m_hWnd != NULL)
        {
            return m_hWnd;
        }
    }

    return NULL;
}