#!/usr/libexec/platform-python

from __future__ import print_function

import sys
sys.path.append('/usr/share/rhn')

try:
    from up2date_client import up2dateAuth, rpcServer
except ImportError:
    # This is not a ULN-enabled system
    sys.exit(123)

import Uptrack
from six.moves.xmlrpc_client import Fault

system_id = up2dateAuth.getSystemId()
if system_id is None:
    # Not registered.
    sys.exit(124)

server = rpcServer.getServer()
try:
    result = server.system.get_ksplice_key(system_id)
except Fault as e:
    print("Unable to retrieve the Ksplice access key from ULN. ({0})".format(str(e)),
          file=sys.stderr)
    sys.exit(126)


access_key = result.get("key")
if access_key and str(access_key) != "-1":
    print(access_key)
    try:
        cfg = Uptrack.UptrackConfig()
        old_access_key = cfg.accesskey
    except Exception:
        old_access_key = None
    if old_access_key and old_access_key != "INSERT_ACCESS_KEY" and old_access_key != access_key:
        print("Your Ksplice access key is {0} according to ULN,".format(access_key), file=sys.stderr)
        print("but {0} uses a different key.".format(Uptrack.UPTRACK_CONFIG_FILE), file=sys.stderr)
else:
    print("Unable to retrieve the Ksplice access key from ULN.", file=sys.stderr)
    sys.exit(125)