Save favicon during reading list distillation

ios/chrome/browser/dom_distiller/favicon_web_state_dispatcher_impl.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 IOS_CHROME_BROWSER_DOM_DISTILLER_FAVICON_WEB_STATE_DISPATCHER_IMPL_H_
+#define IOS_CHROME_BROWSER_DOM_DISTILLER_FAVICON_WEB_STATE_DISPATCHER_IMPL_H_
+
+#include <memory>
+#include <vector>
+
+#include "base/memory/weak_ptr.h"
+#include "components/dom_distiller/ios/favicon_web_state_dispatcher.h"
+
+namespace web {
+class BrowserState;
+}
+
+namespace dom_distiller {
+
+// Implementation of the FaviconWebStateDispatcher.
+class FaviconWebStateDispatcherImpl : public FaviconWebStateDispatcher {
+ public:
+  // Constructor for keeping the WebStates alive for |keep_alive_second|
+  // seconds. If |keep_alive_second| < 0 then the default value is used.
+  FaviconWebStateDispatcherImpl(web::BrowserState* browser_state,
+                                int64_t keep_alive_second);
+  ~FaviconWebStateDispatcherImpl() override;
+
+  // FaviconWebStateDispatcher implementation.
+  web::WebState* RequestWebState() override;
+  void ReturnWebState(web::WebState* web_state) override;
+
+ private:
+  web::BrowserState* browser_state_;
+  // Map of the WebStates currently alive.
+  std::vector<std::unique_ptr<web::WebState>> web_states_;
+  // Time during which the WebState will be kept alive after being returned.
+  int64_t keep_alive_second_;
+  base::WeakPtrFactory<FaviconWebStateDispatcherImpl> weak_ptr_factory_;
+};
+
+}  // namespace dom_distiller
+
+#endif  // IOS_CHROME_BROWSER_DOM_DISTILLER_FAVICON_WEB_STATE_DISPATCHER_IMPL_H_