lib/maintain/maintainer.rb in maintain-0.2.21 vs lib/maintain/maintainer.rb in maintain-0.2.22

@@ -22,11 +22,11 @@
           end
         EOC
       end
       # Now define the state
       if back_end
-        back_end.aggregate(maintainee, name, @attribute, conditions.map{|value| states[value][:value].is_a?(Symbol) ? states[value][:value].to_s : states[value][:value] }, {:force => options[:force]})
+        back_end.aggregate(maintainee, name, @attribute, conditions.select{|value| states.has_key?(value) }.map{|value| states[value][:value].is_a?(Symbol) ? states[value][:value].to_s : states[value][:value] }.compact, {:force => options[:force]})
       end
     end
 
     def aggregates
       @aggregates ||= {}
@@ -154,9 +154,20 @@
       maintainee.class_eval <<-EOC
         def #{@attribute}_#{boolean_method}
           #{@attribute}.#{boolean_method}
         end
         #{"alias :#{boolean_method} :#{@attribute}_#{boolean_method}" if method_free?(boolean_method) || options[:force]}
+      EOC
+
+      # Last but not least, add bang methods to automatically convert to state. Like boolean
+      # methods above, these only get added if they're not already things that are things.
+      bang_method = "#{name}!"
+      # Override any attribute_state! methods
+      maintainee.class_eval <<-EOC
+        def #{@attribute}_#{bang_method}
+          #{@attribute}.#{bang_method}
+        end
+        #{"alias :#{bang_method} :#{@attribute}_#{bang_method}" if method_free?(bang_method) || options[:force]}
       EOC
     end
 
     def states
       @states ||= {}
\ No newline at end of file