electron/brightray/browser/notification_presenter.h

// 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 BRIGHTRAY_BROWSER_NOTIFICATION_PRESENTER_H_
#define BRIGHTRAY_BROWSER_NOTIFICATION_PRESENTER_H_

#include <set>

#include "base/memory/weak_ptr.h"

namespace brightray {

class Notification;
class NotificationDelegate;

class NotificationPresenter {
 public:
  static NotificationPresenter* Create();

  virtual ~NotificationPresenter();

  base::WeakPtr<Notification> CreateNotification(
      NotificationDelegate* delegate);

  std::set<Notification*> notifications() const { return notifications_; }

 protected:
  NotificationPresenter();
  virtual Notification* CreateNotificationObject(
      NotificationDelegate* delegate) = 0;

 private:
  friend class Notification;

  void RemoveNotification(Notification* notification);

  std::set<Notification*> notifications_;

  DISALLOW_COPY_AND_ASSIGN(NotificationPresenter);
};

}  // namespace brightray

#endif  // BRIGHTRAY_BROWSER_NOTIFICATION_PRESENTER_H_