Easily specify bundle or documents location for the HTML Template

PdfReportKit/NSURLWebViewProtocol.h

@@ -0,0 +1,15 @@
+//
+//  NSURLWebViewProtocol.h
+//
+//  Created by Aaron Wardle on 29/11/2013.
+//  http://www.aaronwardle.oc.uk
+//
+
+#import <Foundation/Foundation.h>
+#define kDocumentsImageUrl @"documentsimage"
+#define kBundleImageUrl @"bundleimage"
+
+
+@interface NSURLWebViewProtocol : NSURLProtocol
+
+@end

PdfReportKit/NSURLWebViewProtocol.m

@@ -0,0 +1,88 @@
+//
+//  NSURLWebViewProtocol.m
+//
+//  Created by Aaron Wardle on 29/11/2013.
+//  http://www.aaronwardle.co.uk
+//
+
+#import "NSURLWebViewProtocol.h"
+
+@implementation NSURLWebViewProtocol
+
+
++ (BOOL)canInitWithRequest:(NSURLRequest *)request {
+    if ([request.URL.scheme caseInsensitiveCompare:kDocumentsImageUrl] == NSOrderedSame) {
+        return YES;
+    }
+
+    if ([request.URL.scheme caseInsensitiveCompare:kBundleImageUrl] == NSOrderedSame) {
+        return YES;
+    }
+
+    return NO;
+}
+
+
++ (NSURLRequest *)canonicalRequestForRequest:(NSURLRequest *)request {
+    return request;
+}
+
+- (void)startLoading {
+
+    NSString *fileExtension;
+    NSString *fileName;
+    NSString *urlPrefix;
+    NSString *filePath;
+
+    [self extractFileName:&fileName fileExtension:&fileExtension urlPrefix:&urlPrefix];
+
+
+    NSURLResponse *response =[[NSURLResponse alloc]initWithURL:self.request.URL
+                                                      MIMEType:nil expectedContentLength:-1
+                                              textEncodingName:nil];
+
+
+    if ([urlPrefix isEqualToString:kBundleImageUrl]) {
+        filePath = [[NSBundle mainBundle] pathForResource:fileName ofType:fileExtension];
+    } else {
+        filePath = [NSSearchPathForDirectoriesInDomains(NSDocumentDirectory, NSUserDomainMask, YES) objectAtIndex:0];
+        filePath = [filePath stringByAppendingPathComponent:[NSString stringWithFormat:@"%@.%@", fileName, fileExtension]];
+    }
+
+    NSData *data = [NSData dataWithContentsOfFile:filePath];
+
+    [[self client] URLProtocol:self didReceiveResponse:response cacheStoragePolicy:NSURLCacheStorageNotAllowed];
+    [[self client] URLProtocol:self didLoadData:data];
+    [[self client] URLProtocolDidFinishLoading:self];
+
+}
+
+
+- (void)extractFileName:(NSString**)fileName fileExtension:(NSString **)extension urlPrefix:(NSString **)urlPrefix {
+
+    NSString *urlString = self.request.URL.absoluteString;
+    *extension = [urlString pathExtension];
+
+
+    NSRange isRange = [urlString rangeOfString:kBundleImageUrl options:NSCaseInsensitiveSearch];
+    if (isRange.location == 0) {
+        *urlPrefix = kBundleImageUrl;
+    } else {
+        isRange = [urlString rangeOfString:kDocumentsImageUrl options:NSCaseInsensitiveSearch];
+        if (isRange.location == 0) {
+            *urlPrefix = kDocumentsImageUrl;
+        }
+    }
+
+    urlString = [urlString stringByReplacingOccurrencesOfString:[NSString stringWithFormat:@"%@://", *urlPrefix] withString:@""];
+
+    *fileName = [urlString stringByDeletingPathExtension];
+
+}
+
+
+- (void)stopLoading {
+
+}
+
+@end

PdfReportKit/PRKRenderHtmlOperation.m

@@ -14,6 +14,7 @@
 
 #import "PRKRenderHtmlOperation.h"
 #import "PRKGenerator.h"
+#import "NSURLWebViewProtocol.h"
 
 @implementation PRKRenderHtmlOperation
 
@@ -27,6 +28,9 @@ - (id)initWithHtmlContent:(NSString *)html andSectionType: (PRKSectionType)secti
 
         renderingWebView = [[UIWebView alloc] init];
         renderingWebView.delegate = self;
+
+        // Register the custom NSURL WebView Protocol
+        [NSURLWebViewProtocol registerClass:[NSURLWebViewProtocol class]];
     }
 
     return self;

README.md

@@ -74,6 +74,19 @@ Then you must conform to `PRKGeneratorDataSource` and `PRKGeneratorDelegate` in
 		// Report generated!!! Now we can use NSData as we want!
 	}
 
+## Specifying Images from the App Bundle or Documents Location
+Within the html template you can now specify if an image for the PDF is located in the App Bundle Path or Documents Location (i.e. an image downloaded by your app)
+
+To do this within your template use the following image prefixes:
+
+```html
+<img src="bundleimage://yourimage.png" />
+
+or 
+
+<img src="documentsimage://yourdownloadedimage.png" />
+```
+
 ## Result
 Here it is your wonderful report generated in few simple steps (you can skip all the steps above and see the result just by executing the example application provided)!
 ![Report](http://img706.imageshack.us/img706/6599/captureylw.png)