1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_NOTIFICATIONS_MESSAGE_CENTER_NOTIFICATION_MANAGER_H_
#define CHROME_BROWSER_NOTIFICATIONS_MESSAGE_CENTER_NOTIFICATION_MANAGER_H_
#include <map>
#include <string>
#include "base/memory/scoped_ptr.h"
#include "chrome/browser/notifications/notification.h"
#include "chrome/browser/notifications/notification_ui_manager.h"
#include "chrome/browser/notifications/notification_ui_manager_impl.h"
#include "ui/message_center/message_center.h"
#include "ui/message_center/message_center_tray_delegate.h"
class MessageCenterSettingsController;
class Notification;
class Profile;
// This class extends NotificationUIManagerImpl and delegates actual display
// of notifications to MessageCenter, doing necessary conversions.
class MessageCenterNotificationManager
: public NotificationUIManagerImpl,
public message_center::MessageCenter::Delegate {
public:
explicit MessageCenterNotificationManager(
message_center::MessageCenter* message_center);
virtual ~MessageCenterNotificationManager();
// NotificationUIManager
virtual bool DoesIdExist(const std::string& notification_id) OVERRIDE;
virtual bool CancelById(const std::string& notification_id) OVERRIDE;
virtual bool CancelAllBySourceOrigin(const GURL& source_origin) OVERRIDE;
virtual bool CancelAllByProfile(Profile* profile) OVERRIDE;
virtual void CancelAll() OVERRIDE;
// NotificationUIManagerImpl
virtual bool ShowNotification(const Notification& notification,
Profile* profile) OVERRIDE;
virtual bool UpdateNotification(const Notification& notification,
Profile* profile) OVERRIDE;
// MessageCenter::Delegate
virtual void NotificationRemoved(const std::string& notification_id,
bool by_user) OVERRIDE;
virtual void DisableExtension(const std::string& notification_id) OVERRIDE;
virtual void DisableNotificationsFromSource(
const std::string& notification_id) OVERRIDE;
virtual void ShowSettings(const std::string& notification_id) OVERRIDE;
virtual void ShowSettingsDialog(gfx::NativeView context) OVERRIDE;
virtual void OnClicked(const std::string& notification_id) OVERRIDE;
virtual void OnButtonClicked(const std::string& notification_id,
int button_index) OVERRIDE;
private:
typedef base::Callback<void(const gfx::Image&)> SetImageCallback;
class ImageDownloads
: public base::SupportsWeakPtr<ImageDownloads> {
public:
explicit ImageDownloads(message_center::MessageCenter* message_center);
virtual ~ImageDownloads();
void StartDownloads(const Notification& notification);
void StartDownloadWithImage(const Notification& notification,
const gfx::Image* image,
const GURL& url,
int size,
const SetImageCallback& callback);
void StartDownloadByKey(const Notification& notification,
const char* key,
int size,
const SetImageCallback& callback);
// FaviconHelper callback.
void DownloadComplete(const SetImageCallback& callback,
int download_id,
const GURL& image_url,
int requested_size,
const std::vector<SkBitmap>& bitmaps);
private:
// Weak reference to global message center.
message_center::MessageCenter* message_center_;
DISALLOW_COPY_AND_ASSIGN(ImageDownloads);
};
// This class keeps a set of original Notification objects and corresponding
// Profiles, so when MessageCenter calls back with a notification_id, this
// class has necessary mapping to other source info - for example, it calls
// NotificationDelegate supplied by client when someone clicks on a
// Notification in MessageCenter. Likewise, if a Profile or Extension is
// being removed, the map makes it possible to revoke the notifications from
// MessageCenter. To keep that set, we use the private ProfileNotification
// class that stores a superset of all information about a notification.
// TODO(dimich): Consider merging all 4 types (Notification,
// QueuedNotification, ProfileNotification and NotificationList::Notification)
// into a single class.
class ProfileNotification {
public:
ProfileNotification(Profile* profile,
const Notification& notification,
message_center::MessageCenter* message_center);
virtual ~ProfileNotification();
void StartDownloads();
Profile* profile() const { return profile_; }
const Notification& notification() const { return notification_; }
// Returns extension_id if the notification originates from an extension,
// empty string otherwise.
std::string GetExtensionId();
private:
// Weak, guaranteed not to be used after profile removal by parent class.
Profile* profile_;
Notification notification_;
// Track the downloads for this notification so the notification can be
// updated properly.
scoped_ptr<ImageDownloads> downloads_;
};
scoped_ptr<message_center::MessageCenterTrayDelegate> tray_;
message_center::MessageCenter* message_center_; // Weak, global.
// Use a map by notification_id since this mapping is the most often used.
typedef std::map<std::string, ProfileNotification*> NotificationMap;
NotificationMap profile_notifications_;
scoped_ptr<MessageCenterSettingsController> settings_controller_;
// Helpers that add/remove the notification from local map and MessageCenter.
// They take ownership of profile_notification object.
void AddProfileNotification(ProfileNotification* profile_notification);
void RemoveProfileNotification(ProfileNotification* profile_notification,
bool by_user);
ProfileNotification* FindProfileNotification(const std::string& id) const;
DISALLOW_COPY_AND_ASSIGN(MessageCenterNotificationManager);
};
#endif // CHROME_BROWSER_NOTIFICATIONS_MESSAGE_CENTER_NOTIFICATION_MANAGER_H_
|
