traynotify.cpp

Go to the documentation of this file.

/*
 * Copyright 2003, 2004, 2005 Martin Fuchs
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */


 //
 // Explorer clone
 //
 // traynotify.cpp
 //
 // Martin Fuchs, 22.08.2003
 //


#include <precomp.h>

#include "traynotify.h"

#include "../notifyhook/notifyhook.h"

NotifyHook::NotifyHook()
 :  WM_GETMODULEPATH(InstallNotifyHook())
{
}

NotifyHook::~NotifyHook()
{
    DeinstallNotifyHook();
}

void NotifyHook::GetModulePath(HWND hwnd, HWND hwndCallback)
{
    PostMessage(hwnd, WM_GETMODULEPATH, (WPARAM)hwndCallback, 0);
}

bool NotifyHook::ModulePathCopyData(LPARAM lparam, HWND* phwnd, String& path)
{
    char buffer[MAX_PATH];

    int l = GetWindowModulePathCopyData(lparam, phwnd, buffer, COUNTOF(buffer));

    if (l) {
        path.assign(buffer, l);
        return true;
    } else
        return false;
}


NotifyIconIndex::NotifyIconIndex(NOTIFYICONDATA* pnid)
{
    _hWnd = pnid->hWnd;
    _uID = pnid->uID;

     // special handling for windows task manager
    if ((int)_uID < 0)
        _uID = 0;
}

NotifyIconIndex::NotifyIconIndex()
{
    _hWnd = 0;
    _uID = 0;
}


NotifyInfo::NotifyInfo()
{
    _idx = -1;
    _hIcon = 0;
    _dwState = 0;
    _uCallbackMessage = 0;
    _version = 0;

    _mode = NIM_AUTO;
    _lastChange = GetTickCount();
}


 // WCHAR versions von NOTIFYICONDATA
#define NID_SIZE_W6  sizeof(NOTIFYICONDATAW)                                        // _WIN32_IE = 0x600
#define NID_SIZE_W5 (sizeof(NOTIFYICONDATAW)-sizeof(GUID))                          // _WIN32_IE = 0x500
#define NID_SIZE_W3 (sizeof(NOTIFYICONDATAW)-sizeof(GUID)-(128-64)*sizeof(WCHAR))   // _WIN32_IE < 0x500

 // CHAR versions von NOTIFYICONDATA
#define NID_SIZE_A6  sizeof(NOTIFYICONDATAA)
#define NID_SIZE_A5 (sizeof(NOTIFYICONDATAA)-sizeof(GUID))
#define NID_SIZE_A3 (sizeof(NOTIFYICONDATAA)-sizeof(GUID)-(128-64)*sizeof(CHAR))

bool NotifyInfo::modify(NOTIFYICONDATA* pnid)
{
    bool changes = false;

    if (_hWnd!=pnid->hWnd || _uID!=pnid->uID) {
        _hWnd = pnid->hWnd;
        _uID = pnid->uID;

        changes = true;
    }

    if (pnid->uFlags & NIF_MESSAGE) {
        if (_uCallbackMessage != pnid->uCallbackMessage) {
            _uCallbackMessage = pnid->uCallbackMessage;
            changes = true;
        }
    }

    if (pnid->uFlags & NIF_ICON) {
         // Some applications destroy the icon immediatelly after completing the
         // NIM_ADD/MODIFY message, so we have to make a copy of it.
        if (_hIcon)
            DestroyIcon(_hIcon);

        _hIcon = (HICON) CopyImage(pnid->hIcon, IMAGE_ICON, NOTIFYICON_SIZE, NOTIFYICON_SIZE, 0);

        changes = true; 
    }

    if (pnid->uFlags & NIF_STATE) {
        DWORD new_state = (_dwState&~pnid->dwStateMask) | (pnid->dwState&pnid->dwStateMask);

        if (_dwState != new_state) {
            _dwState = new_state;
            changes = true;
        }
    }

     // store tool tip text
    if (pnid->uFlags & NIF_TIP) {
        String new_text;

        if (pnid->cbSize==NID_SIZE_W6 || pnid->cbSize==NID_SIZE_W5 || pnid->cbSize==NID_SIZE_W3) {
             // UNICODE version of NOTIFYICONDATA structure
            LPCWSTR txt = (LPCWSTR)pnid->szTip;
            int max_len = pnid->cbSize==NID_SIZE_W3? 64: 128;

             // get tooltip string length
            int l = 0;
            for(; l<max_len; ++l)
                if (!txt[l])
                    break;

            new_text.assign(txt, l);

            if (new_text != _tipText) {
                _tipText = new_text;
                changes = true;
            }
        } else if (pnid->cbSize==NID_SIZE_A6 || pnid->cbSize==NID_SIZE_A5 || pnid->cbSize==NID_SIZE_A3) {
            LPCSTR txt = (LPCSTR)pnid->szTip;
            int max_len = pnid->cbSize==NID_SIZE_A3? 64: 128;

            int l = 0;
            for(int l=0; l<max_len; ++l)
                if (!txt[l])
                    break;

            new_text.assign(txt, l);

            if (new_text != _tipText) {
                _tipText = new_text;
                changes = true;
            }
        }
    }

    TCHAR title[MAX_PATH];

    DWORD pid;
    GetWindowThreadProcessId(_hWnd, &pid);

     // avoid to send WM_GETTEXT messages to the own process
    if (pid != GetCurrentProcessId())
        if (GetWindowText(_hWnd, title, COUNTOF(title))) {
            if (_windowTitle != title) {
                _windowTitle = title;
                changes = true;
            }
        }

    if (changes) {
        create_name();
        _lastChange = GetTickCount();
    }

    return changes;
}


NotifyArea::NotifyArea(HWND hwnd)
 :  super(hwnd),
    _tooltip(hwnd)
{
    _next_idx = 0;
    _clock_width = 0;
    _last_icon_count = 0;
    _show_hidden = false;
    _hide_inactive = true;
    _show_button = true;
}

NotifyArea::~NotifyArea()
{
    KillTimer(_hwnd, 0);

    write_config();
}

static bool get_hide_clock_from_registry()
{
    HKEY hkeyStuckRects = 0;
    DWORD buffer[10];
    DWORD len = sizeof(buffer);

    bool hide_clock = false;

     // check if the clock should be hidden
    if (!RegOpenKey(HKEY_CURRENT_USER, TEXT("SOFTWARE\\Microsoft\\Windows\\CurrentVersion\\Explorer\\StuckRects2"), &hkeyStuckRects) &&
        !RegQueryValueEx(hkeyStuckRects, TEXT("Settings"), 0, NULL, (LPBYTE)buffer, &len) &&
        len==sizeof(buffer) && buffer[0]==sizeof(buffer))
        hide_clock = buffer[2] & 0x08? true: false;

    if (hkeyStuckRects)
        RegCloseKey(hkeyStuckRects);

    return hide_clock;
}

void NotifyArea::read_config()
{
    bool clock_visible = true;

     // read notification icon settings from XML configuration
    XMLPos cfg_pos = g_Globals.get_cfg();

    if (!g_Globals._SHRestricted || !SHRestricted(REST_HIDECLOCK))
    {
        if (cfg_pos.go_down("desktopbar")) {
            clock_visible = XMLBoolRef(XMLPos(cfg_pos,"options"), "show-clock", !get_hide_clock_from_registry());
            cfg_pos.back();
        }
    }

    if (cfg_pos.go_down("notify-icons")) {
        XMLPos options(cfg_pos, "options");

        _hide_inactive = XMLBool(options, "hide-inactive", true);   
        _show_hidden = XMLBool(options, "show-hidden", false);  
        _show_button = XMLBool(options, "show-button", true);

        XMLChildrenFilter icons(cfg_pos, "icon");

        for(XMLChildrenFilter::iterator it=icons.begin(); it!=icons.end(); ++it) {
            const XMLNode& node = **it;

            NotifyIconConfig cfg;

            cfg._name = node.get("name").c_str();
            cfg._tipText = node.get("text").c_str();
            cfg._windowTitle = node.get("window").c_str();
            cfg._modulePath = node.get("module").c_str();
            const string& mode = node.get("show");

            if (mode == "show")
                cfg._mode = NIM_SHOW;
            else if (mode == "hide")
                cfg._mode = NIM_HIDE;
            else //if (mode == "auto")
                cfg._mode = NIM_HIDE;

            _cfg.push_back(cfg);
        }

        cfg_pos.back();
    }

    show_clock(clock_visible);
}

void NotifyArea::write_config()
{
     // write notification icon settings to XML configuration file
    XMLPos cfg_pos = g_Globals.get_cfg();

    cfg_pos.smart_create("desktopbar");
    XMLBoolRef boolRef(XMLPos(cfg_pos,"options"), "show-clock");
    boolRef = _hwndClock!=0;
    cfg_pos.back();

    cfg_pos.smart_create("notify-icons");

    XMLPos options(cfg_pos, "options");
    XMLBoolRef(options, "hide-inactive") = _hide_inactive;
    XMLBoolRef(options, "show-hidden") = _show_hidden;
    XMLBoolRef(options, "show-button") = _show_button;

    for(NotifyIconCfgList::iterator it=_cfg.begin(); it!=_cfg.end(); ++it) {
        NotifyIconConfig& cfg = *it;

         // search for the corresponding node using the original name
        cfg_pos.smart_create("icon", "name", cfg._name);

         // refresh unique name
        cfg.create_name();

        cfg_pos["name"] = cfg._name.c_str();
        cfg_pos["text"] = cfg._tipText.c_str();
        cfg_pos["window"] = cfg._windowTitle.c_str();
        cfg_pos["module"] = cfg._modulePath.c_str();
        cfg_pos["show"] = string_from_mode(cfg._mode).c_str();

        cfg_pos.back();
    }

    cfg_pos.back(); // smart_create
}

void NotifyArea::show_clock(bool flag)
{
    bool vis = _hwndClock!=0;

    if (vis != flag) {
        if (flag) {
             // create clock window
            _hwndClock = ClockWindow::Create(_hwnd);

            if (_hwndClock) {
                ClientRect clock_size(_hwndClock);
                _clock_width = clock_size.right;
            }
        } else {
            DestroyWindow(_hwndClock);
            _hwndClock = 0;
            _clock_width = 0;
        }

        SendMessage(GetParent(_hwnd), PM_RESIZE_CHILDREN, 0, 0);
    }
}

LRESULT NotifyArea::Init(LPCREATESTRUCT pcs)
{
    if (super::Init(pcs))
        return 1;

    read_config();

    SetTimer(_hwnd, 0, 1000, NULL);

    return 0;
}

HWND NotifyArea::Create(HWND hwndParent)
{
    static BtnWindowClass wcTrayNotify(CLASSNAME_TRAYNOTIFY, CS_DBLCLKS);

    ClientRect clnt(hwndParent);

    return Window::Create(WINDOW_CREATOR(NotifyArea), WS_EX_STATICEDGE,
                            wcTrayNotify, TITLE_TRAYNOTIFY, WS_CHILD|WS_VISIBLE|WS_CLIPCHILDREN,
                            clnt.right-(NOTIFYAREA_WIDTH_DEF+1), 1, NOTIFYAREA_WIDTH_DEF, clnt.bottom-2, hwndParent);
}

LRESULT NotifyArea::WndProc(UINT nmsg, WPARAM wparam, LPARAM lparam)
{
    switch(nmsg) {
      case WM_PAINT:
        Paint();
        break;

      case WM_TIMER: {
        Refresh();

        ClockWindow* clock_window = GET_WINDOW(ClockWindow, _hwndClock);

        if (clock_window)
            clock_window->TimerTick();
        break;}

      case PM_REFRESH:
        Refresh(true);
        break;

      case WM_SIZE: {
        int cx = LOWORD(lparam);
        SetWindowPos(_hwndClock, 0, cx-_clock_width, 0, 0, 0, SWP_NOSIZE|SWP_NOZORDER|SWP_NOACTIVATE);
        break;}

      case PM_GET_WIDTH: {
        int w = _sorted_icons.size()*NOTIFYICON_DIST + NOTIFYAREA_SPACE + _clock_width;
        if (_show_button)
            w += NOTIFYICON_DIST;
        return w;}

      case PM_REFRESH_CONFIG:
        read_config();
        break;

      case WM_CONTEXTMENU: {
        Point pt(lparam);
        POINTS p;
        p.x = (SHORT) pt.x;
        p.y = (SHORT) pt.y;
        ScreenToClient(_hwnd, &pt);

        if (IconHitTest(pt) == _sorted_icons.end()) { // display menu only when no icon clicked
            PopupMenu menu(IDM_NOTIFYAREA);
            SetMenuDefaultItem(menu, 0, MF_BYPOSITION);
            CheckMenuItem(menu, ID_SHOW_HIDDEN_ICONS, MF_BYCOMMAND|(_show_hidden?MF_CHECKED:MF_UNCHECKED));
            CheckMenuItem(menu, ID_SHOW_ICON_BUTTON, MF_BYCOMMAND|(_show_button?MF_CHECKED:MF_UNCHECKED));
            menu.TrackPopupMenu(_hwnd, p);
        }
        break;}

      case WM_COPYDATA: {   // receive NotifyHook answers
        String path;
        HWND hwnd;

        if (_hook.ModulePathCopyData(lparam, &hwnd, path))
            _window_modules[hwnd] = path;
        break;}

      default:
        if (nmsg>=WM_MOUSEFIRST && nmsg<=WM_MOUSELAST) {
             // close startup menu and other popup menus
             // This functionality is missing in MS Windows.
            if (nmsg==WM_LBUTTONDOWN || nmsg==WM_MBUTTONDOWN || nmsg==WM_RBUTTONDOWN
#ifdef WM_XBUTTONDOWN
                || nmsg==WM_XBUTTONDOWN
#endif
                )

                CancelModes();

            Point pt(lparam);
            NotifyIconSet::const_iterator found = IconHitTest(pt);

            if (found != _sorted_icons.end()) {
                const NotifyInfo& entry = const_cast<NotifyInfo&>(*found);  // Why does GCC 3.3 need this additional const_cast ?!

                 // set activation time stamp
                if (nmsg == WM_LBUTTONDOWN ||   // Some programs need PostMessage() instead of SendMessage().
                    nmsg == WM_MBUTTONDOWN ||   // So call SendMessage() only for BUTTONUP and BLCLK messages
#ifdef WM_XBUTTONDOWN
                    nmsg == WM_XBUTTONDOWN ||
#endif
                    nmsg == WM_RBUTTONDOWN) {
                    _icon_map[entry]._lastChange = GetTickCount();
                }

                 // Notify the message if the owner is still alive
                if (IsWindow(entry._hWnd)) {
                    if (nmsg == WM_MOUSEMOVE ||     // avoid to call blocking SendMessage() for merely moving the mouse over icons
                        nmsg == WM_LBUTTONDOWN ||   // Some programs need PostMessage() instead of SendMessage().
                        nmsg == WM_MBUTTONDOWN ||   // So call SendMessage() only for BUTTONUP and BLCLK messages
#ifdef WM_XBUTTONDOWN
                        nmsg == WM_XBUTTONDOWN ||
#endif
                        nmsg == WM_RBUTTONDOWN)
                        PostMessage(entry._hWnd, entry._uCallbackMessage, entry._uID, nmsg);
                    else {
                         // allow SetForegroundWindow() in client process
                        DWORD pid;

                        if (GetWindowThreadProcessId(entry._hWnd, &pid)) {
                             // bind dynamically to AllowSetForegroundWindow() to be compatible to WIN98
                            static DynamicFct<BOOL(WINAPI*)(DWORD)> AllowSetForegroundWindow(TEXT("USER32"), "AllowSetForegroundWindow");

                            if (AllowSetForegroundWindow)
                                (*AllowSetForegroundWindow)(pid);
                        }

                         // use PostMessage() for notifcation icons of Shell Service Objects in the own process
                        if (pid == GetCurrentProcessId())
                            PostMessage(entry._hWnd, entry._uCallbackMessage, entry._uID, nmsg);
                        else
                            SendMessage(entry._hWnd, entry._uCallbackMessage, entry._uID, nmsg);
                    }
                }
                else if (_icon_map.erase(entry))    // delete icons without valid owner window
                    UpdateIcons();
            } else
                 // handle clicks on notification area button "show hidden icons"
                if (_show_button)
                    if (nmsg == WM_LBUTTONDOWN)
                        if (pt.x>=NOTIFYICON_X && pt.x<NOTIFYICON_X+NOTIFYICON_SIZE &&
                            pt.y>=NOTIFYICON_Y && pt.y<NOTIFYICON_Y+NOTIFYICON_SIZE)
                            PostMessage(_hwnd, WM_COMMAND, MAKEWPARAM(ID_SHOW_HIDDEN_ICONS,0), 0);
        }

        return super::WndProc(nmsg, wparam, lparam);
    }

    return 0;
}

int NotifyArea::Command(int id, int code)
{
    switch(id) {
      case ID_SHOW_HIDDEN_ICONS:
        _show_hidden = !_show_hidden;
        UpdateIcons();
        break;

      case ID_SHOW_ICON_BUTTON:
        _show_button = !_show_button;
        UpdateIcons();
        break;

      case ID_CONFIG_NOTIFYAREA:
        Dialog::DoModal(IDD_NOTIFYAREA, WINDOW_CREATOR(TrayNotifyDlg), GetParent(_hwnd));
        break;

      case ID_CONFIG_TIME:
        launch_cpanel(_hwnd, TEXT("timedate.cpl"));
        break;

      default:
        SendParent(WM_COMMAND, MAKELONG(id,code), 0);
    }

    return 0;
}

int NotifyArea::Notify(int id, NMHDR* pnmh)
{
    if (pnmh->code == TTN_GETDISPINFO) {
        LPNMTTDISPINFO pdi = (LPNMTTDISPINFO)pnmh;

        Point pt(GetMessagePos());
        ScreenToClient(_hwnd, &pt);

        if (_show_button &&
            pt.x>=NOTIFYICON_X && pt.x<NOTIFYICON_X+NOTIFYICON_SIZE &&
            pt.y>=NOTIFYICON_Y && pt.y<NOTIFYICON_Y+NOTIFYICON_SIZE)
        {
            static ResString sShowIcons(IDS_SHOW_HIDDEN_ICONS);
            static ResString sHideIcons(IDS_HIDE_ICONS);

            pdi->lpszText = (_show_hidden? sHideIcons: sShowIcons).str();
        } else {
            NotifyIconSet::iterator found = IconHitTest(pt);

            if (found != _sorted_icons.end()) {
                NotifyInfo& entry = const_cast<NotifyInfo&>(*found);    // Why does GCC 3.3 need this additional const_cast ?!

                 // enable multiline tooltips (break at CR/LF and for very long one-line strings)
                SendMessage(pnmh->hwndFrom, TTM_SETMAXTIPWIDTH, 0, 400);

                pdi->lpszText = entry._tipText.str();
            }
        }
    }

    return 0;
}

void NotifyArea::CancelModes()
{
    PostMessage(HWND_BROADCAST, WM_CANCELMODE, 0, 0);

    for(NotifyIconSet::const_iterator it=_sorted_icons.begin(); it!=_sorted_icons.end(); ++it)
        PostMessage(it->_hWnd, WM_CANCELMODE, 0, 0);
}

LRESULT NotifyArea::ProcessTrayNotification(int notify_code, NOTIFYICONDATA* pnid)
{
    switch(notify_code) {
      case NIM_ADD:
      case NIM_MODIFY:
        if ((int)pnid->uID >= 0) {  
            NotifyInfo& entry = _icon_map[pnid];

             // a new entry?
            if (entry._idx == -1)
                entry._idx = ++_next_idx;
        /* equivalent code using iterator::find();
            NotifyIconMap::iterator found = _icon_map.find(pnid);
            NotifyInfo* pentry;
             // a new entry?
            if (found == _icon_map.end()) {
                pentry = &_icon_map[pnid];
                pentry->_idx = ++_next_idx;
            } else {
                pentry = &found->second;
            }
            NotifyInfo& entry = *pentry;
        */
            bool changes = entry.modify(pnid);

            if (DetermineHideState(entry) && entry._mode==NIM_HIDE) {
                entry._dwState |= NIS_HIDDEN;
                changes = true;
            }

            if (changes)
                UpdateIcons();  

            return TRUE;
        }
        break;

      case NIM_DELETE: {
        NotifyIconMap::iterator found = _icon_map.find(pnid);

        if (found != _icon_map.end()) {
            if (found->second._hIcon)
                DestroyIcon(found->second._hIcon);
            _icon_map.erase(found);
            UpdateIcons();
            return TRUE;
        }
        break;}

      case NIM_SETFOCUS:
        SetForegroundWindow(_hwnd);
        return TRUE;

      case NIM_SETVERSION:
        NotifyIconMap::iterator found = _icon_map.find(pnid);

        if (found != _icon_map.end()) {
            found->second._version = pnid->UNION_MEMBER(uVersion);
            return TRUE;
        } else
            return FALSE;
    }

    return FALSE;
}

void NotifyArea::UpdateIcons()
{
    _sorted_icons.clear();

     // sort icon infos by display index
    for(NotifyIconMap::const_iterator it=_icon_map.begin(); it!=_icon_map.end(); ++it) {
        const NotifyInfo& entry = it->second;

        if (_show_hidden || !(entry._dwState & NIS_HIDDEN))
            _sorted_icons.insert(entry);
    }

     // sync tooltip areas to current icon number
    if (_sorted_icons.size() != _last_icon_count) {
        RECT rect = {NOTIFYICON_X, NOTIFYICON_Y, NOTIFYICON_X+NOTIFYICON_SIZE, NOTIFYICON_Y+NOTIFYICON_SIZE};

        size_t tt_idx = 0;

        if (_show_button) {
            _tooltip.add(_hwnd, tt_idx++, rect);

            rect.left += NOTIFYICON_DIST;
            rect.right += NOTIFYICON_DIST;
        }

        size_t icon_cnt = _sorted_icons.size();
        while(tt_idx < icon_cnt) {
            _tooltip.add(_hwnd, tt_idx++, rect);

            rect.left += NOTIFYICON_DIST;
            rect.right += NOTIFYICON_DIST;
        }

        while(tt_idx < _last_icon_count)
            _tooltip.remove(_hwnd, tt_idx++);

        _last_icon_count = _sorted_icons.size();
    }

    SendMessage(GetParent(_hwnd), PM_RESIZE_CHILDREN, 0, 0);

    InvalidateRect(_hwnd, NULL, FALSE); // refresh icon display
    UpdateWindow(_hwnd);
}

#ifndef _NO_ALPHABLEND
#ifdef _MSC_VER
#pragma comment(lib, "msimg32") // for AlphaBlend()
#endif
#endif

void NotifyArea::Paint()
{
    BufferedPaintCanvas canvas(_hwnd);

     // first fill with the background color
    FillRect(canvas, &canvas.rcPaint, GetSysColorBrush(COLOR_BTNFACE));

     // draw icons
    int x = NOTIFYICON_X;
    int y = NOTIFYICON_Y;

    if (_show_button) {
        static SmallIcon leftArrowIcon(IDI_NOTIFY_L);
        static SmallIcon rightArrowIcon(IDI_NOTIFY_R);

        DrawIconEx(canvas, x, y, _show_hidden?rightArrowIcon:leftArrowIcon, NOTIFYICON_SIZE, NOTIFYICON_SIZE, 0, 0, DI_NORMAL);
        x += NOTIFYICON_DIST;
    }

#ifndef _NO_ALPHABLEND
    MemCanvas mem_dc;
    SelectedBitmap bmp(mem_dc, CreateCompatibleBitmap(canvas, NOTIFYICON_SIZE, NOTIFYICON_SIZE));
    RECT rect = {0, 0, NOTIFYICON_SIZE, NOTIFYICON_SIZE};
    BLENDFUNCTION blend = {AC_SRC_OVER, 0, 128, 0}; // 50 % visible
#endif

    for(NotifyIconSet::const_iterator it=_sorted_icons.begin(); it!=_sorted_icons.end(); ++it) {
#ifndef _NO_ALPHABLEND
        if (it->_dwState & NIS_HIDDEN) {
            FillRect(mem_dc, &rect, GetSysColorBrush(COLOR_BTNFACE));
            DrawIconEx(mem_dc, 0, 0, it->_hIcon, NOTIFYICON_SIZE, NOTIFYICON_SIZE, 0, 0, DI_NORMAL);
            AlphaBlend(canvas, x, y, NOTIFYICON_SIZE, NOTIFYICON_SIZE, mem_dc, 0, 0, NOTIFYICON_SIZE, NOTIFYICON_SIZE, blend);
        } else
#endif
            DrawIconEx(canvas, x, y, it->_hIcon, NOTIFYICON_SIZE, NOTIFYICON_SIZE, 0, 0, DI_NORMAL);

        x += NOTIFYICON_DIST;
    }
}

void NotifyArea::Refresh(bool update)
{
     // Look for task icons without valid owner window.
     // This is an extended feature missing in MS Windows.
    for(NotifyIconSet::const_iterator it=_sorted_icons.begin(); it!=_sorted_icons.end(); ++it) {
        const NotifyInfo& entry = *it;

        if (!IsWindow(entry._hWnd))
            if (_icon_map.erase(entry)) // delete icons without valid owner window
                ++update;
    }

    DWORD now = GetTickCount();

     // handle icon hiding
    for(NotifyIconMap::iterator it=_icon_map.begin(); it!=_icon_map.end(); ++it) {
        NotifyInfo& entry = it->second;

        DetermineHideState(entry);

        switch(entry._mode) {
          case NIM_HIDE:
            if (!(entry._dwState & NIS_HIDDEN)) {
                entry._dwState |= NIS_HIDDEN;
                ++update;
            }
            break;

          case NIM_SHOW:
            if (entry._dwState&NIS_HIDDEN) {
                entry._dwState &= ~NIS_HIDDEN;
                ++update;
            }
            break;

          case NIM_AUTO:
             // automatically hide icons after long periods of inactivity
            if (_hide_inactive)
                if (!(entry._dwState & NIS_HIDDEN))
                    if (now-entry._lastChange > ICON_AUTOHIDE_SECONDS*1000) {
                        entry._dwState |= NIS_HIDDEN;
                        ++update;
                    }
            break;
        }
    }

    if (update)
        UpdateIcons();
}

NotifyIconSet::iterator NotifyArea::IconHitTest(const POINT& pos)
{
    if (pos.y<NOTIFYICON_Y || pos.y>=NOTIFYICON_Y+NOTIFYICON_SIZE)
        return _sorted_icons.end();

    NotifyIconSet::iterator it = _sorted_icons.begin();

    int x = NOTIFYICON_X;

    if (_show_button)
        x += NOTIFYICON_DIST;

    for(; it!=_sorted_icons.end(); ++it) {
        //NotifyInfo& entry = const_cast<NotifyInfo&>(*it); // Why does GCC 3.3 need this additional const_cast ?!

        if (pos.x>=x && pos.x<x+NOTIFYICON_SIZE)
            break;

        x += NOTIFYICON_DIST;
    }

    return it;
}


void NotifyIconConfig::create_name()
{
    _name = FmtString(TEXT("'%s' - '%s' - '%s'"), _tipText.c_str(), _windowTitle.c_str(), _modulePath.c_str());
}


bool NotifyIconConfig::match(const NotifyIconConfig& props) const
{
    if (!_tipText.empty() && !props._tipText.empty())
        if (props._tipText == _tipText)
            return true;

    if (!_windowTitle.empty() && !props._windowTitle.empty())
        if (_tcsstr(props._windowTitle, _windowTitle))
            return true;

    if (!_modulePath.empty() && !props._modulePath.empty())
        if (!_tcsicmp(props._modulePath, _modulePath))
            return true;

    return false;
}

bool NotifyArea::DetermineHideState(NotifyInfo& entry)
{
    if (entry._modulePath.empty()) {
        const String& modulePath = _window_modules[entry._hWnd];

         // request module path for new windows (We will get an asynchronous answer by a WM_COPYDATA message.)
        if (!modulePath.empty())
            entry._modulePath = modulePath;
        else
            _hook.GetModulePath(entry._hWnd, _hwnd);
    }

    for(NotifyIconCfgList::const_iterator it=_cfg.begin(); it!=_cfg.end(); ++it) {
        const NotifyIconConfig& cfg = *it;

        if (cfg.match(entry)) {
            entry._mode = cfg._mode;
            return true;
        }
    }

    return false;
}



String string_from_mode(NOTIFYICONMODE mode)
{
    switch(mode) {
      case NIM_SHOW:
        return ResString(IDS_NOTIFY_SHOW);

      case NIM_HIDE:
        return ResString(IDS_NOTIFY_HIDE);

      default:  //case NIM_AUTO
        return ResString(IDS_NOTIFY_AUTOHIDE);
    }
}


TrayNotifyDlg::TrayNotifyDlg(HWND hwnd)
 :  super(hwnd),
    _tree_ctrl(GetDlgItem(hwnd, IDC_NOTIFY_ICONS)),
    _himl(ImageList_Create(GetSystemMetrics(SM_CXSMICON), GetSystemMetrics(SM_CYSMICON), ILC_COLOR24, 3, 0)),
    _pNotifyArea(static_cast<NotifyArea*>(Window::get_window((HWND)SendMessage(g_Globals._hwndDesktopBar, PM_GET_NOTIFYAREA, 0, 0))))
{
    _selectedItem = 0;

    if (_pNotifyArea) {
         // save original icon states and configuration data
        for(NotifyIconMap::const_iterator it=_pNotifyArea->_icon_map.begin(); it!=_pNotifyArea->_icon_map.end(); ++it)
            _icon_states_org[it->first] = IconStatePair(it->second._mode, it->second._dwState);

        _cfg_org = _pNotifyArea->_cfg;
        _show_hidden_org = _pNotifyArea->_show_hidden;
    }

    SetWindowIcon(hwnd, IDI_REACTOS);

    _haccel = LoadAccelerators(g_Globals._hInstance, MAKEINTRESOURCE(IDA_TRAYNOTIFY));

    {
    WindowCanvas canvas(_hwnd);
    HBRUSH hbkgnd = GetStockBrush(WHITE_BRUSH);

    ImageList_AddAlphaIcon(_himl, SmallIcon(IDI_DOT), hbkgnd, canvas);
    ImageList_AddAlphaIcon(_himl, SmallIcon(IDI_DOT_TRANS), hbkgnd, canvas);
    ImageList_AddAlphaIcon(_himl, SmallIcon(IDI_DOT_RED), hbkgnd, canvas);
    }

    (void)TreeView_SetImageList(_tree_ctrl, _himl, TVSIL_NORMAL);

    _resize_mgr.Add(IDC_NOTIFY_ICONS,   RESIZE);
    _resize_mgr.Add(IDC_LABEL1,         MOVE_Y);
    _resize_mgr.Add(IDC_NOTIFY_TOOLTIP, RESIZE_X|MOVE_Y);
    _resize_mgr.Add(IDC_LABEL2,         MOVE_Y);
    _resize_mgr.Add(IDC_NOTIFY_TITLE,   RESIZE_X|MOVE_Y);
    _resize_mgr.Add(IDC_LABEL3,         MOVE_Y);
    _resize_mgr.Add(IDC_NOTIFY_MODULE,  RESIZE_X|MOVE_Y);

    _resize_mgr.Add(IDC_LABEL4,         MOVE_Y);
    _resize_mgr.Add(IDC_NOTIFY_SHOW,    MOVE_Y);
    _resize_mgr.Add(IDC_NOTIFY_HIDE,    MOVE_Y);
    _resize_mgr.Add(IDC_NOTIFY_AUTOHIDE,MOVE_Y);

    _resize_mgr.Add(IDC_PICTURE,        MOVE);
    _resize_mgr.Add(ID_SHOW_HIDDEN_ICONS,MOVE_Y);

    _resize_mgr.Add(IDC_LABEL6,         MOVE_Y);
    _resize_mgr.Add(IDC_LAST_CHANGE,    MOVE_Y);

    _resize_mgr.Add(IDOK,               MOVE);
    _resize_mgr.Add(IDCANCEL,           MOVE);

    _resize_mgr.Resize(+150, +200);

    Refresh();

    SetTimer(_hwnd, 0, 3000, NULL);
    register_pretranslate(hwnd);
}

TrayNotifyDlg::~TrayNotifyDlg()
{
    KillTimer(_hwnd, 0);
    unregister_pretranslate(_hwnd);
    ImageList_Destroy(_himl);
}

void TrayNotifyDlg::Refresh()
{

    HiddenWindow hide(_tree_ctrl);

    TreeView_DeleteAllItems(_tree_ctrl);

    TV_INSERTSTRUCT tvi;

    tvi.hParent = 0;
    tvi.hInsertAfter = TVI_LAST;

    TV_ITEM& tv = tvi.item;
    tv.mask = TVIF_TEXT|TVIF_IMAGE|TVIF_SELECTEDIMAGE;

    ResString str_cur(IDS_ITEMS_CUR);
    tv.pszText = str_cur.str();
    tv.iSelectedImage = tv.iImage = 0;  // IDI_DOT
    _hitemCurrent = TreeView_InsertItem(_tree_ctrl, &tvi);

    ResString str_conf(IDS_ITEMS_CONFIGURED);
    tv.pszText = str_conf.str();
    tv.iSelectedImage = tv.iImage = 2;  // IDI_DOT_RED
    _hitemConfig = TreeView_InsertItem(_tree_ctrl, &tvi);

    tvi.hParent = _hitemCurrent;

    ResString str_visible(IDS_ITEMS_VISIBLE);
    tv.pszText = str_visible.str();
    tv.iSelectedImage = tv.iImage = 0;  // IDI_DOT
    _hitemCurrent_visible = TreeView_InsertItem(_tree_ctrl, &tvi);

    ResString str_hidden(IDS_ITEMS_HIDDEN);
    tv.pszText = str_hidden.str();
    tv.iSelectedImage = tv.iImage = 1;  // IDI_DOT_TRANS
    _hitemCurrent_hidden = TreeView_InsertItem(_tree_ctrl, &tvi);

    if (_pNotifyArea) {
        _info.clear();

        tv.mask |= TVIF_PARAM;

        WindowCanvas canvas(_hwnd);

         // insert current (visible and hidden) items
        for(NotifyIconMap::const_iterator it=_pNotifyArea->_icon_map.begin(); it!=_pNotifyArea->_icon_map.end(); ++it) {
            const NotifyInfo& entry = it->second;

            InsertItem(entry._dwState&NIS_HIDDEN? _hitemCurrent_hidden: _hitemCurrent_visible, TVI_LAST, entry, canvas);
        }

         // insert configured items in tree view
        const NotifyIconCfgList& cfg = _pNotifyArea->_cfg;
        for(NotifyIconCfgList::const_iterator it=cfg.begin(); it!=cfg.end(); ++it) {
            const NotifyIconConfig& cfg_entry = *it;

            HICON hicon = 0;

            if (!cfg_entry._modulePath.empty()) {
                if ((int)ExtractIconEx(cfg_entry._modulePath, 0, NULL, &hicon, 1) <= 0)
                    hicon = 0;

                if (!hicon) {
                    SHFILEINFO sfi;

                    if (SHGetFileInfo(cfg_entry._modulePath, 0, &sfi, sizeof(sfi), SHGFI_ICON|SHGFI_SMALLICON))
                        hicon = sfi.hIcon;
                }
            }

            InsertItem(_hitemConfig, TVI_SORT, cfg_entry, canvas, hicon, cfg_entry._mode);

            if (hicon)
                DestroyIcon(hicon);
        }

        CheckDlgButton(_hwnd, ID_SHOW_HIDDEN_ICONS, _pNotifyArea->_show_hidden? BST_CHECKED: BST_UNCHECKED);
    }

    TreeView_Expand(_tree_ctrl, _hitemCurrent_visible, TVE_EXPAND);
    TreeView_Expand(_tree_ctrl, _hitemCurrent_hidden, TVE_EXPAND);
    TreeView_Expand(_tree_ctrl, _hitemCurrent, TVE_EXPAND);
    TreeView_Expand(_tree_ctrl, _hitemConfig, TVE_EXPAND);

    TreeView_EnsureVisible(_tree_ctrl, _hitemCurrent_visible);
}

void TrayNotifyDlg::InsertItem(HTREEITEM hparent, HTREEITEM after, const NotifyInfo& entry, HDC hdc)
{
    InsertItem(hparent, after, entry, hdc, entry._hIcon, entry._mode);
}

void TrayNotifyDlg::InsertItem(HTREEITEM hparent, HTREEITEM after, const NotifyIconDlgInfo& entry,
                                HDC hdc, HICON hicon, NOTIFYICONMODE mode)
{
    int idx = _info.size() + 1;
    _info[idx] = entry;

    String mode_str = string_from_mode(mode);

    switch(mode) {
      case NIM_SHOW:    mode_str = ResString(IDS_NOTIFY_SHOW);      break;
      case NIM_HIDE:    mode_str = ResString(IDS_NOTIFY_HIDE);      break;
      case NIM_AUTO:    mode_str = ResString(IDS_NOTIFY_AUTOHIDE);
    }

    FmtString txt(TEXT("%s  -  %s  [%s]"), entry._tipText.c_str(), entry._windowTitle.c_str(), mode_str.c_str());

    TV_INSERTSTRUCT tvi;

    tvi.hParent = hparent;
    tvi.hInsertAfter = after;

    TV_ITEM& tv = tvi.item;
    tv.mask = TVIF_TEXT|TVIF_IMAGE|TVIF_SELECTEDIMAGE|TVIF_PARAM;

    tv.lParam = (LPARAM)idx;
    tv.pszText = txt.str();
    tv.iSelectedImage = tv.iImage = ImageList_AddAlphaIcon(_himl, hicon, GetStockBrush(WHITE_BRUSH), hdc);
    (void)TreeView_InsertItem(_tree_ctrl, &tvi);
}

LRESULT TrayNotifyDlg::WndProc(UINT nmsg, WPARAM wparam, LPARAM lparam)
{
    switch(nmsg) {
      case PM_TRANSLATE_MSG: {
        MSG* pmsg = (MSG*) lparam;

        if (TranslateAccelerator(_hwnd, _haccel, pmsg))
            return TRUE;

        return FALSE;}

      case WM_TIMER:
        Refresh();
        break;

      default:
        return super::WndProc(nmsg, wparam, lparam);
    }

    return 0;
}

int TrayNotifyDlg::Command(int id, int code)
{
    if (code == BN_CLICKED) {
        switch(id) {
          case ID_REFRESH:
            Refresh();
            break;

          case IDC_NOTIFY_SHOW:
            SetIconMode(NIM_SHOW);
            break;

          case IDC_NOTIFY_HIDE:
            SetIconMode(NIM_HIDE);
            break;

          case IDC_NOTIFY_AUTOHIDE:
            SetIconMode(NIM_AUTO);
            break;

          case ID_SHOW_HIDDEN_ICONS:
            if (_pNotifyArea)
                SendMessage(*_pNotifyArea, WM_COMMAND, MAKEWPARAM(id,code), 0);
            break;

          case IDOK:
            EndDialog(_hwnd, id);
            break;

          case IDCANCEL:
             // rollback changes
            if (_pNotifyArea) {
                 // restore original icon states and configuration data
                _pNotifyArea->_cfg = _cfg_org;
                _pNotifyArea->_show_hidden = _show_hidden_org;

                for(IconStateMap::const_iterator it=_icon_states_org.begin(); it!=_icon_states_org.end(); ++it) {
                    NotifyInfo& info = _pNotifyArea->_icon_map[it->first];

                    info._mode = it->second.first;
                    info._dwState = it->second.second;
                }

                SendMessage(*_pNotifyArea, PM_REFRESH, 0, 0);
            }

            EndDialog(_hwnd, id);
            break;
        }

        return 0;
    }

    return 1;
}

int TrayNotifyDlg::Notify(int id, NMHDR* pnmh)
{
    switch(pnmh->code) {
      case TVN_SELCHANGED: {
        NMTREEVIEW* pnmtv = (NMTREEVIEW*)pnmh;
        int idx = pnmtv->itemNew.lParam;

        if (idx) {
            RefreshProperties(_info[idx]);
            _selectedItem = pnmtv->itemNew.hItem;
        } else {
            /*
            SetDlgItemText(_hwnd, IDC_NOTIFY_TOOLTIP, NULL);
            SetDlgItemText(_hwnd, IDC_NOTIFY_TITLE, NULL);
            SetDlgItemText(_hwnd, IDC_NOTIFY_MODULE, NULL);
            */
            CheckRadioButton(_hwnd, IDC_NOTIFY_SHOW, IDC_NOTIFY_AUTOHIDE, 0);
        }
        break;}
    }

    return 0;
}

void TrayNotifyDlg::RefreshProperties(const NotifyIconDlgInfo& entry)
{
    SetDlgItemText(_hwnd, IDC_NOTIFY_TOOLTIP, entry._tipText);
    SetDlgItemText(_hwnd, IDC_NOTIFY_TITLE, entry._windowTitle);
    SetDlgItemText(_hwnd, IDC_NOTIFY_MODULE, entry._modulePath);

    CheckRadioButton(_hwnd, IDC_NOTIFY_SHOW, IDC_NOTIFY_AUTOHIDE, IDC_NOTIFY_SHOW+entry._mode);

    String change_str;
    if (entry._lastChange)
        change_str.printf(TEXT("before %d s"), (GetTickCount()-entry._lastChange+500)/1000);
    SetDlgItemText(_hwnd, IDC_LAST_CHANGE, change_str);

    HICON hicon = 0; //get_window_icon_big(entry._hWnd, false);

     // If we could not find an icon associated with the owner window, try to load one from the owning module.
    if (!hicon && !entry._modulePath.empty()) {
        hicon = ExtractIcon(g_Globals._hInstance, entry._modulePath, 0);

        if (!hicon) {
            SHFILEINFO sfi;

            if (SHGetFileInfo(entry._modulePath, 0, &sfi, sizeof(sfi), SHGFI_ICON|SHGFI_LARGEICON))
                hicon = sfi.hIcon;
        }
    }

    if (hicon) {
        SendMessage(GetDlgItem(_hwnd, IDC_PICTURE), STM_SETICON, (LPARAM)hicon, 0);
        DestroyIcon(hicon);
    } else
        SendMessage(GetDlgItem(_hwnd, IDC_PICTURE), STM_SETICON, 0, 0);
}

void TrayNotifyDlg::SetIconMode(NOTIFYICONMODE mode)
{
    int idx = TreeView_GetItemData(_tree_ctrl, _selectedItem);

    if (!idx)
        return;

    NotifyIconConfig& entry = _info[idx];

    if (entry._mode != mode) {
        entry._mode = mode;

         // trigger refresh in notify area and this dialog
        if (_pNotifyArea)
            SendMessage(*_pNotifyArea, PM_REFRESH, 0, 0);
    }

    if (_pNotifyArea) {
        bool found = false;

        NotifyIconCfgList& cfg = _pNotifyArea->_cfg;
        for(NotifyIconCfgList::iterator it=cfg.begin(); it!=cfg.end(); ++it) {
            NotifyIconConfig& cfg_entry = *it;

            if (cfg_entry.match(entry)) {
                cfg_entry._mode = mode;
                ++found;
                break;
            }
        }

        if (!found) {
             // insert new configuration entry
            NotifyIconConfig cfg_entry = entry;

            cfg_entry._mode = mode;

            _pNotifyArea->_cfg.push_back(cfg_entry);
        }
    }

    Refresh();
}


ClockWindow::ClockWindow(HWND hwnd)
 :  super(hwnd),
    _tooltip(hwnd)
{
    *_time = TEXT('\0');
    FormatTime();

    _tooltip.add(_hwnd, _hwnd);
}

HWND ClockWindow::Create(HWND hwndParent)
{
    static BtnWindowClass wcClock(CLASSNAME_CLOCKWINDOW, CS_DBLCLKS);

    ClientRect clnt(hwndParent);

    WindowCanvas canvas(hwndParent);
    FontSelection font(canvas, GetStockFont(ANSI_VAR_FONT));

    RECT rect = {0, 0, 0, 0};
    TCHAR buffer[16];
    // Arbitrary high time so that the created clock window is big enough
    SYSTEMTIME st = { 1601, 1, 0, 1, 23, 59, 59, 999 };

    if (!GetTimeFormat(LOCALE_USER_DEFAULT, TIME_NOSECONDS, &st, NULL, buffer, sizeof(buffer)/sizeof(TCHAR)))
        _tcscpy(buffer, TEXT("00:00"));

    // Calculate the rectangle needed to draw the time (without actually drawing it)
    DrawText(canvas, buffer, -1, &rect, DT_SINGLELINE|DT_NOPREFIX|DT_CALCRECT);
    int clockwindowWidth = rect.right-rect.left + 4;

    return Window::Create(WINDOW_CREATOR(ClockWindow), 0,
                            wcClock, NULL, WS_CHILD|WS_VISIBLE,
                            clnt.right-(clockwindowWidth), 1, clockwindowWidth, clnt.bottom-2, hwndParent);
}

LRESULT ClockWindow::WndProc(UINT nmsg, WPARAM wparam, LPARAM lparam)
{
    switch(nmsg) {
      case WM_PAINT:
        Paint();
        break;

      case WM_LBUTTONDBLCLK:
        launch_cpanel(_hwnd, TEXT("timedate.cpl"));
        break;

      default:
        return super::WndProc(nmsg, wparam, lparam);
    }

    return 0;
}

int ClockWindow::Notify(int id, NMHDR* pnmh)
{
    if (pnmh->code == TTN_GETDISPINFO) {
        LPNMTTDISPINFO pdi = (LPNMTTDISPINFO)pnmh;

        SYSTEMTIME systime;
        TCHAR buffer[64];

        GetLocalTime(&systime);

        if (GetDateFormat(LOCALE_USER_DEFAULT, DATE_LONGDATE, &systime, NULL, buffer, 64))
            _tcscpy(pdi->szText, buffer);
        else
            pdi->szText[0] = '\0';
    }

    return 0;
}

void ClockWindow::TimerTick()
{
    if (FormatTime())
        InvalidateRect(_hwnd, NULL, TRUE);  // refresh displayed time
}

bool ClockWindow::FormatTime()
{
    TCHAR buffer[16];

    if (GetTimeFormat(LOCALE_USER_DEFAULT, TIME_NOSECONDS, NULL, NULL, buffer, sizeof(buffer)/sizeof(TCHAR)))
        if (_tcscmp(buffer, _time)) {
            _tcscpy(_time, buffer);
            return true;    // The text to display has changed.
        }

    return false;   // no change
}

void ClockWindow::Paint()
{
    PaintCanvas canvas(_hwnd);

    FillRect(canvas, &canvas.rcPaint, GetSysColorBrush(COLOR_BTNFACE));

    BkMode bkmode(canvas, TRANSPARENT);
    FontSelection font(canvas, GetStockFont(ANSI_VAR_FONT));

    DrawText(canvas, _time, -1, ClientRect(_hwnd), DT_SINGLELINE|DT_VCENTER|DT_NOPREFIX);
}