Class: Vra::Client

Inherits:
Object
  • Object
show all
Defined in:
lib/vra/client.rb

Constant Summary collapse

REFRESH_TOKEN_URL = 
"/csp/gateway/am/api/login?access_token"
ACCESS_TOKEN_URL = 
"/iaas/api/login"
ROLES_URL = 
"/csp/gateway/am/api/loggedin/user/orgs"

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(opts) ⇒ Client

Returns a new instance of Client.



32
33
34
35
36
37
38
39
40
41
42
43
44
 
# File 'lib/vra/client.rb', line 32

def initialize(opts)
  @base_url      = opts[:base_url]
  @username      = opts[:username]
  @password      = PasswordMasker.new(opts[:password])
  @domain        = fetch_domain(opts)

  @verify_ssl    = opts.fetch(:verify_ssl, true)
  @refresh_token = PasswordMasker.new(nil)
  @access_token  = PasswordMasker.new(nil)
  @page_size     = opts.fetch(:page_size, 20)

  validate_client_options!
end

Instance Attribute Details

#page_size ⇒ Object

Returns the value of attribute page_size.



30
31
32
 
# File 'lib/vra/client.rb', line 30

def page_size
  @page_size
end

Instance Method Details

#access_token ⇒ Object

client methods



64
65
66
 
# File 'lib/vra/client.rb', line 64

def access_token
  @access_token.value
end

#access_token=(value) ⇒ Object 



72
73
74
 
# File 'lib/vra/client.rb', line 72

def access_token=(value)
  @access_token.value = value
end

#authorize! ⇒ Object

Raises:



97
98
99
100
101
 
# File 'lib/vra/client.rb', line 97

def authorize!
  generate_access_token unless authorized?

  raise Vra::Exception::Unauthorized, "Unable to authorize against vRA" unless authorized?
end

#authorized? ⇒ Boolean

Returns:

  • (Boolean) 


103
104
105
106
107
108
 
# File 'lib/vra/client.rb', line 103

def authorized?
  return false if @access_token.value.nil?

  response = http_head(ROLES_URL, :skip_auth)
  response.success?
end

#catalog ⇒ Object

methods to other classes



51
52
53
 
# File 'lib/vra/client.rb', line 51

def catalog
  @catalog ||= Vra::Catalog.new(self)
end

#deployments ⇒ Object 



55
56
57
 
# File 'lib/vra/client.rb', line 55

def deployments
  @deployments ||= Vra::Deployments.new(self)
end

#full_url(path) ⇒ Object 



133
134
135
 
# File 'lib/vra/client.rb', line 133

def full_url(path)
  "#{@base_url}#{path}"
end

#generate_access_token ⇒ Object

Raises:



110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
 
# File 'lib/vra/client.rb', line 110

def generate_access_token
  @refresh_token.value = nil
  @access_token.value = nil
  validate_client_options!

  # VRA 8 has a two-step authentication process - This probably breaks VRA7, who knows?!?
  # First step: Sending Username/Password to get a Refresh Token
  refresh_response = http_post(REFRESH_TOKEN_URL,
                       FFI_Yajl::Encoder.encode(token_params),
                       :skip_auth)
  raise Vra::Exception::Unauthorized, "Unable to get the refresh token: #{refresh_response.body}" unless refresh_response.success_ok?

  refresh_response_body = FFI_Yajl::Parser.parse(refresh_response.body)
  @refresh_token.value = refresh_response_body["refresh_token"]

  # Second Step: Sending the refresh token to a separate endpoint to get an Access Token
  access_response = http_post(ACCESS_TOKEN_URL, "{ \"refreshToken\": \"#{@refresh_token.value}\" }", :skip_auth)
  raise Vra::Exception::Unauthorized, "Unable to get the access token: #{access_response.body}" unless access_response.success_ok?

  access_response_body = FFI_Yajl::Parser.parse(access_response.body)
  @access_token.value = access_response_body["token"]
end

#get_parsed(path) ⇒ Object 



167
168
169
 
# File 'lib/vra/client.rb', line 167

def get_parsed(path)
  FFI_Yajl::Parser.parse(http_get!(path))
end

#http_delete(path, skip_auth = nil) ⇒ Object 



163
164
165
 
# File 'lib/vra/client.rb', line 163

def http_delete(path, skip_auth = nil)
  http_fetch(:delete, path, skip_auth)
end

#http_fetch(method, path, skip_auth = nil) ⇒ Object 



137
138
139
140
141
142
143
144
145
146
147
148
 
# File 'lib/vra/client.rb', line 137

def http_fetch(method, path, skip_auth = nil)
  authorize! unless skip_auth

  response = Vra::Http.execute(method: method,
                               url: full_url(path),
                               headers: request_headers,
                               verify_ssl: @verify_ssl)
rescue => e
  raise_http_exception(e, path)
else
  response
end

#http_get(path, skip_auth = nil) ⇒ Object 



154
155
156
 
# File 'lib/vra/client.rb', line 154

def http_get(path, skip_auth = nil)
  http_fetch(:get, path, skip_auth)
end

#http_get!(path) ⇒ Object 



158
159
160
161
 
# File 'lib/vra/client.rb', line 158

def http_get!(path)
  response = http_get(path)
  response.body
end

#http_get_paginated_array!(path, filter = nil) ⇒ Object 



171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
 
# File 'lib/vra/client.rb', line 171

def http_get_paginated_array!(path, filter = nil)
  items = []
  page = 0
  base_path = path + "?$top=#{page_size}"
  base_path += "&#{filter}" if filter

  loop do
    response = get_parsed("#{base_path}&$skip=#{page * page_size}")
    items += response["content"]

    page += 1
    break if page >= response["totalPages"]
  end

  if items.uniq!
    raise Vra::Exception::DuplicateItemsDetected,
          "Duplicate items were returned by the vRA API. " \
          "Increase your page size to avoid this vRA API bug. " \
          "See https://github.com/chef-partners/vmware-vra-gem#pagination " \
          "for more information."
  end

  items
end

#http_head(path, skip_auth = nil) ⇒ Object 



150
151
152
 
# File 'lib/vra/client.rb', line 150

def http_head(path, skip_auth = nil)
  http_fetch(:head, path, skip_auth)
end

#http_post(path, payload, skip_auth = nil) ⇒ Object 



196
197
198
199
200
201
202
203
204
205
206
207
208
 
# File 'lib/vra/client.rb', line 196

def http_post(path, payload, skip_auth = nil)
  authorize! unless skip_auth

  response = Vra::Http.execute(method: :post,
                               url: full_url(path),
                               headers: request_headers,
                               payload: payload,
                               verify_ssl: @verify_ssl)
rescue => e
  raise_http_exception(e, path)
else
  response
end

#http_post!(path, payload) ⇒ Object 



210
211
212
213
 
# File 'lib/vra/client.rb', line 210

def http_post!(path, payload)
  response = http_post(path, payload)
  response.body
end

#raise_http_exception(caught_exception, path) ⇒ Object

Raises:

  • (exception) 


215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
 
# File 'lib/vra/client.rb', line 215

def raise_http_exception(caught_exception, path)
  raise unless caught_exception.respond_to?(:http_code)

  klass = if caught_exception.http_code == 404
            Vra::Exception::HTTPNotFound
          else
            Vra::Exception::HTTPError
          end

  exception = klass.new(code: caught_exception.http_code,
                        body: caught_exception.response,
                        klass: caught_exception.class,
                        path: path)

  message = exception.errors.empty? ? caught_exception.message : exception.errors.join(", ")
  raise exception, message
end

#refresh_token ⇒ Object 



68
69
70
 
# File 'lib/vra/client.rb', line 68

def refresh_token
  @refresh_token.value
end

#refresh_token=(value) ⇒ Object 



76
77
78
 
# File 'lib/vra/client.rb', line 76

def refresh_token=(value)
  @refresh_token.value = value
end

#request_headers ⇒ Object 



88
89
90
91
92
93
94
95
 
# File 'lib/vra/client.rb', line 88

def request_headers
  headers                   = {}
  headers["Accept"]         = "application/json"
  headers["Content-Type"]   = "application/json"
  headers["Authorization"]  = "Bearer " + @access_token.value unless @access_token.value.nil?

  headers
end

#token_params ⇒ Object 



80
81
82
83
84
85
86
 
# File 'lib/vra/client.rb', line 80

def token_params
  {
    'username': @username,
    'password': @password.value,
    'domain': @domain,
  }
end

#valid_uri?(uri) ⇒ Boolean

Returns:

  • (Boolean) 


240
241
242
243
244
245
 
# File 'lib/vra/client.rb', line 240

def valid_uri?(uri)
  uri = URI.parse(uri)
  uri.is_a?(URI::HTTP)
rescue URI::InvalidURIError
  false
end

#validate_client_options! ⇒ Object

Raises:

  • (ArgumentError) 


233
234
235
236
237
238
 
# File 'lib/vra/client.rb', line 233

def validate_client_options!
  raise ArgumentError, "Username and password are required" if @username.nil? || @password.value.nil?
  raise ArgumentError, "A domain is required" if @domain.nil?
  raise ArgumentError, "A base URL is required" if @base_url.nil?
  raise ArgumentError, "Base URL #{@base_url} is not a valid URI." unless valid_uri?(@base_url)
end