// Copyright (c) 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 CEF_LIBCEF_BROWSER_IOTHREAD_STATE_H_
#define CEF_LIBCEF_BROWSER_IOTHREAD_STATE_H_
#pragma once

#include "cef/include/cef_request_context.h"
#include "cef/include/cef_request_context_handler.h"
#include "cef/include/cef_scheme.h"
#include "cef/libcef/browser/request_context_handler_map.h"
#include "content/public/browser/browser_thread.h"

namespace content {
struct GlobalRenderFrameHostId;
}

class GURL;

// Stores state that will be accessed on the IO thread. Life span is controlled
// by CefBrowserContext. Created on the UI thread but accessed and destroyed on
// the IO thread. See browser_context.h for an object relationship diagram.
class CefIOThreadState : public base::RefCountedThreadSafe<
                             CefIOThreadState,
                             content::BrowserThread::DeleteOnIOThread> {
 public:
  CefIOThreadState();

  CefIOThreadState(const CefIOThreadState&) = delete;
  CefIOThreadState& operator=(const CefIOThreadState&) = delete;

  // See comments in CefRequestContextHandlerMap.
  void AddHandler(const content::GlobalRenderFrameHostId& global_id,
                  CefRefPtr<CefRequestContextHandler> handler);
  void RemoveHandler(const content::GlobalRenderFrameHostId& global_id);
  CefRefPtr<CefRequestContextHandler> GetHandler(
      const content::GlobalRenderFrameHostId& global_id,
      bool require_frame_match) const;

  // Manage scheme handler factories associated with this context.
  void RegisterSchemeHandlerFactory(const std::string& scheme_name,
                                    const std::string& domain_name,
                                    CefRefPtr<CefSchemeHandlerFactory> factory);
  void ClearSchemeHandlerFactories();
  CefRefPtr<CefSchemeHandlerFactory> GetSchemeHandlerFactory(const GURL& url);

 private:
  friend struct content::BrowserThread::DeleteOnThread<
      content::BrowserThread::IO>;
  friend class base::DeleteHelper<CefIOThreadState>;

  ~CefIOThreadState();

  // Map IDs to CefRequestContextHandler objects.
  CefRequestContextHandlerMap handler_map_;

  // Map (scheme, domain) to factories.
  using SchemeHandlerFactoryMap = std::map<std::pair<std::string, std::string>,
                                           CefRefPtr<CefSchemeHandlerFactory>>;
  SchemeHandlerFactoryMap scheme_handler_factory_map_;
};

#endif  // CEF_LIBCEF_BROWSER_IOTHREAD_STATE_H_