Implement tab indicator for WebUSB.

chrome/browser/ui/tabs/tab_utils.cc

 // Copyright (c) 2012 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/ui/tabs/tab_utils.h"
 
 #include <utility>
 
 #include "base/command_line.h"
 #include "base/strings/string16.h"
 #include "build/build_config.h"
 #include "chrome/browser/media/media_capture_devices_dispatcher.h"
 #include "chrome/browser/media/media_stream_capture_indicator.h"
 #include "chrome/browser/themes/theme_properties.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
+#include "chrome/browser/usb/usb_tab_helper.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/grit/generated_resources.h"
 #include "content/public/browser/web_contents.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/theme_provider.h"
 #include "ui/gfx/animation/multi_animation.h"
 #include "ui/gfx/image/image.h"
 #include "ui/gfx/vector_icons_public.h"
 #include "ui/native_theme/common_theme.h"
 #include "ui/native_theme/native_theme.h"
@@ skipping 129 matching lines @@
   if (indicator.get()) {
     if (indicator->IsBeingMirrored(contents))
       return TabAlertState::TAB_CAPTURING;
     if (indicator->IsCapturingUserMedia(contents))
       return TabAlertState::MEDIA_RECORDING;
   }
 
   if (contents->IsBluetoothDeviceConnected())
     return TabAlertState::BLUETOOTH_CONNECTED;
 
+  UsbTabHelper* usb_tab_helper = UsbTabHelper::FromWebContents(contents);
+  if (usb_tab_helper && usb_tab_helper->IsDeviceConnected())
+    return TabAlertState::USB_CONNECTED;
+
   if (contents->IsAudioMuted())
     return TabAlertState::AUDIO_MUTING;
   if (contents->WasRecentlyAudible())
     return TabAlertState::AUDIO_PLAYING;
 
   return TabAlertState::NONE;
 }
 
 gfx::Image GetTabAlertIndicatorImage(TabAlertState alert_state,
                                      SkColor button_color) {
 #if defined(OS_MACOSX)
   ui::ResourceBundle* rb = &ui::ResourceBundle::GetSharedInstance();
   switch (alert_state) {
     case TabAlertState::AUDIO_PLAYING:
       return rb->GetNativeImageNamed(IDR_TAB_AUDIO_INDICATOR);
     case TabAlertState::AUDIO_MUTING:
       return rb->GetNativeImageNamed(IDR_TAB_AUDIO_MUTING_INDICATOR);
     case TabAlertState::MEDIA_RECORDING:
       return rb->GetNativeImageNamed(IDR_TAB_RECORDING_INDICATOR);
     case TabAlertState::TAB_CAPTURING:
       return rb->GetNativeImageNamed(IDR_TAB_CAPTURE_INDICATOR);
     case TabAlertState::BLUETOOTH_CONNECTED:
       return rb->GetNativeImageNamed(IDR_TAB_BLUETOOTH_INDICATOR);
+    case TabAlertState::USB_CONNECTED:
+      return rb->GetNativeImageNamed(IDR_TAB_USB_INDICATOR);
     case TabAlertState::NONE:
       break;
   }
 #else
   gfx::VectorIconId icon_id = gfx::VectorIconId::VECTOR_ICON_NONE;
   switch (alert_state) {
     case TabAlertState::AUDIO_PLAYING:
       icon_id = gfx::VectorIconId::TAB_AUDIO;
       break;
     case TabAlertState::AUDIO_MUTING:
       icon_id = gfx::VectorIconId::TAB_AUDIO_MUTING;
       break;
     case TabAlertState::MEDIA_RECORDING:
       icon_id = gfx::VectorIconId::TAB_MEDIA_RECORDING;
       break;
     case TabAlertState::TAB_CAPTURING:
       icon_id = gfx::VectorIconId::TAB_MEDIA_CAPTURING;
       break;
     case TabAlertState::BLUETOOTH_CONNECTED:
       icon_id = gfx::VectorIconId::TAB_BLUETOOTH_CONNECTED;
       break;
+    case TabAlertState::USB_CONNECTED:
+      icon_id = gfx::VectorIconId::TAB_USB_CONNECTED;
+      break;
     case TabAlertState::NONE:
       break;
   }
   if (icon_id != gfx::VectorIconId::VECTOR_ICON_NONE)
     return gfx::Image(gfx::CreateVectorIcon(icon_id, 16, button_color));
 #endif
   NOTREACHED();
   return gfx::Image();
 }
 
 gfx::Image GetTabAlertIndicatorAffordanceImage(TabAlertState alert_state,
                                                SkColor button_color) {
   switch (alert_state) {
     case TabAlertState::AUDIO_PLAYING:
       return GetTabAlertIndicatorImage(TabAlertState::AUDIO_MUTING,
                                        button_color);
     case TabAlertState::AUDIO_MUTING:
     case TabAlertState::NONE:
     case TabAlertState::MEDIA_RECORDING:
     case TabAlertState::TAB_CAPTURING:
     case TabAlertState::BLUETOOTH_CONNECTED:
+    case TabAlertState::USB_CONNECTED:
       return GetTabAlertIndicatorImage(alert_state, button_color);
   }
   NOTREACHED();
   return GetTabAlertIndicatorImage(alert_state, button_color);
 }
 
 scoped_ptr<gfx::Animation> CreateTabAlertIndicatorFadeAnimation(
     TabAlertState alert_state) {
   if (alert_state == TabAlertState::MEDIA_RECORDING ||
       alert_state == TabAlertState::TAB_CAPTURING) {
@@ skipping 37 matching lines @@
           IDS_TOOLTIP_TAB_ALERT_STATE_MEDIA_RECORDING));
       break;
     case TabAlertState::TAB_CAPTURING:
       result.append(
           l10n_util::GetStringUTF16(IDS_TOOLTIP_TAB_ALERT_STATE_TAB_CAPTURING));
       break;
     case TabAlertState::BLUETOOTH_CONNECTED:
       result.append(l10n_util::GetStringUTF16(
           IDS_TOOLTIP_TAB_ALERT_STATE_BLUETOOTH_CONNECTED));
       break;
+    case TabAlertState::USB_CONNECTED:
+      result.append(
+          l10n_util::GetStringUTF16(IDS_TOOLTIP_TAB_ALERT_STATE_USB_CONNECTED));
+      break;
     case TabAlertState::NONE:
       NOTREACHED();
       break;
   }
   return result;
 }
 
 bool AreExperimentalMuteControlsEnabled() {
   return base::CommandLine::ForCurrentProcess()->HasSwitch(
       switches::kEnableTabAudioMuting);
 }
 
 bool CanToggleAudioMute(content::WebContents* contents) {
   switch (GetTabAlertStateForContents(contents)) {
     case TabAlertState::NONE:
     case TabAlertState::AUDIO_PLAYING:
     case TabAlertState::AUDIO_MUTING:
       return true;
     case TabAlertState::MEDIA_RECORDING:
     case TabAlertState::TAB_CAPTURING:
     case TabAlertState::BLUETOOTH_CONNECTED:
+    case TabAlertState::USB_CONNECTED:
       return false;
   }
   NOTREACHED();
   return false;
 }
 
 TabMutedReason GetTabAudioMutedReason(content::WebContents* contents) {
   LastMuteMetadata::CreateForWebContents(contents);  // Ensures metadata exists.
   LastMuteMetadata* const metadata =
       LastMuteMetadata::FromWebContents(contents);
@@ skipping 57 matching lines @@
                      const std::vector<int>& indices) {
   for (std::vector<int>::const_iterator i = indices.begin(); i != indices.end();
        ++i) {
     if (!tab_strip.GetWebContentsAt(*i)->IsAudioMuted())
       return false;
   }
   return true;
 }
 
 }  // namespace chrome