lib/humidifier/reservoir/config.rb in humidifier-reservoir-0.1.0 vs lib/humidifier/reservoir/config.rb in humidifier-reservoir-0.2.0

@@ -12,11 +12,11 @@
       def files_for(name)
         raise Error, 'You must configure a stack path' if stack_path.nil?
         Dir["#{stack_path}/#{name}/*.yml"]
       end
 
-      def map(type, opts = {})
-        mappings[type.to_sym] = Mapping.new(opts)
+      def map(type, opts = {}, &block)
+        mappings[type.to_sym] = Mapping.new(opts, &block)
       end
 
       def mapping_for(type)
         mappings[type.to_sym]
       end