Adds routed interface support between RenderFrameHost and RenderFrame

content/browser/web_contents/web_contents_interface_registry_impl.cc

+// 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.
+
+#include "content/browser/web_contents/web_contents_interface_registry_impl.h"
+
+#include "base/bind.h"
+#include "base/memory/ptr_util.h"
+#include "content/public/browser/render_frame_host.h"
+#include "content/public/browser/web_contents.h"
+
+namespace content {
+
+WebContentsInterfaceRegistryImpl::WebContentsInterfaceRegistryImpl(
+    WebContents* web_contents) : WebContentsObserver(web_contents) {
+}
+
+WebContentsInterfaceRegistryImpl::~WebContentsInterfaceRegistryImpl() {
+  for (auto* frame : frames_)
+    RemoveInterfacesFromFrame(frame);
+}
+
+RenderFrameHost* WebContentsInterfaceRegistryImpl::GetCurrentTargetFrame() {
+  DCHECK(current_target_frame_host_);
+  return current_target_frame_host_;
+}
+
+void WebContentsInterfaceRegistryImpl::SetCurrentTargetFrameForTesting(
+    RenderFrameHost* render_frame_host) {
+  current_target_frame_host_ = render_frame_host;
+}
+
+void WebContentsInterfaceRegistryImpl::RenderFrameCreated(
+    RenderFrameHost* render_frame_host) {
+  AddInterfacesToFrame(render_frame_host);
+
+  auto result = frames_.insert(render_frame_host);
+  DCHECK(result.second);
+}
+
+void WebContentsInterfaceRegistryImpl::RenderFrameDeleted(
+    RenderFrameHost* render_frame_host) {
+  auto it = frames_.find(render_frame_host);
+  if (it == frames_.end())
+    return;
+
+  frames_.erase(it);
+  RemoveInterfacesFromFrame(render_frame_host);
+}
+
+void WebContentsInterfaceRegistryImpl::AddFrameInterface(
+    const std::string& name,
+    std::unique_ptr<FrameInterfaceBase> iface) {
+  for (auto* frame : frames_)
+    AddInterfaceToFrame(name, iface.get(), frame);
+
+  auto result =
+      frame_interfaces_.insert(std::make_pair(name, std::move(iface)));
+  DCHECK(result.second);
+}
+
+void WebContentsInterfaceRegistryImpl::WillDispatchForFrame(
+    RenderFrameHost* render_frame_host) {
+  current_target_frame_host_ = render_frame_host;
+}
+
+void WebContentsInterfaceRegistryImpl::AddInterfacesToFrame(
+    RenderFrameHost* render_frame_host) {
+  for (auto& entry : frame_interfaces_)
+    AddInterfaceToFrame(entry.first, entry.second.get(), render_frame_host);
+}
+
+void WebContentsInterfaceRegistryImpl::RemoveInterfacesFromFrame(
+    RenderFrameHost* render_frame_host) {
+  for (auto& entry : frame_interfaces_)
+    render_frame_host->RemoveRoutedInterface(entry.first);
+}
+
+// static
+void WebContentsInterfaceRegistryImpl::AddInterfaceToFrame(
+    const std::string& name,
+    FrameInterfaceBase* iface,
+    RenderFrameHost* render_frame_host) {
+  render_frame_host->AddRoutedInterface(
+      name, base::Bind(&FrameInterfaceBase::AddBinding,
+                       base::Unretained(iface), render_frame_host));
+}
+
+// static
+std::unique_ptr<WebContentsInterfaceRegistry>
+WebContentsInterfaceRegistry::Create(WebContents* web_contents) {
+  return base::MakeUnique<WebContentsInterfaceRegistryImpl>(web_contents);
+}
+
+}  // namespace content