Possible layering for subscription rules

katello-attach-subscription

@@ -837,7 +837,11 @@ def vdcupdate()
     end
     system_details[system['name']]=sys
     # check if the type requested match the host one
-    system_type = KatelloAttachSubscription::FactAnalyzer.system_type(sys)
+    if sys.has_key?('facts')
+      system_type = KatelloAttachSubscription::FactAnalyzer.system_type(sys, sys['facts'])
+    else
+      system_type = KatelloAttachSubscription::FactAnalyzer.system_type_from_host(sys)
+    end
     # for each item in yaml extract sub
     if @options[:debug]
       puts " DEBUG: YAML dump with all definitions for the current system"
@@ -902,8 +906,17 @@ def vdcupdate()
       if @options[:verbose]
         puts "VERBOSE: Start testing facts value for #{sys['name']}"
       end
-      fact_test_passed = KatelloAttachSubscription::HostMatcher.match_host(sys, sub)
+
+      if sub.has_key?('facts')
+        fact_test_passed = KatelloAttachSubscription::HostMatcher.match_host(sys, sub)
+      else
+        if @options[:verbose]
+          puts "  VERBOSE: Fact test passed as no test on it"
+        end
+        fact_test_passed = true
+      end
       unless fact_test_passed
+
         if @options[:verbose]
           puts "  VERBOSE: Host #{sys['name']} doesn't pass facts test. Skip to the next sub entry."
         end
@@ -1093,11 +1106,15 @@ def vdcupdate()
         puts " DEBUG: System '#{system['name']}' in scope, proceeding with assignment of variables"
       end
       # set the desidered subscription to be associated
+      sub_layer = sub['sub_layer'] || "stop_parsing"
+      if @options[:debug]
+        puts " DEBUG: Sub Layer of this sub entry: #{sub_layer}"
+      end
       if sub.has_key?('sub_parsed')
         if has_derived_sub
-          desired_sub_hash = derived_sub
+          desired_sub_hash = KatelloAttachSubscription::Utils.merge_subs(desired_sub_hash, derived_sub, sub_layer)
         else
-          desired_sub_hash = sub['sub_parsed']
+          desired_sub_hash = KatelloAttachSubscription::Utils.merge_subs(desired_sub_hash, sub['sub_parsed'], sub_layer)
         end
       end
       # if "remove_other" has been set, set the flag
@@ -1121,8 +1138,10 @@ def vdcupdate()
         keep_virt_only = sub['keep_virt_only']
       end
 
-      # if the system is found, stop cyclyng over yaml,
-      break
+      # if the system is found and sub_layer is "stop_parsing", stop cyclyng over yaml
+      if sub_layer == "stop_parsing"
+        break
+      end
 
     end
     if skip_host
@@ -1598,7 +1617,8 @@ end
 # count the total number of subs available, consumed and free
 def getsubdetailfor(sub_name)
   detail_hash = {"consumed" => 0, "free" => 0}
-  search_options = KatelloAttachSubscription::Utils.search_args(name: sub_name)
+  search_params = 'name="'+sub_name+'"'
+  search_options = KatelloAttachSubscription::Utils.search_args(search_params)
   parsed_subscription = fetch_all_results(:subscriptions,:index,{:search => search_options, :available_for => "host"})
   detail_hash["available"] = getavailablesubfor(parsed_subscription)
   detail_hash["consumed"] = getconsumedsubfor(parsed_subscription)

lib/katello_attach_subscription/utils.rb

@@ -14,6 +14,25 @@ def self.search_args(search)
         search
       end
     end
+
+    def self.merge_subs(current_sub, sub_to_merge, command)
+      if not current_sub.nil?
+        current = current_sub.clone
+      else
+        current = {}
+      end
+      new_sub = sub_to_merge.clone
+      case command
+      when "override"
+        current = new_sub
+      else
+        if current.nil?
+          current = new_sub
+        else
+          current.merge!(new_sub)
+        end
+      end
+      return current.clone
+    end
   end
 end
-

test/fixtures/merge_sub_file.yaml

@@ -0,0 +1,110 @@
+subs:
+-
+  hostname: .*
+  type: Hypervisor
+  skip_density: true
+  facts:
+    -
+      name: hypervisor::cluster
+      value: example_cluster
+      matcher: regexp
+  sub:
+    rhel:
+      - name = "Red Hat Enterprise Linux for Virtual Datacenters, Premium"
+      - name = "Red Hat Enterprise Linux for Virtual Datacenters, Standard"
+    els:
+      - name = "Red Hat Enterprise Linux Extended Life Cycle Support (Unlimited Guests)"
+    smartmanagement:
+      - name = "Smart Management for Unlimited Guests"
+  sub_parsed:
+    rhel:
+      - d2e16e6ef52a45b7468f1da56bba1953
+      - e78f5438b48b39bcbdea61b73679449d
+      - a98931d104a7fb8f30450547d97e7ca5
+    els:
+      - 7f9a983a540e00931a69382161bdd265
+    smartmanagement:
+      - 439a7d9b0548adbedcce838e37e84ba1
+-
+  hostname: .*
+  type: Hypervisor
+  sub:
+    rhel:
+      - name = "Red Hat Enterprise Linux for Virtual Datacenters, Premium"
+      - name = "Red Hat Enterprise Linux for Virtual Datacenters, Standard"
+    els:
+      - name = "Red Hat Enterprise Linux Extended Life Cycle Support (Unlimited Guests)"
+    smartmanagement:
+      - name = "Smart Management for Unlimited Guests"
+  sub_parsed:
+    rhel:
+      - d2e16e6ef52a45b7468f1da56bba1953
+      - e78f5438b48b39bcbdea61b73679449d
+      - a98931d104a7fb8f30450547d97e7ca5
+    els:
+      - 7f9a983a540e00931a69382161bdd265
+    smartmanagement:
+      - 439a7d9b0548adbedcce838e37e84ba1
+-
+  hostname: .*
+  type: Physical
+  sub_layer: keep_parsing
+  facts:
+    -
+      name: distribution::name
+      value: Server
+      matcher: regexp
+  sub:
+    rhel:
+      - name = "Red Hat Enterprise Linux Server, Standard (Physical or Virtual Nodes)"
+      - name = "Red Hat Enterprise Linux Server, Premium (Physical or Virtual Nodes)"
+    smartmanagement:
+      - name = "Smart Management"
+  sub_parsed:
+    rhel:
+      - b1a5d251fa4fe598cb947ffc42b9cbed
+      - 1337d38747e659ed836548ae6cda7cc2
+    smartmanagement:
+      - f180623caa42379bc4518d06c9c9be05
+-
+  hostname: .*
+  type: Physical
+  sub_layer: keep_parsing
+  facts:
+    -
+      name: distribution::name
+      value: Server
+      matcher: regexp
+    -
+      name: distribution::version
+      value: <6.0
+      matcher: version
+  sub:
+    els:
+      - name = "Red Hat Enterprise Linux Extended Life Cycle Support (Physical or Virtual Nodes)"
+  sub_parsed:
+    els:
+      - 523af537946b79c4f8369ed39ba78605
+-
+  hostname: .*
+  type: Physical
+  sub_layer: override
+  facts:
+    -
+      name: distribution::name
+      value: Client|Workstation
+      matcher: regexp
+  sub:
+    rhel:
+      - name = "Red Hat Enterprise Linux Server for HPC Compute Node, Self-support (1-2 sockets) (Up to 1 guest)"
+    smartmanagement:
+      - name = "Smart Management for Red Hat Enterprise Linux Server for HPC Compute Node (Up to 1 guest)"
+  sub_parsed:
+    rhel:
+      - bb98d4e9c281b175ea84c517b59308ea
+    smartmanagement:
+      - af03af10d57b7b17f26a0130562d6b6e
+-
+  hostname: .*
+  type: Guest
+  auto_attach: true

test/sub_merger_test.rb

@@ -0,0 +1,117 @@
+#!/usr/bin/env ruby
+require 'yaml'
+require_relative '../lib/katello_attach_subscription'
+
+class SubMergerTest
+
+  DUMMY_KEEP_PARSE = {
+    "type" => "Physical",
+    "host" => {
+      'name' => 'physical_example',
+      'facts' => {
+        "distribution::name" => "Red Hat Server",
+        "distribution::version" => "6.1"
+      }
+    }
+  }
+
+  DUMMY_MERGE = {
+    "type" => "Physical",
+    "host" => {
+      'name' => 'physical_example',
+      'facts' => {
+        "distribution::name" => "Red Hat Client",
+        "distribution::version" => "5.2"
+      }
+    }
+  }
+
+  DUMMY_OVERRIDE = {
+    "type" => "Physical",
+    "host" => {
+      'name' => 'physical_example',
+      'facts' => {
+        "distribution::name" => "Red Hat Client"
+      }
+    }
+  }
+
+  DUMMY_STOP = {
+    "type" => "Hypervisor",
+    "host" => {
+      'name' => 'physical_example',
+      'facts' => {
+        'hypervisor::cluster' => 'example_cluster'
+      }
+    }
+  }
+
+  DUMMY_NORMAL = {
+    "type" => "Hypervisor",
+    "host" => {
+      'name' => 'physical_example',
+      'facts' => {}
+    }
+  }
+
+  def test_normal_parse
+    yaml_sub = readSubsFromYAML()
+    parsed_sub = getSubForHost(yaml_sub, DUMMY_NORMAL)
+    puts "NORMAL PARSE"
+    puts "Expected Sub:"
+    p parsed_sub
+  end
+
+  def test_stop_parse
+    yaml_sub = readSubsFromYAML()
+    parsed_sub = getSubForHost(yaml_sub, DUMMY_STOP)
+    puts "PARSE STOP"
+    puts "Expected Sub:"
+    p parsed_sub
+  end
+
+  def test_overdrive_parse
+    yaml_sub = readSubsFromYAML()
+    parsed_sub = getSubForHost(yaml_sub, DUMMY_OVERRIDE)
+    puts "PARSE OVERDRIVE"
+    puts "Expected Sub:"
+    p parsed_sub
+  end
+
+  def test_merge_parse
+    yaml_sub = readSubsFromYAML()
+    parsed_sub = getSubForHost(yaml_sub, DUMMY_MERGE)
+    puts "PARSE OVERDRIVE"
+    puts "Expected Sub:"
+    p parsed_sub
+  end
+
+  def test_keep_parse
+    yaml_sub = readSubsFromYAML()
+    parsed_sub = getSubForHost(yaml_sub, DUMMY_KEEP_PARSE)
+    puts "PARSE OVERDRIVE"
+    puts "Expected Sub:"
+    p parsed_sub
+  end
+
+  def readSubsFromYAML
+    yaml_file = YAML.load_file('fixtures/merge_sub_file.yaml')
+    return yaml_file['sub']
+  end
+
+  def getSubForHost(subs, host)
+    parsed_final = nil
+    subs.each do |single_sub|
+      parsed = single_sub['parsed_sub']
+      if KatelloAttachSubscription::HostMatcher.match_host(host, single_sub)
+        layer_command = single_sub['sub_layer'] || 'stop_parsing'
+        parsed_final = KatelloAttachSubscription::Utils.merge_subs(parsed_final, parsed, layer_command)
+        if layer_command == 'stop_parsing'
+          break
+        end
+      end
+    end
+    return parsed_final.clone
+  end
+
+end