Index: chrome/browser/usb/usb_tab_helper.cc |
diff --git a/chrome/browser/usb/usb_tab_helper.cc b/chrome/browser/usb/usb_tab_helper.cc |
index 152dd0dae5032a48bb0066d0e5945bb3ec4f35c5..5d7b527de7e4b0744e999a59d0c683a2dda3e597 100644 |
--- a/chrome/browser/usb/usb_tab_helper.cc |
+++ b/chrome/browser/usb/usb_tab_helper.cc |
@@ -4,6 +4,8 @@ |
#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" |
@@ -12,6 +14,11 @@ 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) { |
@@ -36,23 +43,53 @@ void UsbTabHelper::CreateDeviceManager( |
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
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())); |
} |
} |