mirror of
https://github.com/owntone/owntone-server.git
synced 2025-11-10 14:09:51 -05:00
Merge from dev-FireflyShell branch
This commit is contained in:
Ron Pedde
270
win32/FireflyShell/NotifyIcon.cpp
Normal file
270
win32/FireflyShell/NotifyIcon.cpp
Normal file
@@ -0,0 +1,270 @@
|
||||
/*
|
||||
*(C) 2006 Roku LLC
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify it
|
||||
* under the terms of the GNU General Public License Version 2 as published
|
||||
* by the Free Software Foundation.
|
||||
*
|
||||
* This program 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 General
|
||||
* Public License for more details.
|
||||
*
|
||||
* Please read README.txt in the same directory as this source file for
|
||||
* further license information.
|
||||
*/
|
||||
|
||||
#include "stdafx.h"
|
||||
#include "resource.h"
|
||||
#include "NotifyIcon.h"
|
||||
#include "FireflyShell.h"
|
||||
|
||||
CNotifyIcon::CNotifyIcon()
|
||||
{
|
||||
ZeroMemory(&m_nid, sizeof(NOTIFYICONDATA));
|
||||
m_nid.cbSize = sizeof(NOTIFYICONDATA);
|
||||
m_nid.uID = ID_SHELLNOTIFY;
|
||||
|
||||
m_running_icon = AtlLoadIcon(IDI_SHELL_RUNNING);
|
||||
m_stopped_icon = AtlLoadIcon(IDI_SHELL_STOPPED);
|
||||
}
|
||||
|
||||
BOOL CNotifyIcon::Create()
|
||||
{
|
||||
m_registered_activation_message = GetApplication()->GetRegisteredActivationMessage();
|
||||
|
||||
RECT rect = {0, 0, 0, 0};
|
||||
|
||||
// Hidden window.
|
||||
if (base::Create(NULL, rect, _T("FireflyShellNotifyIconHidden"), WS_POPUP))
|
||||
{
|
||||
m_nid.uFlags = NIF_ICON|NIF_MESSAGE|NIF_TIP;
|
||||
|
||||
InflictIconState();
|
||||
|
||||
//wcsncpy(niData.szTip, TEXT("Foo?"), sizeof(niData.szTip));
|
||||
m_nid.hWnd = m_hWnd;
|
||||
m_nid.uCallbackMessage = PRIVATE_WM_NOTIFYICON;
|
||||
|
||||
Shell_NotifyIcon(NIM_ADD, &m_nid);
|
||||
|
||||
SetTimer(TIMER_ID, 5000, NULL);
|
||||
|
||||
GetApplication()->ServiceStatusSubscribe(this);
|
||||
|
||||
EnableUserSwitchNotifications();
|
||||
|
||||
return TRUE;
|
||||
}
|
||||
|
||||
return FALSE;
|
||||
}
|
||||
|
||||
void CNotifyIcon::Destroy()
|
||||
{
|
||||
GetApplication()->ServiceStatusUnsubscribe(this);
|
||||
KillTimer(TIMER_ID);
|
||||
Shell_NotifyIcon(NIM_DELETE, &m_nid);
|
||||
DestroyIcon(m_nid.hIcon);
|
||||
|
||||
|
||||
base::DestroyWindow();
|
||||
}
|
||||
|
||||
void CNotifyIcon::PopupBalloon(UINT title_id, UINT text_id, DWORD flags)
|
||||
{
|
||||
CString title, text;
|
||||
title.LoadString(title_id);
|
||||
text.LoadString(text_id);
|
||||
|
||||
m_nid.uFlags |= NIF_INFO;
|
||||
|
||||
SafeStringCopy(m_nid.szInfoTitle, title, 64);
|
||||
SafeStringCopy(m_nid.szInfo, text, 256);
|
||||
|
||||
m_nid.dwInfoFlags = flags;
|
||||
m_nid.uTimeout = 10000;
|
||||
Shell_NotifyIcon(NIM_MODIFY, &m_nid);
|
||||
}
|
||||
|
||||
void CNotifyIcon::Update()
|
||||
{
|
||||
InflictIconState();
|
||||
// I suspect we'll need this line too.
|
||||
// m_nid.uFlags &= ~NIF_INFO;
|
||||
Shell_NotifyIcon(NIM_MODIFY, &m_nid);
|
||||
}
|
||||
|
||||
void CNotifyIcon::InflictIconState()
|
||||
{
|
||||
// Will the icons leak?
|
||||
Service::Status status = GetApplication()->GetServiceStatus();
|
||||
UINT state_id;
|
||||
if (status.IsPending())
|
||||
{
|
||||
state_id = IDS_SERVER_PENDING;
|
||||
m_nid.hIcon = m_stopped_icon; // As good as any?
|
||||
}
|
||||
else if (status.IsRunning())
|
||||
{
|
||||
state_id = IDS_SERVER_RUNNING;
|
||||
m_nid.hIcon = m_running_icon;
|
||||
}
|
||||
else
|
||||
{
|
||||
state_id = IDS_SERVER_STOPPED;
|
||||
m_nid.hIcon = m_stopped_icon;
|
||||
}
|
||||
|
||||
CString tip;
|
||||
tip.LoadString(state_id);
|
||||
SafeStringCopy(m_nid.szTip, tip, 64);
|
||||
}
|
||||
|
||||
void CNotifyIcon::OnServiceStatus(Service::Status old_status, Service::Status new_status)
|
||||
{
|
||||
Update();
|
||||
}
|
||||
|
||||
void CNotifyIcon::OnTimer(UINT id, TIMERPROC proc)
|
||||
{
|
||||
if (id == TIMER_ID)
|
||||
{
|
||||
GetApplication()->CheckServiceStatus();
|
||||
}
|
||||
}
|
||||
|
||||
LRESULT CNotifyIcon::OnNotifyIconMessage(UINT uMsg, WPARAM wParam, LPARAM lParam, BOOL& bHandled)
|
||||
{
|
||||
switch (lParam)
|
||||
{
|
||||
case WM_LBUTTONDBLCLK:
|
||||
GetApplication()->Configure();
|
||||
bHandled = true;
|
||||
return 0L;
|
||||
case WM_RBUTTONDOWN:
|
||||
case WM_CONTEXTMENU:
|
||||
OnContextMenu();
|
||||
bHandled = true;
|
||||
return 0L;
|
||||
}
|
||||
|
||||
return 0L;
|
||||
}
|
||||
|
||||
void CNotifyIcon::OnContextMenu()
|
||||
{
|
||||
HMENU hMenu;
|
||||
|
||||
HINSTANCE hInstance = _Module.GetResourceInstance();
|
||||
hMenu = ::LoadMenu(hInstance, MAKEINTRESOURCE(IDM_CONTEXT));
|
||||
|
||||
POINT pt;
|
||||
GetCursorPos(&pt);
|
||||
|
||||
// See TrackPopupMenu in MSDN.
|
||||
SetForegroundWindow(m_hWnd);
|
||||
//::SetForegroundWindow(m_hWnd);
|
||||
HMENU hPopup = GetSubMenu(hMenu, 0);
|
||||
::SetMenuDefaultItem(hPopup, ID_CONFIGURE, FALSE);
|
||||
TrackPopupMenu(hPopup, TPM_LEFTALIGN | TPM_BOTTOMALIGN, pt.x, pt.y, 0, m_hWnd, NULL);
|
||||
::PostMessage(m_hWnd, WM_NULL, 0, 0);
|
||||
|
||||
::DestroyMenu(hMenu);
|
||||
}
|
||||
|
||||
LRESULT CNotifyIcon::OnConfigure(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/)
|
||||
{
|
||||
GetApplication()->Configure();
|
||||
return 0;
|
||||
}
|
||||
|
||||
LRESULT CNotifyIcon::OnExit(WORD /*wNotifyCode*/, WORD wID, HWND /*hWndCtl*/, BOOL& /*bHandled*/)
|
||||
{
|
||||
/// @todo
|
||||
Application::GetInstance()->Exit();
|
||||
return 0;
|
||||
}
|
||||
|
||||
LRESULT CNotifyIcon::OnRegisteredActivation(UINT, WPARAM, LPARAM, BOOL &bHandled)
|
||||
{
|
||||
ATLTRACE(_T("Activate\n"));
|
||||
bHandled = true;
|
||||
GetApplication()->Configure();
|
||||
|
||||
// We return a magic number so that the caller knows we've been found
|
||||
// and can give up.
|
||||
return m_registered_activation_message;
|
||||
}
|
||||
|
||||
void CNotifyIcon::OnServerEvent(UINT32 id, UINT32 intval, const CString &strval)
|
||||
{
|
||||
// Note that we're running on a different thread here. We need to punt
|
||||
// the event over to the main thread using SendMessage.
|
||||
switch (id)
|
||||
{
|
||||
case 0:
|
||||
break;
|
||||
case 1:
|
||||
case 2:
|
||||
SendMessage(WM_SERVEREVENT, id);
|
||||
break;
|
||||
default:
|
||||
ATLASSERT(false);
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
LRESULT CNotifyIcon::OnServerEvent(UINT, WPARAM wparam, LPARAM, BOOL &bHandled)
|
||||
{
|
||||
bHandled = true;
|
||||
switch (wparam)
|
||||
{
|
||||
case 1:
|
||||
PopupBalloon(IDR_MAINFRAME, IDS_SCAN_START);
|
||||
break;
|
||||
case 2:
|
||||
PopupBalloon(IDR_MAINFRAME, IDS_SCAN_STOP);
|
||||
break;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
void CNotifyIcon::EnableUserSwitchNotifications()
|
||||
{
|
||||
HMODULE h = ::LoadLibrary(_T("WtsApi32.dll"));
|
||||
if (h)
|
||||
{
|
||||
typedef BOOL (WINAPI *Proc)(HWND, DWORD);
|
||||
Proc fn = reinterpret_cast<Proc>(GetProcAddress(h, "WTSRegisterSessionNotification"));
|
||||
if (fn)
|
||||
{
|
||||
(*fn)(m_hWnd, NOTIFY_FOR_THIS_SESSION);
|
||||
}
|
||||
::FreeLibrary(h);
|
||||
}
|
||||
}
|
||||
|
||||
LRESULT CNotifyIcon::OnSessionChange(UINT, WPARAM wparam, LPARAM, BOOL &bHandled)
|
||||
{
|
||||
// Because only one process can get events through the mailslot we
|
||||
// disconnect from it when the user uses XP fast user switching to
|
||||
// switch to a different user.
|
||||
switch (wparam)
|
||||
{
|
||||
case WTS_CONSOLE_CONNECT:
|
||||
case WTS_REMOTE_CONNECT:
|
||||
ATLTRACE("SESSION CONNECT\n");
|
||||
GetApplication()->EnableServerEvents(true);
|
||||
break;
|
||||
|
||||
case WTS_CONSOLE_DISCONNECT:
|
||||
case WTS_REMOTE_DISCONNECT:
|
||||
ATLTRACE("SESSION DISCONNECT\n");
|
||||
GetApplication()->EnableServerEvents(false);
|
||||
break;
|
||||
}
|
||||
bHandled = true;
|
||||
return 0;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user