lib/negroku/tasks/unicorn.rake in negroku-2.0.0.pre2 vs lib/negroku/tasks/unicorn.rake in negroku-2.0.0.pre3

@@ -58,15 +58,18 @@
   namespace :unicorn do
     desc "Upload unicorn configuration file"
     task :setup do
       on release_roles fetch(:unicorn_roles) do
         within "#{shared_path}/config" do
-          config_file = fetch(:unicorn_template)
-          unless File.exists?(config_file)
-            config_file = File.expand_path("../../templates/tasks/unicorn_#{fetch(:unicorn_template_type)}.rb.erb", __FILE__)
+          template_path = fetch(:unicorn_template)
+
+          # user a build in template if the template doesn't exists in the project
+          unless File.exists?(template_path)
+            template_path = "tasks/unicorn_#{fetch(:unicorn_template_type)}.rb.erb"
           end
-          config = ERB.new(File.read(config_file)).result(binding)
-          upload! StringIO.new(config), '/tmp/unicorn.rb'
+
+          config = build_template(template_path, nil, binding)
+          upload! config, '/tmp/unicorn.rb'
 
           execute :mv, '/tmp/unicorn.rb', 'unicorn.rb'
         end
       end
     end