From 433f08d2281a3e03d72dddd8a09eed5a098eaae6 Mon Sep 17 00:00:00 2001
From: JP Simard <jp@jpsim.com>
Date: Tue, 7 Jul 2015 11:37:09 -0700
Subject: [PATCH 1/2] [Templates] Added support for custom templates (addresses
 #20)

---
 CHANGELOG.md                                   | 18 ++++++++++++++++++
 lib/jazzy/config.rb                            | 13 +++++++++++++
 lib/jazzy/doc.rb                               |  3 ++-
 lib/jazzy/{ => templates}/doc.mustache         |  8 ++++----
 .../{partials => templates}/footer.mustache    |  0
 .../{partials => templates}/header.mustache    |  0
 lib/jazzy/{partials => templates}/nav.mustache |  0
 .../{partials => templates}/parameter.mustache |  0
 .../{partials => templates}/task.mustache      |  2 +-
 .../{partials => templates}/tasks.mustache     |  2 +-
 10 files changed, 39 insertions(+), 7 deletions(-)
 rename lib/jazzy/{ => templates}/doc.mustache (89%)
 rename lib/jazzy/{partials => templates}/footer.mustache (100%)
 rename lib/jazzy/{partials => templates}/header.mustache (100%)
 rename lib/jazzy/{partials => templates}/nav.mustache (100%)
 rename lib/jazzy/{partials => templates}/parameter.mustache (100%)
 rename lib/jazzy/{partials => templates}/task.mustache (97%)
 rename lib/jazzy/{partials => templates}/tasks.mustache (80%)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index cf4f32e03..dbaab35e2 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,3 +1,21 @@
+## Master
+
+##### Breaking
+
+* None.
+
+##### Enhancements
+
+* Added support for custom templates: use the `-t`/`--template-directory`
+  argument to jazzy.  
+  [JP Simard](https://github.com/jpsim)
+  [#20](https://github.com/realm/jazzy/issues/20)
+
+##### Bug Fixes
+
+* None.
+
+
 ## 0.2.1
 
 ##### Breaking
diff --git a/lib/jazzy/config.rb b/lib/jazzy/config.rb
index 575f8710e..75102a751 100644
--- a/lib/jazzy/config.rb
+++ b/lib/jazzy/config.rb
@@ -2,6 +2,7 @@
 require 'pathname'
 require 'uri'
 
+require 'jazzy/doc'
 require 'jazzy/podspec_documenter'
 require 'jazzy/source_declaration/access_control_level'
 
@@ -29,6 +30,7 @@ class Config
     attr_accessor :docset_path
     attr_accessor :source_directory
     attr_accessor :excluded_files
+    attr_accessor :template_directory
 
     def initialize
       PodspecDocumenter.configure(self, Dir['*.podspec{,.json}'].first)
@@ -44,12 +46,18 @@ def initialize
       self.skip_undocumented = false
       self.source_directory = Pathname.pwd
       self.excluded_files = []
+      self.template_directory = Pathname(__FILE__).parent + 'templates'
     end
 
     def podspec=(podspec)
       @podspec = PodspecDocumenter.configure(self, podspec)
     end
 
+    def template_directory=(template_directory)
+      @template_directory = template_directory
+      Doc.template_path = template_directory
+    end
+
     # rubocop:disable Metrics/MethodLength
     def self.parse!
       config = new
@@ -161,6 +169,11 @@ def self.parse!
           config.source_directory = Pathname(source_directory)
         end
 
+        opt.on('t', '--template-directory DIRPATH', 'The directory that ' \
+               'contains the mustache templates to use') do |template_directory|
+          config.template_directory = Pathname(template_directory)
+        end
+
         opt.on('--readme FILEPATH',
                'The path to a markdown README file') do |readme|
           config.readme_path = Pathname(readme)
diff --git a/lib/jazzy/doc.rb b/lib/jazzy/doc.rb
index 45fcbb892..3288ba33a 100755
--- a/lib/jazzy/doc.rb
+++ b/lib/jazzy/doc.rb
@@ -1,11 +1,12 @@
 require 'date'
 require 'pathname'
+require 'mustache'
 
 require 'jazzy/gem_version'
 
 module Jazzy
   class Doc < Mustache
-    self.template_path = Pathname(__FILE__).parent.parent
+    self.template_name = 'doc'
 
     def date
       # Fake date is used to keep integration tests consistent
diff --git a/lib/jazzy/doc.mustache b/lib/jazzy/templates/doc.mustache
similarity index 89%
rename from lib/jazzy/doc.mustache
rename to lib/jazzy/templates/doc.mustache
index 336e5629d..454c91b90 100755
--- a/lib/jazzy/doc.mustache
+++ b/lib/jazzy/templates/doc.mustache
@@ -13,7 +13,7 @@
     <a name="//apple_ref/swift/{{dash_type}}/{{name}}" class="dashAnchor"></a>
     {{/dash_type}}
     <a title="{{name}} {{kind}} Reference"></a>
-    {{> jazzy/partials/header}}
+    {{> header}}
     <div class="content-wrapper">
       <p id="breadcrumbs">
         <a href="{{path_to_root}}index.html">{{module_name}} Reference</a>
@@ -22,16 +22,16 @@
       </p>
     </div>
     <div class="content-wrapper">
-      {{> jazzy/partials/nav}}
+      {{> nav}}
       <article class="main-content">
         <section>
           <section class="section">
             {{^hide_name}}<h1>{{name}}</h1>{{/hide_name}}
             {{{overview}}}
           </section>
-          {{> jazzy/partials/tasks}}
+          {{> tasks}}
         </section>
-        {{> jazzy/partials/footer}}
+        {{> footer}}
       </article>
     </div>
   </body>
diff --git a/lib/jazzy/partials/footer.mustache b/lib/jazzy/templates/footer.mustache
similarity index 100%
rename from lib/jazzy/partials/footer.mustache
rename to lib/jazzy/templates/footer.mustache
diff --git a/lib/jazzy/partials/header.mustache b/lib/jazzy/templates/header.mustache
similarity index 100%
rename from lib/jazzy/partials/header.mustache
rename to lib/jazzy/templates/header.mustache
diff --git a/lib/jazzy/partials/nav.mustache b/lib/jazzy/templates/nav.mustache
similarity index 100%
rename from lib/jazzy/partials/nav.mustache
rename to lib/jazzy/templates/nav.mustache
diff --git a/lib/jazzy/partials/parameter.mustache b/lib/jazzy/templates/parameter.mustache
similarity index 100%
rename from lib/jazzy/partials/parameter.mustache
rename to lib/jazzy/templates/parameter.mustache
diff --git a/lib/jazzy/partials/task.mustache b/lib/jazzy/templates/task.mustache
similarity index 97%
rename from lib/jazzy/partials/task.mustache
rename to lib/jazzy/templates/task.mustache
index b3bee63d9..2a4526d8d 100755
--- a/lib/jazzy/partials/task.mustache
+++ b/lib/jazzy/templates/task.mustache
@@ -45,7 +45,7 @@
             <table class="graybox">
               <tbody>
                 {{#parameters}}
-                {{> jazzy/partials/parameter}}
+                {{> parameter}}
                 {{/parameters}}
               </tbody>
             </table>
diff --git a/lib/jazzy/partials/tasks.mustache b/lib/jazzy/templates/tasks.mustache
similarity index 80%
rename from lib/jazzy/partials/tasks.mustache
rename to lib/jazzy/templates/tasks.mustache
index 00b635c03..f97bb437f 100755
--- a/lib/jazzy/partials/tasks.mustache
+++ b/lib/jazzy/templates/tasks.mustache
@@ -1,7 +1,7 @@
 {{#tasks.count}}
 <section class="section task-group-section">
   {{#tasks}}
-  {{> jazzy/partials/task}}
+  {{> task}}
   {{/tasks}}
 </section>
 {{/tasks.count}}

From f68bbe38ca952460183d732bf85eb9e48b8e1d11 Mon Sep 17 00:00:00 2001
From: JP Simard <jp@jpsim.com>
Date: Tue, 7 Jul 2015 11:47:48 -0700
Subject: [PATCH 2/2] updated integration specs

---
 spec/integration_specs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/spec/integration_specs b/spec/integration_specs
index 7c6dff7b0..4ca72204c 160000
--- a/spec/integration_specs
+++ b/spec/integration_specs
@@ -1 +1 @@
-Subproject commit 7c6dff7b0a843c6f619fb59cea97bf35c26f5244
+Subproject commit 4ca72204c624aaea036e486582e36a3529add83a