From 03688b9415cedd7b417610bb266c8edc7d5abcd1 Mon Sep 17 00:00:00 2001
From: Samuel Attard <samuel.r.attard@gmail.com>
Date: Mon, 24 Apr 2017 12:03:05 +1000
Subject: [PATCH] Get image notifications working on Win7 + Win10

---
 atom/browser/api/atom_api_notification_win.cc | 64 +++++++++++++++----
 1 file changed, 52 insertions(+), 12 deletions(-)

diff --git a/atom/browser/api/atom_api_notification_win.cc b/atom/browser/api/atom_api_notification_win.cc
index 93482d481624..c92a1b2ce7fe 100644
--- a/atom/browser/api/atom_api_notification_win.cc
+++ b/atom/browser/api/atom_api_notification_win.cc
@@ -4,15 +4,20 @@
 
 #include "atom/browser/api/atom_api_notification.h"
 
+#include "atom/browser/browser.h"
 #include "atom/browser/ui/notification_delegate_adapter.h"
 #include "atom/browser/ui/win/toast_handler.h"
 #include "atom/browser/ui/win/toast_lib.h"
+#include "base/files/file_util.h"
+#include "base/md5.h"
 #include "base/strings/utf_string_conversions.h"
 #include "browser/notification.h"
 #include "browser/notification_presenter.h"
+#include "browser/win/notification_presenter_win.h"
 #include "browser/win/notification_presenter_win7.h"
 #include "common/string_conversion.h"
 #include "third_party/skia/include/core/SkBitmap.h"
+#include "ui/gfx/codec/png_codec.h"
 #include "url/gurl.h"
 
 namespace atom {
@@ -21,13 +26,49 @@ namespace api {
 
 bool can_toast_ = true;
 bool initialized_ = false;
-brightray::NotificationPresenter* presenter;
+brightray::NotificationPresenterWin7* presenter;
+
+base::ScopedTempDir temp_dir_;
+
+bool SaveIconToPath(const SkBitmap& bitmap, const base::FilePath& path) {
+  std::vector<unsigned char> png_data;
+  if (!gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, false, &png_data))
+    return false;
+
+  char* data = reinterpret_cast<char*>(&png_data[0]);
+  int size = static_cast<int>(png_data.size());
+  return base::WriteFile(path, data, size) == size;
+}
 
 void Notification::Show() {
+  SkBitmap image = *(new SkBitmap);
+  if (has_icon_) {
+    image = *(icon_.ToSkBitmap());
+  }
+
   if (can_toast_) {
     atom::AtomToastHandler* handler = new atom::AtomToastHandler(this);
-    WinToastLib::WinToastTemplate toast = WinToastLib::WinToastTemplate(WinToastLib::WinToastTemplate::TextTwoLines);
-    // toast.setImagePath(L"C:\example.png");
+    WinToastLib::WinToastTemplate::WinToastTemplateType toastType = WinToastLib::WinToastTemplate::TextOneLine;
+    if (!has_icon_) {
+      if (body_ != L"") {
+        toastType = WinToastLib::WinToastTemplate::TextTwoLines;
+      } else {
+        toastType = WinToastLib::WinToastTemplate::TextOneLine;
+      }
+    } else {
+      if (body_ != L"") {
+        toastType = WinToastLib::WinToastTemplate::ImageWithTwoLines;
+      } else {
+        toastType = WinToastLib::WinToastTemplate::ImageWithOneLine;
+      }
+    }
+    WinToastLib::WinToastTemplate toast = WinToastLib::WinToastTemplate(toastType);
+
+    std::string filename = base::MD5String(base::UTF16ToUTF8(icon_path_)) + ".png";
+    base::FilePath savePath = temp_dir_.GetPath().Append(base::UTF8ToUTF16(filename));
+    if (has_icon_ && SaveIconToPath(image, savePath)) {
+      toast.setImagePath(savePath.value());
+    }
     toast.setTextField(title_, WinToastLib::WinToastTemplate::TextField::FirstLine);
     toast.setTextField(body_, WinToastLib::WinToastTemplate::TextField::SecondLine);
     toast.setSilent(silent_);
@@ -38,27 +79,26 @@ void Notification::Show() {
   } else {
     AtomNotificationDelegateAdapter* adapter = new AtomNotificationDelegateAdapter(this);
     auto notif = presenter->CreateNotification(adapter);
-    GURL* u = new GURL;
+    GURL nullUrl = *(new GURL);
     notif->Show(
       title_,
       body_,
       "",
-      u->Resolve(""),
-      *(new SkBitmap),
-      true
+      nullUrl,
+      image,
+      silent_
     );
   }
 }
 
 void Notification::OnInitialProps() {
   if (!initialized_) {
-    WinToastLib::WinToast::instance()->setAppName(L"WinToastExample");
-    WinToastLib::WinToast::instance()->setAppUserModelId(
-                WinToastLib::WinToast::configureAUMI(L"mohabouje", L"wintoast", L"wintoastexample", L"20161006")
-    );
+    Browser* browser = Browser::Get();
+    WinToastLib::WinToast::instance()->setAppName(base::UTF8ToUTF16(browser->GetName()));
+    WinToastLib::WinToast::instance()->setAppUserModelId(browser->GetAppUserModelID());
     can_toast_ = WinToastLib::WinToast::instance()->initialize();
+    temp_dir_.CreateUniqueTempDir();
   }
-  can_toast_ = false;
   if (!can_toast_) {
     presenter = new brightray::NotificationPresenterWin7;
   }