lib/capistrano/figaro_yml/helpers.rb in capistrano-figaro-yml-1.0.2 vs lib/capistrano/figaro_yml/helpers.rb in capistrano-figaro-yml-1.0.3

@@ -3,14 +3,17 @@
 module Capistrano
   module FigaroYml
     module Helpers
 
       def local_figaro_yml(env)
-        @local_figaro_yml ||= YAML.load_file(figaro_yml_local_path)
+        @local_figaro_yml ||= YAML.load(ERB.new(File.read(figaro_yml_local_path)).result)
         local_figaro = {}
+        deployment_env = fetch(:rails_env, env)
 
         @local_figaro_yml.each do |key, value|
-          if key == env or !value.is_a?(Hash)
+          if key == env
+            local_figaro[deployment_env] = @local_figaro_yml[key]
+          elsif !value.is_a?(Hash)
             local_figaro[key] = @local_figaro_yml[key]
           end
         end
 
         local_figaro