#import "TGWallpaperController.h"

#import <LegacyComponents/LegacyComponents.h>
#import "LegacyComponentsInternal.h"

#import "TGModernButton.h"
#import <LegacyComponents/TGRemoteImageView.h>

#import <LegacyComponents/TGWallpaperInfo.h>
#import <LegacyComponents/TGCustomImageWallpaperInfo.h>

//#import "TGDefaultPresentationPallete.h"

#import "TGRemoteImageView.h"

@interface TGWallpaperController () <UIScrollViewDelegate>
{
    TGWallpaperInfo *_wallpaperInfo;
    UIImage *_thumbnailImage;
    TGRemoteImageView *_imageView;
    
    CGSize _adjustingImageSize;
    CGFloat _adjustingImageScale;
    UIScrollView *_scrollView;
    
    TGModernButton *_setButton;
    
    UIView *_panelView;
    
    TGModernButton *_cancelButton;
    
    UIView *_separatorView;
    UIView *_bottomView;
    
    id<LegacyComponentsContext> _context;
}

@end

@implementation TGWallpaperController

- (instancetype)initWithContext:(id<LegacyComponentsContext>)context wallpaperInfo:(TGWallpaperInfo *)wallpaperInfo thumbnailImage:(UIImage *)thumbnailImage
{
    self = [super initWithContext:context];
    if (self != nil)
    {
        _actionHandle = [[ASHandle alloc] initWithDelegate:self releaseOnMainThread:true];
        
        _wallpaperInfo = wallpaperInfo;
        _thumbnailImage = thumbnailImage;
        
        [self setTitleText:TGLocalized(@"BackgroundPreview.Title")];
        
        self.automaticallyManageScrollViewInsets = false;
    }
    return self;
}

- (void)dealloc
{
    
}

#pragma mark -

- (void)viewWillAppear:(BOOL)animated
{
    [super viewWillAppear:animated];
    
    if (self.navigationController == nil && [[UIDevice currentDevice] userInterfaceIdiom] == UIUserInterfaceIdiomPhone)
    {
        [_context animateApplicationStatusBarAppearance:TGStatusBarAppearanceAnimationSlideUp delay:0.0 duration:0.5 completion:^
        {
            [_context setApplicationStatusBarAlpha:0.0f];
        }];
    }
}

- (void)viewWillDisappear:(BOOL)animated
{
    [super viewWillDisappear:animated];
 
    if (self.navigationController == nil && [[UIDevice currentDevice] userInterfaceIdiom] == UIUserInterfaceIdiomPhone)
    {
        [_context setApplicationStatusBarAlpha:1.0f];
        [_context animateApplicationStatusBarAppearance:TGStatusBarAppearanceAnimationSlideDown duration:iosMajorVersion() >= 7 ? 0.23 : 0.3 completion:nil];
    }
}

- (void)loadView
{
    [super loadView];
    
    self.view.clipsToBounds = true;
    self.view.backgroundColor = [UIColor blackColor];
    
    CGSize screenSize = self.view.bounds.size;
    
    _imageView = [[TGRemoteImageView alloc] initWithFrame:CGRectMake(0.0f, 0.0f, screenSize.width, screenSize.height)];
    
    bool imageLoading = false;
    
    UIImage *immediateImage = [_wallpaperInfo image];
    if (immediateImage != nil)
        [_imageView loadImage:immediateImage];
    else
    {
        _imageView.useCache = false;
        _imageView.contentHints = TGRemoteImageContentHintLoadFromDiskSynchronously;
        _imageView.fadeTransition = true;
        _imageView.fadeTransitionDuration = 0.3;
        
        imageLoading = true;
        
        ASHandle *actionHandle = _actionHandle;
        [_imageView setProgressHandler:^(TGRemoteImageView *imageView, float progress)
        {
            if (ABS(progress - 1.0f) < FLT_EPSILON && [imageView currentImage] != nil)
            {
                [actionHandle requestAction:@"imageLoaded" options:nil];
            }
        }];
        [_imageView loadImage:[_wallpaperInfo fullscreenUrl] filter:nil placeholder:_thumbnailImage];
    }
    
    if (_enableWallpaperAdjustment && immediateImage != nil)
    {
        _scrollView = [[UIScrollView alloc] initWithFrame:CGRectMake(0.0f, 0.0f, screenSize.width, screenSize.height)];
        [self.view addSubview:_scrollView];
        
        _scrollView.showsHorizontalScrollIndicator = false;
        _scrollView.showsVerticalScrollIndicator = false;
        _scrollView.delegate = self;
        
        _adjustingImageScale = immediateImage.scale;
        _adjustingImageSize = CGSizeMake(immediateImage.size.width / _adjustingImageScale, immediateImage.size.height / _adjustingImageScale);
        
        _imageView.frame = CGRectMake(0, 0, _adjustingImageSize.width, _adjustingImageSize.height);
        _imageView.contentMode = UIViewContentModeScaleToFill;
        [_scrollView addSubview:_imageView];
        
        [self _adjustScrollView];
        _scrollView.zoomScale = _scrollView.minimumZoomScale;
        
        CGSize contentSize = _scrollView.contentSize;
        CGSize viewSize = _scrollView.frame.size;
        _scrollView.contentOffset = CGPointMake(MAX(0, CGFloor((contentSize.width - viewSize.width) / 2)), MAX(0, CGFloor((contentSize.height - viewSize.height) / 2)));
    }
    else
    {
        _imageView.contentMode = UIViewContentModeScaleAspectFill;
        [self.view addSubview:_imageView];
    }
    
    _panelView = [[UIView alloc] initWithFrame:CGRectMake(0, screenSize.height - 49, screenSize.width, 49)];
    
    TGNavigationBarPallete *pallete = [[LegacyComponentsGlobals provider] navigationBarPallete];
    if (iosMajorVersion() >= 7 && [TGViewController isWidescreen])
    {
        UIToolbar *toolbar = [[UIToolbar alloc] initWithFrame:_panelView.bounds];
        toolbar.autoresizingMask = UIViewAutoresizingFlexibleWidth | UIViewAutoresizingFlexibleHeight;
        [_panelView addSubview:toolbar];
        toolbar.barTintColor = pallete.backgroundColor;
    }
    else
    {
        _panelView.backgroundColor = pallete.backgroundColor;
        _panelView.backgroundColor = [UIColor grayColor];
    }
    
    [self.view addSubview:_panelView];
    
    CGFloat separatorWidth = TGScreenPixel;
    
    _cancelButton = [[TGModernButton alloc] initWithFrame:CGRectMake(0, 0, CGFloor(_panelView.frame.size.width / 2) - separatorWidth, _panelView.frame.size.height)];
    _cancelButton.backgroundColor = [UIColor clearColor];
    //_cancelButton.highlightBackgroundColor = self.presentation.pallete.selectionColor;
    [_cancelButton addTarget:self action:@selector(cancelButtonPressed) forControlEvents:UIControlEventTouchUpInside];
    [_cancelButton setTitle:TGLocalized(@"Common.Cancel") forState:UIControlStateNormal];
    _cancelButton.titleLabel.font = TGSystemFontOfSize(17);
    [_cancelButton setTitleColor:pallete.titleColor];
    [_cancelButton setTitleShadowColor:[UIColor clearColor] forState:UIControlStateNormal];
    [_panelView addSubview:_cancelButton];
    
    _setButton = [[TGModernButton alloc] initWithFrame:CGRectMake(_cancelButton.frame.origin.x + _cancelButton.frame.size.width + separatorWidth, 0, CGFloor(_panelView.frame.size.width / 2), _panelView.frame.size.height)];
    _setButton.backgroundColor = [UIColor clearColor];
    //_setButton.highlightBackgroundColor = self.presentation.pallete.selectionColor;
    _setButton.enabled = !imageLoading;
    [_setButton addTarget:self action:@selector(doneButtonPressed) forControlEvents:UIControlEventTouchUpInside];
    [_setButton setTitle:TGLocalized(@"Wallpaper.Set") forState:UIControlStateNormal];
    _setButton.titleLabel.font = TGSystemFontOfSize(17);
    [_setButton setTitleColor:pallete.titleColor];
    //[_setButton setTitleColor:[self.presentation.pallete.textColor colorWithAlphaComponent:0.4f] forState:UIControlStateDisabled];
    [_setButton setTitleShadowColor:[UIColor clearColor] forState:UIControlStateNormal];
    [_panelView addSubview:_setButton];
    
    _separatorView = [[UIView alloc] initWithFrame:CGRectMake(CGFloor(_panelView.frame.size.width / 2) - separatorWidth, 0, separatorWidth, 49.0f)];
    _separatorView.backgroundColor = pallete.separatorColor;
    [_panelView addSubview:_separatorView];
    
    _bottomView =  [[UIView alloc] initWithFrame:CGRectMake(0.0f, 49.0f, _panelView.frame.size.width, separatorWidth)];
    _bottomView.autoresizingMask = UIViewAutoresizingFlexibleWidth;
    _bottomView.backgroundColor = pallete.separatorColor;
    //[_panelView addSubview:_bottomView];
}

- (void)layoutControllerForSize:(CGSize)size duration:(NSTimeInterval)duration {
    [super layoutControllerForSize:size duration:duration];
    
    CGSize screenSize = self.view.bounds.size;
    
    _imageView.frame = CGRectMake(0.0f, 0.0f, screenSize.width, screenSize.height);
    
    UIImage *immediateImage = [_wallpaperInfo image];
    if (_enableWallpaperAdjustment && immediateImage != nil)
    {
        _scrollView.frame = CGRectMake(0.0f, 0.0f, screenSize.width, screenSize.height);
        
        _scrollView.showsHorizontalScrollIndicator = false;
        _scrollView.showsVerticalScrollIndicator = false;
        _scrollView.delegate = self;
        
        _adjustingImageScale = immediateImage.scale;
        _adjustingImageSize = CGSizeMake(immediateImage.size.width / _adjustingImageScale, immediateImage.size.height / _adjustingImageScale);
        
        _imageView.frame = CGRectMake(0, 0, _adjustingImageSize.width, _adjustingImageSize.height);
        _imageView.contentMode = UIViewContentModeScaleToFill;
        
        [self _adjustScrollView];
        _scrollView.zoomScale = _scrollView.minimumZoomScale;
        
        CGSize contentSize = _scrollView.contentSize;
        CGSize viewSize = _scrollView.frame.size;
        _scrollView.contentOffset = CGPointMake(MAX(0, CGFloor((contentSize.width - viewSize.width) / 2)), MAX(0, CGFloor((contentSize.height - viewSize.height) / 2)));
    }
    else
    {
        _imageView.contentMode = UIViewContentModeScaleAspectFill;
    }
    
    _panelView.frame = CGRectMake(0, screenSize.height - 49.0f - self.controllerSafeAreaInset.bottom, screenSize.width, 49.0f + self.controllerSafeAreaInset.bottom);
    
    CGFloat separatorWidth = TGScreenPixel;
    
    _cancelButton.frame = CGRectMake(0, 0, CGFloor(_panelView.frame.size.width / 2) - separatorWidth, 49.0f);
    
    _setButton.frame = CGRectMake(_cancelButton.frame.origin.x + _cancelButton.frame.size.width + separatorWidth, 0, CGFloor(_panelView.frame.size.width / 2), 49.0f);
    
    _separatorView.frame = CGRectMake(CGFloor(_panelView.frame.size.width / 2) - separatorWidth, 0, separatorWidth, 49.0f + self.controllerSafeAreaInset.bottom);
}

- (void)controllerInsetUpdated:(UIEdgeInsets)previousInset
{
    [super controllerInsetUpdated:previousInset];
    
    _panelView.frame = CGRectMake(0, self.view.bounds.size.height - 49.0f - self.controllerSafeAreaInset.bottom, self.view.bounds.size.width, 49.0f + self.controllerSafeAreaInset.bottom);
    
    CGFloat separatorWidth = TGScreenPixel;
    _separatorView.frame = CGRectMake(CGFloor(_panelView.frame.size.width / 2) - separatorWidth, 0, separatorWidth, 49.0f + self.controllerSafeAreaInset.bottom);
}

- (BOOL)shouldAutorotate
{
    return [[UIDevice currentDevice] userInterfaceIdiom] == UIUserInterfaceIdiomPad;
}

- (UIInterfaceOrientationMask)supportedInterfaceOrientations
{
    return UIInterfaceOrientationMaskPortrait;
}

#pragma mark -

- (void)scrollViewDidZoom:(UIScrollView *)__unused scrollView
{
    [self _adjustScrollView];
}

- (void)scrollViewDidEndZooming:(UIScrollView *)__unused scrollView withView:(UIView *)__unused view atScale:(CGFloat)__unused scale
{
    [self _adjustScrollView];
}

- (UIView *)viewForZoomingInScrollView:(UIScrollView *)__unused scrollView
{
    return _imageView;
}

- (void)_adjustScrollView
{
    CGSize imageSize = _adjustingImageSize;
    CGFloat imageScale = _adjustingImageScale;
    imageSize.width /= imageScale;
    imageSize.height /= imageScale;
    
    CGFloat scaleWidth = _scrollView.frame.size.width / imageSize.width;
    CGFloat scaleHeight = _scrollView.frame.size.height / imageSize.height;
    CGFloat minScale = MAX(scaleWidth, scaleHeight);
    
    if (_scrollView.minimumZoomScale != minScale)
        _scrollView.minimumZoomScale = minScale;
    if (_scrollView.maximumZoomScale != minScale * 3.0f)
        _scrollView.maximumZoomScale = minScale * 3.0f;
    
    CGSize boundsSize = _scrollView.bounds.size;
    CGRect contentsFrame = _imageView.frame;
    
    if (boundsSize.width > contentsFrame.size.width)
        contentsFrame.origin.x = (boundsSize.width - contentsFrame.size.width) / 2.0f;
    else
        contentsFrame.origin.x = 0;
    
    if (boundsSize.height > contentsFrame.size.height)
        contentsFrame.origin.y = (boundsSize.height - contentsFrame.size.height) / 2.0f;
    else
        contentsFrame.origin.y = 0;
    
    _imageView.frame = contentsFrame;
}

#pragma mark

- (void)cancelButtonPressed
{
    if (_customDismiss) {
        _customDismiss();
    } else {
        if (self.navigationController != nil)
            [self.navigationController popViewControllerAnimated:true];
        else
            [self.presentingViewController dismissViewControllerAnimated:true completion:nil];
    }
}

- (void)doneButtonPressed
{
    UIImage *currentImage = [_imageView currentImage];
    if (currentImage != nil)
    {
        TGWallpaperInfo *selectedWallpaperInfo = _wallpaperInfo;
        
        if (_enableWallpaperAdjustment)
        {
            CGSize screenSize = self.view.bounds.size;
            CGFloat screenScale = [UIScreen mainScreen].scale;
            screenSize.width *= screenScale;
            screenSize.height *= screenScale;
            
            CGFloat screenSide = MAX(screenSize.width, screenSize.height);
            
            CGFloat scale = 1.0f / _scrollView.zoomScale;
            
            CGRect visibleRect;
            visibleRect.origin.x = _scrollView.contentOffset.x * scale;
            visibleRect.origin.y = _scrollView.contentOffset.y * scale;
            visibleRect.size.width = _scrollView.bounds.size.width * scale;
            visibleRect.size.height = _scrollView.bounds.size.height * scale;
            
            UIImage *croppedImage = TGFixOrientationAndCrop(currentImage, visibleRect, TGFitSize(visibleRect.size, CGSizeMake(screenSide, screenSide)));
            if (croppedImage != nil)
                selectedWallpaperInfo = [[TGCustomImageWallpaperInfo alloc] initWithImage:croppedImage];
        }
        
        id<TGWallpaperControllerDelegate> delegate = _delegate;
        if ([delegate respondsToSelector:@selector(wallpaperController:didSelectWallpaperWithInfo:)])
            [delegate wallpaperController:self didSelectWallpaperWithInfo:selectedWallpaperInfo];
    }
}

- (void)actionStageActionRequested:(NSString *)action options:(id)__unused options
{
    if ([action isEqualToString:@"imageLoaded"])
    {
        _setButton.enabled = true;
    }
}

@end