Move notification resource loading from content/child to blink

third_party/WebKit/Source/modules/notifications/NotificationResourcesLoader.h

+// Copyright 2016 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 NotificationResourcesLoader_h
+#define NotificationResourcesLoader_h
+
+#include "core/dom/ContextLifecycleObserver.h"
+#include "modules/ModulesExport.h"
+#include "modules/notifications/NotificationImageLoader.h"
+#include "platform/heap/GarbageCollected.h"
+#include "platform/heap/Handle.h"
+#include "platform/heap/HeapAllocator.h"
+#include "platform/heap/ThreadState.h"
+#include "third_party/skia/include/core/SkBitmap.h"
+#include "wtf/Functional.h"
+#include "wtf/OwnPtr.h"
+#include "wtf/PassOwnPtr.h"
+#include "wtf/Vector.h"
+#include <memory>
+
+namespace blink {
+
+class ExecutionContext;
+struct WebNotificationData;
+struct WebNotificationResources;
+
+// Fetches the resources specified in a given WebNotificationData. Uses a
+// callback to notify the caller when all fetches have finished.
+class MODULES_EXPORT NotificationResourcesLoader final
+    : public GarbageCollectedFinalized<NotificationResourcesLoader>
+    , public ContextLifecycleObserver {

[Peter Beverloo, 2016/04/15 11:07:29]

We can remove some magic here by making ServiceWorkerRegistrationNotifications
the ContextLifecycleObserver instead, and calling through to stop() on each
loader in the set.

[Michael van Ouwerkerk, 2016/04/18 11:26:53]

Done.

+    USING_GARBAGE_COLLECTED_MIXIN(NotificationResourcesLoader);
+    USING_PRE_FINALIZER(NotificationResourcesLoader, stop);
+public:
+    // Called when all fetches have finished. Passes a pointer to the
+    // NotificationResourcesLoader so callers that use multiple loaders can use
+    // the same function to handle the callbacks.
+    using CompletionCallback = Function<void(NotificationResourcesLoader*)>;
+
+    NotificationResourcesLoader(ExecutionContext*, PassOwnPtr<CompletionCallback>);
+    ~NotificationResourcesLoader();
+
+    // Starts fetching the resources specified in the given WebNotificationData.
+    // If all the urls for the resources are empty or invalid,
+    // |m_completionCallback| will be run synchronously, otherwise it will be
+    // run asynchronously when all fetches have finished. Should not be called
+    // more than once.
+    void start(const WebNotificationData&);
+
+    // Returns a new WebNotificationResources populated with the resources that
+    // have been fetched.
+    std::unique_ptr<WebNotificationResources> getResources() const;
+
+    // ContextLifecycleObserver interface.
+    void contextDestroyed() override;
+
+    // Stops every loader in |m_imageLoaders|. This is also used as the
+    // pre-finalizer.
+    void stop();
+
+    DECLARE_VIRTUAL_TRACE();
+
+private:
+    void loadImage(const KURL&, PassOwnPtr<NotificationImageLoader::ImageCallback>);
+    void didLoadIcon(const SkBitmap& image);
+    void didLoadBadge(const SkBitmap& image);
+    void didLoadActionIcon(size_t actionIndex, const SkBitmap& image);
+
+    // Decrements |m_pendingRequestCount| and runs |m_completionCallback| if
+    // there are no more pending requests.
+    void didFinishRequest();
+
+    bool m_started;
+    OwnPtr<CompletionCallback> m_completionCallback;
+    int m_pendingRequestCount;
+    HeapVector<Member<NotificationImageLoader>> m_imageLoaders;
+    SkBitmap m_icon;
+    SkBitmap m_badge;
+    Vector<SkBitmap> m_actionIcons;
+};
+
+} // namespace blink
+
+#endif // NotificationResourcesLoader_h