Merge pull request #14 from agios/erb_support

Add support for YAML files that contain ERB
sgruhier · Sep 23, 2013 · 5ecffdb · 5ecffdb
2 parents 37f1a50 + 1be50b9
commit 5ecffdb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/capistrano-db-tasks/database.rb b/lib/capistrano-db-tasks/database.rb
@@ -61,7 +61,7 @@ def initialize(cap_instance)
       @cap.run("cat #{@cap.current_path}/config/database.yml") do |c, s, d|
         @config += d
       end
-      @config = YAML.load(@config)[@cap.rails_env.to_s]
+      @config = YAML.load(ERB.new(@config).result)[@cap.local_rails_env.to_s]
     end
 
     def dump
@@ -86,7 +86,7 @@ def load(file, cleanup)
   class Local < Base
     def initialize(cap_instance)
       super(cap_instance)
-      @config = YAML.load_file(File.join('config', 'database.yml'))[@cap.local_rails_env.to_s]
+      @config = YAML.load(ERB.new(File.read(File.join('config', 'database.yml'))).result)[@cap.local_rails_env.to_s]
       puts "local #{@config}"
     end