Adding new media controls to MediaNotification

content/browser/media/session/media_session_service_impl.cc

 // Copyright 2016 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/session/media_session_service_impl.h"
 
 #include "content/browser/media/session/media_metadata_sanitizer.h"
 #include "content/browser/media/session/media_session_impl.h"
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/web_contents.h"
 
 namespace content {
 
 MediaSessionServiceImpl::MediaSessionServiceImpl(
     RenderFrameHost* render_frame_host)
-    : render_frame_host_(render_frame_host) {}
+    : render_frame_host_(render_frame_host) {
+  MediaSessionImpl* session = GetMediaSession();
+  if (session)
+    session->SetMediaSessionService(this);
+}
 
-MediaSessionServiceImpl::~MediaSessionServiceImpl() = default;
+MediaSessionServiceImpl::~MediaSessionServiceImpl() {
+  MediaSessionImpl* session = GetMediaSession();
+  if (session && session->GetMediaSessionService() == this)
+    session->SetMediaSessionService(nullptr);
+}
 
 // static
 void MediaSessionServiceImpl::Create(
     RenderFrameHost* render_frame_host,
     blink::mojom::MediaSessionServiceRequest request) {
   MediaSessionServiceImpl* impl =
       new MediaSessionServiceImpl(render_frame_host);
   impl->Bind(std::move(request));
 }
 
 void MediaSessionServiceImpl::SetClient(
     blink::mojom::MediaSessionClientPtr client) {
   client_ = std::move(client);
 }
 
 void MediaSessionServiceImpl::SetMetadata(
     const base::Optional<content::MediaMetadata>& metadata) {
   // When receiving a MediaMetadata, the browser process can't trust that it is
   // coming from a known and secure source. It must be processed accordingly.
   if (metadata.has_value() &&
       !MediaMetadataSanitizer::CheckSanity(metadata.value())) {
     render_frame_host_->GetProcess()->ShutdownForBadMessage(
         RenderProcessHost::CrashReportMode::GENERATE_CRASH_DUMP);
     return;
   }
 
-  WebContentsImpl* contents = static_cast<WebContentsImpl*>(
-      WebContentsImpl::FromRenderFrameHost(render_frame_host_));
-  if (contents)
-    MediaSessionImpl::Get(contents)->SetMetadata(metadata);
+  MediaSessionImpl* session = GetMediaSession();
+  if (session)
+    session->SetMetadata(metadata);
 }
 
 void MediaSessionServiceImpl::EnableAction(
     blink::mojom::MediaSessionAction action) {
-  // TODO(zqzhang): Plumb this signal to Java. See https://crbug.com/656563
-  NOTIMPLEMENTED();
+  MediaSessionImpl* session = GetMediaSession();
+  if (session)
+    session->OnMediaSessionEnabledAction(action);
 }
 
 void MediaSessionServiceImpl::DisableAction(
     blink::mojom::MediaSessionAction action) {
-  // TODO(zqzhang): Plumb this signal to Java. See https://crbug.com/656563
-  NOTIMPLEMENTED();
+  MediaSessionImpl* session = GetMediaSession();
+  if (session)
+    session->OnMediaSessionDisabledAction(action);
+}
+
+MediaSessionImpl* MediaSessionServiceImpl::GetMediaSession() {
+  WebContentsImpl* contents = static_cast<WebContentsImpl*>(
+      WebContentsImpl::FromRenderFrameHost(render_frame_host_));
+  if (!contents)
+    return nullptr;
+  if (render_frame_host_ != contents->GetMainFrame())
+    return nullptr;
+  return MediaSessionImpl::Get(contents);
 }
 
 void MediaSessionServiceImpl::Bind(
     blink::mojom::MediaSessionServiceRequest request) {
   binding_.reset(new mojo::Binding<blink::mojom::MediaSessionService>(
       this, std::move(request)));
 }
 
 }  // namespace content