diff options
Diffstat (limited to 'chrome/common/desktop_notifications')
-rw-r--r-- | chrome/common/desktop_notifications/active_notification_tracker.cc | 69 | ||||
-rw-r--r-- | chrome/common/desktop_notifications/active_notification_tracker.h | 50 |
2 files changed, 119 insertions, 0 deletions
diff --git a/chrome/common/desktop_notifications/active_notification_tracker.cc b/chrome/common/desktop_notifications/active_notification_tracker.cc new file mode 100644 index 0000000..230d76a --- /dev/null +++ b/chrome/common/desktop_notifications/active_notification_tracker.cc @@ -0,0 +1,69 @@ +// Copyright (c) 2009 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. + +#include "chrome/common/desktop_notifications/active_notification_tracker.h" + +#include "base/message_loop.h" +#include "webkit/api/public/WebNotification.h" +#include "webkit/api/public/WebNotificationPermissionCallback.h" + +using WebKit::WebNotification; +using WebKit::WebNotificationPermissionCallback; + +bool ActiveNotificationTracker::GetId( + const WebNotification& notification, int& id) { + DCHECK(MessageLoop::current()->type() == MessageLoop::TYPE_UI); + ReverseTable::iterator iter = reverse_notification_table_.find(notification); + if (iter == reverse_notification_table_.end()) + return false; + id = iter->second; + return true; +} + +bool ActiveNotificationTracker::GetNotification( + int id, WebNotification* notification) { + DCHECK(MessageLoop::current()->type() == MessageLoop::TYPE_UI); + WebNotification* lookup = notification_table_.Lookup(id); + if (!lookup) + return false; + + *notification = *lookup; + return true; +} + +int ActiveNotificationTracker::RegisterNotification( + const WebKit::WebNotification& proxy) { + DCHECK(MessageLoop::current()->type() == MessageLoop::TYPE_UI); + WebNotification* notification = new WebNotification(proxy); + int id = notification_table_.Add(notification); + reverse_notification_table_[proxy] = id; + return id; +} + +void ActiveNotificationTracker::UnregisterNotification(int id) { + DCHECK(MessageLoop::current()->type() == MessageLoop::TYPE_UI); + // We want to free the notification after removing it from the table. + scoped_ptr<WebNotification> notification(notification_table_.Lookup(id)); + notification_table_.Remove(id); + DCHECK(notification.get()); + if (notification.get()) + reverse_notification_table_.erase(*notification); +} + +WebNotificationPermissionCallback* ActiveNotificationTracker::GetCallback( + int id) { + DCHECK(MessageLoop::current()->type() == MessageLoop::TYPE_UI); + return callback_table_.Lookup(id); +} + +int ActiveNotificationTracker::RegisterPermissionRequest( + WebNotificationPermissionCallback* callback) { + DCHECK(MessageLoop::current()->type() == MessageLoop::TYPE_UI); + return callback_table_.Add(callback); +} + +void ActiveNotificationTracker::OnPermissionRequestComplete(int id) { + DCHECK(MessageLoop::current()->type() == MessageLoop::TYPE_UI); + callback_table_.Remove(id); +} diff --git a/chrome/common/desktop_notifications/active_notification_tracker.h b/chrome/common/desktop_notifications/active_notification_tracker.h new file mode 100644 index 0000000..4bdd5d9 --- /dev/null +++ b/chrome/common/desktop_notifications/active_notification_tracker.h @@ -0,0 +1,50 @@ +// Copyright (c) 2009 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_COMMON_DESKTOP_NOTIFICATIONS_ACTIVE_NOTIFICATION_TRACKER_H_ +#define CHROME_COMMON_DESKTOP_NOTIFICATIONS_ACTIVE_NOTIFICATION_TRACKER_H_ + +#include <map> + +#include "base/basictypes.h" +#include "base/id_map.h" +#include "base/hash_tables.h" +#include "webkit/api/public/WebNotification.h" + +namespace WebKit { +class WebNotificationPermissionCallback; +} + +// This class manages the set of active Notification objects in either +// a render or worker process. This class should be accessed only on +// the main thread. +class ActiveNotificationTracker { + public: + ActiveNotificationTracker() {} + + // Methods for tracking active notification objects. + int RegisterNotification(const WebKit::WebNotification& notification); + void UnregisterNotification(int id); + bool GetId(const WebKit::WebNotification& notification, int& id); + bool GetNotification(int id, WebKit::WebNotification* notification); + + // Methods for tracking active permission requests. + int RegisterPermissionRequest( + WebKit::WebNotificationPermissionCallback* callback); + void OnPermissionRequestComplete(int id); + WebKit::WebNotificationPermissionCallback* GetCallback(int id); + + private: + typedef std::map<WebKit::WebNotification, int> ReverseTable; + + // Tracking maps for active notifications and permission requests. + IDMap<WebKit::WebNotification> notification_table_; + ReverseTable reverse_notification_table_; + IDMap<WebKit::WebNotificationPermissionCallback> callback_table_; + + DISALLOW_COPY_AND_ASSIGN(ActiveNotificationTracker); +}; + +#endif // CHROME_COMMON_DESKTOP_NOTIFICATIONS_ACTIVE_NOTIFICATION_TRACKER_H_ + |