Add CombinedDesktopMediaList class

chrome/browser/extensions/api/desktop_capture/desktop_capture_base.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/extensions/api/desktop_capture/desktop_capture_base.h"
 
 #include <tuple>
 #include <utility>
 
 #include "base/strings/utf_string_conversions.h"
 #include "build/build_config.h"
 #include "chrome/browser/extensions/extension_tab_util.h"
+#include "chrome/browser/media/combined_desktop_media_list.h"
 #include "chrome/browser/media/desktop_media_list_ash.h"
 #include "chrome/browser/media/desktop_streams_registry.h"
 #include "chrome/browser/media/media_capture_devices_dispatcher.h"
 #include "chrome/browser/media/native_desktop_media_list.h"
+#include "chrome/browser/media/tab_desktop_media_list.h"
 #include "chrome/browser/ui/ash/ash_util.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/browser/web_contents.h"
 #include "third_party/webrtc/modules/desktop_capture/desktop_capture_options.h"
 #include "third_party/webrtc/modules/desktop_capture/screen_capturer.h"
 #include "third_party/webrtc/modules/desktop_capture/window_capturer.h"
 
 namespace extensions {
 
@@ skipping 42 matching lines @@
 bool DesktopCaptureChooseDesktopMediaFunctionBase::Execute(
     const std::vector<api::desktop_capture::DesktopCaptureSourceType>& sources,
     content::WebContents* web_contents,
     const GURL& origin,
     const base::string16 target_name) {
   // Register to be notified when the tab is closed.
   Observe(web_contents);
 
   bool show_screens = false;
   bool show_windows = false;
+  bool show_tabs = false;
 
   for (auto source_type : sources) {
     switch (source_type) {
       case api::desktop_capture::DESKTOP_CAPTURE_SOURCE_TYPE_NONE:
         error_ = kInvalidSourceNameError;
         return false;
 
       case api::desktop_capture::DESKTOP_CAPTURE_SOURCE_TYPE_SCREEN:
         show_screens = true;
         break;
 
       case api::desktop_capture::DESKTOP_CAPTURE_SOURCE_TYPE_WINDOW:
         show_windows = true;
         break;
 
       case api::desktop_capture::DESKTOP_CAPTURE_SOURCE_TYPE_TAB:
+        // TODO(gyzhou) replace next two lines with line show_tabs = true after
+        // a flag for tab capture permission are added.
         error_ = kTabCaptureNotSupportedError;
         return false;
+        break;

[qiangchen, 2016/01/26 01:01:50]

No need to break after return;

[GeorgeZ, 2016/01/27 00:17:37]

Done.

     }
   }
 
-  if (!show_screens && !show_windows) {
+  if (!show_screens && !show_windows && !show_tabs) {
     error_ = kEmptySourcesListError;
     return false;
   }
 
   const gfx::NativeWindow parent_window =
       web_contents->GetTopLevelNativeWindow();
-  scoped_ptr<DesktopMediaList> media_list;
+
+  std::vector<scoped_ptr<DesktopMediaList>> media_lists;
   if (g_picker_factory) {
-    media_list = g_picker_factory->CreateModel(
-        show_screens, show_windows);
+    scoped_ptr<DesktopMediaList> media_list =
+        g_picker_factory->CreateModel(show_screens, show_windows);
+    media_lists.push_back(std::move(media_list));
     picker_ = g_picker_factory->CreatePicker();
   } else {
+    scoped_ptr<DesktopMediaList> screen_window_list;
 #if defined(USE_ASH)
     if (chrome::IsNativeWindowInAsh(parent_window)) {
-      media_list.reset(new DesktopMediaListAsh(
+      screen_window_list.reset(new DesktopMediaListAsh(
           (show_screens ? DesktopMediaListAsh::SCREENS : 0) |
           (show_windows ? DesktopMediaListAsh::WINDOWS : 0)));
     }
 #endif
-    if (!media_list) {
+    if (!screen_window_list) {

[qiangchen, 2016/01/26 01:01:50]

Not your fault.

But I think it would be better to change the whole block as
#if defined...

#else

#endif

rather than doing a runtime check.

[GeorgeZ, 2016/01/27 00:17:37]

I am not sure whether USE_ASH case will never use NativeDesktopMediaList() since
I see the similar code in
https://code.google.com/p/chromium/codesearch#chromium/src/content/browser/re....

       webrtc::DesktopCaptureOptions options =
           webrtc::DesktopCaptureOptions::CreateDefault();
       options.set_disable_effects(false);
       scoped_ptr<webrtc::ScreenCapturer> screen_capturer(
           show_screens ? webrtc::ScreenCapturer::Create(options) : NULL);
       scoped_ptr<webrtc::WindowCapturer> window_capturer(
           show_windows ? webrtc::WindowCapturer::Create(options) : NULL);
 
-      media_list.reset(new NativeDesktopMediaList(std::move(screen_capturer),
-                                                  std::move(window_capturer)));
+      screen_window_list.reset(new NativeDesktopMediaList(
+          std::move(screen_capturer), std::move(window_capturer)));
+    }
+
+    media_lists.push_back(std::move(screen_window_list));
+    if (show_tabs) {
+      scoped_ptr<DesktopMediaList> tab_list(new TabDesktopMediaList());
+      media_lists.push_back(std::move(tab_list));
     }
 
     // DesktopMediaPicker is implemented only for Windows, OSX and
     // Aura Linux builds.
 #if defined(TOOLKIT_VIEWS) || defined(OS_MACOSX)
     picker_ = DesktopMediaPicker::Create();
 #else
     error_ = "Desktop Capture API is not yet implemented for this platform.";
     return false;
 #endif
   }
+
+  scoped_ptr<CombinedDesktopMediaList> combined_list(
+      new CombinedDesktopMediaList(media_lists));
   DesktopMediaPicker::DoneCallback callback = base::Bind(
       &DesktopCaptureChooseDesktopMediaFunctionBase::OnPickerDialogResults,
       this);
 
   picker_->Show(web_contents, parent_window, parent_window,
                 base::UTF8ToUTF16(extension()->name()), target_name,
-                std::move(media_list), callback);
+                std::move(combined_list), callback);
   origin_ = origin;
   return true;
 }
 
 void DesktopCaptureChooseDesktopMediaFunctionBase::WebContentsDestroyed() {
   Cancel();
 }
 
 void DesktopCaptureChooseDesktopMediaFunctionBase::OnPickerDialogResults(
     content::DesktopMediaID source) {
@@ skipping 70 matching lines @@
 
 void DesktopCaptureRequestsRegistry::CancelRequest(int process_id,
                                                    int request_id) {
   RequestsMap::iterator it = requests_.find(RequestId(process_id, request_id));
   if (it != requests_.end())
     it->second->Cancel();
 }
 
 
 }  // namespace extensions