Class: Puppet::Network::HTTP::WEBrickREST

Inherits:
WEBrick::HTTPServlet::AbstractServlet
  • Object
show all
Includes:
Handler
Defined in:
lib/puppet/network/http/webrick/rest.rb

Constant Summary

Constants included from Handler

Handler::DISALLOWED_KEYS

Constants included from Issues

Issues::ENVIRONMENT_NOT_FOUND, Issues::FAILED_AUTHORIZATION, Issues::HANDLER_NOT_FOUND, Issues::MISSING_HEADER_FIELD, Issues::NO_INDIRECTION_REMOTE_REQUESTS, Issues::RESOURCE_NOT_FOUND, Issues::RUNTIME_ERROR, Issues::UNSUPPORTED_FORMAT, Issues::UNSUPPORTED_METHOD

Class Method Summary collapse

Instance Method Summary collapse

Methods included from Handler

#format_to_mime, #process, #register, #resolve_node

Methods included from Authentication

#warn_if_near_expiration

Constructor Details

#initialize(server) ⇒ WEBrickREST

Returns a new instance of WEBrickREST.

Raises:

  • (ArgumentError) 


14
15
16
17
18
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 14

def initialize(server)
  raise ArgumentError, "server is required" unless server
  register([Puppet::Network::HTTP::API::V2.routes, Puppet::Network::HTTP::API::V1.routes])
  super(server)
end

Class Method Details

.mutex ⇒ Object 



10
11
12
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 10

def self.mutex
  @mutex ||= Mutex.new
end

Instance Method Details

#body(request) ⇒ Object 



57
58
59
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 57

def body(request)
  request.body
end

#client_cert(request) ⇒ Object 



61
62
63
64
65
66
67
68
69
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 61

def client_cert(request)
  if cert = request.client_cert
    cert = Puppet::SSL::Certificate.from_instance(cert)
    warn_if_near_expiration(cert)
    cert
  else
    nil
  end
end

#client_information(request) ⇒ Object

Retrieve node/cert/ip information from the request object.



88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 88

def client_information(request)
  result = {}
  if peer = request.peeraddr and ip = peer[3]
    result[:ip] = ip
  end

  # If they have a certificate (which will almost always be true)
  # then we get the hostname from the cert, instead of via IP
  # info
  result[:authenticated] = false
  if cert = request.client_cert and cn = Puppet::Util::SSL.cn_from_subject(cert.subject)
    result[:node] = cn
    result[:authenticated] = true
  else
    result[:node] = resolve_node(result)
  end

  result
end

#headers(request) ⇒ Object 



41
42
43
44
45
46
47
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 41

def headers(request)
  result = {}
  request.each do |k, v|
    result[k.downcase] = v
  end
  result
end

#http_method(request) ⇒ Object 



49
50
51
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 49

def http_method(request)
  request.request_method
end

#params(request) ⇒ Object

Retrieve the request parameters, including authentication information.



21
22
23
24
25
26
27
28
29
30
31
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 21

def params(request)
  params = request.query || {}

  params = Hash[params.collect do |key, value|
    all_values = value.list
    [key, all_values.length == 1 ? value : all_values]
  end]

  params = decode_params(params)
  params.merge(client_information(request))
end

#path(request) ⇒ Object 



53
54
55
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 53

def path(request)
  request.path
end

#service(request, response) ⇒ Object

WEBrick uses a service method to respond to requests. Simply delegate to the handler response method.



35
36
37
38
39
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 35

def service(request, response)
  self.class.mutex.synchronize do
    process(request, response)
  end
end

#set_content_type(response, format) ⇒ Object

Set the specified format as the content type of the response.



72
73
74
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 72

def set_content_type(response, format)
  response["content-type"] = format_to_mime(format)
end

#set_response(response, result, status = 200) ⇒ Object 



76
77
78
79
80
81
82
83
84
85
 
# File 'lib/puppet/network/http/webrick/rest.rb', line 76

def set_response(response, result, status = 200)
  response.status = status
  if status >= 200 and status != 304
    response.body = result
    response["content-length"] = result.stat.size if result.is_a?(File)
  end
  if RUBY_VERSION[0,3] == "1.8"
    response["connection"] = 'close'
  end
end