1 files changed, 142 insertions, 0 deletions
diff --git a/src/ui/common/MCOCIDURLProtocol.mm b/src/ui/common/MCOCIDURLProtocol.mm
new file mode 100644
index 00000000..ad529f54
--- /dev/null
+++ b/src/ui/common/MCOCIDURLProtocol.mm
@@ -0,0 +1,142 @@
+//
+//  MCTCIDURLProtocol.m
+//  testUI
+//
+//  Created by DINH Viêt Hoà on 1/22/13.
+//  Copyright (c) 2013 MailCore. All rights reserved.
+//
+
+#import "MCOCIDURLProtocol.h"
+
+#define MCOCIDURLProtocolDownloadedNotification @"MCOCIDURLProtocolDownloadedNotification"
+
+@implementation MCOCIDURLProtocol
+
++ (void) registerProtocol
+{
+    static dispatch_once_t onceToken = 0;
+    dispatch_once(&onceToken, ^{
+		[NSURLProtocol registerClass:[MCOCIDURLProtocol class]];
+	});
+}
+
++ (BOOL) canInitWithRequest:(NSURLRequest *)theRequest
+{
+    if ([self isCID:[theRequest URL]])
+        return YES;
+    if ([self isXMailcoreImage:[theRequest URL]])
+        return YES;
+    return NO;
+}
+
++ (BOOL) isCID:(NSURL *)url
+{
+	NSString *theScheme = [url scheme];
+	if ([theScheme caseInsensitiveCompare:@"cid"] == NSOrderedSame)
+        return YES;
+    return NO;
+}
+
++ (BOOL) isXMailcoreImage:(NSURL *)url
+{
+	NSString *theScheme = [url scheme];
+	if ([theScheme caseInsensitiveCompare:@"x-mailcore-image"] == NSOrderedSame)
+        return YES;
+    return NO;
+}
+
++ (NSURLRequest *) canonicalRequestForRequest:(NSURLRequest *)request
+{
+    return request;
+}
+
+- (id) initWithRequest:(NSURLRequest *)request cachedResponse:(NSCachedURLResponse *)cachedResponse client:(id < NSURLProtocolClient >)client
+{
+    self = [super initWithRequest:request cachedResponse:cachedResponse client:client];
+    [[NSNotificationCenter defaultCenter] addObserver:self selector:@selector(_downloaded:) name:MCOCIDURLProtocolDownloadedNotification object:nil];
+    return self;
+}
+
+- (void) dealloc
+{
+    [[NSNotificationCenter defaultCenter] removeObserver:self];
+    [super dealloc];
+}
+
+- (NSString *) _partUniqueID
+{
+    return [NSURLProtocol propertyForKey:@"PartUniqueID" inRequest:[self request]];
+}
+
+- (NSData *) _data
+{
+    return [NSURLProtocol propertyForKey:@"Data" inRequest:[self request]];
+}
+
+- (MCOAbstractMessage *) _message
+{
+    return (MCOAbstractMessage *) [NSURLProtocol propertyForKey:@"Message" inRequest:[self request]];
+}
+
+- (void) startLoading
+{
+    //NSLog(@"waiting for %p %@", self, [self _partUniqueID]);
+    if ([self _data] != NULL) {
+        [[self class] partDownloadedMessage:[self _message] partUniqueID:[self _partUniqueID] data:[self _data]];
+    }
+}
+
+- (void) _downloaded:(NSNotification *)notification
+{
+    NSDictionary * userInfo = [notification userInfo];
+    
+    NSString * notifPartID = [userInfo objectForKey:@"PartUniqueID"];
+    MCOAbstractMessage * notifMessage = [userInfo objectForKey:@"Message"];
+    if (notifMessage != [self _message]) {
+        return;
+    }
+    if (![[self _partUniqueID] isEqualToString:notifPartID]) {
+        return;
+    }
+    
+    NSData * data = [userInfo objectForKey:@"Data"];
+    NSURLResponse * response = [[NSURLResponse alloc] initWithURL:[[self request] URL] MIMEType:@"application/data"
+                                            expectedContentLength:[data length] textEncodingName:nil];
+    [[self client] URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
+    [[self client] URLProtocol:self didLoadData:data];
+    [[self client] URLProtocolDidFinishLoading:self];
+    [response release];
+}
+
+- (void) stopLoading
+{
+}
+
++ (void) startLoadingWithMessage:(MCOAbstractMessage *)message
+                    partUniqueID:(NSString *)partUniqueID
+                            data:(NSData *)data
+                         request:(NSMutableURLRequest *)request
+{
+    [NSURLProtocol setProperty:message
+                        forKey:@"Message" inRequest:request];
+    if (data != NULL) {
+        [NSURLProtocol setProperty:data forKey:@"Data" inRequest:request];
+    }
+    [NSURLProtocol setProperty:partUniqueID forKey:@"PartUniqueID" inRequest:request];
+}
+
++ (void) partDownloadedMessage:(MCOAbstractMessage *)message
+                  partUniqueID:(NSString *)partUniqueID
+                          data:(NSData *)data
+{
+    NSMutableDictionary * userInfo = [[NSMutableDictionary alloc] init];
+    [userInfo setObject:message forKey:@"Message"];
+    [userInfo setObject:partUniqueID forKey:@"PartUniqueID"];
+    if (data != NULL) {
+        [userInfo setObject:data forKey:@"Data"];
+    }
+    [[NSNotificationCenter defaultCenter] postNotificationName:MCOCIDURLProtocolDownloadedNotification object:nil userInfo:userInfo];
+    [userInfo release];
+}
+
+@end