Begin adding support for tab mirroring via the MediaStream audio/video capturing

chrome/browser/media/media_stream_devices_menu_model.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/media/media_stream_devices_menu_model.h"
 
 #include <utility>
 
+#include "base/logging.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/app/chrome_command_ids.h"
 #include "chrome/browser/ui/media_stream_infobar_delegate.h"
 #include "content/public/common/media_stream_request.h"
 #include "grit/generated_resources.h"
 #include "ui/base/l10n/l10n_util.h"
 
 MediaStreamDevicesMenuModel::MediaStreamDevicesMenuModel(
     const MediaStreamInfoBarDelegate* delegate)
     : ALLOW_THIS_IN_INITIALIZER_LIST(ui::SimpleMenuModel(this)),
       selected_command_id_audio_(-1),
       selected_command_id_video_(-1),
       always_allow_(false) {
-  bool audio = delegate->HasAudio() && !delegate->GetAudioDevices().empty();
-  bool video = delegate->HasVideo() && !delegate->GetVideoDevices().empty();
+  const content::MediaStreamDevices& audio_devices =
+      delegate->GetAudioDevices();
+  const content::MediaStreamDevices& video_devices =
+      delegate->GetVideoDevices();
+  const bool audio = delegate->HasAudio() && !audio_devices.empty();
+  const bool video = delegate->HasVideo() && !video_devices.empty();
   if (video) {
     // The default command ID is the first element that will be inserted.
     selected_command_id_video_ = commands_.size();
-    AddDevices(delegate->GetVideoDevices());
+    AddDevices(video_devices);
     if (audio)
       AddSeparator(ui::NORMAL_SEPARATOR);
   }
   if (audio) {
     // The default command ID is the first element that will be inserted.
     selected_command_id_audio_ = commands_.size();
-    AddDevices(delegate->GetAudioDevices());
+    AddDevices(audio_devices);
   }
 
-  // Show "always allow" option only for the secure connection.
-  if (delegate->GetSecurityOrigin().SchemeIsSecure())
-    AddAlwaysAllowOption(audio, video);
+  // Show "always allow" option when auto-accepting would be safe in the future.
+  AddAlwaysAllowOption(delegate->IsSafeToAlwaysAllowAudio(),
+                       delegate->IsSafeToAlwaysAllowVideo());
 }
 
 MediaStreamDevicesMenuModel::~MediaStreamDevicesMenuModel() {
 }
 
-bool MediaStreamDevicesMenuModel::GetSelectedDeviceId(
-    content::MediaStreamDeviceType type,
+bool MediaStreamDevicesMenuModel::GetSelectedAudioDeviceId(
     std::string* device_id) const {
-  int command_id = (type == content::MEDIA_STREAM_DEVICE_TYPE_AUDIO_CAPTURE) ?
-      selected_command_id_audio_ : selected_command_id_video_;
-  CommandMap::const_iterator it = commands_.find(command_id);
-  if (it != commands_.end())
+  CommandMap::const_iterator it = commands_.find(selected_command_id_audio_);
+  if (it != commands_.end()) {
     *device_id = it->second.device_id;
-  return (it != commands_.end());
+    return true;
+  } else {
+    return false;
+  }
+}
+
+bool MediaStreamDevicesMenuModel::GetSelectedVideoDeviceId(
+    std::string* device_id) const {
+  CommandMap::const_iterator it = commands_.find(selected_command_id_video_);
+  if (it != commands_.end()) {
+    *device_id = it->second.device_id;
+    return true;
+  } else {
+    return false;
+  }
 }
 
 bool MediaStreamDevicesMenuModel::IsCommandIdChecked(int command_id) const {
   switch (command_id) {
     case IDC_MEDIA_STREAM_DEVICE_ALWAYS_ALLOW:
       return always_allow_;
     default:
       return (selected_command_id_audio_ == command_id ||
               selected_command_id_video_ == command_id);
   }
@@ skipping 11 matching lines @@
 
 void MediaStreamDevicesMenuModel::ExecuteCommand(int command_id) {
   switch (command_id) {
     case IDC_MEDIA_STREAM_DEVICE_ALWAYS_ALLOW:
       always_allow_ = !always_allow_;
       break;
     default:
       CommandMap::iterator it = commands_.find(command_id);
       DCHECK(it != commands_.end());
 
-      if (it->second.type == content::MEDIA_STREAM_DEVICE_TYPE_AUDIO_CAPTURE)
+      if (content::IsAudioMediaType(it->second.type)) {
         selected_command_id_audio_ = command_id;
-      else
+      } else if (content::IsVideoMediaType(it->second.type)) {
         selected_command_id_video_ = command_id;
+      } else {
+        NOTREACHED();
+      }
       break;
   }
 }
 
 void MediaStreamDevicesMenuModel::AddDevices(
     const content::MediaStreamDevices& devices) {
   for (size_t i = 0; i < devices.size(); ++i) {
     int command_id = commands_.size();
     commands_.insert(std::make_pair(command_id, devices[i]));
-    int message_id = (devices[i].type ==
-        content::MEDIA_STREAM_DEVICE_TYPE_AUDIO_CAPTURE) ?
-        IDS_MEDIA_CAPTURE_AUDIO : IDS_MEDIA_CAPTURE_VIDEO;
+    int message_id;
+    if (content::IsAudioMediaType(devices[i].type)) {
+      message_id = IDS_MEDIA_CAPTURE_AUDIO;
+    } else if (content::IsVideoMediaType(devices[i].type)) {
+      message_id = IDS_MEDIA_CAPTURE_VIDEO;
+    } else {
+      NOTIMPLEMENTED();
+      continue;
+    }
     AddCheckItem(command_id,
                  l10n_util::GetStringFUTF16(message_id,
                                             UTF8ToUTF16(devices[i].name)));
   }
 }
 
 void MediaStreamDevicesMenuModel::AddAlwaysAllowOption(bool audio, bool video) {
+  if (!audio && !video) {
+    return;
+  }
+
   int command_id = IDC_MEDIA_STREAM_DEVICE_ALWAYS_ALLOW;
   int message_id = IDS_MEDIA_CAPTURE_ALWAYS_ALLOW_AUDIO_AND_VIDEO;
   if (audio && !video)
     message_id = IDS_MEDIA_CAPTURE_ALWAYS_ALLOW_AUDIO_ONLY;
   else if (!audio && video)
     message_id = IDS_MEDIA_CAPTURE_ALWAYS_ALLOW_VIDEO_ONLY;
 
   AddSeparator(ui::NORMAL_SEPARATOR);
   AddCheckItem(command_id, l10n_util::GetStringUTF16(message_id));
 }