Class: PuppetX::Wildfly::APIClient

Inherits:
Object
  • Object
show all
Defined in:
lib/puppet_x/wildfly/api_client.rb

Instance Method Summary collapse

Constructor Details

#initialize(address, port, user, password, timeout, secure) ⇒ APIClient

Returns a new instance of APIClient.



10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/puppet_x/wildfly/api_client.rb', line 10

def initialize(address, port, user, password, timeout, secure)
  @username = user
  @password = password
  @uri = if secure
           URI.parse "https://#{address}:#{port}/management"
         else
           URI.parse "http://#{address}:#{port}/management"
         end
  @uri.user = CGI.escape(user)
  @uri.password = CGI.escape(password)

  @http_client = Net::HTTP.new @uri.host, @uri.port, nil
  @http_client.read_timeout = timeout
  return unless secure == true
  @http_client.use_ssl = true
  @http_client.verify_mode = OpenSSL::SSL::VERIFY_NONE
end

Instance Method Details

#authz_header ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/puppet_x/wildfly/api_client.rb', line 28

def authz_header
  digest_auth = Net::HTTP::DigestAuth.new
  authz_request = Net::HTTP::Get.new @uri.request_uri

  retried = 0

  # All http_api providers require Wildfly service to be up, but with systemd there is no guarantee that the service is actually running, therefore we need to retry

  begin
    response = @http_client.request authz_request
  rescue => e
    raise e unless retried + 1 < 6
    retried += 1
    sleep 10
    retry
  end

  sleep 0.05 # We've been returned an auth token.  But if we don't wait a small amount of time before using it, we might stil get a 401. :(

  if response['www-authenticate'] =~ /digest/i
    digest_auth.auth_header @uri, response['www-authenticate'], 'POST'
  else
    response['www-authenticate']
  end
end

#submit(body, ignore_failed_outcome = false) ⇒ Object 



54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
 
# File 'lib/puppet_x/wildfly/api_client.rb', line 54

def submit(body, ignore_failed_outcome = false)
  http_request = Net::HTTP::Post.new @uri.request_uri
  http_request.add_field 'Content-type', 'application/json'
  authz = authz_header
  if authz =~ /digest/i
    http_request.add_field 'Authorization', authz
  else
    http_request.basic_auth @username, @password
  end

  http_request.body = body.to_json
  http_response = @http_client.request http_request
  response = JSON.parse(http_response.body)

  unless response['outcome'] == 'success' || ignore_failed_outcome
    raise "Failed with: #{response['failure-description']} for #{body.to_json}"
  end

  response
end