// NotifyIcon.cpp: CNotifyIcon クラスのインプリメンテーション
//
//////////////////////////////////////////////////////////////////////


#include "StdAfx.h"
#include "NotifyIcon.h"


#define WM_NOTIFYICON	(WM_USER + 101UL)


CNotifyIcon::CNotifyIcon(INotifyIconHandler *pEventHandler)
	: m_pEventHandler(pEventHandler)
	, m_hHostHwnd(NULL)
	, m_OldHostWndProc(NULL)
{

}

CNotifyIcon::~CNotifyIcon(void)
{
	RemoveIcon();
}

const BOOL CNotifyIcon::AddIcon(const HWND hHwnd, const HICON hIcon, LPCTSTR lpszTip)
{
	m_hHostHwnd = hHwnd;

	// タスクトレイにアイコン追加
	NOTIFYICONDATA IconData;
	IconData.cbSize = sizeof(IconData);
	IconData.uID = 1UL;
	IconData.hWnd = hHwnd;
	IconData.uFlags = NIF_MESSAGE | NIF_ICON | ((lpszTip)? NIF_TIP : 0UL);
	IconData.hIcon = hIcon;
	IconData.uCallbackMessage = WM_NOTIFYICON;
	IconData.uVersion = NOTIFYICON_VERSION;

	if(lpszTip)::lstrcpy(IconData.szTip, lpszTip);
	if(!::Shell_NotifyIcon(NIM_ADD, &IconData))return FALSE;
	if(!::Shell_NotifyIcon(NIM_SETVERSION, &IconData))return FALSE;
	
	// ホストウィンドウをサブクラス化する
	if(!(m_OldHostWndProc = (WNDPROC)::SetWindowLong(hHwnd, GWL_WNDPROC, (LONG)CNotifyIcon::HostWndProcRaw)))return FALSE;
	::SetWindowLong(hHwnd, GWL_USERDATA, (LONG)this);

	return TRUE;
}

const BOOL CNotifyIcon::RemoveIcon(void)
{
	if(!m_hHostHwnd)return FALSE;

	// サブクラス化解除	
	::SetWindowLong(m_hHostHwnd, GWL_WNDPROC, (LONG)m_OldHostWndProc);

	// タスクトレイからアイコンを削除
	NOTIFYICONDATA IconData;
	IconData.cbSize = sizeof(IconData);
	IconData.uID = 1UL;
	IconData.hWnd = m_hHostHwnd;
	IconData.uFlags = 0UL;

	m_hHostHwnd = NULL;

	return ::Shell_NotifyIcon(NIM_DELETE, &IconData);
}

const BOOL CNotifyIcon::SetIcon(const HICON hIcon)
{
	if(!m_hHostHwnd || !hIcon)return FALSE;

	// タスクトレイのアイコン変更
	NOTIFYICONDATA IconData;
	IconData.cbSize = sizeof(IconData);
	IconData.uID = 1UL;
	IconData.uFlags = NIF_ICON;
	IconData.hWnd = m_hHostHwnd;
	IconData.hIcon = hIcon;

	return ::Shell_NotifyIcon(NIM_MODIFY, &IconData);
}

const BOOL CNotifyIcon::SetTip(LPCTSTR lpszTip)
{
	if(!m_hHostHwnd || !lpszTip)return FALSE;

	// タスクトレイのツールチップテキスト変更
	NOTIFYICONDATA IconData;
	IconData.cbSize = sizeof(IconData);
	IconData.uID = 1UL;
	IconData.hWnd = m_hHostHwnd;
	IconData.uFlags = NIF_TIP;

	::lstrcpy(IconData.szTip, lpszTip);

	return ::Shell_NotifyIcon(NIM_MODIFY, &IconData);
}

const BOOL CNotifyIcon::ShowBalloon(LPCTSTR lpszTitle, LPCTSTR lpszMessage, const DWORD dwBalloonType)
{
	if(!m_hHostHwnd || !lpszTitle || !lpszMessage)return FALSE;

	// バルーンを表示する
	NOTIFYICONDATA IconData;
	IconData.cbSize = sizeof(IconData);
	IconData.uID = 1UL;
	IconData.hWnd = m_hHostHwnd;
	IconData.uFlags = NIF_INFO;

	IconData.uTimeout = 6000UL;

	switch(dwBalloonType){
		case BAL_ERROR :	IconData.dwInfoFlags = NIIF_ERROR;		break;
		case BAL_INFO :		IconData.dwInfoFlags = NIIF_INFO;		break;
		case BAL_WARNING :	IconData.dwInfoFlags = NIIF_WARNING;	break;
		default :			IconData.dwInfoFlags = NIIF_INFO;		break;
		}
	
	::lstrcpy(IconData.szInfoTitle, lpszTitle);
	::lstrcpy(IconData.szInfo, lpszMessage);

	return ::Shell_NotifyIcon(NIM_MODIFY, &IconData);
}

const BOOL CNotifyIcon::HideBalloon(void)
{
	if(!m_hHostHwnd)return FALSE;

	// バルーンを閉じる
	NOTIFYICONDATA IconData;
	IconData.cbSize = sizeof(IconData);
	IconData.uID = 1UL;
	IconData.hWnd = m_hHostHwnd;
	IconData.uFlags = NIF_INFO;
	IconData.dwInfoFlags = NIIF_INFO;
	IconData.uTimeout = 6000UL;
	
	IconData.szInfoTitle[0] = TEXT('\0');
	IconData.szInfo[0] = TEXT('\0');

	return ::Shell_NotifyIcon(NIM_MODIFY, &IconData);
}

LRESULT CALLBACK CNotifyIcon::HostWndProcRaw(HWND hwnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	CNotifyIcon *pThis = reinterpret_cast<CNotifyIcon *>(::GetWindowLong(hwnd, GWL_USERDATA));

	// 扱いやすいようにインスタンス内のメンバを呼ぶ
	if(pThis->HostWndProc(uMsg, wParam, lParam))return TRUE;

	return ::CallWindowProc(pThis->m_OldHostWndProc, hwnd, uMsg, wParam, lParam);
}

const BOOL CNotifyIcon::HostWndProc(UINT uMsg, WPARAM wParam, LPARAM lParam)
{
	if(uMsg == WM_NOTIFYICON){
		// イベントハンドラを呼び出す
		m_pEventHandler->OnNotifyIconEvent(this, LOWORD(lParam));
		return TRUE;
		}
		
	return FALSE;	
}