Factor out a PendingNotificationTracker from the NotificationManager.

content/child/notifications/pending_notifications_tracker.h

+// Copyright 2015 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 CONTENT_CHILD_NOTIFICATIONS_PENDING_NOTIFICATIONS_TRACKER_H_
+#define CONTENT_CHILD_NOTIFICATIONS_PENDING_NOTIFICATIONS_TRACKER_H_
+
+#include <map>
+
+#include "base/id_map.h"
+#include "base/macros.h"
+#include "base/memory/ref_counted.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/memory/weak_ptr.h"
+#include "third_party/WebKit/public/platform/modules/notifications/WebNotificationManager.h"
+
+class SkBitmap;
+
+namespace base {
+class SingleThreadTaskRunner;
+}
+
+namespace blink {
+struct WebNotificationData;
+class WebNotificationDelegate;
+class WebSerializedOrigin;
+}
+
+namespace content {
+
+class NotificationImageLoader;
+class NotificationManager;
+
+// Type definition for the callback signature which is to be invoked when the
+// resources associated with a notification have been fetched.
+using NotificationResourcesFetchedCallback =
+    base::Callback<void(const SkBitmap&)>;
+
+// Tracks all aspects of all pending Web Notifications. Most notably, it's in
+// charge of ensuring that all resource fetches associated with the notification
+// are completed as expected. The data associated with the notifications is
+// stored in this class, to maintain thread integrity of their members.
+//
+// The pending notification tracker is owned by the NotificationManager, and
+// lives on the thread that manager has been associated with.
+class PendingNotificationsTracker {
+ public:
+  explicit PendingNotificationsTracker(
+      scoped_refptr<base::SingleThreadTaskRunner> main_thread_task_runner);
+  ~PendingNotificationsTracker();
+
+  // Adds a page notification to the tracker. Resource fetches for the
+  // notification will be started on asynchronously the main thread.
+  void FetchPageNotificationResources(
+      const blink::WebNotificationData& notification_data,
+      blink::WebNotificationDelegate* delegate,
+      const NotificationResourcesFetchedCallback& callback);
+
+  // Adds a persistent notification to the tracker. Resource fetches for the
+  // notification will be started asynchronously on the main thread.
+  void FetchPersistentNotificationResources(
+      const blink::WebNotificationData& notification_data,
+      const NotificationResourcesFetchedCallback& callback);
+
+  // Cancels all pending and in-fligth fetches for the page notification
+  // identified by |delegate|. Returns if the notification was cancelled.
+  bool CancelPageNotificationFetches(blink::WebNotificationDelegate* delegate);
+
+ private:
+  // To be called on the worker thread when the pending page notification
+  // identified by |notification_id| has finished fetching the icon.
+  void DidFetchPageNotification(blink::WebNotificationDelegate* delegate,
+                                int notification_id,
+                                const SkBitmap& icon);
+
+  // To be called on the worker thread when the pending persistent notification
+  // identified by |notification_id| has finished fetching the icon.
+  void DidFetchPersistentNotification(int notification_id,
+                                      const SkBitmap& icon);
+
+  // Common code for starting to fetch resources associated with any kind of
+  // notification. Will return the id of the pending notification as allocated
+  // in the |pending_notifications_| map.
+  int FetchNotificationResources(
+      const blink::WebNotificationData& notification_data,
+      const NotificationResourcesFetchedCallback& callback,
+      const scoped_refptr<NotificationImageLoader>& image_loader);
+
+  scoped_refptr<base::SingleThreadTaskRunner> main_thread_task_runner_;
+
+  struct PendingNotification;
+
+  // List of the notifications whose resources are still being fetched.
+  IDMap<PendingNotification, IDMapOwnPointer> pending_notifications_;
+
+  // In order to be able to cancel pending page notifications by delegate, store
+  // a mapping of the delegate to the pending notification id as well.
+  std::map<blink::WebNotificationDelegate*, int> delegate_to_pending_id_map_;
+
+  base::WeakPtrFactory<PendingNotificationsTracker> weak_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(PendingNotificationsTracker);
+};
+
+}  // namespace content
+
+#endif  // CONTENT_CHILD_NOTIFICATIONS_PENDING_NOTIFICATIONS_TRACKER_H_