Telegram/Share/TGChatListAvatarSignal.m

#import "TGChatListAvatarSignal.h"

#import "TGColor.h"
#import "TGRoundImage.h"

#import <libkern/OSAtomic.h>
#import <CommonCrypto/CommonDigest.h>

static OSSpinLock imageDataLock;

@implementation TGChatListAvatarSignal

+ (SSignal *)remoteChatListAvatarWithContext:(TGShareContext *)context location:(TGFileLocation *)location
{
    Api48_InputFileLocation_inputFileLocation *inputFileLocation = [Api48_InputFileLocation inputFileLocationWithVolumeId:@(location.volumeId) localId:@(location.localId) secret:@(location.secret)];
    return [[context datacenter:location.datacenterId function:[Api48 upload_getFileWithLocation:inputFileLocation offset:@(0) limit:@(1024 * 1024)]] map:^id(Api48_upload_File *result)
    {
        [context.persistentCache setValue:result.bytes forKey:[[location description] dataUsingEncoding:NSUTF8StringEncoding]];
        
        OSSpinLockLock(&imageDataLock);
        UIImage *image = [[UIImage alloc] initWithData:result.bytes];
        OSSpinLockUnlock(&imageDataLock);
        
        image = TGRoundImage(image, CGSizeMake(40.0f, 40.0f));
        [context.memoryImageCache setImage:image forKey:[location description] attributes:nil];
        return image;
    }];
}

+ (SSignal *)chatListAvatarWithContext:(TGShareContext *)context location:(TGFileLocation *)location
{
    NSString *key = [location description];
    UIImage *image = [context.memoryImageCache imageForKey:key attributes:NULL];
    if (image != nil)
        return [SSignal single:image];
    
    SSignal *loadFromCacheSignal = [[[SSignal alloc] initWithGenerator:^id<SDisposable>(SSubscriber *subscriber)
    {
        NSData *data = [context.persistentCache getValueForKey:[key dataUsingEncoding:NSUTF8StringEncoding]];
        if (data == nil)
            [subscriber putError:nil];
        else
        {
            OSSpinLockLock(&imageDataLock);
            UIImage *image = [[UIImage alloc] initWithData:data];
            OSSpinLockUnlock(&imageDataLock);
            
            image = TGRoundImage(image, CGSizeMake(40.0f, 40.0f));
            [context.memoryImageCache setImage:image forKey:[location description] attributes:nil];
            [subscriber putNext:image];
            [subscriber putCompletion];
        }
        return nil;
    }] startOnThreadPool:[context sharedThreadPool]];
    
    return [loadFromCacheSignal catch:^SSignal *(__unused id error)
    {
        return [self remoteChatListAvatarWithContext:context location:location];
    }];
}

static inline int colorIndexForUid(int32_t uid, int32_t myUserId)
{
    static const int numColors = 8;
    
    int colorIndex = 0;
    
    char buf[16];
    snprintf(buf, 16, "%d%d", (int)uid, (int)myUserId);
    unsigned char digest[CC_MD5_DIGEST_LENGTH];
    CC_MD5(buf, (CC_LONG)strlen(buf), digest);
    colorIndex = ABS(digest[ABS(uid % 16)]) % numColors;
    
    return colorIndex;
}

static inline int colorIndexForGroupId(int64_t groupId)
{
    static const int numColors = 4;
    
    int colorIndex = 0;
    
    char buf[16];
    snprintf(buf, 16, "%lld", groupId);
    unsigned char digest[CC_MD5_DIGEST_LENGTH];
    CC_MD5(buf, (CC_LONG)strlen(buf), digest);
    colorIndex = ABS(digest[ABS(groupId % 16)]) % numColors;
    
    return colorIndex;
}

+ (NSArray *)gradientColorsForPeerId:(TGPeerId)peerId myUserId:(int32_t)myUserId
{
    static OSSpinLock lock = 0;
    static NSMutableDictionary *dict = nil;
    static NSArray *colors = nil;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^
    {
        dict = [[NSMutableDictionary alloc] init];
        colors = @[
            @[TGColorWithHex(0xff516a), TGColorWithHex(0xff885e)],
            @[TGColorWithHex(0xffa85c), TGColorWithHex(0xffcd6a)],
            @[TGColorWithHex(0x54cb68), TGColorWithHex(0xa0de7e)],
            @[TGColorWithHex(0x2a9ef1), TGColorWithHex(0x72d5fd)],
            @[TGColorWithHex(0x665fff), TGColorWithHex(0x82b1ff)],
            @[TGColorWithHex(0xd669ed), TGColorWithHex(0xe0a2f3)],
        ];
    });
    
    OSSpinLockLock(&lock);
    NSNumber *key = [NSNumber numberWithLongLong:(((long long)peerId.namespaceId) << 32) | ((long long)peerId.peerId)];
    NSNumber *index = dict[key];
    if (index == nil)
    {
        if (peerId.namespaceId == TGPeerIdPrivate)
            index = @(colorIndexForUid(peerId.peerId, myUserId));
        else
            index = @(colorIndexForGroupId(-peerId.peerId));
        dict[key] = index;
    }
    OSSpinLockUnlock(&lock);
    
    return colors[[index intValue] % 6];
}

+ (SSignal *)chatListAvatarWithContext:(TGShareContext *)context letters:(NSString *)letters peerId:(TGPeerId)peerId
{
    NSString *key = [[NSString alloc] initWithFormat:@"GradientAvatar-%d.%d-%@", (int)peerId.namespaceId, (int)peerId.peerId, letters];
    UIImage *image = [context.memoryImageCache imageForKey:key attributes:NULL];
    if (image != nil)
        return [SSignal single:image];
    
    return [[[SSignal alloc] initWithGenerator:^id<SDisposable>(SSubscriber *subscriber)
    {
        UIGraphicsBeginImageContextWithOptions(CGSizeMake(40.0f, 40.0f), false, 0.0f);
        CGContextRef contextRef = UIGraphicsGetCurrentContext();
        
        CGContextBeginPath(contextRef);
        CGContextAddEllipseInRect(contextRef, CGRectMake(0.0f, 0.0f, 40.0f, 40.0f));
        CGContextClip(contextRef);
        
        NSArray *gradientColors = [self gradientColorsForPeerId:peerId myUserId:0];
        CGColorRef colors[2] = {
            CGColorRetain(((UIColor *)gradientColors[0]).CGColor),
            CGColorRetain(((UIColor *)gradientColors[1]).CGColor)
        };
        
        CFArrayRef colorsArray = CFArrayCreate(kCFAllocatorDefault, (const void **)&colors, 2, NULL);
        CGFloat locations[2] = {0.0f, 1.0f};
        
        CGColorSpaceRef colorSpace = CGColorSpaceCreateDeviceRGB();
        CGGradientRef gradient = CGGradientCreateWithColors(colorSpace, colorsArray, (CGFloat const *)&locations);
        
        CFRelease(colorsArray);
        CFRelease(colors[0]);
        CFRelease(colors[1]);
        
        CGColorSpaceRelease(colorSpace);
        
        CGContextDrawLinearGradient(contextRef, gradient, CGPointMake(0.0f, 0.0f), CGPointMake(0.0f, 40.0f), 0);
        
        CFRelease(gradient);
        
        UIFont *font = [UIFont systemFontOfSize:letters.length == 1 ? 18.0f : 16.0f];
        CGSize lettersSize = [letters sizeWithAttributes:@{NSFontAttributeName: font}];
        [letters drawAtPoint:CGPointMake((CGFloat)(floor(40.0f - lettersSize.width) / 2.0f), (CGFloat)(floor(40.0f - lettersSize.height) / 2.0f)) withAttributes:@{NSFontAttributeName: font, NSForegroundColorAttributeName: [UIColor whiteColor]}];
        
        UIImage *image = UIGraphicsGetImageFromCurrentImageContext();
        UIGraphicsEndImageContext();
        [context.memoryImageCache setImage:image forKey:key attributes:nil];
        
        [subscriber putNext:image];
        [subscriber putCompletion];
        
        return nil;
    }] startOnThreadPool:context.sharedThreadPool];
}

@end