load template.rb files from local templates

2012-03-03 16:52:49 -08:00 · 2012-03-03 16:52:49 -08:00 · c2f498e26e
parent 70dc453c25
commit c2f498e26e
1 changed files with 8 additions and 1 deletions
--- a/middleman-core/lib/middleman-core/templates/local.rb
+++ b/middleman-core/lib/middleman-core/templates/local.rb
@ -17,5 +17,12 @@ end
 # Iterate over the directories in the templates path and register each one.
 Dir[File.join(Middleman::Templates::Local.source_root, "*")].each do |dir|
  next unless File.directory?(dir)
-  Middleman::Templates.register(File.basename(dir).to_sym, Middleman::Templates::Local)
+  
+  template_file = File.join(dir, "template.rb")
+  
+  if File.exists?(template_file)
+    require template_file
+  else
+    Middleman::Templates.register(File.basename(dir).to_sym, Middleman::Templates::Local)
+  end
 end