Adds routed interface support between RenderFrameHost and RenderFrame

content/browser/web_contents/web_contents_interface_registry_impl.h

+// 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.
+
+#ifndef CONTENT_BROWSER_WEB_CONTENTS_WEB_CONTENTS_INTERFACE_REGISTRY_IMPL_H_
+#define CONTENT_BROWSER_WEB_CONTENTS_WEB_CONTENTS_INTERFACE_REGISTRY_IMPL_H_
+
+#include <map>
+#include <memory>
+#include <string>
+#include <unordered_set>
+
+#include "base/macros.h"
+#include "content/common/content_export.h"
+#include "content/public/browser/web_contents_interface_registry.h"
+#include "content/public/browser/web_contents_observer.h"
+
+namespace content {
+
+class CONTENT_EXPORT WebContentsInterfaceRegistryImpl
+    : public WebContentsInterfaceRegistry,
+      public WebContentsObserver {
+ public:
+  explicit WebContentsInterfaceRegistryImpl(WebContents* web_contents);
+  ~WebContentsInterfaceRegistryImpl() override;
+
+  // WebContentsInterfaceRegistry:
+  RenderFrameHost* GetCurrentTargetFrame() override;
+  base::WeakPtr<WebContentsInterfaceRegistry> GetWeakPtr() override;
+  void SetCurrentTargetFrameForTesting(
+      RenderFrameHost* render_frame_host) override;
+
+ private:
+  // WebContentsObserver:
+  void RenderFrameCreated(RenderFrameHost* render_frame_host) override;
+  void RenderFrameDeleted(RenderFrameHost* render_frame_host) override;
+  void WebContentsDestroyed() override;
+
+  // WebContentsInterfaceRegistry:
+  void AddFrameInterface(const std::string& name,
+                         std::unique_ptr<FrameInterfaceBase> iface) override;
+  void RemoveFrameInterface(const std::string& name) override;
+  void WillDispatchForFrame(RenderFrameHost* render_frame_host) override;
+
+  void AddInterfacesToFrame(RenderFrameHost* render_frame_host);
+  void RemoveInterfacesFromFrame(RenderFrameHost* render_frame_host);
+
+  static void AddInterfaceToFrame(const std::string& name,
+                                  FrameInterfaceBase* iface,
+                                  RenderFrameHost* render_frame_host);
+
+  RenderFrameHost* current_target_frame_host_ = nullptr;
+
+  // A map of all interfaces managed by this registry. All frames belonging to
+  // web_contents() have corresponding routed interfaces attached to them as
+  // long as they and |this| are alive.
+  std::map<std::string, std::unique_ptr<FrameInterfaceBase>> frame_interfaces_;
+
+  // The set of all RenderFrameHost's currently affected by this registry.
+  std::unordered_set<RenderFrameHost*> frames_;
+
+  base::WeakPtrFactory<WebContentsInterfaceRegistryImpl> weak_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(WebContentsInterfaceRegistryImpl);
+};
+
+}  // namespace content
+
+#endif  // CONTENT_BROWSER_WEB_CONTENTS_WEB_CONTENTS_INTERFACE_REGISTRY_IMPL_H_