| OLD | NEW |
| 1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/usb/usb_tab_helper.h" | 5 #include "chrome/browser/usb/usb_tab_helper.h" |
| 6 | 6 |
| 7 #include <utility> |
| 8 |
| 9 #include "base/memory/scoped_ptr.h" |
| 10 #include "chrome/browser/usb/web_usb_permission_bubble.h" |
| 7 #include "chrome/browser/usb/web_usb_permission_provider.h" | 11 #include "chrome/browser/usb/web_usb_permission_provider.h" |
| 8 #include "device/devices_app/usb/device_manager_impl.h" | 12 #include "device/devices_app/usb/device_manager_impl.h" |
| 9 | 13 |
| 10 using content::RenderFrameHost; | 14 using content::RenderFrameHost; |
| 11 using content::WebContents; | 15 using content::WebContents; |
| 12 | 16 |
| 13 DEFINE_WEB_CONTENTS_USER_DATA_KEY(UsbTabHelper); | 17 DEFINE_WEB_CONTENTS_USER_DATA_KEY(UsbTabHelper); |
| 14 | 18 |
| 19 struct FrameUsbServices { |
| 20 scoped_ptr<WebUSBPermissionProvider> permission_provider; |
| 21 scoped_ptr<ChromeWebUsbPermissionBubble> permission_bubble; |
| 22 }; |
| 23 |
| 15 // static | 24 // static |
| 16 UsbTabHelper* UsbTabHelper::GetOrCreateForWebContents( | 25 UsbTabHelper* UsbTabHelper::GetOrCreateForWebContents( |
| 17 WebContents* web_contents) { | 26 WebContents* web_contents) { |
| 18 UsbTabHelper* tab_helper = FromWebContents(web_contents); | 27 UsbTabHelper* tab_helper = FromWebContents(web_contents); |
| 19 if (!tab_helper) { | 28 if (!tab_helper) { |
| 20 CreateForWebContents(web_contents); | 29 CreateForWebContents(web_contents); |
| 21 tab_helper = FromWebContents(web_contents); | 30 tab_helper = FromWebContents(web_contents); |
| 22 } | 31 } |
| 23 return tab_helper; | 32 return tab_helper; |
| 24 } | 33 } |
| 25 | 34 |
| 26 UsbTabHelper::~UsbTabHelper() {} | 35 UsbTabHelper::~UsbTabHelper() {} |
| 27 | 36 |
| 28 void UsbTabHelper::CreateDeviceManager( | 37 void UsbTabHelper::CreateDeviceManager( |
| 29 RenderFrameHost* render_frame_host, | 38 RenderFrameHost* render_frame_host, |
| 30 mojo::InterfaceRequest<device::usb::DeviceManager> request) { | 39 mojo::InterfaceRequest<device::usb::DeviceManager> request) { |
| 31 DCHECK(WebContents::FromRenderFrameHost(render_frame_host) == web_contents()); | 40 DCHECK(WebContents::FromRenderFrameHost(render_frame_host) == web_contents()); |
| 32 device::usb::PermissionProviderPtr permission_provider; | 41 device::usb::PermissionProviderPtr permission_provider; |
| 33 GetPermissionProvider(render_frame_host, | 42 GetPermissionProvider(render_frame_host, |
| 34 mojo::GetProxy(&permission_provider)); | 43 mojo::GetProxy(&permission_provider)); |
| 35 device::usb::DeviceManagerImpl::Create(permission_provider.Pass(), | 44 device::usb::DeviceManagerImpl::Create(permission_provider.Pass(), |
| 36 request.Pass()); | 45 request.Pass()); |
| 37 } | 46 } |
| 38 | 47 |
| 48 void UsbTabHelper::CreatePermissionBubble( |
| 49 content::RenderFrameHost* render_frame_host, |
| 50 mojo::InterfaceRequest<webusb::WebUsbPermissionBubble> request) { |
| 51 GetPermissionBubble(render_frame_host, request.Pass()); |
| 52 } |
| 53 |
| 39 UsbTabHelper::UsbTabHelper(WebContents* web_contents) | 54 UsbTabHelper::UsbTabHelper(WebContents* web_contents) |
| 40 : content::WebContentsObserver(web_contents) {} | 55 : content::WebContentsObserver(web_contents) {} |
| 41 | 56 |
| 42 void UsbTabHelper::RenderFrameDeleted(RenderFrameHost* render_frame_host) { | 57 void UsbTabHelper::RenderFrameDeleted(RenderFrameHost* render_frame_host) { |
| 43 permission_provider_.erase(render_frame_host); | 58 frame_usb_services_.erase(render_frame_host); |
| 59 } |
| 60 |
| 61 FrameUsbServices* UsbTabHelper::GetFrameUsbService( |
| 62 content::RenderFrameHost* render_frame_host) { |
| 63 FrameUsbServicesMap::const_iterator it = |
| 64 frame_usb_services_.find(render_frame_host); |
| 65 if (it == frame_usb_services_.end()) { |
| 66 scoped_ptr<FrameUsbServices> frame_usb_services(new FrameUsbServices()); |
| 67 it = (frame_usb_services_.insert( |
| 68 std::make_pair(render_frame_host, frame_usb_services.Pass()))) |
| 69 .first; |
| 70 } |
| 71 return it->second.get(); |
| 44 } | 72 } |
| 45 | 73 |
| 46 void UsbTabHelper::GetPermissionProvider( | 74 void UsbTabHelper::GetPermissionProvider( |
| 47 RenderFrameHost* render_frame_host, | 75 RenderFrameHost* render_frame_host, |
| 48 mojo::InterfaceRequest<device::usb::PermissionProvider> request) { | 76 mojo::InterfaceRequest<device::usb::PermissionProvider> request) { |
| 49 const auto it = permission_provider_.find(render_frame_host); | 77 FrameUsbServices* frame_usb_services = GetFrameUsbService(render_frame_host); |
| 50 if (it == permission_provider_.end()) { | 78 if (!frame_usb_services->permission_provider) { |
| 51 scoped_ptr<WebUSBPermissionProvider> permission_provider( | 79 frame_usb_services->permission_provider.reset( |
| 52 new WebUSBPermissionProvider(render_frame_host)); | 80 new WebUSBPermissionProvider(render_frame_host)); |
| 53 permission_provider->Bind(request.Pass()); | |
| 54 permission_provider_[render_frame_host] = permission_provider.Pass(); | |
| 55 } else { | |
| 56 it->second->Bind(request.Pass()); | |
| 57 } | 81 } |
| 82 frame_usb_services->permission_provider->Bind(request.Pass()); |
| 58 } | 83 } |
| 84 |
| 85 void UsbTabHelper::GetPermissionBubble( |
| 86 content::RenderFrameHost* render_frame_host, |
| 87 mojo::InterfaceRequest<webusb::WebUsbPermissionBubble> request) { |
| 88 FrameUsbServices* frame_usb_services = GetFrameUsbService(render_frame_host); |
| 89 if (!frame_usb_services->permission_bubble) { |
| 90 frame_usb_services->permission_bubble.reset( |
| 91 new ChromeWebUsbPermissionBubble(render_frame_host)); |
| 92 } |
| 93 frame_usb_services->permission_bubble->Bind(request.Pass()); |
| 94 } |
| OLD | NEW |