Implement tab indicator for WebUSB.

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 <utility>
 
 #include "base/memory/scoped_ptr.h"
+#include "chrome/browser/ui/browser_finder.h"
+#include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/browser/usb/web_usb_permission_provider.h"
 #include "device/usb/mojo/device_manager_impl.h"
 
 #if defined(OS_ANDROID)
 #include "chrome/browser/android/usb/web_usb_chooser_service_android.h"
 #else
 #include "chrome/browser/usb/web_usb_chooser_service.h"
 #endif  // defined(OS_ANDROID)
 
 using content::RenderFrameHost;
@@ skipping 30 matching lines @@
   device::usb::DeviceManagerImpl::Create(
       GetPermissionProvider(render_frame_host), std::move(request));
 }
 
 void UsbTabHelper::CreateChooserService(
     content::RenderFrameHost* render_frame_host,
     mojo::InterfaceRequest<device::usb::ChooserService> request) {
   GetChooserService(render_frame_host, std::move(request));
 }
 
+void UsbTabHelper::IncrementConnectionCount() {
+  device_connection_count_++;
+  NotifyTabStateChanged();
+}
+
+void UsbTabHelper::DecrementConnectionCount() {
+  DCHECK_GT(device_connection_count_, 0);
+  device_connection_count_--;
+  NotifyTabStateChanged();
+}
+
+bool UsbTabHelper::IsDeviceConnected() const {
+  return device_connection_count_ > 0;
+}
+
 UsbTabHelper::UsbTabHelper(WebContents* web_contents)
-    : content::WebContentsObserver(web_contents) {}
+    : content::WebContentsObserver(web_contents), device_connection_count_(0) {}
 
 void UsbTabHelper::RenderFrameDeleted(RenderFrameHost* render_frame_host) {
   frame_usb_services_.erase(render_frame_host);
 }
 
 FrameUsbServices* UsbTabHelper::GetFrameUsbService(
     content::RenderFrameHost* render_frame_host) {
   FrameUsbServicesMap::const_iterator it =
       frame_usb_services_.find(render_frame_host);
   if (it == frame_usb_services_.end()) {
@@ skipping 22 matching lines @@
   if (!frame_usb_services->chooser_service) {
     frame_usb_services->chooser_service.reset(
 #if defined(OS_ANDROID)
         new WebUsbChooserServiceAndroid(render_frame_host));
 #else
         new WebUsbChooserService(render_frame_host));
 #endif  // defined(OS_ANDROID)
   }
   frame_usb_services->chooser_service->Bind(std::move(request));
 }
+
+void UsbTabHelper::NotifyTabStateChanged() const {
+  // TODO(https://crbug.com/601627): Implement tab indicator for Android.
+#if !defined(OS_ANDROID)
+  Browser* browser = chrome::FindBrowserWithWebContents(web_contents());
+  if (browser) {
+    TabStripModel* tab_strip_model = browser->tab_strip_model();
+    tab_strip_model->UpdateWebContentsStateAt(
+        tab_strip_model->GetIndexOfWebContents(web_contents()),
+        TabStripModelObserver::ALL);
+  }
+#endif
+}