chrome/browser/ui/views/bubble/bubble_contents_wrapper_service.h

// Copyright 2021 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 CHROME_BROWSER_UI_VIEWS_BUBBLE_BUBBLE_CONTENTS_WRAPPER_SERVICE_H_
#define CHROME_BROWSER_UI_VIEWS_BUBBLE_BUBBLE_CONTENTS_WRAPPER_SERVICE_H_

#include <memory>
#include <string>

#include "base/containers/flat_map.h"
#include "chrome/browser/ui/views/bubble/bubble_contents_wrapper.h"
#include "components/keyed_service/core/keyed_service.h"

class GURL;
class Profile;

// This service maintains a 1-to-1 mapping of BubbleContentsWrapper to WebUI URL
// hostname for a given profile. This allows multiple WebUI bubbles across
// profile windows to share the same BubbleContentsWrapper instance, minimizing
// the resource cost of persisting the wrapped WebContents. Persisting a
// WebContents is done to improve initialization delays for WebUI bubbles.
// A KeyedService is used to ensure any references to the Profile from the
// wrapped WebContents instances are removed before Profile destruction.
class BubbleContentsWrapperService : public KeyedService {
 public:
  explicit BubbleContentsWrapperService(Profile* profile);
  ~BubbleContentsWrapperService() override;

  template <typename T>
  void InitBubbleContentsWrapper(const GURL& webui_url,
                                 int task_manager_string_id,
                                 bool enable_extension_apis) {
    // If replacing an existing BubbleContentsWrapper make sure it has no
    // associated host.
    auto it = web_contents_map_.find(webui_url.host());
    if (it != web_contents_map_.end() && it->second->GetHost()) {
      it->second->CloseUI();
      DCHECK(!it->second->GetHost());
    }

    auto contents_wrapper = std::make_unique<BubbleContentsWrapperT<T>>(
        webui_url, profile_, task_manager_string_id, enable_extension_apis);
    contents_wrapper->ReloadWebContents();
    web_contents_map_.insert({webui_url.host(), std::move(contents_wrapper)});
  }

  void Shutdown() override;

  BubbleContentsWrapper* GetBubbleContentsWrapperFromURL(const GURL& webui_url);

 private:
  using WebContentsMap =
      base::flat_map<std::string, std::unique_ptr<BubbleContentsWrapper>>;

  // Profile associated with this service.
  Profile* const profile_;

  // Associates BubbleContentsWrapper instances with their WebUI URL hostname.
  WebContentsMap web_contents_map_;
};

#endif  // CHROME_BROWSER_UI_VIEWS_BUBBLE_BUBBLE_CONTENTS_WRAPPER_SERVICE_H_