From b043c749c10865a722570661159c3f645ddf38ca Mon Sep 17 00:00:00 2001
From: Tony Arnold <tony@thecocoabots.com>
Date: Sat, 6 May 2017 14:59:02 +1000
Subject: [PATCH] Add the ability to initialise a DownView using a custom
 template bundle

---
 Source/Views/DownView.swift | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/Source/Views/DownView.swift b/Source/Views/DownView.swift
index 84009b75..17e989c2 100644
--- a/Source/Views/DownView.swift
+++ b/Source/Views/DownView.swift
@@ -20,12 +20,22 @@ open class DownView: WKWebView {
      - parameter frame:               The frame size of the web view
      - parameter markdownString:      A string containing CommonMark Markdown
      - parameter openLinksInBrowser:  Whether or not to open links using an external browser
+     - parameter templateBundle:      Optional custom template bundle. Leaving this as `nil` will use the bundle included with Down.
      - parameter didLoadSuccessfully: Optional callback for when the web content has loaded successfully
 
      - returns: An instance of Self
      */
-    public init(frame: CGRect, markdownString: String, openLinksInBrowser: Bool = true, didLoadSuccessfully: DownViewClosure? = nil) throws {
+    public init(frame: CGRect, markdownString: String, openLinksInBrowser: Bool = true, templateBundle: Bundle? = nil, didLoadSuccessfully: DownViewClosure? = nil) throws {
         self.didLoadSuccessfully = didLoadSuccessfully
+
+        if let templateBundle = templateBundle {
+            self.bundle = templateBundle
+        } else {
+            let classBundle = Bundle(for: DownView.self)
+            let url = classBundle.url(forResource: "DownView", withExtension: "bundle")!
+            self.bundle = Bundle(url: url)!
+        }
+
         super.init(frame: frame, configuration: WKWebViewConfiguration())
 
         if openLinksInBrowser || didLoadSuccessfully != nil { navigationDelegate = self }
@@ -58,11 +68,7 @@ open class DownView: WKWebView {
 
     // MARK: - Private Properties
 
-    fileprivate let bundle: Bundle = {
-        let bundle = Bundle(for: DownView.self)
-        let url = bundle.url(forResource: "DownView", withExtension: "bundle")!
-        return Bundle(url: url)!
-    }()
+    let bundle: Bundle
 
     fileprivate lazy var baseURL: URL = {
         return self.bundle.url(forResource: "index", withExtension: "html")!