Sha256: 0f81a798e47d27a54b1a93b8659b8f14caed994d4ec3caf4d5ecf8c002674432

Contents?: true

Size: 1.82 KB

Versions: 44

Compression:

Stored size: 1.82 KB

Contents

require 'puppet/node/inventory'
require 'puppet/indirector/yaml'

class Puppet::Node::Inventory::Yaml < Puppet::Indirector::Yaml
  desc "Return node names matching the fact query"

  # Return the path to a given node's file.
  def yaml_dir_path
    base = Puppet.run_mode.master? ? Puppet[:yamldir] : Puppet[:clientyamldir]
    File.join(base, 'facts', '*.yaml')
  end

  def node_matches?(facts, options)
    options.each do |key, value|
      type, name, operator = key.to_s.split(".")
      operator ||= 'eq'

      return false unless node_matches_option?(type, name, operator, value, facts)
    end
    return true
  end

  def search(request)
    node_names = []
    Dir.glob(yaml_dir_path).each do |file|
      facts = YAML.load_file(file)
      node_names << facts.name if node_matches?(facts, request.options)
    end
    node_names
  end

  private

  def node_matches_option?(type, name, operator, value, facts)
    case type
    when "meta"
      case name
      when "timestamp"
        compare_timestamp(operator, facts.timestamp, Time.parse(value))
      end
    when "facts"
      compare_facts(operator, facts.values[name], value)
    end
  end

  def compare_facts(operator, value1, value2)
    return false unless value1

    case operator
    when "eq"
      value1.to_s == value2.to_s
    when "le"
      value1.to_f <= value2.to_f
    when "ge"
      value1.to_f >= value2.to_f
    when "lt"
      value1.to_f < value2.to_f
    when "gt"
      value1.to_f > value2.to_f
    when "ne"
      value1.to_s != value2.to_s
    end
  end

  def compare_timestamp(operator, value1, value2)
    case operator
    when "eq"
      value1 == value2
    when "le"
      value1 <= value2
    when "ge"
      value1 >= value2
    when "lt"
      value1 < value2
    when "gt"
      value1 > value2
    when "ne"
      value1 != value2
    end
  end
end

Version data entries

44 entries across 44 versions & 4 rubygems

Version Path
puppet-parse-0.1.4 lib/vendor/puppet/indirector/inventory/yaml.rb
puppet-parse-0.1.3 lib/vendor/puppet/indirector/inventory/yaml.rb
puppet-parse-0.1.2 lib/vendor/puppet/indirector/inventory/yaml.rb
puppet-parse-0.1.1 lib/vendor/puppet/indirector/inventory/yaml.rb
puppet-2.7.26 lib/puppet/indirector/inventory/yaml.rb
puppet-2.7.25 lib/puppet/indirector/inventory/yaml.rb
puppet-2.7.24 lib/puppet/indirector/inventory/yaml.rb
puppet-2.7.23 lib/puppet/indirector/inventory/yaml.rb
puppet-2.7.22 lib/puppet/indirector/inventory/yaml.rb
puppet-parse-0.1.0 lib/vendor/puppet/indirector/inventory/yaml.rb
puppet-parse-0.0.6 lib/vendor/puppet/indirector/inventory/yaml.rb
puppet-2.7.21 lib/puppet/indirector/inventory/yaml.rb
puppet-parse-0.0.5 lib/vendor/puppet/indirector/inventory/yaml.rb
puppet-parse-0.0.4 lib/vendor/puppet/indirector/inventory/yaml.rb
puppet-parse-0.0.2 lib/vendor/puppet/indirector/inventory/yaml.rb
puppet-2.7.20 lib/puppet/indirector/inventory/yaml.rb
puppet-2.7.20.rc1 lib/puppet/indirector/inventory/yaml.rb
librarian-puppet-0.9.4 vendor/gems/ruby/1.8/gems/puppet-2.7.18/lib/puppet/indirector/inventory/yaml.rb
puppet-2.7.19 lib/puppet/indirector/inventory/yaml.rb
supply_drop-0.11.0 examples/vendored-puppet/vendor/puppet-2.7.8/lib/puppet/indirector/inventory/yaml.rb