Windows Toasts: Icons, Events, Cleanup, Documentation
- Enable documentations - Add a small usage example to the header - Final cleanups
This commit is contained in:
		
					parent
					
						
							
								498bff9e5a
							
						
					
				
			
			
				commit
				
					
						86ea0759d8
					
				
			
		
					 4 changed files with 107 additions and 304 deletions
				
			
		| 
						 | 
				
			
			@ -1,30 +1,26 @@
 | 
			
		|||
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
 | 
			
		||||
// Copyright (c) 2013 Patrick Reynolds <piki@github.com>. All rights reserved.
 | 
			
		||||
// Copyright (c) 2015 Felix Rieseberg <feriese@microsoft.com> and Jason Poon <jpoon@microsoft.com>. All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a BSD-style license that can be
 | 
			
		||||
// found in the LICENSE-CHROMIUM file.
 | 
			
		||||
 | 
			
		||||
#include "browser/win/notification_presenter_win.h"
 | 
			
		||||
 | 
			
		||||
#include "base/bind.h"
 | 
			
		||||
#include "base/logging.h"
 | 
			
		||||
#include "base/files/file_enumerator.h"
 | 
			
		||||
#include "base/strings/string_util.h"
 | 
			
		||||
#include "base/strings/utf_string_conversions.h"
 | 
			
		||||
#include "base/win/windows_version.h"
 | 
			
		||||
#include "content/public/browser/desktop_notification_delegate.h"
 | 
			
		||||
#include "content/public/common/platform_notification_data.h"
 | 
			
		||||
#include "third_party/skia/include/core/SkBitmap.h"
 | 
			
		||||
#include "common/application_info.h"
 | 
			
		||||
 | 
			
		||||
#include <stdlib.h>
 | 
			
		||||
#include <stdio.h>
 | 
			
		||||
#include <vector>
 | 
			
		||||
#include "third_party/skia/include/core/SkBitmap.h"
 | 
			
		||||
#include "base/win/windows_version.h"
 | 
			
		||||
 | 
			
		||||
// Windows Header
 | 
			
		||||
#define WIN32_LEAN_AND_MEAN
 | 
			
		||||
 | 
			
		||||
#include <windows.h>
 | 
			
		||||
#include <wrl/client.h>
 | 
			
		||||
#include <windows.ui.notifications.h>
 | 
			
		||||
#include <wrl/client.h>
 | 
			
		||||
#include <wrl/implements.h>
 | 
			
		||||
#include <winstring.h>
 | 
			
		||||
#include "windows_toast_notification.h"
 | 
			
		||||
| 
						 | 
				
			
			@ -59,17 +55,19 @@ void NotificationPresenterWin::ShowNotification(
 | 
			
		|||
  
 | 
			
		||||
  std::wstring title = data.title;
 | 
			
		||||
  std::wstring body = data.body;
 | 
			
		||||
  std::string iconPath = data.icon.spec();  
 | 
			
		||||
  std::string appName = GetApplicationName();
 | 
			
		||||
  char* img = NULL;
 | 
			
		||||
  
 | 
			
		||||
  WinToasts::WindowsToastNotification* wtn = new WinToasts::WindowsToastNotification(appName.c_str());
 | 
			
		||||
  wtn->ShowNotification(title.c_str(), body.c_str(), img);
 | 
			
		||||
  WinToasts::WindowsToastNotification* wtn = new WinToasts::WindowsToastNotification(appName.c_str(), delegate_ptr.release());
 | 
			
		||||
  wtn->ShowNotification(title.c_str(), body.c_str(), iconPath);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void NotificationPresenterWin::CancelNotification() {
 | 
			
		||||
  // No can do.
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void NotificationPresenterWin::DeleteNotification() {
 | 
			
		||||
  // No can do.
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
}  // namespace brightray
 | 
			
		||||
| 
						 | 
				
			
			@ -1,8 +1,19 @@
 | 
			
		|||
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
 | 
			
		||||
// Copyright (c) 2013 Patrick Reynolds <piki@github.com>. All rights reserved.
 | 
			
		||||
// Copyright (c) 2015 Felix Rieseberg <feriese@microsoft.com> and Jason Poon <jpoon@microsoft.com>. All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a BSD-style license that can be
 | 
			
		||||
// found in the LICENSE-CHROMIUM file.
 | 
			
		||||
 | 
			
		||||
// Usage Example (JavaScript:
 | 
			
		||||
// var windowsNotification = new Notification("Test Title", {
 | 
			
		||||
// 	body: "Hi, I'm an example body. How are you?",
 | 
			
		||||
// 	icon: "file:///C:/Path/To/Your/Image.png"
 | 
			
		||||
// });
 | 
			
		||||
 | 
			
		||||
// windowsNotification.onshow  = function () { console.log("Notification shown") };
 | 
			
		||||
// windowsNotification.onclick = function () { console.log("Notification clicked") };
 | 
			
		||||
// windowsNotification.onclose = function () { console.log("Notification dismissed") };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
#ifndef BRIGHTRAY_BROWSER_NOTIFICATION_PRESENTER_WIN_H_
 | 
			
		||||
#define BRIGHTRAY_BROWSER_NOTIFICATION_PRESENTER_WIN_H_
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,53 +1,47 @@
 | 
			
		|||
// Copyright (c) 2015 Felix Rieseberg <feriese@microsoft.com>. All rights reserved.
 | 
			
		||||
// Copyright (c) 2015 Felix Rieseberg <feriese@microsoft.com> and Jason Poon <jpoon@microsoft.com>. All rights reserved.
 | 
			
		||||
// Copyright (c) 2015 Ryan McShane <rmcshane@bandwidth.com> and Brandon Smith <bsmith@bandwidth.com>
 | 
			
		||||
// Thanks to both of those folks mentioned above who first thought up a bunch of this code
 | 
			
		||||
// and released it as MIT to the world.
 | 
			
		||||
 | 
			
		||||
#include "windows_toast_notification.h"
 | 
			
		||||
#include "content/public/browser/desktop_notification_delegate.h"
 | 
			
		||||
 | 
			
		||||
#include <Psapi.h>
 | 
			
		||||
#include <stdio.h>
 | 
			
		||||
#include <ShObjidl.h>
 | 
			
		||||
#include <propvarutil.h>
 | 
			
		||||
#include <propkey.h>
 | 
			
		||||
#include <wincrypt.h>
 | 
			
		||||
#include <string>
 | 
			
		||||
 | 
			
		||||
using namespace WinToasts;
 | 
			
		||||
using namespace Windows::Foundation;
 | 
			
		||||
 | 
			
		||||
#define BREAK_IF_BAD(hr)    if(!SUCCEEDED(hr)) break;
 | 
			
		||||
#define SHORTCUT_FORMAT     "\\Microsoft\\Windows\\Start Menu\\Programs\\%s.lnk"
 | 
			
		||||
 | 
			
		||||
char WindowsToastNotification::s_appName[MAX_PATH] = {};
 | 
			
		||||
char WindowsToastNotification::s_tempDirPath[MAX_PATH] = {};
 | 
			
		||||
 | 
			
		||||
namespace WinToasts {
 | 
			
		||||
 | 
			
		||||
WindowsToastNotification::WindowsToastNotification(const char* appName)
 | 
			
		||||
WindowsToastNotification::WindowsToastNotification(const char* appName, content::DesktopNotificationDelegate* delegate)
 | 
			
		||||
{
 | 
			
		||||
    sprintf_s(s_appName, ARRAYSIZE(s_appName), "%s", appName);
 | 
			
		||||
    m_eventHandler = NULL;
 | 
			
		||||
    m_tempFilePath = NULL;
 | 
			
		||||
    n_delegate = delegate;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
WindowsToastNotification::~WindowsToastNotification()
 | 
			
		||||
{
 | 
			
		||||
    if (m_eventHandler){
 | 
			
		||||
    if (m_eventHandler) {
 | 
			
		||||
        delete m_eventHandler;
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    if (m_tempFilePath){
 | 
			
		||||
        DeleteTempImageFile();
 | 
			
		||||
        delete m_tempFilePath;
 | 
			
		||||
    if (n_delegate) {
 | 
			
		||||
        delete n_delegate;
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
/*static*/ void WindowsToastNotification::InitSystemProps(char* appName, char* tempDirPath)
 | 
			
		||||
{
 | 
			
		||||
 | 
			
		||||
    sprintf_s(s_tempDirPath, ARRAYSIZE(s_tempDirPath), "%s", tempDirPath);
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
void WindowsToastNotification::ShowNotification(const WCHAR* title, const WCHAR* msg, const char* img)
 | 
			
		||||
void WindowsToastNotification::ShowNotification(const WCHAR* title, const WCHAR* msg, std::string iconPath)
 | 
			
		||||
{        
 | 
			
		||||
    // Init using WRL
 | 
			
		||||
    Windows::Foundation::Initialize(RO_INIT_MULTITHREADED);
 | 
			
		||||
| 
						 | 
				
			
			@ -55,14 +49,10 @@ void WindowsToastNotification::ShowNotification(const WCHAR* title, const WCHAR*
 | 
			
		|||
    HSTRING toastNotifMgrStr = NULL;
 | 
			
		||||
    HSTRING appId = NULL;
 | 
			
		||||
    HSTRING toastNotifStr = NULL;
 | 
			
		||||
    HRESULT hr;
 | 
			
		||||
 | 
			
		||||
    // Create a shortcut in the user's app data folder
 | 
			
		||||
    // (requirement for Windows 8 to show notifications - Windows 10 is fine without it)
 | 
			
		||||
    // TODO: This shoudl be cleaned up
 | 
			
		||||
    // HRESULT hr = EnsureShortcut();
 | 
			
		||||
 | 
			
		||||
    do{         
 | 
			
		||||
        HRESULT hr = CreateHString(RuntimeClass_Windows_UI_Notifications_ToastNotificationManager, &toastNotifMgrStr);
 | 
			
		||||
    do {         
 | 
			
		||||
        hr = CreateHString(RuntimeClass_Windows_UI_Notifications_ToastNotificationManager, &toastNotifMgrStr);
 | 
			
		||||
        BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
        ComPtr<IToastNotificationManagerStatics> toastMgr;
 | 
			
		||||
| 
						 | 
				
			
			@ -70,7 +60,7 @@ void WindowsToastNotification::ShowNotification(const WCHAR* title, const WCHAR*
 | 
			
		|||
        BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
        ComPtr<IXmlDocument> toastXml;
 | 
			
		||||
        hr = GetToastXml(toastMgr.Get(), title, msg, img, &toastXml);
 | 
			
		||||
        hr = GetToastXml(toastMgr.Get(), title, msg, iconPath, &toastXml);
 | 
			
		||||
        BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
        WCHAR wAppName[MAX_PATH];
 | 
			
		||||
| 
						 | 
				
			
			@ -99,52 +89,48 @@ void WindowsToastNotification::ShowNotification(const WCHAR* title, const WCHAR*
 | 
			
		|||
        hr = notifier->Show(toast.Get());
 | 
			
		||||
        BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
        //m_cbFn.Reset(Isolate::GetCurrent(), cbFn);
 | 
			
		||||
 | 
			
		||||
        //todo:
 | 
			
		||||
        //cbPtr->NotificationDisplayed();
 | 
			
		||||
        n_delegate->NotificationDisplayed();
 | 
			
		||||
    } while (FALSE);
 | 
			
		||||
 | 
			
		||||
    if (toastNotifMgrStr != NULL){
 | 
			
		||||
    if (toastNotifMgrStr != NULL) {
 | 
			
		||||
        WindowsDeleteString(toastNotifMgrStr);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if (appId != NULL){
 | 
			
		||||
    if (appId != NULL) {
 | 
			
		||||
        WindowsDeleteString(appId);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if (toastNotifStr != NULL){
 | 
			
		||||
    if (toastNotifStr != NULL) {
 | 
			
		||||
        WindowsDeleteString(toastNotifStr);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    // if (!SUCCEEDED(hr)){
 | 
			
		||||
    //     delete this;
 | 
			
		||||
    // }
 | 
			
		||||
    if (!SUCCEEDED(hr)) {
 | 
			
		||||
        delete this;
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
void WindowsToastNotification::NotificationClicked()
 | 
			
		||||
{
 | 
			
		||||
    // Local<Function> fn = Local<Function>::New(Isolate::GetCurrent(), m_cbFn);
 | 
			
		||||
    // Handle<Value> args[1];
 | 
			
		||||
    // fn->Call(Isolate::GetCurrent()->GetCurrentContext()->Global(), 0, args);
 | 
			
		||||
    // delete this;
 | 
			
		||||
    delete this;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
void WindowsToastNotification::NotificationDismissed()
 | 
			
		||||
{
 | 
			
		||||
    delete this;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::GetToastXml(
 | 
			
		||||
    IToastNotificationManagerStatics* toastMgr,
 | 
			
		||||
    const WCHAR* title,
 | 
			
		||||
    const WCHAR* msg,
 | 
			
		||||
    std::string iconPath,
 | 
			
		||||
    IXmlDocument** toastXml) {
 | 
			
		||||
 
 | 
			
		||||
HRESULT WindowsToastNotification::GetToastXml(IToastNotificationManagerStatics* toastMgr, const WCHAR* title, const WCHAR* msg, const char* img, IXmlDocument** toastXml)
 | 
			
		||||
{
 | 
			
		||||
    HRESULT hr;
 | 
			
		||||
    ToastTemplateType templateType;
 | 
			
		||||
    if (title == NULL || msg == NULL){
 | 
			
		||||
    if (title == NULL || msg == NULL) {
 | 
			
		||||
        // Single line toast
 | 
			
		||||
        templateType = img == NULL ? ToastTemplateType_ToastText01 : ToastTemplateType_ToastImageAndText01;
 | 
			
		||||
        templateType = iconPath.length() == 0 ? ToastTemplateType_ToastText01 : ToastTemplateType_ToastImageAndText01;
 | 
			
		||||
        hr = toastMgr->GetTemplateContent(templateType, toastXml);
 | 
			
		||||
        if (SUCCEEDED(hr)) {
 | 
			
		||||
            const WCHAR* text = title != NULL ? title : msg;
 | 
			
		||||
| 
						 | 
				
			
			@ -152,18 +138,17 @@ HRESULT WindowsToastNotification::GetToastXml(IToastNotificationManagerStatics*
 | 
			
		|||
        }
 | 
			
		||||
    } else {
 | 
			
		||||
        // Title and body toast
 | 
			
		||||
        templateType = img == NULL ? ToastTemplateType_ToastText02 : ToastTemplateType_ToastImageAndText02;
 | 
			
		||||
        templateType = iconPath.length() == 0 ? ToastTemplateType_ToastText02 : ToastTemplateType_ToastImageAndText02;
 | 
			
		||||
        hr = toastMgr->GetTemplateContent(templateType, toastXml);
 | 
			
		||||
        if (SUCCEEDED(hr)) {
 | 
			
		||||
            hr = SetXmlText(*toastXml, title, msg);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if (img != NULL && SUCCEEDED(hr)){
 | 
			
		||||
    if (iconPath.length() != 0 && SUCCEEDED(hr)) {
 | 
			
		||||
        // Toast has image
 | 
			
		||||
        hr = CreateTempImageFile(img);
 | 
			
		||||
        if (SUCCEEDED(hr)){
 | 
			
		||||
            hr = SetXmlImage(*toastXml);
 | 
			
		||||
        if (SUCCEEDED(hr)) {
 | 
			
		||||
            hr = SetXmlImage(*toastXml, iconPath);
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        // Don't stop a notification from showing just because an image couldn't be displayed. By default the app icon will be shown.
 | 
			
		||||
| 
						 | 
				
			
			@ -173,7 +158,6 @@ HRESULT WindowsToastNotification::GetToastXml(IToastNotificationManagerStatics*
 | 
			
		|||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::SetXmlText(IXmlDocument* doc, const WCHAR* text)
 | 
			
		||||
{
 | 
			
		||||
    HSTRING tag = NULL;
 | 
			
		||||
| 
						 | 
				
			
			@ -190,14 +174,13 @@ HRESULT WindowsToastNotification::SetXmlText(IXmlDocument* doc, const WCHAR* tex
 | 
			
		|||
        hr = AppendTextToXml(doc, node.Get(), text);
 | 
			
		||||
    } while (FALSE);
 | 
			
		||||
 | 
			
		||||
    if (tag != NULL){
 | 
			
		||||
    if (tag != NULL) {
 | 
			
		||||
        WindowsDeleteString(tag);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::SetXmlText(IXmlDocument* doc, const WCHAR* title, const WCHAR* body)
 | 
			
		||||
{
 | 
			
		||||
    HSTRING tag = NULL;
 | 
			
		||||
| 
						 | 
				
			
			@ -219,21 +202,21 @@ HRESULT WindowsToastNotification::SetXmlText(IXmlDocument* doc, const WCHAR* tit
 | 
			
		|||
        hr = AppendTextToXml(doc, node.Get(), body);
 | 
			
		||||
    } while (FALSE);
 | 
			
		||||
 | 
			
		||||
    if (tag != NULL){
 | 
			
		||||
    if (tag != NULL) {
 | 
			
		||||
        WindowsDeleteString(tag);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::SetXmlImage(IXmlDocument* doc)
 | 
			
		||||
HRESULT WindowsToastNotification::SetXmlImage(IXmlDocument* doc, std::string iconPath)
 | 
			
		||||
{
 | 
			
		||||
    HSTRING tag = NULL;
 | 
			
		||||
    HSTRING src = NULL;
 | 
			
		||||
    HSTRING imgPath = NULL;
 | 
			
		||||
    HRESULT hr = CreateHString(L"image", &tag);
 | 
			
		||||
    do{
 | 
			
		||||
 | 
			
		||||
    do {
 | 
			
		||||
        BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
        ComPtr<IXmlNodeList> nodeList;
 | 
			
		||||
| 
						 | 
				
			
			@ -256,7 +239,7 @@ HRESULT WindowsToastNotification::SetXmlImage(IXmlDocument* doc)
 | 
			
		|||
        BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
        WCHAR xmlPath[MAX_PATH];
 | 
			
		||||
        swprintf(xmlPath, ARRAYSIZE(xmlPath), L"file:///%S", m_tempFilePath);
 | 
			
		||||
        swprintf(xmlPath, ARRAYSIZE(xmlPath), L"%S", iconPath);
 | 
			
		||||
        hr = CreateHString(xmlPath, &imgPath);
 | 
			
		||||
        BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -272,20 +255,19 @@ HRESULT WindowsToastNotification::SetXmlImage(IXmlDocument* doc)
 | 
			
		|||
        hr = srcAttr->AppendChild(srcNode.Get(), &childNode);
 | 
			
		||||
    } while (FALSE);
 | 
			
		||||
 | 
			
		||||
    if (tag != NULL){
 | 
			
		||||
    if (tag != NULL) {
 | 
			
		||||
        WindowsDeleteString(tag);
 | 
			
		||||
    }
 | 
			
		||||
    if (src != NULL){
 | 
			
		||||
    if (src != NULL) {
 | 
			
		||||
        WindowsDeleteString(src);
 | 
			
		||||
    }
 | 
			
		||||
    if (imgPath != NULL){
 | 
			
		||||
    if (imgPath != NULL) {
 | 
			
		||||
        WindowsDeleteString(imgPath);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::GetTextNodeList(HSTRING* tag, IXmlDocument* doc, IXmlNodeList** nodeList, UINT32 reqLength)
 | 
			
		||||
{
 | 
			
		||||
    HRESULT hr = CreateHString(L"text", tag);
 | 
			
		||||
| 
						 | 
				
			
			@ -304,7 +286,7 @@ HRESULT WindowsToastNotification::GetTextNodeList(HSTRING* tag, IXmlDocument* do
 | 
			
		|||
        }
 | 
			
		||||
    } while (FALSE);
 | 
			
		||||
 | 
			
		||||
    if (!SUCCEEDED(hr)){
 | 
			
		||||
    if (!SUCCEEDED(hr)) {
 | 
			
		||||
        // Allow the caller to delete this string on success
 | 
			
		||||
        WindowsDeleteString(*tag);
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -312,12 +294,11 @@ HRESULT WindowsToastNotification::GetTextNodeList(HSTRING* tag, IXmlDocument* do
 | 
			
		|||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::AppendTextToXml(IXmlDocument* doc, IXmlNode* node, const WCHAR* text)
 | 
			
		||||
{
 | 
			
		||||
    HSTRING str = NULL;
 | 
			
		||||
    HRESULT hr = CreateHString(text, &str);
 | 
			
		||||
    do{
 | 
			
		||||
    do {
 | 
			
		||||
        BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
        ComPtr<IXmlText> xmlText;
 | 
			
		||||
| 
						 | 
				
			
			@ -332,20 +313,20 @@ HRESULT WindowsToastNotification::AppendTextToXml(IXmlDocument* doc, IXmlNode* n
 | 
			
		|||
        hr = node->AppendChild(textNode.Get(), &appendNode);
 | 
			
		||||
    } while (FALSE);
 | 
			
		||||
 | 
			
		||||
    if (str != NULL){
 | 
			
		||||
    if (str != NULL) {
 | 
			
		||||
        WindowsDeleteString(str);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::SetupCallbacks(IToastNotification* toast)
 | 
			
		||||
{
 | 
			
		||||
    EventRegistrationToken activatedToken, dismissedToken;
 | 
			
		||||
    m_eventHandler = new ToastEventHandler(this);
 | 
			
		||||
    m_eventHandler = new ToastEventHandler(this, n_delegate);
 | 
			
		||||
    ComPtr<ToastEventHandler> eventHandler(m_eventHandler);
 | 
			
		||||
    HRESULT hr = toast->add_Activated(eventHandler.Get(), &activatedToken);
 | 
			
		||||
    
 | 
			
		||||
    if (SUCCEEDED(hr)) {
 | 
			
		||||
        hr = toast->add_Dismissed(eventHandler.Get(), &dismissedToken);
 | 
			
		||||
    }
 | 
			
		||||
| 
						 | 
				
			
			@ -353,7 +334,6 @@ HRESULT WindowsToastNotification::SetupCallbacks(IToastNotification* toast)
 | 
			
		|||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::CreateHString(const WCHAR* source, HSTRING* dest)
 | 
			
		||||
{
 | 
			
		||||
    if (source == NULL || dest == NULL) {
 | 
			
		||||
| 
						 | 
				
			
			@ -364,229 +344,43 @@ HRESULT WindowsToastNotification::CreateHString(const WCHAR* source, HSTRING* de
 | 
			
		|||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::CreateTempImageFile(const char* base64)
 | 
			
		||||
{
 | 
			
		||||
    BYTE* buff = NULL;
 | 
			
		||||
    HRESULT hr = EnsureTempDir();
 | 
			
		||||
    do{
 | 
			
		||||
        BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
        DWORD buffLen;
 | 
			
		||||
        hr = ConvertBase64DataToImage(base64, &buff, &buffLen);
 | 
			
		||||
        BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
        hr = WriteToFile(buff, buffLen);
 | 
			
		||||
    } while (FALSE);
 | 
			
		||||
 | 
			
		||||
    if (buff != NULL){
 | 
			
		||||
        delete buff;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::ConvertBase64DataToImage(const char* base64, BYTE** buff, DWORD* buffLen)
 | 
			
		||||
{
 | 
			
		||||
    HRESULT hr = E_FAIL;
 | 
			
		||||
    *buffLen = 0;
 | 
			
		||||
    DWORD reqSize;
 | 
			
		||||
    if (CryptStringToBinary((LPCWSTR)base64, 0, CRYPT_STRING_BASE64, NULL, &reqSize, NULL, NULL)){
 | 
			
		||||
        *buff = new BYTE[reqSize];
 | 
			
		||||
        if (CryptStringToBinary((LPCWSTR)base64, 0, CRYPT_STRING_BASE64, *buff, &reqSize, NULL, NULL)){
 | 
			
		||||
            *buffLen = reqSize;
 | 
			
		||||
            hr = S_OK;
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::WriteToFile(BYTE* buff, DWORD buffLen)
 | 
			
		||||
{
 | 
			
		||||
    HRESULT hr = E_FAIL;
 | 
			
		||||
 | 
			
		||||
    GUID guid;
 | 
			
		||||
    hr = CoCreateGuid(&guid);
 | 
			
		||||
    if (SUCCEEDED(hr)){
 | 
			
		||||
        WCHAR randomName[MAX_PATH];
 | 
			
		||||
        StringFromGUID2(guid, randomName, ARRAYSIZE(randomName));
 | 
			
		||||
 | 
			
		||||
        m_tempFilePath = new char[MAX_PATH];
 | 
			
		||||
        sprintf(m_tempFilePath, "%s\\%S", s_tempDirPath, randomName);
 | 
			
		||||
        HANDLE file = CreateFile((LPCWSTR)m_tempFilePath, GENERIC_WRITE, 0, NULL, CREATE_NEW, FILE_ATTRIBUTE_NORMAL, NULL);
 | 
			
		||||
        if (file != INVALID_HANDLE_VALUE){
 | 
			
		||||
            DWORD bytesWritten;
 | 
			
		||||
            if (WriteFile(file, buff, buffLen, &bytesWritten, NULL)){
 | 
			
		||||
                hr = S_OK;
 | 
			
		||||
            }
 | 
			
		||||
            else{
 | 
			
		||||
                DeleteTempImageFile();
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            CloseHandle(file);
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    if (!SUCCEEDED(hr) && m_tempFilePath){
 | 
			
		||||
        delete m_tempFilePath;
 | 
			
		||||
        m_tempFilePath = NULL;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::EnsureTempDir()
 | 
			
		||||
{
 | 
			
		||||
    // Check for the Dir already existing or easy creation
 | 
			
		||||
    HRESULT hr = E_FAIL;
 | 
			
		||||
    if (CreateDirectory((LPCWSTR)s_tempDirPath, NULL) || GetLastError() == ERROR_ALREADY_EXISTS){
 | 
			
		||||
        hr = S_OK;
 | 
			
		||||
    } else {
 | 
			
		||||
        // NOTE: This makes an assumption that the path is a FULL LENGTH PATH.  Therefore the first folder starts at the
 | 
			
		||||
        // 4th character.  ie. "c:\path\to\temp\dir"
 | 
			
		||||
 | 
			
		||||
        // It's possible that multiple directories need to be created.
 | 
			
		||||
        for (int i = 3; i < ARRAYSIZE(s_tempDirPath); i++) {
 | 
			
		||||
            char c = s_tempDirPath[i];
 | 
			
		||||
            if (c == '\\'){
 | 
			
		||||
                //Substring until this char
 | 
			
		||||
                char* temp = new char[i + 1];
 | 
			
		||||
                strncpy(temp, s_tempDirPath, i);
 | 
			
		||||
                temp[i] = 0;
 | 
			
		||||
                if (!CreateDirectory((LPCWSTR)temp, NULL) && GetLastError() != ERROR_ALREADY_EXISTS){
 | 
			
		||||
                    delete temp;
 | 
			
		||||
                    return hr;
 | 
			
		||||
                }
 | 
			
		||||
 | 
			
		||||
                delete temp;
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
        //Try to create the full path one more time.  This will take care of paths that don't end with a slash
 | 
			
		||||
        if (CreateDirectory((LPCWSTR)s_tempDirPath, NULL) || GetLastError() == ERROR_ALREADY_EXISTS){
 | 
			
		||||
            hr = S_OK;
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::DeleteTempImageFile()
 | 
			
		||||
{
 | 
			
		||||
    if (DeleteFile((LPCWSTR)m_tempFilePath)){
 | 
			
		||||
        return S_OK;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return E_FAIL;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::EnsureShortcut()
 | 
			
		||||
{   
 | 
			
		||||
    HRESULT hr = E_FAIL;
 | 
			
		||||
    char shortcut[MAX_PATH];
 | 
			
		||||
    DWORD charsWritten = GetEnvironmentVariable(L"APPDATA", (LPWSTR)shortcut, MAX_PATH);
 | 
			
		||||
 | 
			
		||||
    if (charsWritten > 0) {
 | 
			
		||||
        char shortcutCat[MAX_PATH];
 | 
			
		||||
        sprintf(shortcutCat, SHORTCUT_FORMAT, s_appName);
 | 
			
		||||
        errno_t concatErr = strcat_s(shortcut, ARRAYSIZE(shortcut), shortcutCat);
 | 
			
		||||
        if (concatErr == 0) {
 | 
			
		||||
            DWORD attr = GetFileAttributes((LPCWSTR)shortcut);
 | 
			
		||||
            bool exists = attr < 0xFFFFFFF;
 | 
			
		||||
            if (exists){
 | 
			
		||||
                hr = S_OK;
 | 
			
		||||
            }
 | 
			
		||||
            else{
 | 
			
		||||
                WCHAR path[MAX_PATH];
 | 
			
		||||
                mbstowcs(path, shortcut, ARRAYSIZE(path));
 | 
			
		||||
                hr = CreateShortcut(path);
 | 
			
		||||
            }
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
HRESULT WindowsToastNotification::CreateShortcut(WCHAR* path)
 | 
			
		||||
{
 | 
			
		||||
    HRESULT hr = E_FAIL;
 | 
			
		||||
    char exePath[MAX_PATH];
 | 
			
		||||
    DWORD charsWritten = GetModuleFileNameEx(GetCurrentProcess(), nullptr, (LPWSTR)exePath, ARRAYSIZE(exePath));
 | 
			
		||||
    if (charsWritten > 0) {
 | 
			
		||||
        PROPVARIANT propVariant;
 | 
			
		||||
        ComPtr<IShellLink> shellLink;
 | 
			
		||||
        hr = CoCreateInstance(CLSID_ShellLink, nullptr, CLSCTX_INPROC_SERVER, IID_PPV_ARGS(&shellLink));
 | 
			
		||||
        do{
 | 
			
		||||
            BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
            hr = shellLink->SetPath((LPCWSTR)exePath);
 | 
			
		||||
            BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
            ComPtr<IPropertyStore> propStore;
 | 
			
		||||
            hr = shellLink.As(&propStore);
 | 
			
		||||
            BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
            WCHAR wAppName[MAX_PATH];
 | 
			
		||||
            swprintf(wAppName, ARRAYSIZE(wAppName), L"%S", s_appName);
 | 
			
		||||
            hr = InitPropVariantFromString(wAppName, &propVariant);
 | 
			
		||||
            BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
            hr = propStore->SetValue(PKEY_AppUserModel_ID, propVariant);
 | 
			
		||||
            BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
            hr = propStore->Commit();
 | 
			
		||||
            BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
            ComPtr<IPersistFile> persistFile;
 | 
			
		||||
            hr = shellLink.As(&persistFile);
 | 
			
		||||
            BREAK_IF_BAD(hr);
 | 
			
		||||
 | 
			
		||||
            hr = persistFile->Save(path, TRUE);
 | 
			
		||||
        } while (FALSE);
 | 
			
		||||
 | 
			
		||||
        PropVariantClear(&propVariant);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    return hr;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
ToastEventHandler::ToastEventHandler(WindowsToastNotification* notification)
 | 
			
		||||
/*
 | 
			
		||||
/ Toast Event Handler
 | 
			
		||||
*/
 | 
			
		||||
ToastEventHandler::ToastEventHandler(WindowsToastNotification* notification, content::DesktopNotificationDelegate* delegate)
 | 
			
		||||
{
 | 
			
		||||
    m_ref = 1;
 | 
			
		||||
    m_notification = notification;
 | 
			
		||||
    n_delegate = delegate;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
ToastEventHandler::~ToastEventHandler()
 | 
			
		||||
{
 | 
			
		||||
    // Empty
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
IFACEMETHODIMP ToastEventHandler::Invoke(IToastNotification* sender, IInspectable* args)
 | 
			
		||||
{
 | 
			
		||||
    //Notification "activated" (clicked)
 | 
			
		||||
    if (m_notification != NULL){
 | 
			
		||||
    // Notification "activated" (clicked)
 | 
			
		||||
    n_delegate->NotificationClick();
 | 
			
		||||
    
 | 
			
		||||
    if (m_notification != NULL) {
 | 
			
		||||
        m_notification->NotificationClicked();
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    return S_OK;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
IFACEMETHODIMP ToastEventHandler::Invoke(IToastNotification* sender, IToastDismissedEventArgs* e)
 | 
			
		||||
{   
 | 
			
		||||
    //Notification dismissed
 | 
			
		||||
    if (m_notification != NULL){
 | 
			
		||||
    // Notification dismissed
 | 
			
		||||
    n_delegate->NotificationClosed();
 | 
			
		||||
 | 
			
		||||
    if (m_notification != NULL) {
 | 
			
		||||
        m_notification->NotificationDismissed();
 | 
			
		||||
        
 | 
			
		||||
    }
 | 
			
		||||
    
 | 
			
		||||
    return S_OK;
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -1,12 +1,22 @@
 | 
			
		|||
// Copyright (c) 2015 Felix Rieseberg <feriese@microsoft.com>. All rights reserved.
 | 
			
		||||
// Copyright (c) 2015 Ryan McShane <rmcshane@bandwidth.com> and Brandon Smith <bsmith@bandwidth.com>
 | 
			
		||||
// Thanks to both of those folks mentioned above who first thought up a bunch of this code
 | 
			
		||||
// and released it as MIT to the world.
 | 
			
		||||
 | 
			
		||||
#ifndef __WINDOWS_TOAST_NOTIFICATION_H__
 | 
			
		||||
#define __WINDOWS_TOAST_NOTIFICATION_H__
 | 
			
		||||
 | 
			
		||||
#define WIN32_LEAN_AND_MEAN
 | 
			
		||||
 | 
			
		||||
#include "content/public/browser/desktop_notification_delegate.h"
 | 
			
		||||
#include "content/public/common/platform_notification_data.h"
 | 
			
		||||
#include "base/bind.h"
 | 
			
		||||
 | 
			
		||||
#include <windows.h>
 | 
			
		||||
#include <windows.ui.notifications.h>
 | 
			
		||||
#include <wrl/client.h>
 | 
			
		||||
#include <wrl/implements.h>
 | 
			
		||||
#include <string>
 | 
			
		||||
 | 
			
		||||
using namespace Microsoft::WRL;
 | 
			
		||||
using namespace ABI::Windows::UI::Notifications;
 | 
			
		||||
| 
						 | 
				
			
			@ -24,35 +34,24 @@ namespace WinToasts{
 | 
			
		|||
    {
 | 
			
		||||
    private:
 | 
			
		||||
        static char s_appName[MAX_PATH];
 | 
			
		||||
        static char s_tempDirPath[MAX_PATH];
 | 
			
		||||
 | 
			
		||||
        char* m_tempFilePath;
 | 
			
		||||
        ToastEventHandler* m_eventHandler;
 | 
			
		||||
        content::DesktopNotificationDelegate* n_delegate;
 | 
			
		||||
 | 
			
		||||
        HRESULT GetToastXml(IToastNotificationManagerStatics* toastMgr, const WCHAR* title, const WCHAR* msg, const char* img, IXmlDocument** toastXml);
 | 
			
		||||
        HRESULT GetToastXml(IToastNotificationManagerStatics* toastMgr, const WCHAR* title, const WCHAR* msg, std::string iconPath, IXmlDocument** toastXml);
 | 
			
		||||
        HRESULT SetXmlText(IXmlDocument* doc, const WCHAR* text);
 | 
			
		||||
        HRESULT SetXmlText(IXmlDocument* doc, const WCHAR* title, const WCHAR* body);
 | 
			
		||||
        HRESULT SetXmlImage(IXmlDocument* doc);
 | 
			
		||||
        HRESULT SetXmlImage(IXmlDocument* doc, std::string iconPath);
 | 
			
		||||
        HRESULT GetTextNodeList(HSTRING* tag, IXmlDocument* doc, IXmlNodeList** nodeList, UINT32 reqLength);
 | 
			
		||||
        HRESULT AppendTextToXml(IXmlDocument* doc, IXmlNode* node, const WCHAR* text);
 | 
			
		||||
        HRESULT SetupCallbacks(IToastNotification* toast);
 | 
			
		||||
        HRESULT CreateHString(const WCHAR* source, HSTRING* dest);
 | 
			
		||||
        HRESULT CreateTempImageFile(const char* base64);
 | 
			
		||||
        HRESULT ConvertBase64DataToImage(const char* base64, BYTE** buff, DWORD* buffLen);
 | 
			
		||||
        HRESULT WriteToFile(BYTE* buff, DWORD buffLen);
 | 
			
		||||
        HRESULT EnsureTempDir();
 | 
			
		||||
        HRESULT DeleteTempImageFile();
 | 
			
		||||
        HRESULT EnsureShortcut();
 | 
			
		||||
        HRESULT CreateShortcut(WCHAR* path);
 | 
			
		||||
 | 
			
		||||
    public:
 | 
			
		||||
        WindowsToastNotification(const char* appName);
 | 
			
		||||
        WindowsToastNotification(const char* appName, content::DesktopNotificationDelegate* delegate);
 | 
			
		||||
        ~WindowsToastNotification();
 | 
			
		||||
        void ShowNotification(const WCHAR* title, const WCHAR* msg, const char* img);
 | 
			
		||||
        void ShowNotification(const WCHAR* title, const WCHAR* msg, std::string iconPath);
 | 
			
		||||
        void NotificationClicked();
 | 
			
		||||
        void NotificationDismissed();
 | 
			
		||||
 | 
			
		||||
        static void InitSystemProps(char* appName, char* tempDirPath);
 | 
			
		||||
    };
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			@ -62,9 +61,10 @@ namespace WinToasts{
 | 
			
		|||
    private:
 | 
			
		||||
        ULONG m_ref;
 | 
			
		||||
        WindowsToastNotification* m_notification;
 | 
			
		||||
        content::DesktopNotificationDelegate* n_delegate;
 | 
			
		||||
 | 
			
		||||
    public:
 | 
			
		||||
        ToastEventHandler(WindowsToastNotification* notification);
 | 
			
		||||
        ToastEventHandler(WindowsToastNotification* notification, content::DesktopNotificationDelegate* delegate);
 | 
			
		||||
        ~ToastEventHandler();
 | 
			
		||||
        IFACEMETHODIMP Invoke(IToastNotification* sender, IInspectable* args);
 | 
			
		||||
        IFACEMETHODIMP Invoke(IToastNotification* sender, IToastDismissedEventArgs* e);
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue