Sha256: abaed496045c0f476c7a1c59c204332dfa2fe8ddfc00fcff8248c9ba6c82b644

Contents?: true

Size: 1.78 KB

Versions: 72

Compression:

Stored size: 1.78 KB

Contents

require 'puppet/node/facts'
require 'puppet/indirector/rest'

class Puppet::Node::Facts::Rest < Puppet::Indirector::REST
  desc "Find and save facts about nodes over HTTP via REST."

  def find(request)
    return super unless use_http_client?

    session = Puppet.lookup(:http_session)
    api = session.route_to(:puppet)
    _, facts = api.get_facts(
      request.key,
      environment: request.environment.to_s
    )
    facts
  rescue Puppet::HTTP::ResponseError => e
    if e.response.code == 404
      return nil unless request.options[:fail_on_404]

      _, body = parse_response(e.response.nethttp)
      msg = _("Find %{uri} resulted in 404 with the message: %{body}") % { uri: elide(e.response.url.path, 100), body: body }
      raise Puppet::Error, msg
    else
      raise convert_to_http_error(e.response.nethttp)
    end
  end

  def save(request)
    raise ArgumentError, _("PUT does not accept options") unless request.options.empty?

    return legacy_save(request) unless use_http_client?

    session = Puppet.lookup(:http_session)
    api = session.route_to(:puppet)
    api.put_facts(
      request.key,
      facts: request.instance,
      environment: request.environment.to_s
    )

    # preserve existing behavior
    nil
  rescue Puppet::HTTP::ResponseError => e
    # always raise even if fail_on_404 is false
    raise convert_to_http_error(e.response.nethttp)
  end

  private

  def legacy_save(request)
    response = do_request(request) do |req|
      http_put(req, IndirectedRoutes.request_to_uri(req), req.instance.render, headers.merge({ "Content-Type" => req.instance.mime }))
    end

    if is_http_200?(response)
      content_type, body = parse_response(response)
      deserialize_save(content_type, body)
    else
      raise convert_to_http_error(response)
    end
  end
end

Version data entries

72 entries across 72 versions & 1 rubygems

Version Path
puppet-6.29.0 lib/puppet/indirector/facts/rest.rb
puppet-6.29.0-x86-mingw32 lib/puppet/indirector/facts/rest.rb
puppet-6.29.0-x64-mingw32 lib/puppet/indirector/facts/rest.rb
puppet-6.29.0-universal-darwin lib/puppet/indirector/facts/rest.rb
puppet-6.28.0 lib/puppet/indirector/facts/rest.rb
puppet-6.28.0-x86-mingw32 lib/puppet/indirector/facts/rest.rb
puppet-6.28.0-x64-mingw32 lib/puppet/indirector/facts/rest.rb
puppet-6.28.0-universal-darwin lib/puppet/indirector/facts/rest.rb
puppet-6.27.0 lib/puppet/indirector/facts/rest.rb
puppet-6.27.0-x86-mingw32 lib/puppet/indirector/facts/rest.rb
puppet-6.27.0-x64-mingw32 lib/puppet/indirector/facts/rest.rb
puppet-6.27.0-universal-darwin lib/puppet/indirector/facts/rest.rb
puppet-6.26.0 lib/puppet/indirector/facts/rest.rb
puppet-6.26.0-x86-mingw32 lib/puppet/indirector/facts/rest.rb
puppet-6.26.0-x64-mingw32 lib/puppet/indirector/facts/rest.rb
puppet-6.26.0-universal-darwin lib/puppet/indirector/facts/rest.rb
puppet-6.25.1 lib/puppet/indirector/facts/rest.rb
puppet-6.25.1-x86-mingw32 lib/puppet/indirector/facts/rest.rb
puppet-6.25.1-x64-mingw32 lib/puppet/indirector/facts/rest.rb
puppet-6.25.1-universal-darwin lib/puppet/indirector/facts/rest.rb