Merge pull request #157 from mikkergimenez/mikkergp/allow_custom_temp…

…lates_in_control_repo Add functionality to allow custom templates in control repo, this all…
voxpupuli · Mar 6, 2018 · baa6be4 · baa6be4
2 parents e11d5c2 + 26b9c3e
commit baa6be4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/lib/onceover/controlrepo.rb b/lib/onceover/controlrepo.rb
@@ -537,7 +537,12 @@ def self.create_dirs_and_log(dir)
     def self.evaluate_template(template_name, bind)
       logger.debug "Evaluating template #{template_name}"
       template_dir = File.expand_path('../../templates', File.dirname(__FILE__))
-      template = File.read(File.expand_path("./#{template_name}", template_dir))
+      if File.file?(File.expand_path("./spec/templates/#{template_name}", @root))
+        puts "Using Custom #{template_name}"
+        template = File.read(File.expand_path("./spec/templates/#{template_name}", @root))
+      else
+        template = File.read(File.expand_path("./#{template_name}", template_dir))
+      end
       ERB.new(template, nil, '-').result(bind)
     end