bolt 0.20.0 → 0.20.2

data/vendored/puppet/lib/puppet/network/format_handler.rb

@@ -0,0 +1,108 @@
+require 'yaml'
+require 'puppet/network'
+require 'puppet/network/format'
+
+module Puppet::Network::FormatHandler
+  class FormatError < Puppet::Error; end
+
+  ALL_MEDIA_TYPES = '*/*'.freeze
+
+  @formats = {}
+
+  def self.create(*args, &block)
+    instance = Puppet::Network::Format.new(*args, &block)
+
+    @formats[instance.name] = instance
+    instance
+  end
+
+  def self.create_serialized_formats(name,options = {},&block)
+    ["application/x-#{name}", "application/#{name}", "text/x-#{name}", "text/#{name}"].each { |mime_type|
+      create name, {:mime => mime_type}.update(options), &block
+    }
+  end
+
+  def self.format(name)
+    @formats[name.to_s.downcase.intern]
+  end
+
+  def self.format_for(name)
+    name = format_to_canonical_name(name)
+    format(name)
+  end
+
+  def self.format_by_extension(ext)
+    @formats.each do |name, format|
+      return format if format.extension == ext
+    end
+    nil
+  end
+
+  # Provide a list of all formats.
+  def self.formats
+    @formats.keys
+  end
+
+  # Return a format capable of handling the provided mime type.
+  def self.mime(mimetype)
+    mimetype = mimetype.to_s.downcase
+    @formats.values.find { |format| format.mime == mimetype }
+  end
+
+  # Return a format name given:
+  #  * a format name
+  #  * a mime-type
+  #  * a format instance
+  def self.format_to_canonical_name(format)
+    case format
+    when Puppet::Network::Format
+      out = format
+    when %r{\w+/\w+}
+      out = mime(format)
+    else
+      out = format(format)
+    end
+
+    if out.nil?
+      raise ArgumentError, _("No format matches the given format name or mime-type (%{format})") % {format: format}
+    end
+
+    out.name
+  end
+
+  # Determine which of the accepted formats should be used given what is supported.
+  #
+  # @param accepted [Array<String, Symbol>] the accepted formats in a form a
+  #   that generally conforms to an HTTP Accept header. Any quality specifiers
+  #   are ignored and instead the formats are simply in strict preference order
+  #   (most preferred is first)
+  # @param supported [Array<Symbol>] the names of the supported formats (the
+  #   most preferred format is first)
+  # @return [Array<Puppet::Network::Format>] the most suitable formats that
+  #   are both accepted and supported
+  # @api private
+  def self.most_suitable_formats_for(accepted, supported)
+    accepted.collect do |format|
+      format.to_s.sub(/;q=.*$/, '')
+    end.collect do |format|
+      if format == ALL_MEDIA_TYPES
+        supported.first
+      else
+        format_to_canonical_name_or_nil(format)
+      end
+    end.compact.find_all do |format|
+      supported.include?(format)
+    end.collect do |format|
+      format_for(format)
+    end
+  end
+
+  # @api private
+  def self.format_to_canonical_name_or_nil(format)
+    format_to_canonical_name(format)
+  rescue ArgumentError
+    nil
+  end
+end
+
+require 'puppet/network/formats'

data/vendored/puppet/lib/puppet/network/format_support.rb

@@ -0,0 +1,134 @@
+require 'puppet/network/format_handler'
+
+# Provides network serialization support when included
+# @api public
+module Puppet::Network::FormatSupport
+  def self.included(klass)
+    klass.extend(ClassMethods)
+  end
+
+  module ClassMethods
+    def convert_from(format, data)
+      get_format(format).intern(self, data)
+    rescue => err
+      #TRANSLATORS "intern" is a function name and should not be translated
+      raise Puppet::Network::FormatHandler::FormatError, _("Could not intern from %{format}: %{err}") % { format: format, err: err }, err.backtrace
+    end
+
+    def convert_from_multiple(format, data)
+      get_format(format).intern_multiple(self, data)
+    rescue => err
+      #TRANSLATORS "intern_multiple" is a function name and should not be translated
+      raise Puppet::Network::FormatHandler::FormatError, _("Could not intern_multiple from %{format}: %{err}") % { format: format, err: err }, err.backtrace
+    end
+
+    def render_multiple(format, instances)
+      get_format(format).render_multiple(instances)
+    rescue => err
+      #TRANSLATORS "render_multiple" is a function name and should not be translated
+      raise Puppet::Network::FormatHandler::FormatError, _("Could not render_multiple to %{format}: %{err}") % { format: format, err: err }, err.backtrace
+    end
+
+    def default_format
+      supported_formats[0]
+    end
+
+    def support_format?(name)
+      Puppet::Network::FormatHandler.format(name).supported?(self)
+    end
+
+    def supported_formats
+      result = format_handler.formats.collect do |f|
+        format_handler.format(f)
+      end.find_all do |f|
+        f.supported?(self)
+      end.sort do |a, b|
+        # It's an inverse sort -- higher weight formats go first.
+        b.weight <=> a.weight
+      end.collect do |f|
+        f.name
+      end
+
+      result = put_preferred_format_first(result)
+
+      Puppet.debug "#{friendly_name} supports formats: #{result.join(' ')}"
+
+      result
+    end
+
+    # @api private
+    def get_format(format_name)
+      format_handler.format_for(format_name)
+    end
+
+    private
+
+    def format_handler
+      Puppet::Network::FormatHandler
+    end
+
+    def friendly_name
+      if self.respond_to? :indirection
+        indirection.name
+      else
+        self
+      end
+    end
+
+    def put_preferred_format_first(list)
+      preferred_format = Puppet.settings[:preferred_serialization_format].to_sym
+      if list.include?(preferred_format)
+        list.delete(preferred_format)
+        list.unshift(preferred_format)
+      else
+        Puppet.debug "Value of 'preferred_serialization_format' (#{preferred_format}) is invalid for #{friendly_name}, using default (#{list.first})"
+      end
+      list
+    end
+  end
+
+  def to_msgpack(*args)
+    to_data_hash.to_msgpack(*args)
+  end
+
+  # @deprecated, use to_json
+  def to_pson(*args)
+    to_data_hash.to_pson(*args)
+  end
+
+  def to_json(*args)
+    to_data_hash.to_json(*args)
+  end
+
+  def render(format = nil)
+    format ||= self.class.default_format
+
+    self.class.get_format(format).render(self)
+  rescue => err
+    #TRANSLATORS "render" is a function name and should not be translated
+    raise Puppet::Network::FormatHandler::FormatError, _("Could not render to %{format}: %{err}") % { format: format, err: err }, err.backtrace
+  end
+
+  def mime(format = nil)
+    format ||= self.class.default_format
+
+    self.class.get_format(format).mime
+  rescue => err
+    #TRANSLATORS "mime" is a function name and should not be translated
+    raise Puppet::Network::FormatHandler::FormatError, _("Could not mime to %{format}: %{err}") % { format: format, err: err }, err.backtrace
+  end
+
+  def support_format?(name)
+    self.class.support_format?(name)
+  end
+
+  # @comment Document to_data_hash here as it is called as a hook from to_msgpack if it exists
+  # @!method to_data_hash(*args)
+  # @api public
+  # @abstract
+  # This method may be implemented to return a hash object that is used for serializing.
+  # The object returned by this method should contain all the info needed to instantiate it again.
+  # If the method exists it will be called from to_msgpack and other serialization methods.
+  # @return [Hash]
+end
+

data/vendored/puppet/lib/puppet/network/formats.rb

@@ -0,0 +1,171 @@
+require 'puppet/network/format_handler'
+require 'json'
+
+Puppet::Network::FormatHandler.create_serialized_formats(:msgpack, :weight => 20, :mime => "application/x-msgpack", :required_methods => [:render_method, :intern_method], :intern_method => :from_data_hash) do
+
+  confine :feature => :msgpack
+
+  def intern(klass, text)
+    data = MessagePack.unpack(text)
+    return data if data.is_a?(klass)
+    klass.from_data_hash(data)
+  end
+
+  def intern_multiple(klass, text)
+    MessagePack.unpack(text).collect do |data|
+      klass.from_data_hash(data)
+    end
+  end
+
+  def render_multiple(instances)
+    instances.to_msgpack
+  end
+end
+
+Puppet::Network::FormatHandler.create_serialized_formats(:yaml) do
+  def intern(klass, text)
+    data = YAML.load(text)
+    data_to_instance(klass, data)
+  end
+
+  def intern_multiple(klass, text)
+    data = YAML.load(text)
+    unless data.respond_to?(:collect)
+      raise Puppet::Network::FormatHandler::FormatError, _("Serialized YAML did not contain a collection of instances when calling intern_multiple")
+    end
+
+    data.collect do |datum|
+      data_to_instance(klass, datum)
+    end
+  end
+
+  def data_to_instance(klass, data)
+    return data if data.is_a?(klass)
+
+    unless data.is_a? Hash
+      raise Puppet::Network::FormatHandler::FormatError, _("Serialized YAML did not contain a valid instance of %{klass}") % { klass: klass }
+    end
+
+    klass.from_data_hash(data)
+  end
+
+  def render(instance)
+    instance.to_yaml
+  end
+
+  # Yaml monkey-patches Array, so this works.
+  def render_multiple(instances)
+    instances.to_yaml
+  end
+
+  def supported?(klass)
+    true
+  end
+end
+
+Puppet::Network::FormatHandler.create(:s, :mime => "text/plain", :charset => Encoding::UTF_8, :extension => "txt")
+
+# By default, to_binary is called to render and from_binary called to intern. Note unlike
+# text-based formats (json, yaml, etc), we don't use to_data_hash for binary.
+Puppet::Network::FormatHandler.create(:binary, :mime => "application/octet-stream", :weight => 1,
+                                      :required_methods => [:render_method, :intern_method]) do
+end
+
+# PSON is deprecated
+Puppet::Network::FormatHandler.create_serialized_formats(:pson, :weight => 10, :required_methods => [:render_method, :intern_method], :intern_method => :from_data_hash) do
+  def intern(klass, text)
+    data_to_instance(klass, PSON.parse(text))
+  end
+
+  def intern_multiple(klass, text)
+    PSON.parse(text).collect do |data|
+      data_to_instance(klass, data)
+    end
+  end
+
+  # PSON monkey-patches Array, so this works.
+  def render_multiple(instances)
+    instances.to_pson
+  end
+
+  # If they pass class information, we want to ignore it.
+  # This is required for compatibility with Puppet 3.x
+  def data_to_instance(klass, data)
+    if data.is_a?(Hash) and d = data['data']
+      data = d
+    end
+    return data if data.is_a?(klass)
+    klass.from_data_hash(data)
+  end
+end
+
+Puppet::Network::FormatHandler.create_serialized_formats(:json, :mime => 'application/json', :charset => Encoding::UTF_8, :weight => 15, :required_methods => [:render_method, :intern_method], :intern_method => :from_data_hash) do
+  def intern(klass, text)
+    data_to_instance(klass, JSON.parse(text))
+  end
+
+  def intern_multiple(klass, text)
+    JSON.parse(text).collect do |data|
+      data_to_instance(klass, data)
+    end
+  end
+
+  # JSON monkey-patches Array, so this works.
+  # https://github.com/ruby/ruby/blob/ruby_1_9_3/ext/json/generator/generator.c#L1416
+  def render_multiple(instances)
+    instances.to_json
+  end
+
+  # Unlike PSON, we do not need to unwrap the data envelope, because legacy 3.x agents
+  # have never supported JSON
+  def data_to_instance(klass, data)
+    return data if data.is_a?(klass)
+    klass.from_data_hash(data)
+  end
+end
+
+# This is really only ever going to be used for Catalogs.
+Puppet::Network::FormatHandler.create_serialized_formats(:dot, :required_methods => [:render_method])
+
+
+Puppet::Network::FormatHandler.create(:console,
+                                      :mime   => 'text/x-console-text',
+                                      :weight => 0) do
+  def json
+    @json ||= Puppet::Network::FormatHandler.format(:json)
+  end
+
+  def render(datum)
+    return datum if datum.is_a?(String) || datum.is_a?(Numeric)
+
+    # Simple hash to table
+    if datum.is_a?(Hash) && datum.keys.all? { |x| x.is_a?(String) || x.is_a?(Numeric) }
+      output = ''
+      column_a = datum.empty? ? 2 : datum.map{ |k,v| k.to_s.length }.max + 2
+      datum.sort_by { |k,v| k.to_s } .each do |key, value|
+        output << key.to_s.ljust(column_a)
+        output << json.render(value).
+          chomp.gsub(/\n */) { |x| x + (' ' * column_a) }
+        output << "\n"
+      end
+      return output
+    end
+
+    # Print one item per line for arrays
+    if datum.is_a? Array
+      output = ''
+      datum.each do |item|
+        output << item.to_s
+        output << "\n"
+      end
+      return output
+    end
+
+    # ...or pretty-print the inspect outcome.
+    JSON.pretty_generate(datum, :quirks_mode => true)
+  end
+
+  def render_multiple(data)
+    data.collect(&:render).join("\n")
+  end
+end

data/vendored/puppet/lib/puppet/network/http.rb

@@ -0,0 +1,30 @@
+module Puppet::Network::HTTP
+  HEADER_ENABLE_PROFILING = "X-Puppet-Profiling"
+  HEADER_PUPPET_VERSION = "X-Puppet-Version"
+
+  MASTER_URL_PREFIX = "/puppet"
+  MASTER_URL_VERSIONS = "v3"
+
+  CA_URL_PREFIX = "/puppet-ca"
+  CA_URL_VERSIONS = "v1"
+
+  require 'puppet/network/authorization'
+  require 'puppet/network/http/issues'
+  require 'puppet/network/http/error'
+  require 'puppet/network/http/route'
+  require 'puppet/network/http/api'
+  require 'puppet/network/http/api/ca'
+  require 'puppet/network/http/api/ca/v1'
+  require 'puppet/network/http/api/master'
+  require 'puppet/network/http/api/master/v3'
+  require 'puppet/network/http/handler'
+  require 'puppet/network/http/response'
+  require 'puppet/network/http/request'
+  require 'puppet/network/http/site'
+  require 'puppet/network/http/session'
+  require 'puppet/network/http/factory'
+  require 'puppet/network/http/nocache_pool'
+  require 'puppet/network/http/pool'
+  require 'puppet/network/http/memory_response'
+  require 'puppet/network/http/compression'
+end

data/vendored/puppet/lib/puppet/network/http/api.rb

@@ -0,0 +1,44 @@
+class Puppet::Network::HTTP::API
+  require 'puppet/version'
+
+  def self.not_found
+    Puppet::Network::HTTP::Route.
+      path(/.*/).
+      any(lambda do |req, res|
+        raise Puppet::Network::HTTP::Error::HTTPNotFoundError.new("No route for #{req.method} #{req.path}", Puppet::Network::HTTP::Issues::HANDLER_NOT_FOUND)
+      end)
+  end
+
+  def self.not_found_upgrade
+    Puppet::Network::HTTP::Route.
+      path(/.*/).
+      any(lambda do |req, res|
+        raise Puppet::Network::HTTP::Error::HTTPNotFoundError.new("Error: Invalid URL - Puppet expects requests that conform to the " +
+                                                                      "/puppet and /puppet-ca APIs.\n\n" +
+                                                                      "Note that Puppet 3 agents aren't compatible with this version; if you're " +
+                                                                      "running Puppet 3, you must either upgrade your agents to match the server " +
+                                                                      "or point them to a server running Puppet 3.\n\n" +
+                                                                      "Master Info:\n" +
+                                                                      "  Puppet version: #{Puppet.version}\n" +
+                                                                      "  Supported /puppet API versions: #{Puppet::Network::HTTP::MASTER_URL_VERSIONS}\n" +
+                                                                      "  Supported /puppet-ca API versions: #{Puppet::Network::HTTP::CA_URL_VERSIONS}",
+                                                                  Puppet::Network::HTTP::Issues::HANDLER_NOT_FOUND)
+      end)
+  end
+
+  def self.master_routes
+    master_prefix = Regexp.new("^#{Puppet::Network::HTTP::MASTER_URL_PREFIX}/")
+    Puppet::Network::HTTP::Route.path(master_prefix).
+      any.
+      chain(Puppet::Network::HTTP::API::Master::V3.routes,
+            Puppet::Network::HTTP::API.not_found)
+  end
+
+  def self.ca_routes
+    ca_prefix = Regexp.new("^#{Puppet::Network::HTTP::CA_URL_PREFIX}/")
+    Puppet::Network::HTTP::Route.path(ca_prefix).
+      any.
+      chain(Puppet::Network::HTTP::API::CA::V1.routes,
+            Puppet::Network::HTTP::API.not_found)
+  end
+end