#import "TGMediaEditingContext.h"

#import "LegacyComponentsInternal.h"
#import "TGStringUtils.h"

#import <LegacyComponents/UIImage+TG.h>
#import "TGPhotoEditorUtils.h"
#import "PGPhotoEditorValues.h"
#import "TGVideoEditAdjustments.h"

#import "TGModernCache.h"
#import "TGMemoryImageCache.h"
#import "TGMediaAsset.h"

#import "TGPaintingData.h"

@interface TGMediaImageUpdate : NSObject

@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly, strong) id representation;

+ (instancetype)imageUpdateWithItem:(id<TGMediaEditableItem>)item representation:(id)representation;

@end


@interface TGMediaAdjustmentsUpdate : NSObject

@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly, strong) id<TGMediaEditAdjustments> adjustments;

+ (instancetype)adjustmentsUpdateWithItem:(id<TGMediaEditableItem>)item adjustments:(id<TGMediaEditAdjustments>)adjustments;

@end


@interface TGMediaCaptionUpdate : NSObject

@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly, strong) NSAttributedString *caption;

+ (instancetype)captionUpdateWithItem:(id<TGMediaEditableItem>)item caption:(NSAttributedString *)caption;

@end


@interface TGMediaTimerUpdate : NSObject

@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly, strong) NSNumber *timer;

+ (instancetype)timerUpdateWithItem:(id<TGMediaEditableItem>)item timer:(NSNumber *)timer;
+ (instancetype)timerUpdate:(NSNumber *)timer;

@end

@interface TGMediaSpoilerUpdate : NSObject

@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly) bool spoiler;

+ (instancetype)spoilerUpdateWithItem:(id<TGMediaEditableItem>)item spoiler:(bool)spoiler;
+ (instancetype)spoilerUpdate:(bool)spoiler;

@end

@interface TGMediaPriceUpdate : NSObject

@property (nonatomic, readonly, strong) id<TGMediaEditableItem> item;
@property (nonatomic, readonly, strong) NSNumber *price;

+ (instancetype)priceUpdateWithItem:(id<TGMediaEditableItem>)item price:(NSNumber *)price;
+ (instancetype)priceUpdate:(NSNumber *)timer;

@end


@interface TGModernCache (Private)

- (void)cleanup;

@end

@interface TGMediaEditingContext ()
{
    NSString *_contextId;
    
    NSMutableDictionary *_captions;
    NSMutableDictionary *_adjustments;
    NSMutableDictionary *_timers;
    NSNumber *_timer;
    
    NSMutableDictionary *_spoilers;
    NSMutableDictionary *_prices;
    
    SQueue *_queue;
    
    NSMutableDictionary *_temporaryRepCache;
    
    TGMemoryImageCache *_imageCache;
    TGMemoryImageCache *_thumbnailImageCache;
    
    TGMemoryImageCache *_paintingImageCache;
    TGMemoryImageCache *_stillPaintingImageCache;
    
    TGMemoryImageCache *_originalImageCache;
    TGMemoryImageCache *_originalThumbnailImageCache;
    
    TGModernCache *_diskCache;
    NSURL *_fullSizeResultsUrl;
    NSURL *_paintingDatasUrl;
    NSURL *_paintingImagesUrl;
    NSURL *_stillPaintingImagesUrl;
    NSURL *_videoPaintingImagesUrl;
    
    NSMutableArray *_storeVideoPaintingImages;
    
    NSMutableDictionary *_faces;
    
    SPipe *_representationPipe;
    SPipe *_thumbnailImagePipe;
    SPipe *_adjustmentsPipe;
    SPipe *_captionPipe;
    SPipe *_timerPipe;
    SPipe *_spoilerPipe;
    SPipe *_pricePipe;
    SPipe *_fullSizePipe;
    SPipe *_cropPipe;
    SPipe *_captionAbovePipe;
    
    NSAttributedString *_forcedCaption;
    
    bool _captionAbove;
}
@end

@implementation TGMediaEditingContext

- (instancetype)init
{
    self = [super init];
    if (self != nil)
    {
        _contextId = [NSString stringWithFormat:@"%ld", lrand48()];
        _queue = [[SQueue alloc] init];

        _captions = [[NSMutableDictionary alloc] init];
        _adjustments = [[NSMutableDictionary alloc] init];
        _timers = [[NSMutableDictionary alloc] init];
        _spoilers = [[NSMutableDictionary alloc] init];
        _prices = [[NSMutableDictionary alloc] init];
        
        _imageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] imageSoftMemoryLimit]
                                                          hardMemoryLimit:[[self class] imageHardMemoryLimit]];
        _thumbnailImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] thumbnailImageSoftMemoryLimit]
                                                                   hardMemoryLimit:[[self class] thumbnailImageHardMemoryLimit]];
        
        _paintingImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] imageSoftMemoryLimit]
                                                                  hardMemoryLimit:[[self class] imageHardMemoryLimit]];
        
        _stillPaintingImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] imageSoftMemoryLimit]
                                                                  hardMemoryLimit:[[self class] imageHardMemoryLimit]];
        
        _originalImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] originalImageSoftMemoryLimit]
                                                                  hardMemoryLimit:[[self class] originalImageHardMemoryLimit]];
        _originalThumbnailImageCache = [[TGMemoryImageCache alloc] initWithSoftMemoryLimit:[[self class] thumbnailImageSoftMemoryLimit]
                                                                           hardMemoryLimit:[[self class] thumbnailImageHardMemoryLimit]];
        
        NSString *diskCachePath = [[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:[[self class] diskCachePath]];
        _diskCache = [[TGModernCache alloc] initWithPath:diskCachePath size:[[self class] diskMemoryLimit]];
        
        _fullSizeResultsUrl = [NSURL fileURLWithPath:[[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:[NSString stringWithFormat:@"photoeditorresults/%@", _contextId]]];
        [[NSFileManager defaultManager] createDirectoryAtPath:_fullSizeResultsUrl.path withIntermediateDirectories:true attributes:nil error:nil];
        
        _paintingImagesUrl = [NSURL fileURLWithPath:[[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:[NSString stringWithFormat:@"paintingimages/%@", _contextId]]];
        [[NSFileManager defaultManager] createDirectoryAtPath:_paintingImagesUrl.path withIntermediateDirectories:true attributes:nil error:nil];
        
        _stillPaintingImagesUrl = [NSURL fileURLWithPath:[[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:@"stillpaintingimages"]];
        [[NSFileManager defaultManager] createDirectoryAtPath:_stillPaintingImagesUrl.path withIntermediateDirectories:true attributes:nil error:nil];
        
        _videoPaintingImagesUrl = [NSURL fileURLWithPath:[[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:@"videopaintingimages"]];
        [[NSFileManager defaultManager] createDirectoryAtPath:_videoPaintingImagesUrl.path withIntermediateDirectories:true attributes:nil error:nil];
        
        _paintingDatasUrl = [NSURL fileURLWithPath:[[[LegacyComponentsGlobals provider] dataStoragePath] stringByAppendingPathComponent:[NSString stringWithFormat:@"paintingdatas/%@", _contextId]]];
        [[NSFileManager defaultManager] createDirectoryAtPath:_paintingDatasUrl.path withIntermediateDirectories:true attributes:nil error:nil];
        
        _storeVideoPaintingImages = [[NSMutableArray alloc] init];
        
        _faces = [[NSMutableDictionary alloc] init];
        
        _temporaryRepCache = [[NSMutableDictionary alloc] init];
        
        _representationPipe = [[SPipe alloc] init];
        _thumbnailImagePipe = [[SPipe alloc] init];
        _adjustmentsPipe = [[SPipe alloc] init];
        _captionPipe = [[SPipe alloc] init];
        _timerPipe = [[SPipe alloc] init];
        _spoilerPipe = [[SPipe alloc] init];
        _pricePipe = [[SPipe alloc] init];
        _fullSizePipe = [[SPipe alloc] init];
        _cropPipe = [[SPipe alloc] init];
        _captionAbovePipe = [[SPipe alloc] init];
    }
    return self;
}

- (void)dealloc
{
    [self cleanup];
}

- (void)cleanup
{
    TGModernCache *diskCache = _diskCache;
    TGDispatchAfter(10.0, dispatch_get_main_queue(), ^{
        [diskCache cleanup];
    });
    
    [[NSFileManager defaultManager] removeItemAtPath:_fullSizeResultsUrl.path error:nil];
    [[NSFileManager defaultManager] removeItemAtPath:_paintingImagesUrl.path error:nil];
    [[NSFileManager defaultManager] removeItemAtPath:_paintingDatasUrl.path error:nil];
}

+ (instancetype)contextForCaptionsOnly
{
    TGMediaEditingContext *context = [[TGMediaEditingContext alloc] init];
    context->_inhibitEditing = true;
    return context;
}

#pragma mark -

- (SSignal *)imageSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
    return [self imageSignalForItem:item withUpdates:true];
}

- (SSignal *)imageSignalForItem:(NSObject<TGMediaEditableItem> *)item withUpdates:(bool)withUpdates
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return [SSignal fail:nil];
    
    SSignal *updateSignal = [[_representationPipe.signalProducer() filter:^bool(TGMediaImageUpdate *update)
    {
        return [update.item.uniqueIdentifier isEqualToString:item.uniqueIdentifier];
    }] map:^id(TGMediaImageUpdate *update)
    {
        return update.representation;
    }];
    
    if ([self _adjustmentsForItemId:itemId] == nil)
    {
        SSignal *signal = [SSignal single:nil];
        if (withUpdates)
            signal = [signal then:updateSignal];
        return signal;
    }
    
    NSString *imageUri = [TGMediaEditingContext _imageUriForItemId:itemId];
    SSignal *signal = [[self _imageSignalForItemId:itemId imageCache:_imageCache imageDiskUri:imageUri synchronous:false] catch:^SSignal *(__unused id error)
    {
        id temporaryRep = [_temporaryRepCache objectForKey:itemId];
        SSignal *signal = [SSignal single:temporaryRep];
        if (withUpdates)
            signal = [signal then:updateSignal];
        return signal;
    }];
    if (withUpdates)
        signal = [signal then:updateSignal];
    return signal;
}

- (SSignal *)thumbnailImageSignalForItem:(id<TGMediaEditableItem>)item
{
    return [self thumbnailImageSignalForIdentifier:item.uniqueIdentifier];
}

- (SSignal *)thumbnailImageSignalForItem:(id<TGMediaEditableItem>)item withUpdates:(bool)withUpdates synchronous:(bool)synchronous
{
    return [self thumbnailImageSignalForIdentifier:item.uniqueIdentifier withUpdates:withUpdates synchronous: synchronous];
}

- (SSignal *)thumbnailImageSignalForIdentifier:(NSString *)identifier {
    return [self thumbnailImageSignalForIdentifier:identifier withUpdates:true synchronous:false];
}

- (SSignal *)thumbnailImageSignalForIdentifier:(NSString *)identifier withUpdates:(bool)withUpdates synchronous:(bool)synchronous {
    NSString *itemId = [self _contextualIdForItemId:identifier];
    if (itemId == nil)
        return [SSignal fail:nil];
    
    SSignal *updateSignal = [[_thumbnailImagePipe.signalProducer() filter:^bool(TGMediaImageUpdate *update)
    {
        return [update.item.uniqueIdentifier isEqualToString:identifier];
    }] map:^id(TGMediaImageUpdate *update)
    {
        return update.representation;
    }];
    
    if ([self _adjustmentsForItemId:itemId] == nil)
    {
        SSignal *signal = [SSignal single:nil];
        if (withUpdates)
            signal = [signal then:updateSignal];
        return signal;
    }
    
    NSString *imageUri = [TGMediaEditingContext _thumbnailImageUriForItemId:itemId];
    SSignal *signal = [[self _imageSignalForItemId:itemId imageCache:_thumbnailImageCache imageDiskUri:imageUri synchronous:synchronous] catch:^SSignal *(__unused id error)
    {
        SSignal *signal = [SSignal single:nil];
        if (withUpdates)
            signal = [signal then:updateSignal];
        return signal;
    }];
    if (withUpdates)
        signal = [signal then:updateSignal];
    return signal;
}

- (SSignal *)fastImageSignalForItem:(NSObject<TGMediaEditableItem> *)item withUpdates:(bool)withUpdates
{
    return [[self thumbnailImageSignalForItem:item withUpdates:false synchronous:true] then:[self imageSignalForItem:item withUpdates:withUpdates]];
}

- (SSignal *)_imageSignalForItemId:(NSString *)itemId imageCache:(TGMemoryImageCache *)imageCache imageDiskUri:(NSString *)imageDiskUri synchronous:(bool)synchronous
{
    if (itemId == nil)
        return [SSignal fail:nil];
    
    SSignal *signal = [[SSignal alloc] initWithGenerator:^id<SDisposable>(SSubscriber *subscriber)
    {
        void (^completionBlock)(UIImage *) = ^(UIImage *result) {
            if (result == nil)
            {
                NSData *imageData = [_diskCache getValueForKey:[imageDiskUri dataUsingEncoding:NSUTF8StringEncoding]];
                if (imageData.length > 0)
                {
                    result = [UIImage imageWithData:imageData];
                    [imageCache setImage:result forKey:itemId attributes:NULL];
                }
            }
            
            if (result != nil)
            {
                [subscriber putNext:result];
                [subscriber putCompletion];
            }
            else
            {
                [subscriber putError:nil];
            }
        };
        
        if (synchronous) {
            UIImage *result = [imageCache imageForKey:itemId attributes:NULL];
            completionBlock(result);
        } else {
            [imageCache imageForKey:itemId attributes:NULL completion:^(UIImage *result) {
                completionBlock(result);
            }];
        }
        
        return nil;
    }];
    
    return synchronous ? signal : [signal startOn:_queue];
}

- (void)_clearPreviousImageForItemId:(NSString *)itemId
{    
    [_imageCache setImage:nil forKey:itemId attributes:NULL];
    
    NSString *imageUri = [[self class] _imageUriForItemId:itemId];
    [_diskCache setValue:[NSData data] forKey:[imageUri dataUsingEncoding:NSUTF8StringEncoding]];
}

- (UIImage *)paintingImageForItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return nil;
    
    UIImage *result = [_paintingImageCache imageForKey:itemId attributes:NULL];
    if (result == nil)
    {
        NSURL *imageUrl = [_paintingImagesUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.png", [TGStringUtils md5:itemId]]];
        UIImage *diskImage = [UIImage imageWithContentsOfFile:imageUrl.path];
        if (diskImage != nil)
        {
            result = diskImage;
            [_paintingImageCache setImage:result forKey:itemId attributes:NULL];
        }
    }

    return result;
}

- (UIImage *)stillPaintingImageForItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return nil;
    
    UIImage *result = [_stillPaintingImageCache imageForKey:itemId attributes:NULL];
    if (result == nil)
    {
        NSURL *imageUrl = [_stillPaintingImagesUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.png", [TGStringUtils md5:itemId]]];
        UIImage *diskImage = [UIImage imageWithContentsOfFile:imageUrl.path];
        if (diskImage != nil)
        {
            result = diskImage;
            [_stillPaintingImageCache setImage:result forKey:itemId attributes:NULL];
        }
    }

    return result;
}

#pragma mark - Caption

- (NSAttributedString *)captionForItem:(id<TGMediaEditableItem>)item
{
    if (_forcedCaption != nil)
        return _forcedCaption;
    
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return nil;
    
    return _captions[itemId];
}

- (void)setCaption:(NSAttributedString *)caption forItem:(id<TGMediaEditableItem>)item
{
    if (_forcedCaption != nil)
    {
        _forcedCaption = caption;
        _captionPipe.sink([TGMediaCaptionUpdate captionUpdateWithItem:item caption:caption]);
        return;
    }
    
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return;
    
    if (caption.length > 0)
        _captions[itemId] = caption;
    else
        [_captions removeObjectForKey:itemId];
    
    _captionPipe.sink([TGMediaCaptionUpdate captionUpdateWithItem:item caption:caption]);
}

- (bool)isForcedCaption {
    return _forcedCaption.string.length > 0;
}

- (SSignal *)forcedCaption
{
    __weak TGMediaEditingContext *weakSelf = self;
    SSignal *updateSignal = [_captionPipe.signalProducer() map:^NSAttributedString *(TGMediaCaptionUpdate *update)
    {
        __strong TGMediaEditingContext *strongSelf = weakSelf;
        if (strongSelf.isForcedCaption) {
            return strongSelf->_forcedCaption;
        } else {
            return nil;
        }
    }];
    
    return [[SSignal single:_forcedCaption] then:updateSignal];
}

- (void)setForcedCaption:(NSAttributedString *)caption
{
    [self setForcedCaption:caption skipUpdate:false];
}

- (void)setForcedCaption:(NSAttributedString *)caption skipUpdate:(bool)skipUpdate
{
    _forcedCaption = caption;
    
    if (!skipUpdate) {
        _captionPipe.sink([TGMediaCaptionUpdate captionUpdateWithItem:nil caption:caption]);
    }
}

- (SSignal *)captionSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *uniqueIdentifier = item.uniqueIdentifier;
    SSignal *updateSignal = [[_captionPipe.signalProducer() filter:^bool(TGMediaCaptionUpdate *update)
    {
        return [update.item.uniqueIdentifier isEqualToString:uniqueIdentifier];
    }] map:^NSAttributedString *(TGMediaCaptionUpdate *update)
    {
        return update.caption;
    }];
    
    return [[SSignal single:[self captionForItem:item]] then:updateSignal];
}

#pragma mark -

- (id<TGMediaEditAdjustments>)adjustmentsForItem:(id<TGMediaEditableItem>)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return nil;
    
    return [self _adjustmentsForItemId:itemId];
}

- (id<TGMediaEditAdjustments>)_adjustmentsForItemId:(NSString *)itemId
{
    if (itemId == nil)
        return nil;
    
    return _adjustments[itemId];
}

- (void)setAdjustments:(id<TGMediaEditAdjustments>)adjustments forItem:(id<TGMediaEditableItem>)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return;
    
    id<TGMediaEditAdjustments> previousAdjustments = _adjustments[itemId];
    
    if (adjustments != nil)
        _adjustments[itemId] = adjustments;
    else
        [_adjustments removeObjectForKey:itemId];

    bool cropChanged = false;
    if (![previousAdjustments cropAppliedForAvatar:false] && [adjustments cropAppliedForAvatar:false])
        cropChanged = true;
    else if ([previousAdjustments cropAppliedForAvatar:false] && ![adjustments cropAppliedForAvatar:false])
        cropChanged = true;
    else if ([previousAdjustments cropAppliedForAvatar:false] && [adjustments cropAppliedForAvatar:false] && ![previousAdjustments isCropEqualWith:adjustments])
        cropChanged = true;
    
    if (cropChanged)
        _cropPipe.sink(@true);
    
    _adjustmentsPipe.sink([TGMediaAdjustmentsUpdate adjustmentsUpdateWithItem:item adjustments:adjustments]);
}

- (SSignal *)adjustmentsSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
    SSignal *updateSignal = [[_adjustmentsPipe.signalProducer() filter:^bool(TGMediaAdjustmentsUpdate *update)
    {
        return [update.item.uniqueIdentifier isEqualToString:item.uniqueIdentifier];
    }] map:^id<TGMediaEditAdjustments>(TGMediaAdjustmentsUpdate *update)
    {
        return update.adjustments;
    }];
    
    return [[SSignal single:[self adjustmentsForItem:item]] then:updateSignal];
}

- (SSignal *)cropAdjustmentsUpdatedSignal
{
    return _cropPipe.signalProducer();
}

- (SSignal *)adjustmentsUpdatedSignal
{
    return [_adjustmentsPipe.signalProducer() map:^id(__unused id value)
    {
        return @true;
    }];
}

#pragma mark -

- (NSNumber *)timerForItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return nil;
    
    return [self _timerForItemId:itemId];
}

- (NSNumber *)_timerForItemId:(NSString *)itemId
{
    if (itemId == nil)
        return nil;
    
    return _timers[itemId];
}

- (void)setTimer:(NSNumber *)timer forItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return;
    
    if (timer.integerValue != 0)
        _timers[itemId] = timer;
    else
        [_timers removeObjectForKey:itemId];

    _timerPipe.sink([TGMediaTimerUpdate timerUpdateWithItem:item timer:timer]);
}

- (SSignal *)timerSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
    SSignal *updateSignal = [[_timerPipe.signalProducer() filter:^bool(TGMediaTimerUpdate *update)
    {
        return [update.item.uniqueIdentifier isEqualToString:item.uniqueIdentifier];
    }] map:^NSNumber *(TGMediaTimerUpdate *update)
    {
        return update.timer;
    }];
    
    return [[SSignal single:[self timerForItem:item]] then:updateSignal];
}

- (SSignal *)timersUpdatedSignal
{
    return [_timerPipe.signalProducer() map:^id(__unused id value)
    {
        return @true;
    }];
}

#pragma mark -

- (bool)spoilerForItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return nil;
    
    return [self _spoilerForItemId:itemId];
}

- (bool)_spoilerForItemId:(NSString *)itemId
{
    if (itemId == nil)
        return nil;
    
    return _spoilers[itemId];
}

- (void)setSpoiler:(bool)spoiler forItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return;
    
    if (spoiler)
        _spoilers[itemId] = @true;
    else
        [_spoilers removeObjectForKey:itemId];

    _spoilerPipe.sink([TGMediaSpoilerUpdate spoilerUpdateWithItem:item spoiler:spoiler]);
}

- (SSignal *)spoilerSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
    SSignal *updateSignal = [[_spoilerPipe.signalProducer() filter:^bool(TGMediaSpoilerUpdate *update)
    {
        return [update.item.uniqueIdentifier isEqualToString:item.uniqueIdentifier];
    }] map:^NSNumber *(TGMediaSpoilerUpdate *update)
    {
        return @(update.spoiler);
    }];
    
    return [[SSignal single:@([self spoilerForItem:item])] then:updateSignal];
}

- (SSignal *)spoilerSignalForIdentifier:(NSString *)identifier
{
    SSignal *updateSignal = [[_spoilerPipe.signalProducer() filter:^bool(TGMediaSpoilerUpdate *update)
    {
        return [update.item.uniqueIdentifier isEqualToString:identifier];
    }] map:^NSNumber *(TGMediaSpoilerUpdate *update)
    {
        return @(update.spoiler);
    }];
    
    return [[SSignal single:@([self _spoilerForItemId:identifier])] then:updateSignal];
}

- (SSignal *)spoilersUpdatedSignal
{
    return [_spoilerPipe.signalProducer() map:^id(__unused id value)
    {
        return @true;
    }];
}

#pragma mark -

- (NSNumber *)priceForItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return nil;
    
    return [self _priceForItemId:itemId];
}

- (NSNumber *)_priceForItemId:(NSString *)itemId
{
    if (itemId == nil)
        return nil;
    
    return _prices[itemId];
}

- (void)setPrice:(NSNumber *)price forItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return;
    
    if (price.integerValue != 0)
        _prices[itemId] = price;
    else
        [_prices removeObjectForKey:itemId];

    _pricePipe.sink([TGMediaPriceUpdate priceUpdateWithItem:item price:price]);
}

- (SSignal *)priceSignalForItem:(NSObject<TGMediaEditableItem> *)item
{
    SSignal *updateSignal = [[_pricePipe.signalProducer() filter:^bool(TGMediaPriceUpdate *update)
    {
        return [update.item.uniqueIdentifier isEqualToString:item.uniqueIdentifier];
    }] map:^NSNumber *(TGMediaPriceUpdate *update)
    {
        return update.price;
    }];
    
    return [[SSignal single:[self priceForItem:item]] then:updateSignal];
}

- (SSignal *)priceSignalForIdentifier:(NSString *)identifier
{
    SSignal *updateSignal = [[_pricePipe.signalProducer() filter:^bool(TGMediaPriceUpdate *update)
    {
        return [update.item.uniqueIdentifier isEqualToString:identifier];
    }] map:^NSNumber *(TGMediaPriceUpdate *update)
    {
        return update.price;
    }];
    
    return [[SSignal single:[self _priceForItemId:identifier]] then:updateSignal];
}

- (SSignal *)pricesUpdatedSignal
{
    return [_pricePipe.signalProducer() map:^id(__unused id value)
    {
        return @true;
    }];
}


#pragma mark -

- (void)setImage:(UIImage *)image thumbnailImage:(UIImage *)thumbnailImage forItem:(id<TGMediaEditableItem>)item synchronous:(bool)synchronous
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    
    if (itemId == nil)
        return;
    
    void (^block)(void) = ^
    {
        [_temporaryRepCache removeObjectForKey:itemId];
        
        NSString *imageUri = [[self class] _imageUriForItemId:itemId];
        [_imageCache setImage:image forKey:itemId attributes:NULL];
        if (image != nil)
        {
            NSData *imageData = UIImageJPEGRepresentation(image, 0.95f);
            [_diskCache setValue:imageData forKey:[imageUri dataUsingEncoding:NSUTF8StringEncoding]];
        }
        _representationPipe.sink([TGMediaImageUpdate imageUpdateWithItem:item representation:image]);
    
        NSString *thumbnailImageUri = [[self class] _thumbnailImageUriForItemId:itemId];
        [_thumbnailImageCache setImage:thumbnailImage forKey:itemId attributes:NULL];
        if (thumbnailImage != nil)
        {
            NSData *imageData = UIImageJPEGRepresentation(thumbnailImage, 0.87f);
            [_diskCache setValue:imageData forKey:[thumbnailImageUri dataUsingEncoding:NSUTF8StringEncoding]];
        }
        if ([item isKindOfClass:[TGMediaAsset class]] && ((TGMediaAsset *)item).isVideo)
            _thumbnailImagePipe.sink([TGMediaImageUpdate imageUpdateWithItem:item representation:thumbnailImage]);
    };
    
    if (synchronous)
        [_queue dispatchSync:block];
    else
        [_queue dispatch:block];
}

- (bool)setPaintingData:(NSData *)data entitiesData:(NSData *)entitiesData image:(UIImage *)image stillImage:(UIImage *)stillImage forItem:(NSObject<TGMediaEditableItem> *)item dataUrl:(NSURL **)dataOutUrl entitiesDataUrl:(NSURL **)entitiesDataOutUrl imageUrl:(NSURL **)imageOutUrl forVideo:(bool)video
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    
    if (itemId == nil)
        return false;
    
    NSURL *imagesDirectory = video ? _videoPaintingImagesUrl : _paintingImagesUrl;
    NSURL *imageUrl = [imagesDirectory URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.png", [TGStringUtils md5:itemId]]];
    NSURL *dataUrl = [_paintingDatasUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.dat", [TGStringUtils md5:itemId]]];
    NSURL *entitiesDataUrl = [_paintingDatasUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@_entities.dat", [TGStringUtils md5:itemId]]];
    
    [_paintingImageCache setImage:image forKey:itemId attributes:NULL];
    
    NSData *imageData = UIImagePNGRepresentation(image);
    [[NSFileManager defaultManager] removeItemAtURL:imageUrl error:nil];
    bool imageSuccess = [imageData writeToURL:imageUrl options:NSDataWritingAtomic error:nil];
    [[NSFileManager defaultManager] removeItemAtURL:dataUrl error:nil];
    bool dataSuccess = [data writeToURL:dataUrl options:NSDataWritingAtomic error:nil];
    bool entitiesDataSuccess = [entitiesData writeToURL:entitiesDataUrl options:NSDataWritingAtomic error:nil];
    
    if (imageSuccess && imageOutUrl != NULL)
        *imageOutUrl = imageUrl;
    
    if (dataSuccess && dataOutUrl != NULL)
        *dataOutUrl = dataUrl;
    
    if (entitiesDataSuccess && entitiesDataOutUrl != NULL)
        *entitiesDataOutUrl = entitiesDataUrl;
        
    if (video)
        [_storeVideoPaintingImages addObject:imageUrl];
    
    if (stillImage != nil) {
        NSURL *stillImageUrl = [_stillPaintingImagesUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.png", [TGStringUtils md5:itemId]]];
        [_stillPaintingImageCache setImage:stillImage forKey:itemId attributes:NULL];
        
        NSData *stillImageData = UIImagePNGRepresentation(stillImage);
        [[NSFileManager defaultManager] removeItemAtURL:stillImageUrl error:nil];
        [stillImageData writeToURL:stillImageUrl options:NSDataWritingAtomic error:nil];
        
        if (video)
            [_storeVideoPaintingImages addObject:stillImageUrl];
    }
    
    return (image == nil || imageSuccess) && (data == nil || dataSuccess);
}

- (void)clearPaintingData
{
    for (NSURL *url in _storeVideoPaintingImages)
    {
        [[NSFileManager defaultManager] removeItemAtURL:url error:NULL];
    }
}

- (bool)isCaptionAbove {
    return _captionAbove;
}

- (SSignal *)captionAbove
{
    __weak TGMediaEditingContext *weakSelf = self;
    SSignal *updateSignal = [_captionAbovePipe.signalProducer() map:^NSNumber *(NSNumber *update)
    {
        __strong TGMediaEditingContext *strongSelf = weakSelf;
        return @(strongSelf->_captionAbove);
    }];
    
    return [[SSignal single:@(_captionAbove)] then:updateSignal];
}

- (void)setCaptionAbove:(bool)captionAbove
{
    _captionAbove = captionAbove;
    _captionAbovePipe.sink(@(captionAbove));
}

- (SSignal *)facesForItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return [SSignal fail:nil];
    
    NSArray *faces = _faces[itemId];
    
    return [[SSignal alloc] initWithGenerator:^id<SDisposable>(SSubscriber *subscriber)
    {
        [subscriber putNext:faces];
        [subscriber putCompletion];
                           
        return nil;
    }];
}

- (void)setFaces:(NSArray *)faces forItem:(NSObject<TGMediaEditableItem> *)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    if (itemId == nil)
        return;
    
    if (faces.count > 0)
        _faces[itemId] = faces;
    else
        [_faces removeObjectForKey:itemId];
}

- (void)setFullSizeImage:(UIImage *)image forItem:(id<TGMediaEditableItem>)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    
    if (itemId == nil)
        return;
    
    NSData *imageData = UIImageJPEGRepresentation(image, 0.7f);
    NSURL *url = [_fullSizeResultsUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.jpg", [TGStringUtils md5:itemId]]];
    
    bool succeed = [imageData writeToURL:url options:NSDataWritingAtomic error:nil];
    if (succeed)
        _fullSizePipe.sink(itemId);
}

- (NSURL *)_fullSizeImageUrlForItem:(id<TGMediaEditableItem>)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    
    if (itemId == nil)
        return nil;
    
    NSURL *url = [_fullSizeResultsUrl URLByAppendingPathComponent:[NSString stringWithFormat:@"%@.jpg", [TGStringUtils md5:itemId]]];
    if ([[NSFileManager defaultManager] fileExistsAtPath:url.path])
        return url;
    
    return nil;
}

- (SSignal *)fullSizeImageUrlForItem:(id<TGMediaEditableItem>)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    id<TGMediaEditAdjustments> adjustments = [self adjustmentsForItem:item];
    
    if (![adjustments isKindOfClass:[PGPhotoEditorValues class]])
        return [SSignal complete];
    
    PGPhotoEditorValues *editorValues = (PGPhotoEditorValues *)adjustments;
    if (![editorValues toolsApplied] && ![editorValues hasPainting])
        return [SSignal complete];
    
    if ([editorValues.paintingData hasAnimation])
        return [SSignal complete];
    
    NSURL *url = [self _fullSizeImageUrlForItem:item];
    if (url != nil)
        return [SSignal single:url];
    
    return [[[_fullSizePipe.signalProducer() filter:^bool(NSString *identifier)
    {
        return [identifier isEqualToString:itemId];
    }] mapToSignal:^SSignal *(__unused id next)
    {
        NSURL *url = [self _fullSizeImageUrlForItem:item];
        if (url != nil)
            return [SSignal single:url];
        else
            return [SSignal complete];
    }] timeout:5.0 onQueue:_queue orSignal:[SSignal complete]];
}

- (void)setTemporaryRep:(id)rep forItem:(id<TGMediaEditableItem>)item
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    
    if (itemId == nil)
        return;
    
    UIImage *thumbnailImage = nil;
    if ([rep isKindOfClass:[UIImage class]])
    {
        UIImage *image = (UIImage *)rep;
        image.degraded = true;
        image.edited = true;
        
        CGSize fillSize = TGPhotoThumbnailSizeForCurrentScreen();
        fillSize.width = CGCeil(fillSize.width);
        fillSize.height = CGCeil(fillSize.height);
        
        CGSize size = TGScaleToFillSize(image.size, fillSize);
        
        UIGraphicsBeginImageContextWithOptions(size, true, 0.0f);
        CGContextRef context = UIGraphicsGetCurrentContext();
        CGContextSetInterpolationQuality(context, kCGInterpolationMedium);
        
        [image drawInRect:CGRectMake(0, 0, size.width, size.height)];
        
        thumbnailImage = UIGraphicsGetImageFromCurrentImageContext();
        UIGraphicsEndImageContext();
    }
    
    [_queue dispatchSync:^
    {
        [self _clearPreviousImageForItemId:itemId];
        
        if (rep != nil)
            [_temporaryRepCache setObject:rep forKey:itemId];
        else
            [_temporaryRepCache removeObjectForKey:itemId];
        
        _representationPipe.sink([TGMediaImageUpdate imageUpdateWithItem:item representation:rep]);
        
        if (thumbnailImage != nil)
        {
            [_thumbnailImageCache setImage:thumbnailImage forKey:itemId attributes:NULL];
            _thumbnailImagePipe.sink([TGMediaImageUpdate imageUpdateWithItem:item representation:thumbnailImage]);
        }
    }];
}

#pragma mark - Original Images

- (void)requestOriginalImageForItem:(id<TGMediaEditableItem>)item completion:(void (^)(UIImage *))completion
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    
    if (itemId == nil)
    {
        if (completion != nil)
            completion(nil);
        return;
    }
    
    __block UIImage *result = [_originalImageCache imageForKey:itemId attributes:NULL];
    if (result != nil)
    {
        if (completion != nil)
            completion(result);
    }
    else
    {
        [_queue dispatch:^
        {
            NSString *originalImageUri = [[self class] _originalImageUriForItemId:itemId];
            NSData *imageData = [_diskCache getValueForKey:[originalImageUri dataUsingEncoding:NSUTF8StringEncoding]];
            if (imageData != nil)
            {
                result = [UIImage imageWithData:imageData];
                
                [_originalImageCache setImage:result forKey:itemId attributes:NULL];
            }
            
            if (completion != nil)
                completion(result);
        }];
    }
}

- (void)requestOriginalThumbnailImageForItem:(id<TGMediaEditableItem>)item completion:(void (^)(UIImage *))completion
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    
    if (itemId == nil)
    {
        if (completion != nil)
            completion(nil);
        return;
    }
    
    __block UIImage *result = [_originalThumbnailImageCache imageForKey:itemId attributes:NULL];
    if (result != nil)
    {
        if (completion != nil)
            completion(result);
    }
    else
    {
        [_queue dispatch:^
        {
            NSString *originalThumbnailImageUri = [[self class] _originalThumbnailImageUriForItemId:itemId];
            NSData *imageData = [_diskCache getValueForKey:[originalThumbnailImageUri dataUsingEncoding:NSUTF8StringEncoding]];
            if (imageData != nil)
            {
                result = [UIImage imageWithData:imageData];
                
                [_originalThumbnailImageCache setImage:result forKey:itemId attributes:NULL];
            }
            
            if (completion != nil)
                completion(result);
        }];
    }
}

- (void)setOriginalImage:(UIImage *)image forItem:(id<TGMediaEditableItem>)item synchronous:(bool)synchronous
{
    NSString *itemId = [self _contextualIdForItemId:item.uniqueIdentifier];
    
    if (itemId == nil || image == nil)
        return;
    
    if ([_originalImageCache imageForKey:itemId attributes:NULL] != nil)
        return;
    
    void (^block)(void) = ^
    {
        if (image != nil)
        {
            NSString *originalImageUri = [[self class] _originalImageUriForItemId:itemId];
            NSData *existingImageData = [_diskCache getValueForKey:[originalImageUri dataUsingEncoding:NSUTF8StringEncoding]];
            if (existingImageData.length > 0)
                return;
            
            [_originalImageCache setImage:image forKey:itemId attributes:NULL];
            NSData *imageData = UIImageJPEGRepresentation(image, 0.95f);
            [_diskCache setValue:imageData forKey:[originalImageUri dataUsingEncoding:NSUTF8StringEncoding]];
            
            CGFloat thumbnailImageSide = TGPhotoThumbnailSizeForCurrentScreen().width;
            CGSize targetSize = TGScaleToSize(image.size, CGSizeMake(thumbnailImageSide, thumbnailImageSide));
            
            UIGraphicsBeginImageContextWithOptions(targetSize, true, 0.0f);
            [image drawInRect:CGRectMake(0, 0, targetSize.width, targetSize.height)];
            UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
            UIGraphicsEndImageContext();
            
            [_originalThumbnailImageCache setImage:image forKey:itemId attributes:NULL];
            NSString *originalThumbnailImageUri = [[self class] _originalThumbnailImageUriForItemId:itemId];
            NSData *thumbnailImageData = UIImageJPEGRepresentation(image, 0.87f);
            [_diskCache setValue:thumbnailImageData forKey:[originalThumbnailImageUri dataUsingEncoding:NSUTF8StringEncoding]];
        }
    };
    
    if (synchronous)
        [_queue dispatchSync:block];
    else
        [_queue dispatch:block];
}

+ (NSString *)_originalImageUriForItemId:(NSString *)itemId
{
    return [NSString stringWithFormat:@"photo-editor-original://%@", itemId];
}

+ (NSString *)_originalThumbnailImageUriForItemId:(NSString *)itemId
{
    return [NSString stringWithFormat:@"photo-editor-original-thumb://%@", itemId];
}

#pragma mark - URI

- (NSString *)_contextualIdForItemId:(NSString *)itemId
{
    if (itemId == nil)
        return nil;
    
    return itemId;
    //return [NSString stringWithFormat:@"%@hm%@", _contextId, itemId];
}

+ (NSString *)_imageUriForItemId:(NSString *)itemId
{
    return [NSString stringWithFormat:@"%@://%@", [self imageUriScheme], itemId];
}

+ (NSString *)_thumbnailImageUriForItemId:(NSString *)itemId
{
    return [NSString stringWithFormat:@"%@://%@", [self thumbnailImageUriScheme], itemId];
}

#pragma mark - Constants

+ (NSString *)imageUriScheme
{
    return @"photo-editor";
}

+ (NSString *)thumbnailImageUriScheme
{
    return @"photo-editor-thumb";
}

+ (NSString *)diskCachePath
{
    return @"photoeditorcache_v1";
}

+ (NSUInteger)diskMemoryLimit
{
    return 512 * 1024 * 1024;
}

+ (NSUInteger)imageSoftMemoryLimit
{
    return 13 * 1024 * 1024;
}

+ (NSUInteger)imageHardMemoryLimit
{
    return 15 * 1024 * 1024;
}

+ (NSUInteger)originalImageSoftMemoryLimit
{
    return 12 * 1024 * 1024;
}

+ (NSUInteger)originalImageHardMemoryLimit
{
    return 14 * 1024 * 1024;
}

+ (NSUInteger)thumbnailImageSoftMemoryLimit
{
    return 2 * 1024 * 1024;
}

+ (NSUInteger)thumbnailImageHardMemoryLimit
{
    return 3 * 1024 * 1024;
}

@end


@implementation TGMediaImageUpdate

+ (instancetype)imageUpdateWithItem:(id<TGMediaEditableItem>)item representation:(id)representation
{
    TGMediaImageUpdate *update = [[TGMediaImageUpdate alloc] init];
    update->_item = item;
    update->_representation = representation;
    return update;
}

@end


@implementation TGMediaAdjustmentsUpdate

+ (instancetype)adjustmentsUpdateWithItem:(id<TGMediaEditableItem>)item adjustments:(id<TGMediaEditAdjustments>)adjustments
{
    TGMediaAdjustmentsUpdate *update = [[TGMediaAdjustmentsUpdate alloc] init];
    update->_item = item;
    update->_adjustments = adjustments;
    return update;
}

@end

@implementation TGMediaCaptionUpdate

+ (instancetype)captionUpdateWithItem:(id<TGMediaEditableItem>)item caption:(NSAttributedString *)caption
{
    TGMediaCaptionUpdate *update = [[TGMediaCaptionUpdate alloc] init];
    update->_item = item;
    update->_caption = caption;
    return update;
}

@end

@implementation TGMediaTimerUpdate

+ (instancetype)timerUpdateWithItem:(id<TGMediaEditableItem>)item timer:(NSNumber *)timer
{
    TGMediaTimerUpdate *update = [[TGMediaTimerUpdate alloc] init];
    update->_item = item;
    update->_timer = timer;
    return update;
}

+ (instancetype)timerUpdate:(NSNumber *)timer
{
    TGMediaTimerUpdate *update = [[TGMediaTimerUpdate alloc] init];
    update->_timer = timer;
    return update;
}

@end


@implementation TGMediaSpoilerUpdate

+ (instancetype)spoilerUpdateWithItem:(id<TGMediaEditableItem>)item spoiler:(bool)spoiler
{
    TGMediaSpoilerUpdate *update = [[TGMediaSpoilerUpdate alloc] init];
    update->_item = item;
    update->_spoiler = spoiler;
    return update;
}

+ (instancetype)spoilerUpdate:(bool)spoiler
{
    TGMediaSpoilerUpdate *update = [[TGMediaSpoilerUpdate alloc] init];
    update->_spoiler = spoiler;
    return update;
}

@end


@implementation TGMediaPriceUpdate

+ (instancetype)priceUpdateWithItem:(id<TGMediaEditableItem>)item price:(NSNumber *)price
{
    TGMediaPriceUpdate *update = [[TGMediaPriceUpdate alloc] init];
    update->_item = item;
    update->_price = price;
    return update;
}

+ (instancetype)priceUpdate:(NSNumber *)price
{
    TGMediaPriceUpdate *update = [[TGMediaPriceUpdate alloc] init];
    update->_price = price;
    return update;
}

@end