lib/outliers/collection.rb in outliers-0.1.1 vs lib/outliers/collection.rb in outliers-0.2.0

@@ -15,15 +15,23 @@
 
     def self.verifications
        Outliers::Verifications::Shared.verifications + self.resource_class.verifications 
     end
 
+    def self.filters
+      []
+    end
+
     def self.resource_class
       array = self.to_s.gsub(/Collection$/, '').split('::')
       array.inject(Object) {|o,c| o.const_get c}
     end
 
+    def to_s
+      self.class.to_human
+    end
+
     def initialize(provider)
       @targets   = []
       @provider = provider
       @logger   = Outliers.logger
     end
@@ -38,17 +46,43 @@
       @logger.info "Excluding the following resources: '#{exclusions.join(',')}'."
       save = all.reject {|u| exclusions.include? u.public_send key}
       @all = save
     end
 
+    def filter(args)
+      name  = args.keys.first
+      value = args.fetch name
+
+      logger.info "Applying filter '#{name}' with value '#{value}'."
+
+      unless self.public_methods.include? "filter_#{name}".to_sym
+        raise Exceptions::UnknownFilter.new "Unknown filter '#{name}'."
+      end
+
+      filtered_list = self.public_send "filter_#{name}", value
+
+      logger.warn "No resources match filter." unless filtered_list.any?
+
+      @all = filtered_list
+    end
+
     def verify(name, arguments={})
       name << "?" unless name =~ /^.*\?$/
 
-      logger.debug "Verifying resources '#{all_by_key.join(', ')}'."
+      unless all.any?
+        return { failing_resources: [], passing_resources: [] }
+      end
 
+      logger.info "Verifying '#{name}'."
+      logger.debug "Target resources '#{all_by_key.join(', ')}'."
+
+      unless verification_exists? name
+        raise Exceptions::UnknownVerification.new "Unkown verification '#{name}'."
+      end
+
       if collection_verification? name
-        send_verification self, name, arguments
+        send_collection_verification name, arguments
       else
         send_resources_verification name, arguments
       end
     end
 
@@ -64,10 +98,17 @@
       self.class.resource_class
     end
 
     private
 
+    def verification_exists?(name)
+      m = resource_class.instance_methods - resource_class.class.instance_methods
+      m += Outliers::Verifications::Shared.instance_methods
+      m -= [:source, :id, :method_missing]
+      m.include? name.to_sym
+    end
+
     def all_by_key
       all.map {|r| r.public_send key}
     end
 
     def connect
@@ -80,21 +121,10 @@
 
     def collection_verification?(name)
       self.public_methods.include? name.to_sym
     end
 
-    def send_resources_verification(verification, arguments)
-      set_target_resources verification if targets.any?
-
-      results = map do |resource|
-        result = send_verification resource, verification, arguments
-        logger.debug "Verification of resource '#{resource.id}' #{result ? 'passed' : 'failed'}."
-        result
-      end
-      !results.include? false
-    end
-
     def set_target_resources(verification)
       logger.info "Verifying target '#{targets.join(', ')}'."
 
       @all = all.select {|r| targets.include? r.id }
 
@@ -103,15 +133,30 @@
       end
 
       @all
     end
 
+    def send_resources_verification(verification, arguments)
+      set_target_resources verification if targets.any?
+
+      failing_resources = reject do |resource|
+        result = send_verification resource, verification, arguments
+        logger.debug "Verification of resource '#{resource.id}' #{result ? 'passed' : 'failed'}."
+        result
+      end
+      { failing_resources: failing_resources, passing_resources: all - failing_resources }
+    end
+
+    def send_collection_verification(verification, arguments)
+      failing_resources = send_verification(self, verification, arguments)
+      { failing_resources: failing_resources, passing_resources: all - failing_resources }
+    end
+
     def send_verification(object, verification, arguments) 
       if object.method(verification).arity.zero?
         if arguments.any?
           raise Outliers::Exceptions::NoArgumentRequired.new "Verification '#{verification}' does not require an arguments."
         end
-
         object.public_send verification
       else
         if arguments.none?
           raise Outliers::Exceptions::ArgumentRequired.new "Verification '#{verification}' requires arguments."
         end