/////////////////////////////////////////////////////////////////////////////
// TrayIcon.cpp : implementation file
//
// The tray icon can be instantiated using either the constructor or by
// declaring the object and creating (and displaying) it later on in the
// program. eg.
//
//        CTrayIcon m_TrayIcon;    // Member variable of some class
//        
//        ... 
//        // in some member function maybe...
//        m_TrayIcon.Create(pParentWnd, WM_MY_NOTIFY, "Click here", 
//                          hIcon, nTrayIconID);
    
#include "stdafx.h"
#include "TrayIcon.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

IMPLEMENT_DYNAMIC(CTrayIcon, CObject)

/////////////////////////////////////////////////////////////////////////////
// CTrayIcon construction/creation/destruction
CTrayIcon::CTrayIcon()
{
    memset(&m_tnd, 0, sizeof(m_tnd));
    m_bEnabled = FALSE;
    m_bHidden  = FALSE;
}

CTrayIcon::CTrayIcon(CWnd* pWnd, UINT uCallbackMessage, LPCTSTR szToolTip, 
                     HICON icon, UINT uID)
{
    Create(pWnd, uCallbackMessage, szToolTip, icon, uID);
    m_bHidden = FALSE;
}

BOOL CTrayIcon::Create(CWnd* pWnd, UINT uCallbackMessage, LPCTSTR szToolTip, 
                       HICON icon, UINT uID)
{
    // this is only for Windows 95 (or higher)
    VERIFY( m_bEnabled = ( GetVersion() & 0xff ) >= 4 );
    if( !m_bEnabled ) return FALSE;

    //Make sure Notification window is valid
    VERIFY( m_bEnabled = ( pWnd && ::IsWindow( pWnd->GetSafeHwnd() ) ) );
    if( !m_bEnabled ) return FALSE;
    
    //Make sure we avoid conflict with other messages
    ASSERT( uCallbackMessage >= WM_USER );

    //Tray only supports tooltip text up to 64 characters
    ASSERT( _tcslen(szToolTip) <= 64 );

    // load up the NOTIFYICONDATA structure
    m_tnd.cbSize			= sizeof(NOTIFYICONDATA);
    m_tnd.hWnd				= pWnd->GetSafeHwnd();
    m_tnd.uID				= uID;
    m_tnd.hIcon				= icon;
    m_tnd.uFlags			= NIF_MESSAGE | NIF_ICON | NIF_TIP;
    m_tnd.uCallbackMessage	= uCallbackMessage;
    
	strcpy (m_tnd.szTip, szToolTip);

    // Set the tray icon
    VERIFY( m_bEnabled = Shell_NotifyIcon( NIM_ADD, &m_tnd ) );
    return m_bEnabled;
}

CTrayIcon::~CTrayIcon()
{
    RemoveIcon();
}


/////////////////////////////////////////////////////////////////////////////
// CTrayIcon icon manipulation

void CTrayIcon::MoveToRight()
{
    HideIcon();
    ShowIcon();
}

void CTrayIcon::RemoveIcon()
{
    if( !m_bEnabled ) return;

    m_tnd.uFlags = 0;
    Shell_NotifyIcon( NIM_DELETE, &m_tnd );
    m_bEnabled = FALSE;
}

void CTrayIcon::HideIcon()
{
    if( m_bEnabled && !m_bHidden ) {
        m_tnd.uFlags = NIF_ICON;
        Shell_NotifyIcon( NIM_DELETE, &m_tnd );
        m_bHidden = TRUE;
    }
}

void CTrayIcon::ShowIcon()
{
    if( m_bEnabled && m_bHidden ) {
        m_tnd.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
        Shell_NotifyIcon( NIM_ADD, &m_tnd );
        m_bHidden = FALSE;
    }
}

BOOL CTrayIcon::SetIcon(HICON hIcon)
{
    if( !m_bEnabled ) return FALSE;

    m_tnd.uFlags = NIF_ICON;
    m_tnd.hIcon = hIcon;

    return Shell_NotifyIcon( NIM_MODIFY, &m_tnd );
}

BOOL CTrayIcon::SetIcon(LPCTSTR lpszIconName)
{
    HICON hIcon = AfxGetApp()->LoadIcon(lpszIconName);

    return SetIcon(hIcon);
}

BOOL CTrayIcon::SetIcon(UINT nIDResource)
{
    HICON hIcon = AfxGetApp()->LoadIcon(nIDResource);

    return SetIcon(hIcon);
}

BOOL CTrayIcon::SetStandardIcon(LPCTSTR lpIconName)
{
    HICON hIcon = LoadIcon(NULL, lpIconName);

    return SetIcon(hIcon);
}

BOOL CTrayIcon::SetStandardIcon(UINT nIDResource)
{
    HICON hIcon = LoadIcon(NULL, MAKEINTRESOURCE(nIDResource));

    return SetIcon(hIcon);
}
 
HICON CTrayIcon::GetIcon() const
{
    HICON hIcon = NULL;
    if( m_bEnabled )
        hIcon = m_tnd.hIcon;

    return hIcon;
}

/////////////////////////////////////////////////////////////////////////////
// CTrayIcon tooltip text manipulation

BOOL CTrayIcon::SetTooltipText(LPCTSTR pszTip)
{
    if( !m_bEnabled ) return FALSE;

    m_tnd.uFlags = NIF_TIP;
    _tcscpy( m_tnd.szTip, pszTip );

    return Shell_NotifyIcon( NIM_MODIFY, &m_tnd );
}

BOOL CTrayIcon::SetTooltipText(UINT nID)
{
    CString strText;
    VERIFY( strText.LoadString(nID) );

    return SetTooltipText(strText);
}

CString CTrayIcon::GetTooltipText() const
{
    CString strText;
    if( m_bEnabled )
        strText = m_tnd.szTip;

    return strText;
}

/////////////////////////////////////////////////////////////////////////////
// CTrayIcon notification window stuff

BOOL CTrayIcon::SetNotificationWnd(CWnd* pWnd)
{
    if( !m_bEnabled ) return FALSE;

    //Make sure Notification window is valid
    ASSERT( pWnd && ::IsWindow( pWnd->GetSafeHwnd() ) );

    m_tnd.hWnd = pWnd->GetSafeHwnd();
    m_tnd.uFlags = 0;

    return Shell_NotifyIcon( NIM_MODIFY, &m_tnd );
}

CWnd* CTrayIcon::GetNotificationWnd() const
{
    return CWnd::FromHandle(m_tnd.hWnd);
}

/////////////////////////////////////////////////////////////////////////////
// CTrayIcon implentation of OnTrayNotification

LRESULT CTrayIcon::OnTrayNotification(UINT wParam, LONG lParam) 
{
    //Return quickly if its not for this tray icon
    if( wParam != m_tnd.uID )
        return 0L;

    CMenu menu, *pSubMenu;

    // Clicking with right button brings up a context menu
    if( LOWORD(lParam) == WM_RBUTTONUP ) {    
        if( !menu.LoadMenu( m_tnd.uID ) ) return 0;
        if( !(pSubMenu = menu.GetSubMenu(0) ) ) return 0;

        // Make first menu item the default (bold font)
        ::SetMenuDefaultItem(pSubMenu->m_hMenu, 0, TRUE);

        //Display and track the popup menu
        CPoint pos;
        GetCursorPos(&pos);

        ::SetForegroundWindow( m_tnd.hWnd );  
        ::TrackPopupMenu( pSubMenu->m_hMenu, TPM_RIGHTALIGN | TPM_BOTTOMALIGN | TPM_RIGHTBUTTON, pos.x, pos.y, 0, m_tnd.hWnd, NULL );

        ::PostMessage(m_tnd.hWnd, WM_NULL, 0, 0);

        menu.DestroyMenu();
    } 
    else if (LOWORD(lParam) == WM_LBUTTONDBLCLK) {
        if( !menu.LoadMenu(m_tnd.uID) ) return 0;
        if( !(pSubMenu = menu.GetSubMenu(0) ) ) return 0;

        // double click received, the default action is to execute first menu item
		::SetForegroundWindow(m_tnd.hWnd);
		::SendMessage( m_tnd.hWnd, WM_COMMAND, pSubMenu->GetMenuItemID(0), 0 );

        menu.DestroyMenu();
    }

    return 1;
}