[Notification] fix "Cancel Desktop Notification" for all platform. and implement it for win8 (toast notification)

jefry-vcube · jefry-vcube · commit ec801e3782f3 · 2014-11-18T16:39:58.000+08:00
diff --git a/src/nw_notification_manager_toast_win.cc b/src/nw_notification_manager_toast_win.cc
@@ -37,10 +37,8 @@
 
 #include <intsafe.h>
 #include <strsafe.h>
-#include <wrl\client.h>
 #include <wrl\implements.h>
 
-using namespace Microsoft::WRL;
 using namespace Windows::Foundation;
 
 namespace nw {
@@ -198,6 +196,7 @@ IFACEMETHODIMP ToastEventHandler::Invoke(_In_ IToastNotification* /* sender */,
     BOOL succeeded = nw::NotificationManager::getSingleton()->DesktopNotificationPostClose(_render_process_id, _render_frame_id, _notification_id, tdr == ToastDismissalReason_UserCanceled);
     hr = succeeded ? S_OK : E_FAIL;
   }
+  nw::NotificationManager::getSingleton()->CancelDesktopNotification(_render_process_id, _render_frame_id, _notification_id);
   return hr;
 }
 
@@ -326,37 +325,28 @@ HRESULT NotificationManagerToastWin::CreateToastXml(_In_ IToastNotificationManag
 HRESULT NotificationManagerToastWin::CreateToast(_In_ IToastNotificationManagerStatics *toastManager, _In_ IXmlDocument *xml, 
   const int render_process_id, const int render_frame_id, const int notification_id)
 {
-  base::string16 appID;
-  if (content::Shell::GetPackage()->root()->GetString("app-id", &appID) == false)
-    content::Shell::GetPackage()->root()->GetString(switches::kmName, &appID);
-
-  ComPtr<IToastNotifier> notifier;
-  HRESULT hr = toastManager->CreateToastNotifierWithId(StringReferenceWrapper(appID.c_str(), appID.length()).Get(), &notifier);
+  ComPtr<IToastNotificationFactory> factory;
+  HRESULT hr = GetActivationFactory(StringReferenceWrapper(RuntimeClass_Windows_UI_Notifications_ToastNotification).Get(), &factory);
   if (SUCCEEDED(hr))
   {
-    ComPtr<IToastNotificationFactory> factory;
-    hr = GetActivationFactory(StringReferenceWrapper(RuntimeClass_Windows_UI_Notifications_ToastNotification).Get(), &factory);
+    ComPtr<IToastNotification>& toast = notification_map_[notification_id];
+    hr = factory->CreateToastNotification(xml, &toast);
     if (SUCCEEDED(hr))
     {
-      ComPtr<IToastNotification> toast;
-      hr = factory->CreateToastNotification(xml, &toast);
+      // Register the event handlers
+      EventRegistrationToken activatedToken, dismissedToken, failedToken;
+      ComPtr<ToastEventHandler> eventHandler = new ToastEventHandler(render_process_id, render_frame_id, notification_id);
+
+      hr = toast->add_Activated(eventHandler.Get(), &activatedToken);
       if (SUCCEEDED(hr))
       {
-        // Register the event handlers
-        EventRegistrationToken activatedToken, dismissedToken, failedToken;
-        ComPtr<ToastEventHandler> eventHandler = new ToastEventHandler(render_process_id, render_frame_id, notification_id);
-
-        hr = toast->add_Activated(eventHandler.Get(), &activatedToken);
+        hr = toast->add_Dismissed(eventHandler.Get(), &dismissedToken);
         if (SUCCEEDED(hr))
         {
-          hr = toast->add_Dismissed(eventHandler.Get(), &dismissedToken);
+          hr = toast->add_Failed(eventHandler.Get(), &failedToken);
           if (SUCCEEDED(hr))
           {
-            hr = toast->add_Failed(eventHandler.Get(), &failedToken);
-            if (SUCCEEDED(hr))
-            {
-              hr = notifier->Show(toast.Get());
-            }
+            hr = notifier_->Show(toast.Get());
           }
         }
       }
@@ -381,6 +371,15 @@ bool NotificationManagerToastWin::IsSupported() {
 }
 
 NotificationManagerToastWin::NotificationManagerToastWin() {
+  HRESULT hr = GetActivationFactory(StringReferenceWrapper(RuntimeClass_Windows_UI_Notifications_ToastNotificationManager).Get(), &toastStatics_);
+  if (SUCCEEDED(hr))
+  {
+    base::string16 appID;
+    if (content::Shell::GetPackage()->root()->GetString("app-id", &appID) == false)
+      content::Shell::GetPackage()->root()->GetString(switches::kmName, &appID);
+
+    HRESULT hr = toastStatics_->CreateToastNotifierWithId(StringReferenceWrapper(appID.c_str(), appID.length()).Get(), &notifier_);
+  }
 }
 
 NotificationManagerToastWin::~NotificationManagerToastWin() {
@@ -395,24 +394,28 @@ bool NotificationManagerToastWin::AddDesktopNotification(const content::ShowDesk
   if (host == NULL)
     return false;
 
-  ComPtr<IToastNotificationManagerStatics> toastStatics;
-  HRESULT hr = GetActivationFactory(StringReferenceWrapper(RuntimeClass_Windows_UI_Notifications_ToastNotificationManager).Get(), &toastStatics);
-  if (SUCCEEDED(hr))
-  {
+  
     ComPtr<IXmlDocument> toastXml;
-    hr = CreateToastXml(toastStatics.Get(), params, &toastXml);
+    HRESULT hr = CreateToastXml(toastStatics_.Get(), params, &toastXml);
     if (SUCCEEDED(hr))
     {
-      hr = CreateToast(toastStatics.Get(), toastXml.Get(), render_process_id, render_frame_id, notification_id);
+      hr = CreateToast(toastStatics_.Get(), toastXml.Get(), render_process_id, render_frame_id, notification_id);
       if (SUCCEEDED(hr))
         DesktopNotificationPostDisplay(render_process_id, render_frame_id, notification_id);
     }
-  }
+  
   
   return SUCCEEDED(hr);
 }
 
 bool NotificationManagerToastWin::CancelDesktopNotification(int render_process_id, int render_frame_id, int notification_id) {
-  return true;
+  std::map<int, ComPtr<IToastNotification>>::iterator i = notification_map_.find(notification_id);
+  if (i == notification_map_.end())
+    return false;
+  
+  ComPtr<IToastNotification> toast = i->second;
+  notification_map_.erase(i);
+
+  return SUCCEEDED(notifier_->Hide(toast.Get()));
 }
 } // namespace nw
diff --git a/src/nw_notification_manager_toast_win.h b/src/nw_notification_manager_toast_win.h
@@ -22,13 +22,20 @@
 
 #include "content/nw/src/nw_notification_manager.h"
 #include <windows.ui.notifications.h>
+#include <wrl\client.h>
+
 
 namespace nw {
+  using namespace Microsoft::WRL;
   using namespace ABI::Windows::UI::Notifications;
   using namespace ABI::Windows::Data::Xml::Dom;
 
 class NotificationManagerToastWin : public NotificationManager{
 
+  ComPtr<IToastNotificationManagerStatics> toastStatics_;
+  ComPtr<IToastNotifier> notifier_;
+  std::map<int, ComPtr<IToastNotification>> notification_map_;
+
   // internal function for AddDesktopNotification
   virtual bool AddDesktopNotification(const content::ShowDesktopNotificationHostMsgParams& params,
     const int render_process_id, const int render_frame_id, const int notification_id, 
diff --git a/src/shell_content_browser_client.cc b/src/shell_content_browser_client.cc
@@ -495,6 +495,15 @@ ShellContentBrowserClient::CreateQuotaPermissionContext() {
   return new ShellQuotaPermissionContext();
 }
 
+void CancelDesktopNotification(int render_process_id, int render_frame_id, int notification_id) {
+  nw::NotificationManager *notificationManager = nw::NotificationManager::getSingleton();
+  if (notificationManager == NULL) {
+    NOTIMPLEMENTED();
+    return;
+  }
+  notificationManager->CancelDesktopNotification(render_process_id, render_frame_id, notification_id);
+}
+
 void ShellContentBrowserClient::ShowDesktopNotification(
       const ShowDesktopNotificationHostMsgParams& params,
       RenderFrameHost* render_frame_host,
@@ -511,6 +520,7 @@ void ShellContentBrowserClient::ShowDesktopNotification(
                                               render_frame_host->GetRoutingID(),
                                               delegate->notification_id(),
                                               false);
+  *cancel_callback = base::Bind(&CancelDesktopNotification, process->GetID(), render_frame_host->GetRoutingID(), delegate->notification_id());
 #else
   NOTIMPLEMENTED();
 #endif
