Axsi Mini Shell

import logging
import time

from uaclient import actions, exceptions, lock, messages, util
from uaclient.clouds import AutoAttachCloudInstance
from uaclient.clouds.gcp import UAAutoAttachGCPInstance
from uaclient.clouds.identity import cloud_instance_factory
from uaclient.config import UAConfig

LOG = logging.getLogger("ua.daemon")


def start():
    try:
        util.subp(["systemctl", "start", "ubuntu-advantage.service"])
    except exceptions.ProcessExecutionError as e:
        LOG.warning(e)


def stop():
    try:
        util.subp(["systemctl", "stop", "ubuntu-advantage.service"])
    except exceptions.ProcessExecutionError as e:
        LOG.warning(e)


def attempt_auto_attach(cfg: UAConfig, cloud: AutoAttachCloudInstance):
    try:
        with lock.SpinLock(
            cfg=cfg, lock_holder="ua.daemon.attempt_auto_attach"
        ):
            actions.auto_attach(cfg, cloud)
    except exceptions.LockHeldError as e:
        LOG.error(e)
        cfg.add_notice(
            "",
            messages.NOTICE_DAEMON_AUTO_ATTACH_LOCK_HELD.format(
                operation=e.lock_holder
            ),
        )
        LOG.debug("Failed to auto attach")
        return
    except Exception as e:
        LOG.exception(e)
        cfg.add_notice("", messages.NOTICE_DAEMON_AUTO_ATTACH_FAILED)
        lock.clear_lock_file_if_present()
        LOG.debug("Failed to auto attach")
        return
    LOG.debug("Successful auto attach")


def poll_for_pro_license(cfg: UAConfig):
    if util.is_config_value_true(
        config=cfg.cfg, path_to_value="features.disable_auto_attach"
    ):
        LOG.debug("Configured to not auto attach, shutting down")
        return
    if cfg.is_attached:
        LOG.debug("Already attached, shutting down")
        return
    if not util.is_current_series_lts():
        LOG.debug("Not on LTS, shutting down")
        return

    try:
        cloud = cloud_instance_factory()
    except exceptions.CloudFactoryError:
        LOG.debug("Not on cloud, shutting down")
        return

    if not isinstance(cloud, UAAutoAttachGCPInstance):
        LOG.debug("Not on gcp, shutting down")
        return

    if not cloud.should_poll_for_pro_license():
        LOG.debug("Not on supported instance, shutting down")
        return

    try:
        pro_license_present = cloud.is_pro_license_present(
            wait_for_change=False
        )
    except exceptions.CancelProLicensePolling:
        LOG.debug("Cancelling polling")
        return
    except exceptions.DelayProLicensePolling:
        # Continue to polling loop anyway and handle error there if it occurs
        # again
        pass
    else:
        if pro_license_present:
            attempt_auto_attach(cfg, cloud)
            return

    if not cfg.poll_for_pro_license:
        LOG.debug("Configured to not poll for pro license, shutting down")
        return

    while True:
        try:
            start = time.time()
            pro_license_present = cloud.is_pro_license_present(
                wait_for_change=True
            )
            end = time.time()
        except exceptions.CancelProLicensePolling:
            LOG.debug("Cancelling polling")
            return
        except exceptions.DelayProLicensePolling:
            time.sleep(cfg.polling_error_retry_delay)
            continue
        else:
            if cfg.is_attached:
                # This could have changed during the long poll or sleep
                LOG.debug("Already attached, shutting down")
                return
            if pro_license_present:
                attempt_auto_attach(cfg, cloud)
                return
            if end - start < 10:
                LOG.debug(
                    "wait_for_change returned quickly and no pro license"
                    " present. Waiting {} seconds before polling again".format(
                        cfg.polling_error_retry_delay
                    )
                )
                time.sleep(cfg.polling_error_retry_delay)
                continue