|
require 'rest_client'
|
|
|
|
module HammerCLICsv
|
|
class CsvCommand
|
|
class SubscriptionsCommand < BaseCommand
|
|
include ::HammerCLICsv::Utils::Subscriptions
|
|
|
|
command_name 'subscriptions'
|
|
desc 'import or export subscriptions'
|
|
|
|
def self.supported?
|
|
true
|
|
end
|
|
|
|
option %w(--in-portal), :flag, _('Import subscription comment lines into portal'),
|
|
:hidden => true
|
|
option %w(--portal-username), 'PORTAL_USERNAME', 'Portal username',
|
|
:hidden => true
|
|
option %w(--portal-password), 'PORTAL_PASSWORD', 'Portal password',
|
|
:hidden => true
|
|
option %w(--portal-hock), 'PORTAL_HOCK', 'Portal subscription creation address',
|
|
:hidden => true
|
|
option %w(--portal), 'PORTAL', 'Portal subscription access address',
|
|
:default => "https://subscription.rhn.redhat.com:443",
|
|
:hidden => true
|
|
|
|
ORGANIZATION = 'Organization'
|
|
MANIFEST = 'Manifest File'
|
|
|
|
def export(csv)
|
|
csv << [NAME, ORGANIZATION, MANIFEST, SUBS_NAME, SUBS_QUANTITY, SUBS_SKU, SUBS_CONTRACT, SUBS_ACCOUNT, SUBS_START, SUBS_END]
|
|
@api.resource(:organizations).call(:index, {:per_page => 999999})['results'].each do |organization|
|
|
next if option_organization && organization['name'] != option_organization
|
|
organization = @api.resource(:organizations).call(:show, {'id' => organization['id']})
|
|
export_manifest(csv, organization)
|
|
export_subscriptions(csv, organization)
|
|
end
|
|
end
|
|
|
|
def export_manifest(csv, organization)
|
|
info = organization['owner_details']['upstreamConsumer']
|
|
return if info.nil?
|
|
|
|
csv << ["Manifest Name", organization['name'], info['name']]
|
|
csv << ["Manifest URL", organization['name'], "https://#{info['webUrl']}#{info['uuid']}"]
|
|
end
|
|
|
|
def export_subscriptions(csv, organization)
|
|
@api.resource(:subscriptions).call(:index, {
|
|
'per_page' => 999999,
|
|
'search' => option_search,
|
|
'organization_id' => organization['id']
|
|
})['results'].each do |subscription|
|
|
next if subscription['product_id'].to_i != 0 # Red Hat subs do not have number SKU
|
|
name = subscription['host'].nil? ? "Subscription" : "Guest Subscription for Host '#{subscription['host']['name']}'"
|
|
quantity = subscription['quantity'] < 0 ? "Unlimited" : subscription['quantity']
|
|
csv << [name, organization['name'], nil, subscription['name'],
|
|
quantity, subscription['product_id'], subscription['contract_number'],
|
|
subscription['account_number'],
|
|
subscription['start_date'], subscription['end_date']]
|
|
end
|
|
end
|
|
|
|
def import
|
|
if option_in_portal?
|
|
import_into_portal
|
|
else
|
|
thread_import do |line|
|
|
if line[NAME] == 'Manifest' && line[MANIFEST] && !line[MANIFEST].empty?
|
|
import_manifest(line[ORGANIZATION], line[MANIFEST])
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
def import_manifest(organization_name, filename)
|
|
return if option_organization && organization_name != option_organization
|
|
print(_("Importing manifest '%{filename}' into organization '%{organization}'...") % {:filename => filename, :organization => organization_name}) if option_verbose?
|
|
args = %W{
|
|
--server #{ @server } --username #{ @username } --password #{ @password }
|
|
subscription upload --file #{ filename }
|
|
--organization-id #{ foreman_organization(:name => organization_name) }
|
|
}
|
|
hammer.run(args)
|
|
puts(_("done")) if option_verbose?
|
|
end
|
|
|
|
def import_into_portal
|
|
raise _("--portal-username and --portal-password required") unless option_portal_username && option_portal_password
|
|
raise _("--portal required") unless option_portal
|
|
|
|
@manifests = {}
|
|
|
|
thread_import do |line|
|
|
return if option_organization && line[ORGANIZATION] != option_organization
|
|
|
|
@manifests[line[ORGANIZATION]] ||= {}
|
|
import_subscription(line)
|
|
end
|
|
|
|
@manifests.each do |organization, manifest|
|
|
print _("Downloading manifest for organization '%{organization}...") % {:organization => organization} if option_verbose?
|
|
api = rest_client("/subscription/consumers/#{manifest[:manifest]['uuid']}/export")
|
|
data = api.get({'accept' => 'application/zip'})
|
|
filename = manifest[:file] || "#{manifest[:name]}.zip"
|
|
print _("writing to file '%{filename}'...") % {:filename => filename} if option_verbose?
|
|
File.open(filename, 'w') do |f|
|
|
f.binmode
|
|
f.write data
|
|
end
|
|
puts _("done") if option_verbose?
|
|
import_manifest(organization, filename)
|
|
end
|
|
end
|
|
|
|
def import_subscription(line)
|
|
case line[NAME]
|
|
when "Manifest Name"
|
|
print _("Checking manifest '%{name}'...") % {:name => line[MANIFEST]} if option_verbose?
|
|
@manifests[line[ORGANIZATION]][:name] = line[MANIFEST]
|
|
@manifests[line[ORGANIZATION]][:manifest] = get_or_create_manifest(line)
|
|
puts _("done") if option_verbose?
|
|
when "Manifest URL"
|
|
# ignore
|
|
when "Manifest"
|
|
@manifests[line[ORGANIZATION]][:file] = line[MANIFEST]
|
|
when "Subscription"
|
|
manifest = @manifests[line[ORGANIZATION]][:manifest]
|
|
raise _('Manifest Name row is required before updating from Subscription rows') unless manifest
|
|
line[SUBS_QUANTITY] = line[SUBS_QUANTITY].to_i #guarantee integer for future calculations
|
|
add_subscription(line, manifest)
|
|
else
|
|
# ignore
|
|
end
|
|
end
|
|
|
|
def add_subscription(line, manifest)
|
|
if find_existing_subscription(line, manifest)
|
|
puts _("'%{name}' of quantity %{quantity} already attached") %
|
|
{:name => line[SUBS_NAME], :quantity => line[SUBS_QUANTITY]} if option_verbose?
|
|
return
|
|
end
|
|
print _("Attaching '%{name}' of quantity %{quantity}...") %
|
|
{:name => line[SUBS_NAME], :quantity => line[SUBS_QUANTITY]} if option_verbose?
|
|
manifest['available_subscriptions'] ||= get_available_subscriptions(manifest)
|
|
attach_subscription(line, manifest)
|
|
puts _('done')
|
|
end
|
|
|
|
def attach_subscription(line, manifest)
|
|
manifest['available_subscriptions'].each do |subscription|
|
|
if subscription['productId'] == line[SUBS_SKU] && subscription['quantity'] >= line[SUBS_QUANTITY]
|
|
api = rest_client("/subscription/consumers/#{manifest['uuid']}/entitlements?pool=#{subscription['id']}&quantity=#{line[SUBS_QUANTITY]}")
|
|
results = api.post({}.to_json)
|
|
subscription['quantity'] -= line[SUBS_QUANTITY]
|
|
return
|
|
end
|
|
end
|
|
print _('subscription unavailable...')
|
|
end
|
|
|
|
def get_available_subscriptions(manifest)
|
|
api = rest_client("/subscription/pools/?consumer=#{manifest['uuid']}&listall=false")
|
|
JSON.parse(api.get)
|
|
end
|
|
|
|
def find_existing_subscription(line, manifest)
|
|
manifest['subscriptions'].each do |subscription|
|
|
if !subscription['csv_matched'] && subscription['pool']['productId'] == line[SUBS_SKU] && subscription['quantity'] == line[SUBS_QUANTITY]
|
|
subscription['csv_matched'] = true
|
|
return true
|
|
end
|
|
end
|
|
false
|
|
end
|
|
|
|
def get_or_create_manifest(line)
|
|
manifest = get_existing_manifest(line)
|
|
if manifest
|
|
if manifest['subscriptions'].nil?
|
|
api = rest_client("/subscription/consumers/#{manifest['uuid']}/entitlements")
|
|
results = JSON.parse(api.get)
|
|
manifest['subscriptions'] = results
|
|
end
|
|
else
|
|
api = rest_client("/subscription/consumers?owner=#{@manifests[line[ORGANIZATION]][:owner]}")
|
|
body = {
|
|
'name' => line[MANIFEST],
|
|
'type' => 'satellite',
|
|
'facts' => {
|
|
'distributor_version' => 'sat-6.0',
|
|
'system.certificate_version' => '3.2'
|
|
}
|
|
}
|
|
results = api.post(body.to_json,
|
|
{'accept' => 'json', 'content_type' => 'application/json'}
|
|
)
|
|
manifest = JSON.parse(results)
|
|
manifest['subscriptions'] = []
|
|
@manifests[line[ORGANIZATION]][:manifest] = manifest
|
|
end
|
|
manifest
|
|
end
|
|
|
|
def get_existing_manifest(line)
|
|
return @manifests[line[ORGANIZATION]][:manifest] if @manifests[line[ORGANIZATION]][:manifest]
|
|
|
|
unless @manifests[line[ORGANIZATION]][:owner]
|
|
api = rest_client("/subscription/users/#{option_portal_username}/owners")
|
|
@manifests[line[ORGANIZATION]][:owner] = JSON.parse(api.get)[0]['key']
|
|
end
|
|
|
|
api = rest_client("/subscription/owners/#{@manifests[line[ORGANIZATION]][:owner]}/consumers?type=satellite")
|
|
response = JSON.parse(api.get).each do |manifest|
|
|
if manifest['name'] == @manifests[line[ORGANIZATION]][:name]
|
|
@manifests[line[ORGANIZATION]][:manifest] = manifest
|
|
break
|
|
end
|
|
end
|
|
@manifests[line[ORGANIZATION]][:manifest]
|
|
end
|
|
|
|
def rest_client(path)
|
|
options = {
|
|
:headers => {
|
|
'accept' => 'application/json',
|
|
'accept-language' => HammerCLI::I18n.locale,
|
|
'content-type' => 'application/json'
|
|
},
|
|
:user => option_portal_username,
|
|
:password => option_portal_password,
|
|
:verify_ssl => OpenSSL::SSL::VERIFY_NONE
|
|
}
|
|
|
|
RestClient::Resource.new(option_portal + path, options)
|
|
end
|
|
end
|
|
end
|
|
end
|
