#import "TGMediaPickerGalleryGifItemView.h"
#import "TGMediaPickerGalleryGifItem.h"

#import "TGPhotoEditorUtils.h"
#import "TGStringUtils.h"
#import "TGFont.h"

#import "TGModernAnimatedImagePlayer.h"

#import "TGMessageImageViewOverlayView.h"

#import "TGMediaAssetImageSignals.h"

@interface TGMediaPickerGalleryGifItemView ()
{
    UIView *_containerView;
    
    CGSize _imageSize;
    
    UITapGestureRecognizer *_tapGestureRecognizer;
    
    SMetaDisposable *_gifDataDisposable;
    TGModernAnimatedImagePlayer *_player;
    
    TGMessageImageViewOverlayView *_progressView;
    bool _progressVisible;
    void (^_currentAvailabilityObserver)(bool);
    
    UILabel *_fileInfoLabel;
    SMetaDisposable *_attributesDisposable;
    
    bool _imageAvailable;
}
@end

@implementation TGMediaPickerGalleryGifItemView

- (instancetype)initWithFrame:(CGRect)frame
{
    self = [super initWithFrame:frame];
    if (self != nil)
    {
        _containerView = [[UIView alloc] initWithFrame:self.bounds];
        _containerView.clipsToBounds = true;
        [self addSubview:_containerView];
        
        _imageView = [[TGModernGalleryImageItemImageView alloc] init];
        [_containerView addSubview:_imageView];
        
        _fileInfoLabel = [[UILabel alloc] initWithFrame:CGRectMake(0, 0, 200, 20)];
        _fileInfoLabel.autoresizingMask = UIViewAutoresizingFlexibleWidth;
        _fileInfoLabel.backgroundColor = [UIColor clearColor];
        _fileInfoLabel.font = TGSystemFontOfSize(13.0f);
        _fileInfoLabel.textAlignment = NSTextAlignmentCenter;
        _fileInfoLabel.textColor = [UIColor whiteColor];
        
        _tapGestureRecognizer = [[UITapGestureRecognizer alloc] initWithTarget:self action:@selector(singleTap)];
        [_containerView addGestureRecognizer:_tapGestureRecognizer];
        
        _gifDataDisposable = [[SMetaDisposable alloc] init];
    }
    return self;
}

- (void)dealloc
{
    [_gifDataDisposable dispose];
    [_attributesDisposable dispose];
}

- (void)setFrame:(CGRect)frame
{
    [super setFrame:frame];
    
    if (_containerView == nil)
        return;
    
    _containerView.frame = self.bounds;
    
    [self _layoutPlayerView];
}

- (void)prepareForRecycle
{
    [_imageView reset];
    [_player stop];
    _player = nil;
    
    _imageAvailable = false;
    [self setProgressVisible:false value:0.0f animated:false];
}

- (void)setItem:(TGMediaPickerGalleryGifItem *)item synchronously:(bool)synchronously
{
    [super setItem:item synchronously:synchronously];
    
    _imageSize = item.asset.dimensions;
    [self _layoutPlayerView];
    
    if (item.asset == nil)
    {
        [self.imageView reset];
    }
    else
    {
        SSignal *imageSignal = [TGMediaAssetImageSignals imageForAsset:item.asset imageType:(item.immediateThumbnailImage != nil) ? TGMediaAssetImageTypeScreen : TGMediaAssetImageTypeFastScreen size:CGSizeMake(1280, 1280)];
        
        if (item.immediateThumbnailImage != nil)
            imageSignal = [[SSignal single:item.immediateThumbnailImage] then:imageSignal];

        [self.imageView setSignal:imageSignal];
        
        __weak TGMediaPickerGalleryGifItemView *weakSelf = self;
        [_gifDataDisposable setDisposable:[[TGMediaAssetImageSignals imageDataForAsset:item.asset] startWithNext:^(id next)
        {
            __strong TGMediaPickerGalleryGifItemView *strongSelf = weakSelf;
            if (strongSelf == nil)
                return;
            
            if ([next isKindOfClass:[NSNumber class]])
            {
                float value = [next floatValue];
                [strongSelf setProgressVisible:value < 1.0f - FLT_EPSILON value:value animated:true];
            }
            else if ([next isKindOfClass:[TGMediaAssetImageData class]])
            {
                [strongSelf setProgressVisible:false value:1.0f animated:true];
                
                TGMediaAssetImageData *data = (TGMediaAssetImageData *)next;
                [strongSelf _playWithData:data.imageData];
                
                strongSelf->_imageAvailable = true;
                if (strongSelf->_currentAvailabilityObserver != nil)
                    strongSelf->_currentAvailabilityObserver(true);
            }
        }]];
    }
    
    if (!item.asFile)
        return;
    
    _fileInfoLabel.text = nil;
    
    if (_attributesDisposable == nil)
        _attributesDisposable = [[SMetaDisposable alloc] init];
    
    __weak TGMediaPickerGalleryGifItemView *weakSelf = self;
    [_attributesDisposable setDisposable:[[[TGMediaAssetImageSignals fileAttributesForAsset:item.asset] deliverOn:[SQueue mainQueue]] startWithNext:^(TGMediaAssetImageFileAttributes *next)
    {
        __strong TGMediaPickerGalleryGifItemView *strongSelf = weakSelf;
        if (strongSelf == nil)
            return;
        
        NSString *extension = @"GIF";
        NSString *fileSize = [TGStringUtils stringForFileSize:next.fileSize precision:2];
        NSString *dimensions = [NSString stringWithFormat:@"%dx%d", (int)next.dimensions.width, (int)next.dimensions.height];
        
        strongSelf->_fileInfoLabel.text = [NSString stringWithFormat:@"%@ • %@ • %@", extension, fileSize, dimensions];
    }]];
}

- (void)_playWithData:(NSData *)data
{
    [self.imageView setSignal:nil];

    _player = [[TGModernAnimatedImagePlayer alloc] initWithSize:_imageSize data:data];
    __weak TGMediaPickerGalleryGifItemView *weakSelf = self;
    _player.frameReady = ^(UIImage *image)
    {
        __strong TGMediaPickerGalleryGifItemView *strongSelf = weakSelf;
        if (strongSelf != nil)
            [strongSelf->_imageView loadUri:@"embedded-image://" withOptions:@{TGImageViewOptionEmbeddedImage: image}];
    };
    [_player play];
}

- (void)singleTap
{
    if ([self.item conformsToProtocol:@protocol(TGModernGallerySelectableItem)])
    {
        TGMediaSelectionContext *selectionContext = ((id<TGModernGallerySelectableItem>)self.item).selectionContext;
        id<TGMediaSelectableItem> item = ((id<TGModernGallerySelectableItem>)self.item).selectableMediaItem;
        
        [selectionContext toggleItemSelection:item animated:true sender:nil];
    }
    else
    {
        id<TGModernGalleryItemViewDelegate> delegate = self.delegate;
        if ([delegate respondsToSelector:@selector(itemViewDidRequestInterfaceShowHide:)])
            [delegate itemViewDidRequestInterfaceShowHide:self];
    }
}

- (void)setProgressVisible:(bool)progressVisible value:(CGFloat)value animated:(bool)animated
{
    _progressVisible = progressVisible;
    
    if (progressVisible && _progressView == nil)
    {
        _progressView = [[TGMessageImageViewOverlayView alloc] initWithFrame:CGRectMake(0.0f, 0.0f, 50.0f, 50.0f)];
        _progressView.userInteractionEnabled = false;
        
        _progressView.frame = (CGRect){{CGFloor((self.frame.size.width - _progressView.frame.size.width) / 2.0f), CGFloor((self.frame.size.height - _progressView.frame.size.height) / 2.0f)}, _progressView.frame.size};
    }
    
    if (progressVisible)
    {
        if (_progressView.superview == nil)
            [_containerView addSubview:_progressView];
        
        _progressView.alpha = 1.0f;
    }
    else if (_progressView.superview != nil)
    {
        if (animated)
        {
            [UIView animateWithDuration:0.3 delay:0.0 options:UIViewAnimationOptionBeginFromCurrentState animations:^
             {
                 _progressView.alpha = 0.0f;
             } completion:^(BOOL finished)
             {
                 if (finished)
                     [_progressView removeFromSuperview];
             }];
        }
        else
            [_progressView removeFromSuperview];
    }
    
    [_progressView setProgress:value cancelEnabled:false animated:animated];
}

- (SSignal *)contentAvailabilityStateSignal
{
    __weak TGMediaPickerGalleryGifItemView *weakSelf = self;
    return [[SSignal alloc] initWithGenerator:^id<SDisposable>(SSubscriber *subscriber)
    {
        __strong TGMediaPickerGalleryGifItemView *strongSelf = weakSelf;
        if (strongSelf != nil)
        {
            [subscriber putNext:@(strongSelf->_imageAvailable)];
            strongSelf->_currentAvailabilityObserver = ^(bool available)
            {
                [subscriber putNext:@(available)];
            };
        }
        
        return nil;
    }];
}

- (UIView *)footerView
{
    if (((TGMediaPickerGalleryItem *)self.item).asFile)
        return _fileInfoLabel;
    
    return nil;
}

- (void)_layoutPlayerView
{
    if (CGSizeEqualToSize(_imageSize, CGSizeZero))
        return;
    
    CGSize fittedSize = TGScaleToSize(_imageSize, self.frame.size);
    
    _imageView.frame = CGRectMake((self.frame.size.width - fittedSize.width) / 2.0f, (self.frame.size.height - fittedSize.height) / 2.0f, fittedSize.width, fittedSize.height);
}

- (UIView *)transitionView
{
    return _containerView;
}

- (CGRect)transitionViewContentRect
{
    return [_imageView convertRect:_imageView.bounds toView:[self transitionView]];
}

@end