4

Pods/SDWebImage/SDWebImage/Core/SDWebImageDownloaderResponseModifier.m

@@ -0,0 +1,73 @@
+/*
+* This file is part of the SDWebImage package.
+* (c) Olivier Poitrey <rs@dailymotion.com>
+*
+* For the full copyright and license information, please view the LICENSE
+* file that was distributed with this source code.
+*/
+
+
+#import "SDWebImageDownloaderResponseModifier.h"
+
+@interface SDWebImageDownloaderResponseModifier ()
+
+@property (nonatomic, copy, nonnull) SDWebImageDownloaderResponseModifierBlock block;
+
+@end
+
+@implementation SDWebImageDownloaderResponseModifier
+
+- (instancetype)initWithBlock:(SDWebImageDownloaderResponseModifierBlock)block {
+    self = [super init];
+    if (self) {
+        self.block = block;
+    }
+    return self;
+}
+
++ (instancetype)responseModifierWithBlock:(SDWebImageDownloaderResponseModifierBlock)block {
+    SDWebImageDownloaderResponseModifier *responseModifier = [[SDWebImageDownloaderResponseModifier alloc] initWithBlock:block];
+    return responseModifier;
+}
+
+- (nullable NSURLResponse *)modifiedResponseWithResponse:(nonnull NSURLResponse *)response {
+    if (!self.block) {
+        return nil;
+    }
+    return self.block(response);
+}
+
+@end
+
+@implementation SDWebImageDownloaderResponseModifier (Conveniences)
+
+- (instancetype)initWithStatusCode:(NSInteger)statusCode {
+    return [self initWithStatusCode:statusCode version:nil headers:nil];
+}
+
+- (instancetype)initWithVersion:(NSString *)version {
+    return [self initWithStatusCode:200 version:version headers:nil];
+}
+
+- (instancetype)initWithHeaders:(NSDictionary<NSString *,NSString *> *)headers {
+    return [self initWithStatusCode:200 version:nil headers:headers];
+}
+
+- (instancetype)initWithStatusCode:(NSInteger)statusCode version:(NSString *)version headers:(NSDictionary<NSString *,NSString *> *)headers {
+    version = version ? [version copy] : @"HTTP/1.1";
+    headers = [headers copy];
+    return [self initWithBlock:^NSURLResponse * _Nullable(NSURLResponse * _Nonnull response) {
+        if (![response isKindOfClass:NSHTTPURLResponse.class]) {
+            return response;
+        }
+        NSMutableDictionary *mutableHeaders = [((NSHTTPURLResponse *)response).allHeaderFields mutableCopy];
+        for (NSString *header in headers) {
+            NSString *value = headers[header];
+            mutableHeaders[header] = value;
+        }
+        NSHTTPURLResponse *httpResponse = [[NSHTTPURLResponse alloc] initWithURL:response.URL statusCode:statusCode HTTPVersion:version headerFields:[mutableHeaders copy]];
+        return httpResponse;
+    }];
+}
+
+@end