chrome/android/java/src/org/chromium/chrome/browser/media/router/caf/CafNotificationController.java

// Copyright 2018 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.

package org.chromium.chrome.browser.media.router.caf;

import android.content.Intent;

import com.google.android.gms.cast.CastDevice;
import com.google.android.gms.cast.MediaStatus;
import com.google.android.gms.cast.framework.media.RemoteMediaClient;

import org.chromium.chrome.R;
import org.chromium.chrome.browser.media.ui.MediaNotificationInfo;
import org.chromium.chrome.browser.media.ui.MediaNotificationListener;
import org.chromium.chrome.browser.media.ui.MediaNotificationManager;
import org.chromium.chrome.browser.metrics.MediaNotificationUma;
import org.chromium.chrome.browser.tab.Tab;
import org.chromium.content_public.common.MediaMetadata;

/** Controller for updating media notification for Casting and MediaFling. */
public class CafNotificationController implements MediaNotificationListener {
    private MediaNotificationInfo.Builder mNotificationBuilder;
    private final BaseSessionController mSessionController;

    public CafNotificationController(BaseSessionController sessionController) {
        mSessionController = sessionController;
    }

    /** Called when session started. */
    public void onSessionStarted() {
        Intent contentIntent =
                Tab.createBringTabToFrontIntent(mSessionController.getRouteCreationInfo().tabId);
        if (contentIntent != null) {
            contentIntent.putExtra(MediaNotificationUma.INTENT_EXTRA_NAME,
                    MediaNotificationUma.Source.PRESENTATION);
        }
        mNotificationBuilder =
                new MediaNotificationInfo.Builder()
                        .setPaused(false)
                        .setOrigin(mSessionController.getRouteCreationInfo().origin)
                        // TODO(zqzhang): the same session might have more than one tab id. Should
                        // we track the last foreground alive tab and update the notification with
                        // it?
                        .setTabId(mSessionController.getRouteCreationInfo().tabId)
                        .setPrivate(mSessionController.getRouteCreationInfo().isIncognito)
                        .setActions(MediaNotificationInfo.ACTION_STOP)
                        .setContentIntent(contentIntent)
                        .setNotificationSmallIcon(R.drawable.ic_notification_media_route)
                        .setDefaultNotificationLargeIcon(R.drawable.cast_playing_square)
                        .setId(R.id.presentation_notification)
                        .setListener(this);

        updateNotificationMetadata();
        MediaNotificationManager.show(mNotificationBuilder.build());
    }

    /** Called when session ended. */
    public void onSessionEnded() {
        MediaNotificationManager.clear(R.id.presentation_notification);
        mNotificationBuilder = null;
    }

    /** Called when media status updated. */
    public void onStatusUpdated() {
        if (mNotificationBuilder == null) return;
        if (!mSessionController.isConnected()) return;

        MediaStatus mediaStatus = mSessionController.getRemoteMediaClient().getMediaStatus();
        if (mediaStatus == null) return;

        int playerState = mediaStatus.getPlayerState();
        if (playerState == MediaStatus.PLAYER_STATE_PAUSED
                || playerState == MediaStatus.PLAYER_STATE_PLAYING) {
            mNotificationBuilder.setPaused(playerState != MediaStatus.PLAYER_STATE_PLAYING);
            mNotificationBuilder.setActions(
                    MediaNotificationInfo.ACTION_STOP | MediaNotificationInfo.ACTION_PLAY_PAUSE);
        } else {
            mNotificationBuilder.setActions(MediaNotificationInfo.ACTION_STOP);
        }
        MediaNotificationManager.show(mNotificationBuilder.build());
    }

    /** Called when media metadata updated. */
    public void onMetadataUpdated() {
        if (mNotificationBuilder == null) return;
        updateNotificationMetadata();
        MediaNotificationManager.show(mNotificationBuilder.build());
    }

    private void updateNotificationMetadata() {
        MediaMetadata notificationMetadata = new MediaMetadata("", "", "");
        mNotificationBuilder.setMetadata(notificationMetadata);

        if (!mSessionController.isConnected()) return;

        CastDevice castDevice = mSessionController.getSession().getCastDevice();
        if (castDevice != null) notificationMetadata.setTitle(castDevice.getFriendlyName());

        RemoteMediaClient remoteMediaClient = mSessionController.getRemoteMediaClient();

        com.google.android.gms.cast.MediaInfo info = remoteMediaClient.getMediaInfo();
        if (info == null) return;

        com.google.android.gms.cast.MediaMetadata metadata = info.getMetadata();
        if (metadata == null) return;

        String title = metadata.getString(com.google.android.gms.cast.MediaMetadata.KEY_TITLE);
        if (title != null) notificationMetadata.setTitle(title);

        String artist = metadata.getString(com.google.android.gms.cast.MediaMetadata.KEY_ARTIST);
        if (artist == null) {
            artist = metadata.getString(com.google.android.gms.cast.MediaMetadata.KEY_ALBUM_ARTIST);
        }
        if (artist != null) notificationMetadata.setArtist(artist);

        String album =
                metadata.getString(com.google.android.gms.cast.MediaMetadata.KEY_ALBUM_TITLE);
        if (album != null) notificationMetadata.setAlbum(album);
    }

    /////////////////////////////////////////////////////////////////////////////////////////////
    // MediaNotificationListener implementation.

    @Override
    public void onPlay(int actionSource) {
        if (!mSessionController.isConnected()) return;

        mSessionController.getRemoteMediaClient().play();
    }

    @Override
    public void onPause(int actionSource) {
        if (!mSessionController.isConnected()) return;

        mSessionController.getRemoteMediaClient().pause();
    }

    @Override
    public void onStop(int actionSource) {
        if (!mSessionController.isConnected()) return;

        mSessionController.endSession();
    }

    @Override
    public void onMediaSessionAction(int action) {}
}