Make MediaSession available on Desktop

content/browser/media/media_web_contents_observer.cc

 // Copyright 2014 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 "content/browser/media/media_web_contents_observer.h"
 
 #include <memory>
 
 #include "base/lazy_instance.h"
 #include "build/build_config.h"
 #include "content/browser/media/audible_metrics.h"
 #include "content/browser/media/audio_stream_monitor.h"
+#include "content/browser/media/session/browser_media_session_manager.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/common/media/media_player_delegate_messages.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/web_contents.h"
 #include "device/power_save_blocker/power_save_blocker.h"
 #include "ipc/ipc_message_macros.h"
 
 namespace content {
 
 namespace {
 
 static base::LazyInstance<AudibleMetrics>::Leaky g_audible_metrics =
     LAZY_INSTANCE_INITIALIZER;
 
 }  // anonymous namespace
 
 MediaWebContentsObserver::MediaWebContentsObserver(WebContents* web_contents)
     : WebContentsObserver(web_contents),
       session_controllers_manager_(this) {}
 
 MediaWebContentsObserver::~MediaWebContentsObserver() {}
 
+BrowserMediaSessionManager*
+MediaWebContentsObserver::GetMediaSessionManager(
+    RenderFrameHost* render_frame_host) {
+  auto it = media_session_managers_.find(render_frame_host);
+  if (it != media_session_managers_.end())
+    return it->second;
+
+  BrowserMediaSessionManager* manager =
+      new BrowserMediaSessionManager(render_frame_host);
+  media_session_managers_.set(render_frame_host, base::WrapUnique(manager));
+  return manager;
+}
+
+void MediaWebContentsObserver::SetMediaSessionManagerForTest(
+    RenderFrameHost* render_frame_host,
+    std::unique_ptr<BrowserMediaSessionManager> manager) {
+  media_session_managers_.set(render_frame_host, std::move(manager));
+}
+
 void MediaWebContentsObserver::WebContentsDestroyed() {
   g_audible_metrics.Get().UpdateAudibleWebContentsState(web_contents(), false);
 }
 
 void MediaWebContentsObserver::RenderFrameDeleted(
     RenderFrameHost* render_frame_host) {
   ClearPowerSaveBlockers(render_frame_host);
   session_controllers_manager_.RenderFrameDeleted(render_frame_host);
+  media_session_managers_.erase(render_frame_host);
 }
 
 void MediaWebContentsObserver::MaybeUpdateAudibleState() {
   if (!AudioStreamMonitor::monitoring_available())
     return;
 
   AudioStreamMonitor* audio_stream_monitor =
       static_cast<WebContentsImpl*>(web_contents())->audio_stream_monitor();
 
   if (audio_stream_monitor->WasRecentlyAudible()) {
@@ skipping 197 matching lines @@
   if (it == player_map->end())
     return;
 
   for (int delegate_id : it->second)
     removed_players->insert(MediaPlayerId(render_frame_host, delegate_id));
 
   player_map->erase(it);
 }
 
 }  // namespace content