WebPProtocol.m
2.58 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
//
// Copyright (c) 2013-2014 Scott Talbot.
// Copyright (c) 2014 Darryl Pogue.
#import "WebPProtocol.h"
#import "STWebPStreamingDecoder.h"
@interface WebPProtocol () <NSURLConnectionDelegate>
@property (nonatomic, strong) NSURLConnection *connection;
@property (nonatomic, strong) STWebPStreamingDecoder *decoder;
@end
@implementation WebPProtocol
+ (BOOL)canInitWithRequest:(NSURLRequest *)request {
if ([NSURLProtocol propertyForKey:@"WebPProtocolHandledKey" inRequest:request]) {
return NO;
}
NSString* const requestFiletype = [[[request URL] pathExtension] lowercaseString];
BOOL webpEnable = [[[[request URL] absoluteString] lowercaseString] containsString:@"/format/webp"];
return [@"webp" isEqualToString:requestFiletype] || webpEnable;
}
+ (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request {
return request;
}
+ (BOOL)requestIsCacheEquivalent:(NSURLRequest *)a toRequest:(NSURLRequest *)b {
return [super requestIsCacheEquivalent:a toRequest:b];
}
- (void)startLoading {
NSMutableURLRequest *newRequest = [self.request mutableCopy];
[NSURLProtocol setProperty:@YES forKey:@"WebPProtocolHandledKey" inRequest:newRequest];
self.connection = [NSURLConnection connectionWithRequest:newRequest delegate:self];
}
- (void)stopLoading {
[self.connection cancel];
self.connection = nil;
}
#pragma mark - NSURLConnectionDelegate
- (void)connection:(NSURLConnection *)connection didReceiveResponse:(NSURLResponse *)response {
self.decoder = [[STWebPStreamingDecoder alloc] init];
[self.client URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageAllowed];
}
- (void)connection:(NSURLConnection *)connection didReceiveData:(NSData *)data {
[self.decoder updateWithData:data];
}
- (void)connectionDidFinishLoading:(NSURLConnection *)connection {
NSError *error = nil;
UIImage *image = [self.decoder imageWithScale:1 error:&error];
if (!image) {
[self.client URLProtocol:self didFailWithError:error];
return;
}
NSData *imagePNGData = UIImagePNGRepresentation(image);
[self.client URLProtocol:self didLoadData:imagePNGData];
self.decoder = nil;
[self.client URLProtocolDidFinishLoading:self];
}
- (void)connection:(NSURLConnection *)connection didFailWithError:(NSError *)error {
[self.client URLProtocol:self didFailWithError:error];
}
@end