Add support for original erb templates

tokzk · Jan 13, 2017 · 99578ff · 99578ff
1 parent ba6d99b
commit 99578ff
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -44,6 +44,7 @@ jekyll-amazon:
                 # Affiliate account (e.g. `'us'` for United States).
                 # default: 'us'
                 # 'br'/'ca'/'cn'/'de'/'es'/'fr'/'in'/'it'/'jp'/'mx/'uk'/'us'
+  template_dir: '_templates' # original template directory
 ```
 
 

diff --git a/lib/jekyll-amazon/amazon_tag.rb b/lib/jekyll-amazon/amazon_tag.rb
@@ -120,6 +120,7 @@ def setup(context)
         @site   = context.registers[:site]
         @config = @site.config['jekyll-amazon'] || {}
         country = @config['country'] || DEFAULT_COUNTRY
+        @template_dir = @config['template_dir']
         AmazonResultCache.instance.setup(country)
       end
 
@@ -138,7 +139,11 @@ def parse_options(markup)
       end
 
       def render_from_file(type, item)
-        file = File.expand_path("../../templates/#{type}.erb", __dir__)
+        if @template_dir
+          template_file = File.expand_path("#{type}.erb", @template_dir)
+          file = template_file if File.exist? template_file
+        end
+        file ||= File.expand_path("../../templates/#{type}.erb", __dir__)
         ERB.new(open(file).read).result(binding)
       end