extensions/browser/view_type_utils.cc

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

#include "extensions/browser/view_type_utils.h"

#include "base/lazy_instance.h"
#include "content/public/browser/web_contents.h"
#include "extensions/browser/extension_web_contents_observer.h"
#include "extensions/browser/extensions_browser_client.h"

using content::WebContents;

namespace extensions {

namespace {

const char kViewTypeUserDataKey[] = "ViewTypeUserData";

class ViewTypeUserData : public base::SupportsUserData::Data {
 public:
  explicit ViewTypeUserData(mojom::ViewType type) : type_(type) {}
  ~ViewTypeUserData() override {}
  mojom::ViewType type() { return type_; }

 private:
  mojom::ViewType type_;
};

}  // namespace

mojom::ViewType GetViewType(WebContents* tab) {
  if (!tab)
    return mojom::ViewType::kInvalid;

  ViewTypeUserData* user_data = static_cast<ViewTypeUserData*>(
      tab->GetUserData(&kViewTypeUserDataKey));

  return user_data ? user_data->type() : mojom::ViewType::kInvalid;
}

void SetViewType(WebContents* tab, mojom::ViewType type) {
  tab->SetUserData(&kViewTypeUserDataKey,
                   std::make_unique<ViewTypeUserData>(type));

  ExtensionsBrowserClient::Get()->AttachExtensionTaskManagerTag(tab, type);

  if (auto* ewco = ExtensionWebContentsObserver::GetForWebContents(tab)) {
    tab->ForEachRenderFrameHost(
        [ewco, type](content::RenderFrameHost* frame_host) {
          if (mojom::LocalFrame* local_frame = ewco->GetLocalFrame(frame_host))
            local_frame->NotifyRenderViewType(type);
        });
  }
}

}  // namespace extensions