#include "stdafx.h"

#include <conio.h>
#include <windows.h>
#include <Shlwapi.h>

#pragma comment(lib, "shlwapi.lib")

typedef struct tagNOTIFYITEM
{
    PWSTR pszExeName;
    PWSTR pszTip;
    HICON hIcon;
    HWND hWnd;
    DWORD dwPreference;
    UINT uID;
    GUID guidItem;
} NOTIFYITEM, *PNOTIFYITEM;

class __declspec(uuid("D782CCBA-AFB0-43F1-94DB-FDA3779EACCB")) INotificationCB : public IUnknown
{
    virtual HRESULT __stdcall Notify (ULONG, NOTIFYITEM *) = 0;
};

class __declspec(uuid("FB852B2C-6BAD-4605-9551-F15F87830935")) ITrayNotify : public IUnknown
{
public:
    virtual HRESULT __stdcall RegisterCallback(INotificationCB* callback) = 0;
    virtual HRESULT __stdcall SetPreference(const NOTIFYITEM* notify_item) = 0;
    virtual HRESULT __stdcall EnableAutoTray(BOOL enabled) = 0;
};

class __declspec(uuid("D133CE13-3537-48BA-93A7-AFCD5D2053B4")) ITrayNotifyWindows8 : public IUnknown
{
public:
    virtual HRESULT __stdcall RegisterCallback(INotificationCB* callback, unsigned long*) = 0;
    virtual HRESULT __stdcall UnregisterCallback(unsigned long*) = 0;
    virtual HRESULT __stdcall SetPreference(NOTIFYITEM const*) = 0;
    virtual HRESULT __stdcall EnableAutoTray(BOOL) = 0;
    virtual HRESULT __stdcall DoAction(BOOL) = 0;
};

const CLSID CLSID_TrayNotify = {0x25DEAD04, 0x1EAC, 0x4911, {0x9E, 0x3A, 0xAD, 0x0A, 0x4A, 0xB5, 0x60, 0xFD}};

class NotificationMgr : public INotificationCB
{
public:
    NotificationMgr(){
        m_pTrayNotify = NULL;
    }

    HRESULT __stdcall QueryInterface (
        REFIID riid,
        PVOID *ppv)
    {
        if (ppv == NULL) 
            return E_POINTER;

        if (riid == __uuidof (INotificationCB)) {
            *ppv = (INotificationCB *) this;
        }else if (riid == IID_IUnknown) {
            *ppv = (IUnknown *) this;
        }else{
            return E_NOINTERFACE;
        }

        ((IUnknown *) *ppv) -> AddRef ();

        return S_OK;
    }

    ULONG __stdcall AddRef (VOID)
    {
        return 1;
    }

    ULONG __stdcall Release (VOID)
    {
        return 1;
    }
    HRESULT __stdcall Notify (ULONG Event, NOTIFYITEM * NotifyItem){
        printf("event:%d Preference:%d id:%d path:%ls\n", Event, NotifyItem->dwPreference, NotifyItem->uID, NotifyItem->pszExeName);
        if (StrStrIW(NotifyItem->pszExeName, L"qq.exe")){
            if (m_pTrayNotify){
                NotifyItem->dwPreference = 2;
                m_pTrayNotify->SetPreference(NotifyItem);
            }
        }
        return S_OK;
    }

public:
    ITrayNotify * m_pTrayNotify;
};

int _tmain(int argc, _TCHAR* argv[])
{
    ITrayNotify *pTrayNotify;
    NOTIFYITEM NotifyItem = {0};
    NotificationMgr NotiMgr;

    CoInitialize(NULL);

    //
    // for win7
    //
    
    HRESULT hr = CoCreateInstance (
       CLSID_TrayNotify,
       NULL,
       CLSCTX_LOCAL_SERVER,
       __uuidof(ITrayNotify),
       (PVOID *)&pTrayNotify);
    
    if (!SUCCEEDED(hr)){
        printf("create instance of ITrayNotify error\n");
        return -1;
    }
    
    //
    // register callback
    //
    
    NotiMgr.m_pTrayNotify = pTrayNotify;
    hr = pTrayNotify->RegisterCallback(&NotiMgr);
    
    printf("over\n");

    pTrayNotify->RegisterCallback(NULL);
    pTrayNotify->Release();
    
    _getch();

	return 0;
}