72 lines
2.1 KiB
Objective-C
72 lines
2.1 KiB
Objective-C
/*
|
|
* 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 "SDWebImageDownloaderRequestModifier.h"
|
|
|
|
@interface SDWebImageDownloaderRequestModifier ()
|
|
|
|
@property (nonatomic, copy, nonnull) SDWebImageDownloaderRequestModifierBlock block;
|
|
|
|
@end
|
|
|
|
@implementation SDWebImageDownloaderRequestModifier
|
|
|
|
- (instancetype)initWithBlock:(SDWebImageDownloaderRequestModifierBlock)block {
|
|
self = [super init];
|
|
if (self) {
|
|
self.block = block;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
+ (instancetype)requestModifierWithBlock:(SDWebImageDownloaderRequestModifierBlock)block {
|
|
SDWebImageDownloaderRequestModifier *requestModifier = [[SDWebImageDownloaderRequestModifier alloc] initWithBlock:block];
|
|
return requestModifier;
|
|
}
|
|
|
|
- (NSURLRequest *)modifiedRequestWithRequest:(NSURLRequest *)request {
|
|
if (!self.block) {
|
|
return nil;
|
|
}
|
|
return self.block(request);
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation SDWebImageDownloaderRequestModifier (Conveniences)
|
|
|
|
- (instancetype)initWithMethod:(NSString *)method {
|
|
return [self initWithMethod:method headers:nil body:nil];
|
|
}
|
|
|
|
- (instancetype)initWithHeaders:(NSDictionary<NSString *,NSString *> *)headers {
|
|
return [self initWithMethod:nil headers:headers body:nil];
|
|
}
|
|
|
|
- (instancetype)initWithBody:(NSData *)body {
|
|
return [self initWithMethod:nil headers:nil body:body];
|
|
}
|
|
|
|
- (instancetype)initWithMethod:(NSString *)method headers:(NSDictionary<NSString *,NSString *> *)headers body:(NSData *)body {
|
|
method = method ? [method copy] : @"GET";
|
|
headers = [headers copy];
|
|
body = [body copy];
|
|
return [self initWithBlock:^NSURLRequest * _Nullable(NSURLRequest * _Nonnull request) {
|
|
NSMutableURLRequest *mutableRequest = [request mutableCopy];
|
|
mutableRequest.HTTPMethod = method;
|
|
mutableRequest.HTTPBody = body;
|
|
for (NSString *header in headers) {
|
|
NSString *value = headers[header];
|
|
[mutableRequest setValue:value forHTTPHeaderField:header];
|
|
}
|
|
return [mutableRequest copy];
|
|
}];
|
|
}
|
|
|
|
@end
|