mirror of
https://github.com/puppetlabs/vmfloaty.git
synced 2026-01-26 05:28:40 -05:00
a3d8484124
- If ABS queries returns a body for 200 or 202, floaty will print it this is useful in the new version of ABS, since it shows the progress for ondemand requests (AWS or vmpooler) - removed the queue_place and querying the queue for a 'get' request this queue_place number was misleading since it was just a redis index and did not represent well where the request was in the queue - Also added a flag option --continue to be used when the cli was interrupted for example with ctrl-c
119 lines
3.5 KiB
Ruby
119 lines
3.5 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require 'vmfloaty/errors'
|
|
require 'vmfloaty/http'
|
|
require 'faraday'
|
|
require 'json'
|
|
|
|
class NonstandardPooler
|
|
def self.list(verbose, url, os_filter = nil)
|
|
conn = Http.get_conn(verbose, url)
|
|
|
|
response = conn.get 'status'
|
|
response_body = JSON.parse(response.body)
|
|
os_list = response_body.keys.sort
|
|
os_list.delete 'ok'
|
|
|
|
os_filter ? os_list.select { |i| i[/#{os_filter}/] } : os_list
|
|
end
|
|
|
|
def self.list_active(verbose, url, token, _user)
|
|
status = Auth.token_status(verbose, url, token)
|
|
status['reserved_hosts'] || []
|
|
end
|
|
|
|
def self.retrieve(verbose, os_type, token, url, _user, _options, ondemand = nil, _continue = nil)
|
|
conn = Http.get_conn(verbose, url)
|
|
conn.headers['X-AUTH-TOKEN'] = token if token
|
|
|
|
os_string = os_type.map { |os, num| Array(os) * num }.flatten.join('+')
|
|
raise MissingParamError, 'No operating systems provided to obtain.' if os_string.empty?
|
|
|
|
response = conn.post "host/#{os_string}"
|
|
|
|
res_body = JSON.parse(response.body)
|
|
|
|
if res_body['ok']
|
|
res_body
|
|
elsif response.status == 401
|
|
raise AuthError, "HTTP #{response.status}: The token provided could not authenticate to the pooler.\n#{res_body}"
|
|
else
|
|
raise "HTTP #{response.status}: Failed to obtain VMs from the pooler at #{url}/host/#{os_string}. #{res_body}"
|
|
end
|
|
end
|
|
|
|
def self.modify(verbose, url, hostname, token, modify_hash)
|
|
raise TokenError, 'Token provided was nil; Request cannot be made to modify VM' if token.nil?
|
|
|
|
modify_hash.each do |key, _value|
|
|
raise ModifyError, "Configured service type does not support modification of #{key}" unless %i[reason reserved_for_reason].include? key
|
|
end
|
|
|
|
if modify_hash[:reason]
|
|
# "reason" is easier to type than "reserved_for_reason", but nspooler needs the latter
|
|
modify_hash[:reserved_for_reason] = modify_hash.delete :reason
|
|
end
|
|
|
|
conn = Http.get_conn(verbose, url)
|
|
conn.headers['X-AUTH-TOKEN'] = token
|
|
|
|
response = conn.put do |req|
|
|
req.url "host/#{hostname}"
|
|
req.body = modify_hash.to_json
|
|
end
|
|
|
|
response.body.empty? ? {} : JSON.parse(response.body)
|
|
end
|
|
|
|
def self.disk(_verbose, _url, _hostname, _token, _disk)
|
|
raise ModifyError, 'Configured service type does not support modification of disk space'
|
|
end
|
|
|
|
def self.snapshot(_verbose, _url, _hostname, _token)
|
|
raise ModifyError, 'Configured service type does not support snapshots'
|
|
end
|
|
|
|
def self.revert(_verbose, _url, _hostname, _token, _snapshot_sha)
|
|
raise ModifyError, 'Configured service type does not support snapshots'
|
|
end
|
|
|
|
def self.delete(verbose, url, hosts, token, _user)
|
|
raise TokenError, 'Token provided was nil; Request cannot be made to delete VM' if token.nil?
|
|
|
|
conn = Http.get_conn(verbose, url)
|
|
|
|
conn.headers['X-AUTH-TOKEN'] = token if token
|
|
|
|
response_body = {}
|
|
|
|
hosts = hosts.split(',') unless hosts.is_a? Array
|
|
hosts.each do |host|
|
|
response = conn.delete "host/#{host}"
|
|
res_body = JSON.parse(response.body)
|
|
response_body[host] = res_body
|
|
end
|
|
|
|
response_body
|
|
end
|
|
|
|
def self.status(verbose, url)
|
|
conn = Http.get_conn(verbose, url)
|
|
|
|
response = conn.get '/status'
|
|
JSON.parse(response.body)
|
|
end
|
|
|
|
def self.summary(verbose, url)
|
|
conn = Http.get_conn(verbose, url)
|
|
|
|
response = conn.get '/summary'
|
|
JSON.parse(response.body)
|
|
end
|
|
|
|
def self.query(verbose, url, hostname)
|
|
conn = Http.get_conn(verbose, url)
|
|
|
|
response = conn.get "host/#{hostname}"
|
|
JSON.parse(response.body)
|
|
end
|
|
end
|