Index: content/browser/web_contents/web_contents_interface_registry_impl.cc |
diff --git a/content/browser/web_contents/web_contents_interface_registry_impl.cc b/content/browser/web_contents/web_contents_interface_registry_impl.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..7db171c78645bbf686db02335088ee2631f9ca34 |
--- /dev/null |
+++ b/content/browser/web_contents/web_contents_interface_registry_impl.cc |
@@ -0,0 +1,95 @@ |
+// 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 |