From 5fa20bc3081232c03070a0d3ad8c3f848dd706a0 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Rapha=C3=ABl=20Pinson?= <raphael.pinson@camptocamp.com>
Date: Wed, 27 May 2015 15:14:39 +0200
Subject: [PATCH] Allow multiple facts in filter

---
 lib/puppet/catalog-diff/searchfacts.rb | 21 ++++++++++++---------
 1 file changed, 12 insertions(+), 9 deletions(-)

diff --git a/lib/puppet/catalog-diff/searchfacts.rb b/lib/puppet/catalog-diff/searchfacts.rb
index c11b114..2f441a8 100644
--- a/lib/puppet/catalog-diff/searchfacts.rb
+++ b/lib/puppet/catalog-diff/searchfacts.rb
@@ -4,14 +4,13 @@
 module Puppet::CatalogDiff
   class SearchFacts
 
-    def initialize(args)
-      @args = args
+    def initialize(facts)
+      @facts = Hash[facts.split(',').map { |f| f.split('=') }]
     end
 
     def find_nodes(options = {})
      # Pull all nodes from the yaml cache
      # Then validate they are active nodes against the rest of puppetdb api
-     yaml_cache = find_nodes_local(*@args.split("="))
      if options[:use_puppetdb]
        active_nodes = find_nodes_puppetdb()
      else
@@ -21,13 +20,14 @@ def find_nodes(options = {})
        raise "No active nodes were returned from your fact search"
      end
      if options[:filter_local]
+       yaml_cache = find_nodes_local()
        yaml_cache.select { |node| active_nodes.include?(node) }
      else
        active_nodes
      end
     end
 
-    def find_nodes_local(fact,value)
+    def find_nodes_local
       Puppet[:clientyamldir] = Puppet[:yamldir]
       if Puppet::Node.respond_to? :terminus_class
         Puppet::Node.terminus_class = :yaml
@@ -36,7 +36,9 @@ def find_nodes_local(fact,value)
         Puppet::Node.indirection.terminus_class = :yaml
         nodes = Puppet::Node.indirection.search("*")
       end
-      unless filtered =  nodes.select {|n| n.parameters[fact] == value }.map{ |n| n.name }
+      unless filtered =  nodes.select {|n|
+          @facts.select { |f, v| n[f] == v }.size == @facts.size
+        }.map{ |n| n.name }
         raise "No matching nodes found using yaml terminus"
       end
       filtered
@@ -44,7 +46,8 @@ def find_nodes_local(fact,value)
 
 
     def find_nodes_rest(server)
-        endpoint = "/v2/facts_search/search?facts.#{@args}"
+        query = @facts.map { |k, v| "facts.#{k}=#{v}" }.join('&')
+        endpoint = "/v2/facts_search/search?#{query}"
 
         begin
           connection = Puppet::Network::HttpPool.http_instance(server,'8140')
@@ -63,9 +66,9 @@ def find_nodes_rest(server)
 
     def find_nodes_puppetdb()
         connection = Puppet::Network::HttpPool.http_instance(Puppet::Util::Puppetdb.server,'8081')
-        fact_query = @args.split("=")
-        #json_query = URI.escape(["=", ["fact", fact_query[0]], fact_query[1]].to_json)
-        json_query = URI.escape(["=", ["node","active"], true].to_json)
+        base_query = ["and", ["=", ["node","active"], true]]
+        query = base_query.concat(@facts.map { |k, v| ["=", ["fact", k], v] })
+        json_query = URI.escape(query.to_json)
         unless filtered = PSON.load(connection.request_get("/v2/nodes/?query=#{json_query}", {"Accept" => 'application/json'}).body)
           raise "Error parsing json output of puppet search"
         end