/*
 *Copyright (C) 1994-2000 The XFree86 Project, Inc. All Rights Reserved.
 *
 *Permission is hereby granted, free of charge, to any person obtaining
 * a copy of this software and associated documentation files (the
 *"Software"), to deal in the Software without restriction, including
 *without limitation the rights to use, copy, modify, merge, publish,
 *distribute, sublicense, and/or sell copies of the Software, and to
 *permit persons to whom the Software is furnished to do so, subject to
 *the following conditions:
 *
 *The above copyright notice and this permission notice shall be
 *included in all copies or substantial portions of the Software.
 *
 *THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 *EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 *MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
 *NONINFRINGEMENT. IN NO EVENT SHALL THE XFREE86 PROJECT BE LIABLE FOR
 *ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF
 *CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
 *WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 *
 *Except as contained in this notice, the name of the XFree86 Project
 *shall not be used in advertising or otherwise to promote the sale, use
 *or other dealings in this Software without prior written authorization
 *from the XFree86 Project.
 *
 * Authors:	Early Ehlinger
 *		Harold L Hunt II
 */

#ifdef HAVE_XWIN_CONFIG_H
#include <xwin-config.h>
#endif
#include "win.h"
#include <shellapi.h>
#include "winprefs.h"

static NOTIFYICONDATA nid;
/*
 * Initialize the tray icon
 */

void
winInitNotifyIcon(winPrivScreenPtr pScreenPriv, Bool Modify)
{
    winScreenInfo *pScreenInfo = pScreenPriv->pScreenInfo;
    char HostName[256];

    if (!Modify)
    {
        nid.cbSize = sizeof(NOTIFYICONDATA);
        nid.hWnd = pScreenPriv->hwndScreen;
        nid.uID = pScreenInfo->dwScreen;
        nid.uFlags = NIF_ICON | NIF_MESSAGE | NIF_TIP;
        nid.uCallbackMessage = WM_TRAYICON;
        nid.hIcon = winTaskbarIcon();

        /* Save handle to the icon so it can be freed later */
        pScreenPriv->hiconNotifyIcon = nid.hIcon;
    }

    gethostname(HostName,256);

    /* Set display and screen-specific tooltip text */
    if (g_pszQueryHost)
    {
        snprintf(nid.szTip,
                 sizeof(nid.szTip),
                 "%s - %s:%s.%d - %d clients",
	               g_pszQueryHost,
	               HostName,
                 display, 
                 (int) pScreenInfo->dwScreen,
                 pScreenPriv->iConnectedClients);
    }
    else
    {
        snprintf (nid.szTip,
                  sizeof (nid.szTip),
                  "%s:%s.%d - %d clients",
                  HostName,
                  display, 
                  (int) pScreenInfo->dwScreen,
                  pScreenPriv->iConnectedClients);
    }

    /* Add the tray icon */
    if (!Shell_NotifyIcon ((Modify) ? NIM_MODIFY : NIM_ADD, &nid))
        ErrorF("winInitNotifyIcon - Shell_NotifyIcon Failed\n");
}

/*
 * Delete the tray icon
 */

void
winDeleteNotifyIcon(winPrivScreenPtr pScreenPriv)
{
    winScreenInfo *pScreenInfo = pScreenPriv->pScreenInfo;

    if (!pScreenPriv->hiconNotifyIcon)
        return;

    /* Delete the tray icon */
    Shell_NotifyIcon (NIM_DELETE, &nid);

    /* Free the icon that was loaded */
    DestroyIcon (pScreenPriv->hiconNotifyIcon);

    pScreenPriv->hiconNotifyIcon = NULL;
}

/*
 * Process messages intended for the tray icon
 */

LRESULT
winHandleIconMessage(HWND hwnd, UINT message,
                     WPARAM wParam, LPARAM lParam, winPrivScreenPtr pScreenPriv)
{
#if defined(XWIN_MULTIWINDOWEXTWM) || defined(XWIN_MULTIWINDOW)
    winScreenInfo *pScreenInfo = pScreenPriv->pScreenInfo;
#endif

    switch (lParam) {
    case WM_MOUSEMOVE:
    {
      static int PrevNrClients;
      int NrClients=GetLiveClients(pScreenPriv);
      if (NrClients!=PrevNrClients)
      {
        PrevNrClients=NrClients;
        winInitNotifyIcon (pScreenPriv, TRUE);
      }
    }
    break;
    case WM_LBUTTONUP:
        /* Restack and bring all windows to top */
        SetForegroundWindow (pScreenPriv->hwndScreen);

#ifdef XWIN_MULTIWINDOWEXTWM
        if (pScreenInfo->fMWExtWM)
            winMWExtWMRestackWindows(pScreenInfo->pScreen);
#endif
        break;

    case WM_LBUTTONDBLCLK:
        /* Display Exit dialog box */
        winDisplayExitDialog(pScreenPriv);
        break;

    case WM_RBUTTONUP:
    {
        POINT ptCursor;
        HMENU hmenuPopup;
        HMENU hmenuTray;

        /* Get cursor position */
        GetCursorPos(&ptCursor);

        /* Load tray icon menu resource */
        hmenuPopup = LoadMenu(g_hInstance, MAKEINTRESOURCE(IDM_TRAYICON_MENU));
        if (!hmenuPopup)
            ErrorF("winHandleIconMessage - LoadMenu failed\n");

        /* Get actual tray icon menu */
        hmenuTray = GetSubMenu(hmenuPopup, 0);

#ifdef XWIN_MULTIWINDOW
        /* Check for MultiWindow mode */
        if (pScreenInfo->fMultiWindow) {
            MENUITEMINFO mii = { 0 };

            /* Root is shown, remove the check box */

            /* Setup menu item info structure */
            mii.cbSize = sizeof(MENUITEMINFO);
            mii.fMask = MIIM_STATE;
            mii.fState = MFS_CHECKED;

            /* Unheck box if root is shown */
            if (pScreenPriv->fRootWindowShown)
                mii.fState = MFS_UNCHECKED;

            /* Set menu state */
            SetMenuItemInfo(hmenuTray, ID_APP_HIDE_ROOT, FALSE, &mii);
        }
        else
#endif
        {
            /* Remove Hide Root Window button */
            RemoveMenu(hmenuTray, ID_APP_HIDE_ROOT, MF_BYCOMMAND);
        }

        SetupRootMenu(hmenuTray);

        /*
         * NOTE: This three-step procedure is required for
         * proper popup menu operation.  Without the
         * call to SetForegroundWindow the
         * popup menu will often not disappear when you click
         * outside of it.  Without the PostMessage the second
         * time you display the popup menu it might immediately
         * disappear.
         */
        SetForegroundWindow(hwnd);
        TrackPopupMenuEx(hmenuTray,
                         TPM_LEFTALIGN | TPM_BOTTOMALIGN | TPM_RIGHTBUTTON,
                         ptCursor.x, ptCursor.y, hwnd, NULL);
        PostMessage(hwnd, WM_NULL, 0, 0);

        /* Free menu */
        DestroyMenu(hmenuPopup);
    }
        break;
    }

    return 0;
}