#!/usr/bin/env python

# File: sat6ShowHostSubscriptions.py
# Authors: Rich Jerrido <rjerrido@outsidaz.org>
#          Christopher Hornberger <christopher.hornberger@camptocamp.com>  
#
# Purpose: given an hostname and login to Satelite, show me all the 
#   hosts and their subscriptions.
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, version 2 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.

import json
import getpass
import urllib2
import base64
import sys
import ssl
from optparse import OptionParser

parser = OptionParser()
parser.add_option("-l", "--login", dest="login", help="Login user", metavar="LOGIN")
parser.add_option("-p", "--password", dest="password", help="Password for specified user. Will prompt if omitted", metavar="PASSWORD")
parser.add_option("-s", "--server", dest="server", help="FQDN of sat6 instance", metavar="SERVER")
(options, args) = parser.parse_args()

if not ( options.login and options.server):
	print "Must specify a login & server (will prompt for password if omitted).  See usage:"
	parser.print_help()
	print "\nExample usage: ./sat6ShowHostSubscriptions.py -l admin -s sat6.example.com "
	sys.exit(1)
else:
	login = options.login
	password = options.password
	server = options.server

if not password: password = getpass.getpass("%s's password:" % login)


if hasattr(ssl, '_create_unverified_context'):
	    ssl._create_default_https_context = ssl._create_unverified_context

#### Get the list of hosts
url = "https://" + server + "/api/v2/hosts?per_page=9999"
try:
 	request = urllib2.Request(url)
	print "Attempting to connect: " + url
	base64string = base64.encodestring('%s:%s' % (login, password)).strip()
	request.add_header("Authorization", "Basic %s" % base64string)
	result = urllib2.urlopen(request)
except urllib2.URLError, e:
	print "Error: cannot connect to the API: %s" % (e)
	print "Check your URL & try to login using the same user/pass via the WebUI and check the error!"
	sys.exit(1)
except:
	print "FATAL Error - %s" % (e)
	sys.exit(2)

hostdata = json.load(result)
for host in hostdata['results']:
	detailedurl = "https://" + server + "/api/v2/hosts/" + str(host["id"]) + "/subscriptions/"
	try:
		sysinfo = urllib2.Request(detailedurl)
		base64string = base64.encodestring('%s:%s' % (login, password)).strip()
		sysinfo.add_header("Authorization", "Basic %s" % base64string)
		sysresult = urllib2.urlopen(sysinfo)
		packagedata = json.load(sysresult)
		if packagedata.has_key("errors"):
			print packagedata["errors"]
		if packagedata["results"]:
			for package in packagedata['results']:
				print "%s; %s" % (host["certname"], package["name"])
	except urllib2.HTTPError as e:
		if str(e) == "HTTP Error 400: Bad Request":
			msg = e.read().decode()
			errormsg = json.loads(msg)
			print "%s; %s" % (host["certname"], str(errormsg["displayMessage"]))
		if str(e) != "HTTP Error 400: Bad Request":
			raise
			sys.exit(2)
                
sys.exit(0)