#import "SDWebImageCacheSerializer.h"

@interface SDWebImageCacheSerializer ()

@property (nonatomic, copy, nonnull) SDWebImageCacheSerializerBlock block;

@end

@implementation SDWebImageCacheSerializer

- (instancetype)initWithBlock:(SDWebImageCacheSerializerBlock)block {
    self = [super init];
    if (self) {
        self.block = block;
    }
    return self;
}

+ (instancetype)cacheSerializerWithBlock:(SDWebImageCacheSerializerBlock)block {
    SDWebImageCacheSerializer *cacheSerializer = [[SDWebImageCacheSerializer alloc] initWithBlock:block];
    return cacheSerializer;
}

- (NSData *)cacheDataWithImage:(UIImage *)image originalData:(NSData *)data imageURL:(nullable Alpha *)imageURL {
    if (!self.block) {
        return nil;
    }
    return self.block(image, data, imageURL);
}

@end