Refactor the notification code

brightray/browser/mac/cocoa_notification.h

@@ -0,0 +1,59 @@
+// Copyright (c) 2015 GitHub, Inc.
+// Use of this source code is governed by the MIT license that can be
+// found in the LICENSE file.
+
+#ifndef BROWSER_MAC_COCOA_NOTIFICATION_H_
+#define BROWSER_MAC_COCOA_NOTIFICATION_H_
+
+#include <set>
+
+#import <Foundation/Foundation.h>
+
+#include "base/mac/scoped_nsobject.h"
+#include "base/memory/weak_ptr.h"
+#include "base/strings/string16.h"
+#include "content/public/browser/desktop_notification_delegate.h"
+
+@class NotificationDelegate;
+class SkBitmap;
+
+namespace brightray {
+
+class CocoaNotification {
+ public:
+  static CocoaNotification* FromNSNotification(
+      NSUserNotification* notification);
+
+  CocoaNotification(
+      scoped_ptr<content::DesktopNotificationDelegate> delegate);
+  ~CocoaNotification();
+
+  void ShowNotification(const base::string16& title,
+                        const base::string16& msg,
+                        const SkBitmap& icon);
+  void DismissNotification();
+
+  void NotifyDisplayed();
+  void NotifyClick();
+
+  base::WeakPtr<CocoaNotification> GetWeakPtr() {
+    return weak_factory_.GetWeakPtr();
+  }
+
+ private:
+  static void Cleanup();
+
+  scoped_ptr<content::DesktopNotificationDelegate> delegate_;
+  base::scoped_nsobject<NSUserNotification> notification_;
+
+  base::WeakPtrFactory<CocoaNotification> weak_factory_;
+
+  static base::scoped_nsobject<NotificationDelegate> notification_delegate_;
+  static std::set<CocoaNotification*> notifications_;
+
+  DISALLOW_COPY_AND_ASSIGN(CocoaNotification);
+};
+
+}  // namespace brightray
+
+#endif  // BROWSER_MAC_COCOA_NOTIFICATION_H_