828 lines
22 KiB
C++
828 lines
22 KiB
C++
/*
|
|
This file is part of Telegram Desktop,
|
|
the official desktop application for the Telegram messaging service.
|
|
|
|
For license and copyright information please follow this link:
|
|
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
|
|
*/
|
|
#include "platform/win/notifications_manager_win.h"
|
|
|
|
#include "window/notifications_utilities.h"
|
|
#include "window/window_session_controller.h"
|
|
#include "base/platform/win/base_windows_wrl.h"
|
|
#include "base/platform/base_platform_info.h"
|
|
#include "platform/win/windows_app_user_model_id.h"
|
|
#include "platform/win/windows_event_filter.h"
|
|
#include "platform/win/windows_dlls.h"
|
|
#include "history/history.h"
|
|
#include "core/application.h"
|
|
#include "core/core_settings.h"
|
|
#include "main/main_session.h"
|
|
#include "mainwindow.h"
|
|
#include "windows_quiethours_h.h"
|
|
|
|
#include <Shobjidl.h>
|
|
#include <shellapi.h>
|
|
|
|
#ifndef __MINGW32__
|
|
#include "base/platform/win/wrl/wrl_implements_h.h"
|
|
#include <windows.ui.notifications.h>
|
|
|
|
HICON qt_pixmapToWinHICON(const QPixmap &);
|
|
|
|
using namespace Microsoft::WRL;
|
|
using namespace ABI::Windows::UI::Notifications;
|
|
using namespace ABI::Windows::Data::Xml::Dom;
|
|
using namespace Windows::Foundation;
|
|
#endif // !__MINGW32__
|
|
|
|
namespace Platform {
|
|
namespace Notifications {
|
|
|
|
#ifndef __MINGW32__
|
|
namespace {
|
|
|
|
using base::Platform::GetActivationFactory;
|
|
using base::Platform::StringReferenceWrapper;
|
|
|
|
bool init() {
|
|
if (!IsWindows8OrGreater()) {
|
|
return false;
|
|
}
|
|
if ((Dlls::SetCurrentProcessExplicitAppUserModelID == nullptr)
|
|
|| (Dlls::PropVariantToString == nullptr)
|
|
|| !base::Platform::SupportsWRL()) {
|
|
return false;
|
|
}
|
|
|
|
if (!AppUserModelId::validateShortcut()) {
|
|
return false;
|
|
}
|
|
|
|
auto appUserModelId = AppUserModelId::getId();
|
|
if (!SUCCEEDED(Dlls::SetCurrentProcessExplicitAppUserModelID(appUserModelId))) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
HRESULT SetNodeValueString(_In_ HSTRING inputString, _In_ IXmlNode *node, _In_ IXmlDocument *xml) {
|
|
ComPtr<IXmlText> inputText;
|
|
|
|
HRESULT hr = xml->CreateTextNode(inputString, &inputText);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
ComPtr<IXmlNode> inputTextNode;
|
|
|
|
hr = inputText.As(&inputTextNode);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
ComPtr<IXmlNode> pAppendedChild;
|
|
return node->AppendChild(inputTextNode.Get(), &pAppendedChild);
|
|
}
|
|
|
|
HRESULT SetAudioSilent(_In_ IXmlDocument *toastXml) {
|
|
ComPtr<IXmlNodeList> nodeList;
|
|
HRESULT hr = toastXml->GetElementsByTagName(StringReferenceWrapper(L"audio").Get(), &nodeList);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
ComPtr<IXmlNode> audioNode;
|
|
hr = nodeList->Item(0, &audioNode);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
if (audioNode) {
|
|
ComPtr<IXmlElement> audioElement;
|
|
hr = audioNode.As(&audioElement);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
hr = audioElement->SetAttribute(StringReferenceWrapper(L"silent").Get(), StringReferenceWrapper(L"true").Get());
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
} else {
|
|
ComPtr<IXmlElement> audioElement;
|
|
hr = toastXml->CreateElement(StringReferenceWrapper(L"audio").Get(), &audioElement);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
hr = audioElement->SetAttribute(StringReferenceWrapper(L"silent").Get(), StringReferenceWrapper(L"true").Get());
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
ComPtr<IXmlNode> audioNode;
|
|
hr = audioElement.As(&audioNode);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
ComPtr<IXmlNodeList> nodeList;
|
|
hr = toastXml->GetElementsByTagName(StringReferenceWrapper(L"toast").Get(), &nodeList);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
ComPtr<IXmlNode> toastNode;
|
|
hr = nodeList->Item(0, &toastNode);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
ComPtr<IXmlNode> appendedNode;
|
|
hr = toastNode->AppendChild(audioNode.Get(), &appendedNode);
|
|
}
|
|
return hr;
|
|
}
|
|
|
|
HRESULT SetImageSrc(_In_z_ const wchar_t *imagePath, _In_ IXmlDocument *toastXml) {
|
|
wchar_t imageSrc[MAX_PATH] = L"file:///";
|
|
HRESULT hr = StringCchCat(imageSrc, ARRAYSIZE(imageSrc), imagePath);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
ComPtr<IXmlNodeList> nodeList;
|
|
hr = toastXml->GetElementsByTagName(StringReferenceWrapper(L"image").Get(), &nodeList);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
ComPtr<IXmlNode> imageNode;
|
|
hr = nodeList->Item(0, &imageNode);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
ComPtr<IXmlNamedNodeMap> attributes;
|
|
hr = imageNode->get_Attributes(&attributes);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
ComPtr<IXmlNode> srcAttribute;
|
|
hr = attributes->GetNamedItem(StringReferenceWrapper(L"src").Get(), &srcAttribute);
|
|
if (!SUCCEEDED(hr)) return hr;
|
|
|
|
return SetNodeValueString(StringReferenceWrapper(imageSrc).Get(), srcAttribute.Get(), toastXml);
|
|
}
|
|
|
|
typedef ABI::Windows::Foundation::ITypedEventHandler<ToastNotification*, ::IInspectable *> DesktopToastActivatedEventHandler;
|
|
typedef ABI::Windows::Foundation::ITypedEventHandler<ToastNotification*, ToastDismissedEventArgs*> DesktopToastDismissedEventHandler;
|
|
typedef ABI::Windows::Foundation::ITypedEventHandler<ToastNotification*, ToastFailedEventArgs*> DesktopToastFailedEventHandler;
|
|
|
|
class ToastEventHandler final : public Implements<
|
|
DesktopToastActivatedEventHandler,
|
|
DesktopToastDismissedEventHandler,
|
|
DesktopToastFailedEventHandler> {
|
|
public:
|
|
using NotificationId = Manager::NotificationId;
|
|
|
|
// We keep a weak pointer to a member field of native notifications manager.
|
|
ToastEventHandler(
|
|
const std::shared_ptr<Manager*> &guarded,
|
|
NotificationId id)
|
|
: _id(id)
|
|
, _weak(guarded) {
|
|
}
|
|
|
|
void performOnMainQueue(FnMut<void(Manager *manager)> task) {
|
|
const auto weak = _weak;
|
|
crl::on_main(weak, [=, task = std::move(task)]() mutable {
|
|
task(*weak.lock());
|
|
});
|
|
}
|
|
|
|
// DesktopToastActivatedEventHandler
|
|
IFACEMETHODIMP Invoke(_In_ IToastNotification *sender, _In_ IInspectable* args) {
|
|
const auto my = _id;
|
|
performOnMainQueue([my](Manager *manager) {
|
|
manager->notificationActivated(my);
|
|
});
|
|
return S_OK;
|
|
}
|
|
|
|
// DesktopToastDismissedEventHandler
|
|
IFACEMETHODIMP Invoke(_In_ IToastNotification *sender, _In_ IToastDismissedEventArgs *e) {
|
|
ToastDismissalReason tdr;
|
|
if (SUCCEEDED(e->get_Reason(&tdr))) {
|
|
switch (tdr) {
|
|
case ToastDismissalReason_ApplicationHidden:
|
|
break;
|
|
case ToastDismissalReason_UserCanceled:
|
|
case ToastDismissalReason_TimedOut:
|
|
default:
|
|
const auto my = _id;
|
|
performOnMainQueue([my](Manager *manager) {
|
|
manager->clearNotification(my);
|
|
});
|
|
break;
|
|
}
|
|
}
|
|
return S_OK;
|
|
}
|
|
|
|
// DesktopToastFailedEventHandler
|
|
IFACEMETHODIMP Invoke(_In_ IToastNotification *sender, _In_ IToastFailedEventArgs *e) {
|
|
const auto my = _id;
|
|
performOnMainQueue([my](Manager *manager) {
|
|
manager->clearNotification(my);
|
|
});
|
|
return S_OK;
|
|
}
|
|
|
|
// IUnknown
|
|
IFACEMETHODIMP_(ULONG) AddRef() {
|
|
return InterlockedIncrement(&_refCount);
|
|
}
|
|
|
|
IFACEMETHODIMP_(ULONG) Release() {
|
|
auto refCount = InterlockedDecrement(&_refCount);
|
|
if (refCount == 0) {
|
|
delete this;
|
|
}
|
|
return refCount;
|
|
}
|
|
|
|
IFACEMETHODIMP QueryInterface(_In_ REFIID riid, _COM_Outptr_ void **ppv) {
|
|
if (IsEqualIID(riid, IID_IUnknown))
|
|
*ppv = static_cast<IUnknown*>(static_cast<DesktopToastActivatedEventHandler*>(this));
|
|
else if (IsEqualIID(riid, __uuidof(DesktopToastActivatedEventHandler)))
|
|
*ppv = static_cast<DesktopToastActivatedEventHandler*>(this);
|
|
else if (IsEqualIID(riid, __uuidof(DesktopToastDismissedEventHandler)))
|
|
*ppv = static_cast<DesktopToastDismissedEventHandler*>(this);
|
|
else if (IsEqualIID(riid, __uuidof(DesktopToastFailedEventHandler)))
|
|
*ppv = static_cast<DesktopToastFailedEventHandler*>(this);
|
|
else *ppv = nullptr;
|
|
|
|
if (*ppv) {
|
|
reinterpret_cast<IUnknown*>(*ppv)->AddRef();
|
|
return S_OK;
|
|
}
|
|
|
|
return E_NOINTERFACE;
|
|
}
|
|
|
|
private:
|
|
ULONG _refCount = 0;
|
|
NotificationId _id;
|
|
std::weak_ptr<Manager*> _weak;
|
|
|
|
};
|
|
|
|
auto Checked = false;
|
|
auto InitSucceeded = false;
|
|
|
|
void Check() {
|
|
InitSucceeded = init();
|
|
}
|
|
|
|
bool QuietHoursEnabled = false;
|
|
DWORD QuietHoursValue = 0;
|
|
|
|
[[nodiscard]] bool UseQuietHoursRegistryEntry() {
|
|
static const bool result = [] {
|
|
// Taken from QSysInfo.
|
|
OSVERSIONINFO result = { sizeof(OSVERSIONINFO), 0, 0, 0, 0,{ '\0' } };
|
|
if (const auto library = GetModuleHandle(L"ntdll.dll")) {
|
|
using RtlGetVersionFunction = NTSTATUS(NTAPI*)(LPOSVERSIONINFO);
|
|
const auto RtlGetVersion = reinterpret_cast<RtlGetVersionFunction>(
|
|
GetProcAddress(library, "RtlGetVersion"));
|
|
if (RtlGetVersion) {
|
|
RtlGetVersion(&result);
|
|
}
|
|
}
|
|
// At build 17134 (Redstone 4) the "Quiet hours" was replaced
|
|
// by "Focus assist" and it looks like it doesn't use registry.
|
|
return (result.dwMajorVersion == 10
|
|
&& result.dwMinorVersion == 0
|
|
&& result.dwBuildNumber < 17134);
|
|
}();
|
|
return result;
|
|
}
|
|
|
|
// Thanks https://stackoverflow.com/questions/35600128/get-windows-quiet-hours-from-win32-or-c-sharp-api
|
|
void QueryQuietHours() {
|
|
if (!UseQuietHoursRegistryEntry()) {
|
|
// There are quiet hours in Windows starting from Windows 8.1
|
|
// But there were several reports about the notifications being shut
|
|
// down according to the registry while no quiet hours were enabled.
|
|
// So we try this method only starting with Windows 10.
|
|
return;
|
|
}
|
|
|
|
LPCWSTR lpKeyName = L"Software\\Microsoft\\Windows\\CurrentVersion\\Notifications\\Settings";
|
|
LPCWSTR lpValueName = L"NOC_GLOBAL_SETTING_TOASTS_ENABLED";
|
|
HKEY key;
|
|
auto result = RegOpenKeyEx(HKEY_CURRENT_USER, lpKeyName, 0, KEY_READ, &key);
|
|
if (result != ERROR_SUCCESS) {
|
|
return;
|
|
}
|
|
|
|
DWORD value = 0, type = 0, size = sizeof(value);
|
|
result = RegQueryValueEx(key, lpValueName, 0, &type, (LPBYTE)&value, &size);
|
|
RegCloseKey(key);
|
|
|
|
auto quietHoursEnabled = (result == ERROR_SUCCESS) && (value == 0);
|
|
if (QuietHoursEnabled != quietHoursEnabled) {
|
|
QuietHoursEnabled = quietHoursEnabled;
|
|
QuietHoursValue = value;
|
|
LOG(("Quiet hours changed, entry value: %1").arg(value));
|
|
} else if (QuietHoursValue != value) {
|
|
QuietHoursValue = value;
|
|
LOG(("Quiet hours value changed, was value: %1, entry value: %2").arg(QuietHoursValue).arg(value));
|
|
}
|
|
}
|
|
|
|
bool FocusAssistBlocks = false;
|
|
|
|
// Thanks https://www.withinrafael.com/2019/09/19/determine-if-your-app-is-in-a-focus-assist-profiles-priority-list/
|
|
void QueryFocusAssist() {
|
|
ComPtr<IQuietHoursSettings> quietHoursSettings;
|
|
auto hr = CoCreateInstance(
|
|
CLSID_QuietHoursSettings,
|
|
nullptr,
|
|
CLSCTX_LOCAL_SERVER,
|
|
IID_PPV_ARGS(&quietHoursSettings));
|
|
if (!SUCCEEDED(hr) || !quietHoursSettings) {
|
|
return;
|
|
}
|
|
|
|
auto profileId = LPWSTR{};
|
|
const auto guardProfileId = gsl::finally([&] {
|
|
if (profileId) CoTaskMemFree(profileId);
|
|
});
|
|
hr = quietHoursSettings->get_UserSelectedProfile(&profileId);
|
|
if (!SUCCEEDED(hr) || !profileId) {
|
|
return;
|
|
}
|
|
const auto profileName = QString::fromWCharArray(profileId);
|
|
if (profileName.endsWith(".alarmsonly", Qt::CaseInsensitive)) {
|
|
if (!FocusAssistBlocks) {
|
|
LOG(("Focus Assist: Alarms Only."));
|
|
FocusAssistBlocks = true;
|
|
}
|
|
return;
|
|
} else if (!profileName.endsWith(".priorityonly", Qt::CaseInsensitive)) {
|
|
if (!profileName.endsWith(".unrestricted", Qt::CaseInsensitive)) {
|
|
LOG(("Focus Assist Warning: Unknown profile '%1'"
|
|
).arg(profileName));
|
|
}
|
|
if (FocusAssistBlocks) {
|
|
LOG(("Focus Assist: Unrestricted."));
|
|
FocusAssistBlocks = false;
|
|
}
|
|
return;
|
|
}
|
|
const auto appUserModelId = std::wstring(AppUserModelId::getId());
|
|
auto blocked = true;
|
|
const auto guard = gsl::finally([&] {
|
|
if (FocusAssistBlocks != blocked) {
|
|
LOG(("Focus Assist: %1, AppUserModelId: %2, Blocks: %3"
|
|
).arg(profileName
|
|
).arg(QString::fromStdWString(appUserModelId)
|
|
).arg(Logs::b(blocked)));
|
|
FocusAssistBlocks = blocked;
|
|
}
|
|
});
|
|
|
|
ComPtr<IQuietHoursProfile> profile;
|
|
hr = quietHoursSettings->GetProfile(profileId, &profile);
|
|
if (!SUCCEEDED(hr) || !profile) {
|
|
return;
|
|
}
|
|
|
|
UINT32 count = 0;
|
|
auto apps = (LPWSTR*)nullptr;
|
|
const auto guardApps = gsl::finally([&] {
|
|
if (apps) CoTaskMemFree(apps);
|
|
});
|
|
hr = profile->GetAllowedApps(&count, &apps);
|
|
if (!SUCCEEDED(hr) || !apps) {
|
|
return;
|
|
}
|
|
for (UINT32 i = 0; i < count; i++) {
|
|
auto app = apps[i];
|
|
const auto guardApp = gsl::finally([&] {
|
|
if (app) CoTaskMemFree(app);
|
|
});
|
|
if (app == appUserModelId) {
|
|
blocked = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
QUERY_USER_NOTIFICATION_STATE UserNotificationState = QUNS_ACCEPTS_NOTIFICATIONS;
|
|
|
|
void QueryUserNotificationState() {
|
|
if (Dlls::SHQueryUserNotificationState != nullptr) {
|
|
QUERY_USER_NOTIFICATION_STATE state;
|
|
if (SUCCEEDED(Dlls::SHQueryUserNotificationState(&state))) {
|
|
UserNotificationState = state;
|
|
}
|
|
}
|
|
}
|
|
|
|
static constexpr auto kQuerySettingsEachMs = 1000;
|
|
crl::time LastSettingsQueryMs = 0;
|
|
|
|
void QuerySystemNotificationSettings() {
|
|
auto ms = crl::now();
|
|
if (LastSettingsQueryMs > 0 && ms <= LastSettingsQueryMs + kQuerySettingsEachMs) {
|
|
return;
|
|
}
|
|
LastSettingsQueryMs = ms;
|
|
QueryQuietHours();
|
|
QueryFocusAssist();
|
|
QueryUserNotificationState();
|
|
}
|
|
|
|
} // namespace
|
|
#endif // !__MINGW32__
|
|
|
|
bool SkipAudioForCustom() {
|
|
QuerySystemNotificationSettings();
|
|
|
|
return (UserNotificationState == QUNS_NOT_PRESENT)
|
|
|| (UserNotificationState == QUNS_PRESENTATION_MODE)
|
|
|| Core::App().screenIsLocked();
|
|
}
|
|
|
|
bool SkipToastForCustom() {
|
|
QuerySystemNotificationSettings();
|
|
|
|
return (UserNotificationState == QUNS_PRESENTATION_MODE)
|
|
|| (UserNotificationState == QUNS_RUNNING_D3D_FULL_SCREEN);
|
|
}
|
|
|
|
bool SkipFlashBounceForCustom() {
|
|
return SkipToastForCustom();
|
|
}
|
|
|
|
bool Supported() {
|
|
#ifndef __MINGW32__
|
|
if (!Checked) {
|
|
Checked = true;
|
|
Check();
|
|
}
|
|
return InitSucceeded;
|
|
#endif // !__MINGW32__
|
|
|
|
return false;
|
|
}
|
|
|
|
bool Enforced() {
|
|
return false;
|
|
}
|
|
|
|
bool ByDefault() {
|
|
return false;
|
|
}
|
|
|
|
void Create(Window::Notifications::System *system) {
|
|
#ifndef __MINGW32__
|
|
if (Core::App().settings().nativeNotifications() && Supported()) {
|
|
auto result = std::make_unique<Manager>(system);
|
|
if (result->init()) {
|
|
system->setManager(std::move(result));
|
|
return;
|
|
}
|
|
}
|
|
#endif // !__MINGW32__
|
|
system->setManager(nullptr);
|
|
}
|
|
|
|
#ifndef __MINGW32__
|
|
class Manager::Private {
|
|
public:
|
|
using Type = Window::Notifications::CachedUserpics::Type;
|
|
|
|
explicit Private(Manager *instance, Type type);
|
|
bool init();
|
|
|
|
bool showNotification(
|
|
not_null<PeerData*> peer,
|
|
std::shared_ptr<Data::CloudImageView> &userpicView,
|
|
MsgId msgId,
|
|
const QString &title,
|
|
const QString &subtitle,
|
|
const QString &msg,
|
|
bool hideNameAndPhoto,
|
|
bool hideReplyButton);
|
|
void clearAll();
|
|
void clearFromHistory(not_null<History*> history);
|
|
void clearFromSession(not_null<Main::Session*> session);
|
|
void beforeNotificationActivated(NotificationId id);
|
|
void afterNotificationActivated(
|
|
NotificationId id,
|
|
not_null<Window::SessionController*> window);
|
|
void clearNotification(NotificationId id);
|
|
|
|
~Private();
|
|
|
|
private:
|
|
Window::Notifications::CachedUserpics _cachedUserpics;
|
|
|
|
std::shared_ptr<Manager*> _guarded;
|
|
|
|
ComPtr<IToastNotificationManagerStatics> _notificationManager;
|
|
ComPtr<IToastNotifier> _notifier;
|
|
ComPtr<IToastNotificationFactory> _notificationFactory;
|
|
|
|
struct NotificationPtr {
|
|
NotificationPtr() {
|
|
}
|
|
NotificationPtr(const ComPtr<IToastNotification> &ptr) : p(ptr) {
|
|
}
|
|
|
|
ComPtr<IToastNotification> p;
|
|
};
|
|
base::flat_map<FullPeer, base::flat_map<MsgId, NotificationPtr>> _notifications;
|
|
|
|
};
|
|
|
|
Manager::Private::Private(Manager *instance, Type type)
|
|
: _cachedUserpics(type)
|
|
, _guarded(std::make_shared<Manager*>(instance)) {
|
|
}
|
|
|
|
bool Manager::Private::init() {
|
|
if (!SUCCEEDED(GetActivationFactory(StringReferenceWrapper(RuntimeClass_Windows_UI_Notifications_ToastNotificationManager).Get(), &_notificationManager))) {
|
|
return false;
|
|
}
|
|
|
|
auto appUserModelId = AppUserModelId::getId();
|
|
if (!SUCCEEDED(_notificationManager->CreateToastNotifierWithId(StringReferenceWrapper(appUserModelId, wcslen(appUserModelId)).Get(), &_notifier))) {
|
|
return false;
|
|
}
|
|
|
|
if (!SUCCEEDED(GetActivationFactory(StringReferenceWrapper(RuntimeClass_Windows_UI_Notifications_ToastNotification).Get(), &_notificationFactory))) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
Manager::Private::~Private() {
|
|
clearAll();
|
|
|
|
_notifications.clear();
|
|
if (_notificationManager) _notificationManager.Reset();
|
|
if (_notifier) _notifier.Reset();
|
|
if (_notificationFactory) _notificationFactory.Reset();
|
|
}
|
|
|
|
void Manager::Private::clearAll() {
|
|
if (!_notifier) {
|
|
return;
|
|
}
|
|
|
|
auto temp = base::take(_notifications);
|
|
for (const auto &[key, notifications] : base::take(_notifications)) {
|
|
for (const auto &[msgId, notification] : notifications) {
|
|
_notifier->Hide(notification.p.Get());
|
|
}
|
|
}
|
|
}
|
|
|
|
void Manager::Private::clearFromHistory(not_null<History*> history) {
|
|
if (!_notifier) {
|
|
return;
|
|
}
|
|
|
|
auto i = _notifications.find(FullPeer{
|
|
.sessionId = history->session().uniqueId(),
|
|
.peerId = history->peer->id
|
|
});
|
|
if (i != _notifications.cend()) {
|
|
auto temp = base::take(i->second);
|
|
_notifications.erase(i);
|
|
|
|
for (const auto &[msgId, notification] : temp) {
|
|
_notifier->Hide(notification.p.Get());
|
|
}
|
|
}
|
|
}
|
|
|
|
void Manager::Private::clearFromSession(not_null<Main::Session*> session) {
|
|
if (!_notifier) {
|
|
return;
|
|
}
|
|
|
|
const auto sessionId = session->uniqueId();
|
|
for (auto i = _notifications.begin(); i != _notifications.end();) {
|
|
if (i->first.sessionId != sessionId) {
|
|
++i;
|
|
continue;
|
|
}
|
|
const auto temp = base::take(i->second);
|
|
_notifications.erase(i);
|
|
|
|
for (const auto &[msgId, notification] : temp) {
|
|
_notifier->Hide(notification.p.Get());
|
|
}
|
|
}
|
|
}
|
|
|
|
void Manager::Private::beforeNotificationActivated(NotificationId id) {
|
|
clearNotification(id);
|
|
}
|
|
|
|
void Manager::Private::afterNotificationActivated(
|
|
NotificationId id,
|
|
not_null<Window::SessionController*> window) {
|
|
SetForegroundWindow(window->widget()->psHwnd());
|
|
}
|
|
|
|
void Manager::Private::clearNotification(NotificationId id) {
|
|
auto i = _notifications.find(id.full);
|
|
if (i != _notifications.cend()) {
|
|
i->second.remove(id.msgId);
|
|
if (i->second.empty()) {
|
|
_notifications.erase(i);
|
|
}
|
|
}
|
|
}
|
|
|
|
bool Manager::Private::showNotification(
|
|
not_null<PeerData*> peer,
|
|
std::shared_ptr<Data::CloudImageView> &userpicView,
|
|
MsgId msgId,
|
|
const QString &title,
|
|
const QString &subtitle,
|
|
const QString &msg,
|
|
bool hideNameAndPhoto,
|
|
bool hideReplyButton) {
|
|
if (!_notificationManager || !_notifier || !_notificationFactory) {
|
|
return false;
|
|
}
|
|
|
|
ComPtr<IXmlDocument> toastXml;
|
|
bool withSubtitle = !subtitle.isEmpty();
|
|
|
|
HRESULT hr = _notificationManager->GetTemplateContent(
|
|
(withSubtitle
|
|
? ToastTemplateType_ToastImageAndText04
|
|
: ToastTemplateType_ToastImageAndText02),
|
|
&toastXml);
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
hr = SetAudioSilent(toastXml.Get());
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
const auto userpicKey = hideNameAndPhoto
|
|
? InMemoryKey()
|
|
: peer->userpicUniqueKey(userpicView);
|
|
const auto userpicPath = _cachedUserpics.get(userpicKey, peer, userpicView);
|
|
const auto userpicPathWide = QDir::toNativeSeparators(userpicPath).toStdWString();
|
|
|
|
hr = SetImageSrc(userpicPathWide.c_str(), toastXml.Get());
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
ComPtr<IXmlNodeList> nodeList;
|
|
hr = toastXml->GetElementsByTagName(StringReferenceWrapper(L"text").Get(), &nodeList);
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
UINT32 nodeListLength;
|
|
hr = nodeList->get_Length(&nodeListLength);
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
if (nodeListLength < (withSubtitle ? 3U : 2U)) return false;
|
|
|
|
{
|
|
ComPtr<IXmlNode> textNode;
|
|
hr = nodeList->Item(0, &textNode);
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
std::wstring wtitle = title.toStdWString();
|
|
hr = SetNodeValueString(StringReferenceWrapper(wtitle.data(), wtitle.size()).Get(), textNode.Get(), toastXml.Get());
|
|
if (!SUCCEEDED(hr)) return false;
|
|
}
|
|
if (withSubtitle) {
|
|
ComPtr<IXmlNode> textNode;
|
|
hr = nodeList->Item(1, &textNode);
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
std::wstring wsubtitle = subtitle.toStdWString();
|
|
hr = SetNodeValueString(StringReferenceWrapper(wsubtitle.data(), wsubtitle.size()).Get(), textNode.Get(), toastXml.Get());
|
|
if (!SUCCEEDED(hr)) return false;
|
|
}
|
|
{
|
|
ComPtr<IXmlNode> textNode;
|
|
hr = nodeList->Item(withSubtitle ? 2 : 1, &textNode);
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
std::wstring wmsg = msg.toStdWString();
|
|
hr = SetNodeValueString(StringReferenceWrapper(wmsg.data(), wmsg.size()).Get(), textNode.Get(), toastXml.Get());
|
|
if (!SUCCEEDED(hr)) return false;
|
|
}
|
|
|
|
ComPtr<IToastNotification> toast;
|
|
hr = _notificationFactory->CreateToastNotification(toastXml.Get(), &toast);
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
const auto key = FullPeer{
|
|
.sessionId = peer->session().uniqueId(),
|
|
.peerId = peer->id,
|
|
};
|
|
const auto notificationId = NotificationId{
|
|
.full = key,
|
|
.msgId = msgId
|
|
};
|
|
|
|
EventRegistrationToken activatedToken, dismissedToken, failedToken;
|
|
ComPtr<ToastEventHandler> eventHandler(new ToastEventHandler(
|
|
_guarded,
|
|
notificationId));
|
|
|
|
hr = toast->add_Activated(eventHandler.Get(), &activatedToken);
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
hr = toast->add_Dismissed(eventHandler.Get(), &dismissedToken);
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
hr = toast->add_Failed(eventHandler.Get(), &failedToken);
|
|
if (!SUCCEEDED(hr)) return false;
|
|
|
|
auto i = _notifications.find(key);
|
|
if (i != _notifications.cend()) {
|
|
auto j = i->second.find(msgId);
|
|
if (j != i->second.end()) {
|
|
ComPtr<IToastNotification> notify = j->second.p;
|
|
i->second.erase(j);
|
|
_notifier->Hide(notify.Get());
|
|
i = _notifications.find(key);
|
|
}
|
|
}
|
|
if (i == _notifications.cend()) {
|
|
i = _notifications.emplace(
|
|
key,
|
|
base::flat_map<MsgId, NotificationPtr>()).first;
|
|
}
|
|
hr = _notifier->Show(toast.Get());
|
|
if (!SUCCEEDED(hr)) {
|
|
i = _notifications.find(key);
|
|
if (i != _notifications.cend() && i->second.empty()) {
|
|
_notifications.erase(i);
|
|
}
|
|
return false;
|
|
}
|
|
i->second.emplace(msgId, toast);
|
|
|
|
return true;
|
|
}
|
|
|
|
Manager::Manager(Window::Notifications::System *system) : NativeManager(system)
|
|
, _private(std::make_unique<Private>(this, Private::Type::Rounded)) {
|
|
}
|
|
|
|
bool Manager::init() {
|
|
return _private->init();
|
|
}
|
|
|
|
void Manager::clearNotification(NotificationId id) {
|
|
_private->clearNotification(id);
|
|
}
|
|
|
|
Manager::~Manager() = default;
|
|
|
|
void Manager::doShowNativeNotification(
|
|
not_null<PeerData*> peer,
|
|
std::shared_ptr<Data::CloudImageView> &userpicView,
|
|
MsgId msgId,
|
|
const QString &title,
|
|
const QString &subtitle,
|
|
const QString &msg,
|
|
bool hideNameAndPhoto,
|
|
bool hideReplyButton) {
|
|
_private->showNotification(
|
|
peer,
|
|
userpicView,
|
|
msgId,
|
|
title,
|
|
subtitle,
|
|
msg,
|
|
hideNameAndPhoto,
|
|
hideReplyButton);
|
|
}
|
|
|
|
void Manager::doClearAllFast() {
|
|
_private->clearAll();
|
|
}
|
|
|
|
void Manager::doClearFromHistory(not_null<History*> history) {
|
|
_private->clearFromHistory(history);
|
|
}
|
|
|
|
void Manager::doClearFromSession(not_null<Main::Session*> session) {
|
|
_private->clearFromSession(session);
|
|
}
|
|
|
|
void Manager::onBeforeNotificationActivated(NotificationId id) {
|
|
_private->beforeNotificationActivated(id);
|
|
}
|
|
|
|
void Manager::onAfterNotificationActivated(
|
|
NotificationId id,
|
|
not_null<Window::SessionController*> window) {
|
|
_private->afterNotificationActivated(id, window);
|
|
}
|
|
|
|
bool Manager::doSkipAudio() const {
|
|
return SkipAudioForCustom()
|
|
|| QuietHoursEnabled
|
|
|| FocusAssistBlocks;
|
|
}
|
|
|
|
bool Manager::doSkipToast() const {
|
|
return false;
|
|
}
|
|
|
|
bool Manager::doSkipFlashBounce() const {
|
|
return SkipFlashBounceForCustom()
|
|
|| QuietHoursEnabled
|
|
|| FocusAssistBlocks;
|
|
}
|
|
#endif // !__MINGW32__
|
|
|
|
} // namespace Notifications
|
|
} // namespace Platform
|