// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "components/system_media_controls/mac/now_playing_info_center_delegate_cocoa.h"

#import <MediaPlayer/MediaPlayer.h>

@interface NowPlayingInfoCenterDelegateCocoa ()

// Initialize the |nowPlayingInfo_| dictionary with values.
- (void)initializeNowPlayingInfoValues;

// Give MPNowPlayingInfoCenter the updated nowPlayingInfo_ dictionary.
- (void)updateNowPlayingInfo;

@end

@implementation NowPlayingInfoCenterDelegateCocoa {
  NSMutableDictionary* __strong _nowPlayingInfo;
}

- (instancetype)init {
  if (self = [super init]) {
    _nowPlayingInfo = [[NSMutableDictionary alloc] init];
    [self resetNowPlayingInfo];
    [self updateNowPlayingInfo];
  }
  return self;
}

- (void)resetNowPlayingInfo {
  [_nowPlayingInfo removeAllObjects];
  [self initializeNowPlayingInfoValues];
  [MPNowPlayingInfoCenter defaultCenter].nowPlayingInfo = nil;
}

- (void)setPlaybackState:(MPNowPlayingPlaybackState)state {
  [MPNowPlayingInfoCenter defaultCenter].playbackState = state;
}

- (void)setTitle:(NSString*)title {
  [_nowPlayingInfo setObject:title forKey:MPMediaItemPropertyTitle];
}

- (void)setArtist:(NSString*)artist {
  [_nowPlayingInfo setObject:artist forKey:MPMediaItemPropertyArtist];
}

- (void)setAlbum:(NSString*)album {
  [_nowPlayingInfo setObject:album forKey:MPMediaItemPropertyAlbumTitle];
}

- (void)setPlaybackRate:(NSNumber*)rate {
  [_nowPlayingInfo setObject:rate forKey:MPNowPlayingInfoPropertyPlaybackRate];
}

- (void)setCurrentPlaybackDate:(NSDate*)date {
  [_nowPlayingInfo setObject:date
                      forKey:MPNowPlayingInfoPropertyCurrentPlaybackDate];
}

- (void)setElapsedPlaybackTime:(NSNumber*)time {
  [_nowPlayingInfo setObject:time
                      forKey:MPNowPlayingInfoPropertyElapsedPlaybackTime];
}

- (void)setDuration:(NSNumber*)duration {
  [_nowPlayingInfo setObject:duration
                      forKey:MPMediaItemPropertyPlaybackDuration];
}

- (void)setThumbnail:(NSImage*)image {
  MPMediaItemArtwork* artwork = [[MPMediaItemArtwork alloc]
      initWithBoundsSize:image.size
          requestHandler:^NSImage* _Nonnull(CGSize aSize) {
            return image;
          }];
  [_nowPlayingInfo setObject:artwork forKey:MPMediaItemPropertyArtwork];
}

- (void)clearPosition {
  [_nowPlayingInfo
      removeObjectForKey:MPNowPlayingInfoPropertyCurrentPlaybackDate];
  [_nowPlayingInfo
      removeObjectForKey:MPNowPlayingInfoPropertyElapsedPlaybackTime];
  [_nowPlayingInfo removeObjectForKey:MPNowPlayingInfoPropertyPlaybackRate];
  [_nowPlayingInfo removeObjectForKey:MPMediaItemPropertyPlaybackDuration];

  // ClearMetadata deliberately publishes nil. Do not replace it with the
  // default-filled internal dictionary when a null position follows during
  // media session teardown.
  if ([MPNowPlayingInfoCenter defaultCenter].nowPlayingInfo) {
    [self updateNowPlayingInfo];
  }
}

- (void)clearMetadata {
  // Reset our internal dictionary to have default values.
  [self initializeNowPlayingInfoValues];

  // In some cases, setting defaultCenter to a dictionary of default values
  // causes the menu bar's media icon to persist with completely blank metadata
  // after a track ends, or on navigation away from the page that was playing
  // media. See crbug.com/359628047 for more information.
  // To avoid this, set defaultCenter to nil as recommended here:
  // https://developer.apple.com/documentation/mediaplayer/mpnowplayinginfocenter/1615903-nowplayinginfo
  [MPNowPlayingInfoCenter defaultCenter].nowPlayingInfo = nil;
}

- (void)initializeNowPlayingInfoValues {
  [_nowPlayingInfo setObject:@0
                      forKey:MPNowPlayingInfoPropertyElapsedPlaybackTime];
  [_nowPlayingInfo setObject:@0 forKey:MPNowPlayingInfoPropertyPlaybackRate];
  [_nowPlayingInfo setObject:@0 forKey:MPMediaItemPropertyPlaybackDuration];
  [_nowPlayingInfo setObject:@"" forKey:MPMediaItemPropertyTitle];
  [_nowPlayingInfo setObject:@"" forKey:MPMediaItemPropertyArtist];
  [_nowPlayingInfo setObject:@"" forKey:MPMediaItemPropertyAlbumTitle];
  [_nowPlayingInfo removeObjectForKey:MPMediaItemPropertyArtwork];
}

- (void)updateNowPlayingInfo {
  [MPNowPlayingInfoCenter defaultCenter].nowPlayingInfo = _nowPlayingInfo;
}

@end
