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

@@ -4,18 +4,26 @@
   module FigaroYml
     module Helpers
 
       def local_figaro_yml(env)
         @local_figaro_yml ||= YAML.load_file(figaro_yml_local_path)
-        @local_figaro_yml[env]
+        local_figaro = {}
+
+        @local_figaro_yml.each do |key, value|
+          if key == env or !value.is_a?(Hash)
+            local_figaro[key] = @local_figaro_yml[key]
+          end
+        end
+
+        local_figaro
       end
 
       def figaro_yml_env
         fetch(:figaro_yml_env).to_s
       end
 
       def figaro_yml_content
-        { figaro_yml_env => local_figaro_yml(figaro_yml_env) }.to_yaml
+        local_figaro_yml(figaro_yml_env).to_yaml
       end
 
       # error helpers
 
       def check_git_tracking_error