supply_drop 0.5.0 → 0.6.0

data/examples/vendored-puppet/vendor/puppet-2.7.8/lib/puppet/indirector/indirection.rb

@@ -0,0 +1,317 @@
+require 'puppet/util/docs'
+require 'puppet/indirector/envelope'
+require 'puppet/indirector/request'
+
+# The class that connects functional classes with their different collection
+# back-ends.  Each indirection has a set of associated terminus classes,
+# each of which is a subclass of Puppet::Indirector::Terminus.
+class Puppet::Indirector::Indirection
+  include Puppet::Util::Docs
+
+  @@indirections = []
+
+  # Find an indirection by name.  This is provided so that Terminus classes
+  # can specifically hook up with the indirections they are associated with.
+  def self.instance(name)
+    @@indirections.find { |i| i.name == name }
+  end
+
+  # Return a list of all known indirections.  Used to generate the
+  # reference.
+  def self.instances
+    @@indirections.collect { |i| i.name }
+  end
+
+  # Find an indirected model by name.  This is provided so that Terminus classes
+  # can specifically hook up with the indirections they are associated with.
+  def self.model(name)
+    return nil unless match = @@indirections.find { |i| i.name == name }
+    match.model
+  end
+
+  attr_accessor :name, :model
+
+  attr_reader :termini
+
+  # Create and return our cache terminus.
+  def cache
+    raise(Puppet::DevError, "Tried to cache when no cache class was set") unless cache_class
+    terminus(cache_class)
+  end
+
+  # Should we use a cache?
+  def cache?
+    cache_class ? true : false
+  end
+
+  attr_reader :cache_class
+  # Define a terminus class to be used for caching.
+  def cache_class=(class_name)
+    validate_terminus_class(class_name) if class_name
+    @cache_class = class_name
+  end
+
+  # This is only used for testing.
+  def delete
+    @@indirections.delete(self) if @@indirections.include?(self)
+  end
+
+  # Set the time-to-live for instances created through this indirection.
+  def ttl=(value)
+    raise ArgumentError, "Indirection TTL must be an integer" unless value.is_a?(Fixnum)
+    @ttl = value
+  end
+
+  # Default to the runinterval for the ttl.
+  def ttl
+    @ttl ||= Puppet[:runinterval].to_i
+  end
+
+  # Calculate the expiration date for a returned instance.
+  def expiration
+    Time.now + ttl
+  end
+
+  # Generate the full doc string.
+  def doc
+    text = ""
+
+    text += scrub(@doc) + "\n\n" if @doc
+
+    if s = terminus_setting
+      text += "* **Terminus Setting**: #{terminus_setting}"
+    end
+
+    text
+  end
+
+  def initialize(model, name, options = {})
+    @model = model
+    @name = name
+    @termini = {}
+
+    @cache_class = nil
+    @terminus_class = nil
+
+    raise(ArgumentError, "Indirection #{@name} is already defined") if @@indirections.find { |i| i.name == @name }
+    @@indirections << self
+
+    if mod = options[:extend]
+      extend(mod)
+      options.delete(:extend)
+    end
+
+    # This is currently only used for cache_class and terminus_class.
+    options.each do |name, value|
+      begin
+        send(name.to_s + "=", value)
+      rescue NoMethodError
+        raise ArgumentError, "#{name} is not a valid Indirection parameter"
+      end
+    end
+  end
+
+  # Set up our request object.
+  def request(*args)
+    Puppet::Indirector::Request.new(self.name, *args)
+  end
+
+  # Return the singleton terminus for this indirection.
+  def terminus(terminus_name = nil)
+    # Get the name of the terminus.
+    raise Puppet::DevError, "No terminus specified for #{self.name}; cannot redirect" unless terminus_name ||= terminus_class
+
+    termini[terminus_name] ||= make_terminus(terminus_name)
+  end
+
+  # This can be used to select the terminus class.
+  attr_accessor :terminus_setting
+
+  # Determine the terminus class.
+  def terminus_class
+    unless @terminus_class
+      if setting = self.terminus_setting
+        self.terminus_class = Puppet.settings[setting].to_sym
+      else
+        raise Puppet::DevError, "No terminus class nor terminus setting was provided for indirection #{self.name}"
+      end
+    end
+    @terminus_class
+  end
+
+  def reset_terminus_class
+    @terminus_class = nil
+  end
+
+  # Specify the terminus class to use.
+  def terminus_class=(klass)
+    validate_terminus_class(klass)
+    @terminus_class = klass
+  end
+
+  # This is used by terminus_class= and cache=.
+  def validate_terminus_class(terminus_class)
+    raise ArgumentError, "Invalid terminus name #{terminus_class.inspect}" unless terminus_class and terminus_class.to_s != ""
+    unless Puppet::Indirector::Terminus.terminus_class(self.name, terminus_class)
+      raise ArgumentError, "Could not find terminus #{terminus_class} for indirection #{self.name}"
+    end
+  end
+
+  # Expire a cached object, if one is cached.  Note that we don't actually
+  # remove it, we expire it and write it back out to disk.  This way people
+  # can still use the expired object if they want.
+  def expire(key, *args)
+    request = request(:expire, key, *args)
+
+    return nil unless cache?
+
+    return nil unless instance = cache.find(request(:find, key, *args))
+
+    Puppet.info "Expiring the #{self.name} cache of #{instance.name}"
+
+    # Set an expiration date in the past
+    instance.expiration = Time.now - 60
+
+    cache.save(request(:save, instance, *args))
+  end
+
+  # Search for an instance in the appropriate terminus, caching the
+  # results if caching is configured..
+  def find(key, *args)
+    request = request(:find, key, *args)
+    terminus = prepare(request)
+
+    if result = find_in_cache(request)
+      return result
+    end
+
+    # Otherwise, return the result from the terminus, caching if appropriate.
+    if ! request.ignore_terminus? and result = terminus.find(request)
+      result.expiration ||= self.expiration if result.respond_to?(:expiration)
+      if cache? and request.use_cache?
+        Puppet.info "Caching #{self.name} for #{request.key}"
+        cache.save request(:save, result, *args)
+      end
+
+      return terminus.respond_to?(:filter) ? terminus.filter(result) : result
+    end
+
+    nil
+  end
+
+  # Search for an instance in the appropriate terminus, and return a
+  # boolean indicating whether the instance was found.
+  def head(key, *args)
+    request = request(:head, key, *args)
+    terminus = prepare(request)
+
+    # Look in the cache first, then in the terminus.  Force the result
+    # to be a boolean.
+    !!(find_in_cache(request) || terminus.head(request))
+  end
+
+  def find_in_cache(request)
+    # See if our instance is in the cache and up to date.
+    return nil unless cache? and ! request.ignore_cache? and cached = cache.find(request)
+    if cached.expired?
+      Puppet.info "Not using expired #{self.name} for #{request.key} from cache; expired at #{cached.expiration}"
+      return nil
+    end
+
+    Puppet.debug "Using cached #{self.name} for #{request.key}"
+    cached
+  rescue => detail
+    puts detail.backtrace if Puppet[:trace]
+    Puppet.err "Cached #{self.name} for #{request.key} failed: #{detail}"
+    nil
+  end
+
+  # Remove something via the terminus.
+  def destroy(key, *args)
+    request = request(:destroy, key, *args)
+    terminus = prepare(request)
+
+    result = terminus.destroy(request)
+
+    if cache? and cached = cache.find(request(:find, key, *args))
+      # Reuse the existing request, since it's equivalent.
+      cache.destroy(request)
+    end
+
+    result
+  end
+
+  # Search for more than one instance.  Should always return an array.
+  def search(key, *args)
+    request = request(:search, key, *args)
+    terminus = prepare(request)
+
+    if result = terminus.search(request)
+      raise Puppet::DevError, "Search results from terminus #{terminus.name} are not an array" unless result.is_a?(Array)
+      result.each do |instance|
+        next unless instance.respond_to? :expiration
+        instance.expiration ||= self.expiration
+      end
+      return result
+    end
+  end
+
+  # Save the instance in the appropriate terminus.  This method is
+  # normally an instance method on the indirected class.
+  def save(instance, key = nil)
+    request = request(:save, key, instance)
+    terminus = prepare(request)
+
+    result = terminus.save(request)
+
+    # If caching is enabled, save our document there
+    cache.save(request) if cache?
+
+    result
+  end
+
+  private
+
+  # Check authorization if there's a hook available; fail if there is one
+  # and it returns false.
+  def check_authorization(request, terminus)
+    # At this point, we're assuming authorization makes no sense without
+    # client information.
+    return unless request.node
+
+    # This is only to authorize via a terminus-specific authorization hook.
+    return unless terminus.respond_to?(:authorized?)
+
+    unless terminus.authorized?(request)
+      msg = "Not authorized to call #{request.method} on #{request}"
+      msg += " with #{request.options.inspect}" unless request.options.empty?
+      raise ArgumentError, msg
+    end
+  end
+
+  # Setup a request, pick the appropriate terminus, check the request's authorization, and return it.
+  def prepare(request)
+    # Pick our terminus.
+    if respond_to?(:select_terminus)
+      unless terminus_name = select_terminus(request)
+        raise ArgumentError, "Could not determine appropriate terminus for #{request}"
+      end
+    else
+      terminus_name = terminus_class
+    end
+
+    dest_terminus = terminus(terminus_name)
+    check_authorization(request, dest_terminus)
+
+    dest_terminus
+  end
+
+  # Create a new terminus instance.
+  def make_terminus(terminus_class)
+    # Load our terminus class.
+    unless klass = Puppet::Indirector::Terminus.terminus_class(self.name, terminus_class)
+      raise ArgumentError, "Could not find terminus #{terminus_class} for indirection #{self.name}"
+    end
+    klass.new
+  end
+end

data/examples/vendored-puppet/vendor/puppet-2.7.8/lib/puppet/indirector/inventory/yaml.rb

@@ -0,0 +1,81 @@
+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

data/examples/vendored-puppet/vendor/puppet-2.7.8/lib/puppet/indirector/key/ca.rb

@@ -0,0 +1,12 @@
+require 'puppet/indirector/ssl_file'
+require 'puppet/ssl/key'
+
+class Puppet::SSL::Key::Ca < Puppet::Indirector::SslFile
+  desc "Manage the CA's private on disk.  This terminus *only* works
+    with the CA key, because that's the only key that the CA ever interacts
+    with."
+
+  store_in :privatekeydir
+
+  store_ca_at :cakey
+end

data/examples/vendored-puppet/vendor/puppet-2.7.8/lib/puppet/indirector/key/file.rb

@@ -0,0 +1,42 @@
+require 'puppet/indirector/ssl_file'
+require 'puppet/ssl/key'
+
+class Puppet::SSL::Key::File < Puppet::Indirector::SslFile
+  desc "Manage SSL private and public keys on disk."
+
+  store_in :privatekeydir
+  store_ca_at :cakey
+
+  # Where should we store the public key?
+  def public_key_path(name)
+    if ca?(name)
+      Puppet[:capub]
+    else
+      File.join(Puppet[:publickeydir], name.to_s + ".pem")
+    end
+  end
+
+  # Remove the public key, in addition to the private key
+  def destroy(request)
+    super
+
+    return unless FileTest.exist?(public_key_path(request.key))
+
+    begin
+      File.unlink(public_key_path(request.key))
+    rescue => detail
+      raise Puppet::Error, "Could not remove #{request.key} public key: #{detail}"
+    end
+  end
+
+  # Save the public key, in addition to the private key.
+  def save(request)
+    super
+
+    begin
+      Puppet.settings.writesub(:publickeydir, public_key_path(request.key)) { |f| f.print request.instance.content.public_key.to_pem }
+    rescue => detail
+      raise Puppet::Error, "Could not write #{request.key}: #{detail}"
+    end
+  end
+end

data/examples/vendored-puppet/vendor/puppet-2.7.8/lib/puppet/indirector/ldap.rb

@@ -0,0 +1,78 @@
+require 'puppet/indirector/terminus'
+require 'puppet/util/ldap/connection'
+
+class Puppet::Indirector::Ldap < Puppet::Indirector::Terminus
+  # Perform our ldap search and process the result.
+  def find(request)
+    ldapsearch(search_filter(request.key)) { |entry| return process(entry) } || nil
+  end
+
+  # Process the found entry.  We assume that we don't just want the
+  # ldap object.
+  def process(entry)
+    raise Puppet::DevError, "The 'process' method has not been overridden for the LDAP terminus for #{self.name}"
+  end
+
+  # Default to all attributes.
+  def search_attributes
+    nil
+  end
+
+  def search_base
+    Puppet[:ldapbase]
+  end
+
+  # The ldap search filter to use.
+  def search_filter(name)
+    raise Puppet::DevError, "No search string set for LDAP terminus for #{self.name}"
+  end
+
+  # Find the ldap node, return the class list and parent node specially,
+  # and everything else in a parameter hash.
+  def ldapsearch(filter)
+    raise ArgumentError.new("You must pass a block to ldapsearch") unless block_given?
+
+    found = false
+    count = 0
+
+    begin
+      connection.search(search_base, 2, filter, search_attributes) do |entry|
+        found = true
+        yield entry
+      end
+    rescue SystemExit,NoMemoryError
+      raise
+    rescue Exception => detail
+      if count == 0
+        # Try reconnecting to ldap if we get an exception and we haven't yet retried.
+        count += 1
+        @connection = nil
+        Puppet.warning "Retrying LDAP connection"
+        retry
+      else
+        error = Puppet::Error.new("LDAP Search failed")
+        error.set_backtrace(detail.backtrace)
+        raise error
+      end
+    end
+
+    found
+  end
+
+  # Create an ldap connection.
+  def connection
+    unless @connection
+      raise Puppet::Error, "Could not set up LDAP Connection: Missing ruby/ldap libraries" unless Puppet.features.ldap?
+      begin
+        conn = Puppet::Util::Ldap::Connection.instance
+        conn.start
+        @connection = conn.connection
+      rescue => detail
+        puts detail.backtrace if Puppet[:trace]
+        raise Puppet::Error, "Could not connect to LDAP: #{detail}"
+      end
+    end
+
+    @connection
+  end
+end

data/examples/vendored-puppet/vendor/puppet-2.7.8/lib/puppet/indirector/memory.rb

@@ -0,0 +1,21 @@
+require 'puppet/indirector/terminus'
+
+# Manage a memory-cached list of instances.
+class Puppet::Indirector::Memory < Puppet::Indirector::Terminus
+  def initialize
+    @instances = {}
+  end
+
+  def destroy(request)
+    raise ArgumentError.new("Could not find #{request.key} to destroy") unless @instances.include?(request.key)
+    @instances.delete(request.key)
+  end
+
+  def find(request)
+    @instances[request.key]
+  end
+
+  def save(request)
+    @instances[request.key] = request.instance
+  end
+end

data/examples/vendored-puppet/vendor/puppet-2.7.8/lib/puppet/indirector/node/active_record.rb

@@ -0,0 +1,13 @@
+require 'puppet/rails/host'
+require 'puppet/indirector/active_record'
+require 'puppet/node'
+
+class Puppet::Node::ActiveRecord < Puppet::Indirector::ActiveRecord
+  use_ar_model Puppet::Rails::Host
+
+  def find(request)
+    node = super
+    node.fact_merge
+    node
+  end
+end

data/examples/vendored-puppet/vendor/puppet-2.7.8/lib/puppet/indirector/node/exec.rb

@@ -0,0 +1,48 @@
+require 'puppet/node'
+require 'puppet/indirector/exec'
+
+class Puppet::Node::Exec < Puppet::Indirector::Exec
+  desc "Call an external program to get node information.  See
+  the [External Nodes](http://docs.puppetlabs.com/guides/external_nodes.html) page for more information."
+  include Puppet::Util
+
+  def command
+    command = Puppet[:external_nodes]
+    raise ArgumentError, "You must set the 'external_nodes' parameter to use the external node terminus" unless command != "none"
+    command.split
+  end
+
+  # Look for external node definitions.
+  def find(request)
+    output = super or return nil
+
+    # Translate the output to ruby.
+    result = translate(request.key, output)
+
+    create_node(request.key, result)
+  end
+
+  private
+
+  # Turn our outputted objects into a Puppet::Node instance.
+  def create_node(name, result)
+    node = Puppet::Node.new(name)
+    set = false
+    [:parameters, :classes, :environment].each do |param|
+      if value = result[param]
+        node.send(param.to_s + "=", value)
+        set = true
+      end
+    end
+
+    node.fact_merge
+    node
+  end
+
+  # Translate the yaml string into Ruby objects.
+  def translate(name, output)
+      YAML.load(output).inject({}) { |hash, data| hash[symbolize(data[0])] = data[1]; hash }
+  rescue => detail
+      raise Puppet::Error, "Could not load external node results for #{name}: #{detail}"
+  end
+end