Swiftgram/submodules/LegacyComponents/Sources/TGEmbedVideoPlayerView.m

#import "TGEmbedVideoPlayerView.h"
#import "TGEmbedPlayerState.h"

#import <AVFoundation/AVFoundation.h>

#import <LegacyComponents/TGModernGalleryVideoView.h>

#import <LegacyComponents/TGTimerTarget.h>

@interface TGEmbedVideoPlayerView ()
{
    NSString *_url;
    bool _started;
    
    AVPlayer *_player;
    TGModernGalleryVideoView *_videoView;
    UIImageView *_watermarkView;
    
    NSInteger _playbackTicks;
    bool _playingStarted;
}
@end

@implementation TGEmbedVideoPlayerView

- (instancetype)initWithWebPageAttachment:(TGWebPageMediaAttachment *)webPage thumbnailSignal:(SSignal *)thumbnailSignal alternateCachePathSignal:(SSignal *)alternateCachePathSignal
{
    self = [super initWithWebPageAttachment:webPage thumbnailSignal:thumbnailSignal alternateCachePathSignal:alternateCachePathSignal];
    if (self != nil)
    {
        _url = webPage.embedUrl;
    }
    return self;
}

- (void)dealloc
{
    [_player.currentItem removeObserver:self forKeyPath:@"status"];
    [_player.currentItem removeObserver:self forKeyPath:@"loadedTimeRanges"];
}

- (void)playVideo
{
    [_player play];
    
    TGEmbedPlayerState *state = [TGEmbedPlayerState stateWithPlaying:true duration:self.state.duration position:self.state.position downloadProgress:self.state.downloadProgress buffering:self.state.buffering];
    [self updateState:state];
}

- (void)pauseVideo:(bool)manually
{
    [super pauseVideo:manually];
    [_player pause];
    
    TGEmbedPlayerState *state = [TGEmbedPlayerState stateWithPlaying:false duration:self.state.duration position:self.state.position downloadProgress:self.state.downloadProgress buffering:self.state.buffering];
    [self updateState:state];
}

- (void)seekToPosition:(NSTimeInterval)position
{
    [_player.currentItem seekToTime:CMTimeMake((int64_t)(position * 1000.0), 1000.0)];
    
    TGEmbedPlayerState *newState = [TGEmbedPlayerState stateWithPlaying:self.state.isPlaying duration:self.state.duration position:position downloadProgress:self.state.downloadProgress buffering:self.state.buffering];
    [self updateState:newState];
}

- (void)setupWithEmbedSize:(CGSize)embedSize
{
    [super setupWithEmbedSize:embedSize];
    [self _setupCustomPlayerWithURL:[NSURL URLWithString:_url]];
}

- (TGEmbedPlayerControlsType)_controlsType
{
    return TGEmbedPlayerControlsTypeFull;
}

- (void)_onPageReady
{
    
}

- (void)_didBeginPlayback
{
    [super _didBeginPlayback];
    [self setDimmed:false animated:true shouldDelay:false];
}

- (void)observeValueForKeyPath:(NSString *)keyPath ofObject:(id)__unused object change:(NSDictionary *)__unused change context:(void *)__unused context {
    bool playing = self.state.playing;
    NSTimeInterval position = self.state.position;
    NSTimeInterval duration = self.state.duration;
    CGFloat downloadProgress = self.state.downloadProgress;
    bool buffering = self.state.buffering;
    
    if ([keyPath isEqualToString:@"status"])
    {
        if (_player.currentItem.status == AVPlayerItemStatusReadyToPlay)
        {
            if (duration < DBL_EPSILON)
                duration = CMTimeGetSeconds(_player.currentItem.asset.duration);
            
            if (!_started) {
                _started = true;
                [self setDimmed:true animated:false];
            }
        }
    }
    else if ([keyPath isEqualToString:@"loadedTimeRanges"])
    {
        NSValue *range = _player.currentItem.loadedTimeRanges.firstObject;
        CMTime time = CMTimeRangeGetEnd(range.CMTimeRangeValue);
        NSTimeInterval availableDuration = CMTimeGetSeconds(time);
        if (duration < DBL_EPSILON)
            duration = MAX(0.01, CMTimeGetSeconds(_player.currentItem.asset.duration));
        downloadProgress = MAX(0.0, MIN(1.0, availableDuration / duration));
    }
    
    TGEmbedPlayerState *newState = [TGEmbedPlayerState stateWithPlaying:playing duration:duration position:position downloadProgress:downloadProgress buffering:buffering];
    [self updateState:newState];
}

- (void)_setupCustomPlayerWithURL:(NSURL *)url
{
    AVPlayerItem *item = [AVPlayerItem playerItemWithURL:url];
    AVPlayer *player = [AVPlayer playerWithPlayerItem:item];
    _player = player;
    
    [player.currentItem addObserver:self forKeyPath:@"status" options:0 context:nil];
    [player.currentItem addObserver:self forKeyPath:@"loadedTimeRanges" options:0 context:nil];
    
    UIView *currentView = [self _webView];
    TGModernGalleryVideoView *videoView = [[TGModernGalleryVideoView alloc] initWithFrame:currentView.frame player:player];
    [currentView.superview insertSubview:videoView aboveSubview:currentView];
    
    [self _cleanWebView];
    _videoView = videoView;
    
    __weak TGEmbedVideoPlayerView *weakSelf = self;    
    [player addPeriodicTimeObserverForInterval:CMTimeMake(1, 10) queue:dispatch_get_main_queue() usingBlock:^(CMTime time)
    {
        __strong TGEmbedVideoPlayerView *strongSelf = weakSelf;
        if (strongSelf != nil)
        {
            NSTimeInterval position = CMTimeGetSeconds(time);
            if (!strongSelf->_playingStarted && position > DBL_EPSILON)
            {
                strongSelf->_playbackTicks++;
                if (strongSelf->_playbackTicks > 2)
                {
                    strongSelf->_playingStarted = true;
                    [strongSelf _didBeginPlayback];
                    
                    TGEmbedPlayerState *state = [TGEmbedPlayerState stateWithPlaying:true];
                    [strongSelf updateState:state];
                }
            }
            
            TGEmbedPlayerState *state = strongSelf.state;
            TGEmbedPlayerState *newState = [TGEmbedPlayerState stateWithPlaying:state.playing duration:state.duration position:position downloadProgress:state.downloadProgress buffering:self.state.buffering];
            [strongSelf updateState:newState];
        }
    }];
    
    [player play];
}

- (UIView *)_webView
{
    if (_videoView != nil)
        return _videoView;
    
    return [super _webView];
}


+ (bool)_supportsWebPage:(TGWebPageMediaAttachment *)webPage
{
    NSString *url = webPage.embedUrl;
    NSArray *components = [url componentsSeparatedByString:@"?"];
    if (components.count > 1)
        url = components.firstObject;
    
    return ([url hasSuffix:@".mp4"] || [url hasSuffix:@".mov"]);
}

@end