chromium/ios/chrome/browser/reading_list/model/favicon_web_state_dispatcher_impl.h

// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef IOS_CHROME_BROWSER_READING_LIST_MODEL_FAVICON_WEB_STATE_DISPATCHER_IMPL_H_
#define IOS_CHROME_BROWSER_READING_LIST_MODEL_FAVICON_WEB_STATE_DISPATCHER_IMPL_H_

#include <memory>
#include <vector>

#import "base/memory/raw_ptr.h"
#include "base/memory/weak_ptr.h"
#include "components/reading_list/ios/favicon_web_state_dispatcher.h"

namespace web {
class BrowserState;
}

namespace reading_list {

// 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.
  std::unique_ptr<web::WebState> RequestWebState() override;
  void ReturnWebState(std::unique_ptr<web::WebState> web_state) override;
  void ReleaseAll() override;

 private:
  raw_ptr<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 reading_list

#endif  // IOS_CHROME_BROWSER_READING_LIST_MODEL_FAVICON_WEB_STATE_DISPATCHER_IMPL_H_