Add chrome side webusb permission UI code

chrome/browser/usb/usb_tab_helper.cc

 // Copyright 2015 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 "chrome/browser/usb/usb_tab_helper.h"
 
+#include "base/memory/scoped_ptr.h"
+#include "chrome/browser/usb/web_usb_permission_bubble.h"
 #include "chrome/browser/usb/web_usb_permission_provider.h"
 #include "device/devices_app/usb/device_manager_impl.h"
 
 using content::RenderFrameHost;
 using content::WebContents;
 
 DEFINE_WEB_CONTENTS_USER_DATA_KEY(UsbTabHelper);
 
+struct FrameUsbServices {
+  scoped_ptr<WebUSBPermissionProvider> permission_provider;
+  scoped_ptr<ChromeWebUsbPermissionBubble> permission_bubble;
+};
+
 // static
 UsbTabHelper* UsbTabHelper::GetOrCreateForWebContents(
     WebContents* web_contents) {
   UsbTabHelper* tab_helper = FromWebContents(web_contents);
   if (!tab_helper) {
     CreateForWebContents(web_contents);
     tab_helper = FromWebContents(web_contents);
   }
   return tab_helper;
 }
 
 UsbTabHelper::~UsbTabHelper() {}
 
 void UsbTabHelper::CreateDeviceManager(
     RenderFrameHost* render_frame_host,
     mojo::InterfaceRequest<device::usb::DeviceManager> request) {
   DCHECK(WebContents::FromRenderFrameHost(render_frame_host) == web_contents());
   device::usb::PermissionProviderPtr permission_provider;
   GetPermissionProvider(render_frame_host,
                         mojo::GetProxy(&permission_provider));
   device::usb::DeviceManagerImpl::Create(permission_provider.Pass(),
                                          request.Pass());
 }
 
+void UsbTabHelper::CreatePermissionBubble(
+    content::RenderFrameHost* render_frame_host,
+    mojo::InterfaceRequest<webusb::WebUsbPermissionBubble> request) {
+  GetPermissionBubble(render_frame_host, request.Pass());
+}
+
 UsbTabHelper::UsbTabHelper(WebContents* web_contents)
     : content::WebContentsObserver(web_contents) {}
 
 void UsbTabHelper::RenderFrameDeleted(RenderFrameHost* render_frame_host) {
-  permission_provider_.erase(render_frame_host);
+  frame_usb_services_.erase(render_frame_host);
 }
 
 void UsbTabHelper::GetPermissionProvider(
     RenderFrameHost* render_frame_host,
     mojo::InterfaceRequest<device::usb::PermissionProvider> request) {
-  const auto it = permission_provider_.find(render_frame_host);
-  if (it == permission_provider_.end()) {
-    scoped_ptr<WebUSBPermissionProvider> permission_provider(
+  auto it = frame_usb_services_.find(render_frame_host);
+
+  if (it == frame_usb_services_.end()) {
+    scoped_ptr<FrameUsbServices> frame_usb_services(new FrameUsbServices());
+    it = (frame_usb_services_.insert(render_frame_host,
+                                     frame_usb_services.Pass()))
+             .first;
+  }
+
+  if (!it->second->permission_provider) {
+    it->second->permission_provider.reset(
         new WebUSBPermissionProvider(render_frame_host));
-    permission_provider->Bind(request.Pass());
-    permission_provider_.set(render_frame_host, permission_provider.Pass());
-  } else {
-    it->second->Bind(request.Pass());
+  }
+
+  it->second->permission_provider->Bind(request.Pass());
+}
+
+void UsbTabHelper::GetPermissionBubble(
+    content::RenderFrameHost* render_frame_host,
+    mojo::InterfaceRequest<webusb::WebUsbPermissionBubble> request) {
+  auto it = frame_usb_services_.find(render_frame_host);
+
+  if (it == frame_usb_services_.end()) {

[Reilly Grant (use Gerrit), 2015/10/29 01:11:33]

Create a helper function that prevents duplication of this code.

[juncai, 2015/10/31 04:15:36]

Done.

+    scoped_ptr<FrameUsbServices> frame_usb_services(new FrameUsbServices());
+    it = (frame_usb_services_.insert(render_frame_host,
+                                     frame_usb_services.Pass()))
+             .first;
+  }
+
+  if (!it->second->permission_bubble) {
+    it->second->permission_bubble.reset(
+        new ChromeWebUsbPermissionBubble(render_frame_host, request.Pass()));
   }
 }