//
//  UIImageView+EzvizOpenSDK.m
//  EZOpenSDKDemo
//
//  Created by DeJohn Dong on 15/11/18.
//  Copyright © 2015年 Ezviz. All rights reserved.
//

#import "UIImageView+EzvizOpenSDK.h"
#import <objc/runtime.h>
#import "AFHTTPRequestOperation.h"

@interface EZDataCache : NSCache <EZDataCache>
@end

@interface UIImageView ()
@property (readwrite, nonatomic, strong, setter = ez_setImageRequestOperation:) AFHTTPRequestOperation *ez_imageRequestOperation;
@end

@implementation UIImageView (EzvizOpenSDK)

+ (id <EZDataCache>)sharedDataCache {
    static EZDataCache *_ez_defaultDataCache = nil;
    static dispatch_once_t oncePredicate;
    dispatch_once(&oncePredicate, ^{
        _ez_defaultDataCache = [[EZDataCache alloc] init];
        
        [[NSNotificationCenter defaultCenter] addObserverForName:UIApplicationDidReceiveMemoryWarningNotification object:nil queue:[NSOperationQueue mainQueue] usingBlock:^(NSNotification * __unused notification) {
            [_ez_defaultDataCache removeAllObjects];
        }];
    });
    
#pragma clang diagnostic push
#pragma clang diagnostic ignored "-Wgnu"
    return objc_getAssociatedObject(self, @selector(sharedDataCache)) ?: _ez_defaultDataCache;
#pragma clang diagnostic pop
}

+ (void)setSharedDataCache:(__nullable id <EZDataCache>)dataCache {
    objc_setAssociatedObject(self, @selector(sharedDataCache), dataCache, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

#pragma mark -

+ (NSOperationQueue *)ez_sharedImageRequestOperationQueue {
    static NSOperationQueue *_ez_sharedImageRequestOperationQueue = nil;
    static dispatch_once_t onceToken;
    dispatch_once(&onceToken, ^{
        _ez_sharedImageRequestOperationQueue = [[NSOperationQueue alloc] init];
        _ez_sharedImageRequestOperationQueue.maxConcurrentOperationCount = NSOperationQueueDefaultMaxConcurrentOperationCount;
    });
    
    return _ez_sharedImageRequestOperationQueue;
}

- (AFHTTPRequestOperation *)ez_imageRequestOperation {
    return (AFHTTPRequestOperation *)objc_getAssociatedObject(self, @selector(ez_imageRequestOperation));
}

- (void)ez_setImageRequestOperation:(AFHTTPRequestOperation *)imageRequestOperation {
    objc_setAssociatedObject(self, @selector(ez_imageRequestOperation), imageRequestOperation, OBJC_ASSOCIATION_RETAIN_NONATOMIC);
}

- (void)ez_setImageWithURLRequest:(NSURLRequest *)urlRequest
              placeholderImage:(UIImage *)placeholderImage
                       success:(void (^)(NSURLRequest *request, NSHTTPURLResponse * __nullable response, NSData *data))success
                       failure:(void (^)(NSURLRequest *request, NSHTTPURLResponse * __nullable response, NSError *error))failure
{
    [self cancelImageRequestOperation];
    
    NSData *cachedData = [[[self class] sharedDataCache] cachedDataForRequest:urlRequest];
    if (cachedData) {
        if (success) {
            success(urlRequest, nil, cachedData);
        } else {
            self.image = [[UIImage alloc] initWithData:cachedData];
        }
        
        self.ez_imageRequestOperation = nil;
    } else {
        if (placeholderImage) {
            self.image = placeholderImage;
        }
        
        __weak __typeof(self)weakSelf = self;
        self.ez_imageRequestOperation = [[AFHTTPRequestOperation alloc] initWithRequest:urlRequest];
        [self.ez_imageRequestOperation setCompletionBlockWithSuccess:^(AFHTTPRequestOperation *operation, id responseObject) {
            __strong __typeof(weakSelf)strongSelf = weakSelf;
            if ([[urlRequest URL] isEqual:[strongSelf.ez_imageRequestOperation.request URL]]) {
                if (success) {
                    success(urlRequest, operation.response, responseObject);
                } else if (responseObject) {
                    strongSelf.image = responseObject;
                }
                
                if (operation == strongSelf.ez_imageRequestOperation){
                    strongSelf.ez_imageRequestOperation = nil;
                }
            }
            
            [[[strongSelf class] sharedDataCache] cacheData:responseObject forRequest:urlRequest];
        } failure:^(AFHTTPRequestOperation *operation, NSError *error) {
            __strong __typeof(weakSelf)strongSelf = weakSelf;
            if ([[urlRequest URL] isEqual:[strongSelf.ez_imageRequestOperation.request URL]]) {
                if (failure) {
                    failure(urlRequest, operation.response, error);
                }
                
                if (operation == strongSelf.ez_imageRequestOperation){
                    strongSelf.ez_imageRequestOperation = nil;
                }
            }
        }];
    }
    //处理测试环境下证书不合法的情况，正式线上可以注释该代码。
    [self.ez_imageRequestOperation setWillSendRequestForAuthenticationChallengeBlock:^(NSURLConnection * _Nonnull connection, NSURLAuthenticationChallenge * _Nonnull challenge) {
        if ([challenge.protectionSpace.authenticationMethod isEqualToString:NSURLAuthenticationMethodServerTrust])
        {
            // 证书不认证  全部信任
            [[challenge sender]  useCredential:[NSURLCredential credentialForTrust:challenge.protectionSpace.serverTrust] forAuthenticationChallenge:challenge];
            [[challenge sender]  continueWithoutCredentialForAuthenticationChallenge: challenge];
        }
    }];
    [[[self class] ez_sharedImageRequestOperationQueue] addOperation:self.ez_imageRequestOperation];
}

- (void)cancelImageRequestOperation {
    [self.ez_imageRequestOperation cancel];
    self.ez_imageRequestOperation = nil;
}

@end

#pragma mark -

static inline NSString * EZDataCacheKeyFromURLRequest(NSURLRequest *request) {
    return [[request URL] absoluteString];
}

@implementation EZDataCache

- (NSData *)cachedDataForRequest:(NSURLRequest *)request {
    switch ([request cachePolicy]) {
        case NSURLRequestReloadIgnoringCacheData:
        case NSURLRequestReloadIgnoringLocalAndRemoteCacheData:
            return nil;
        default:
            break;
    }
    
    return [self objectForKey:EZDataCacheKeyFromURLRequest(request)];
}

- (void)cacheData:(NSData *)data forRequest:(NSURLRequest *)request
{
    if (data && request) {
        [self setObject:data forKey:EZDataCacheKeyFromURLRequest(request)];
    }
}

@end