#!/opt/cloudlinux/venv/bin/python3 -bb
# coding=utf-8
#
# Copyright © Cloud Linux GmbH & Cloud Linux Software, Inc 2010-2020 All Rights Reserved
#
# Licensed under CLOUD LINUX LICENSE AGREEMENT
# http://cloudlinux.com/docs/LICENCE.TXT
#
"""
Detects edition change and performs predefined actions.
"""
import sys
import argparse
import logging

from clcommon.lib.cledition import CLEditions
from clsentry import init_sentry_client

from clconfigure.watcher import (
    PKG_VERSION_TINY,
    SENTRY_DSN,
    check,
    save_edition
)

from clconfigure import setup_logger

LOGS_PATH = '/var/log/cloudlinux/cloudlinux-edition-watcher.log'

if __name__ == "__main__":
    init_sentry_client('cloudlinux-edition-watcher', PKG_VERSION_TINY, SENTRY_DSN, custom_length=7000)

    # Make sure we log to stderr - this script can be run manually, not just from cron
    watcher_logger = logging.getLogger(None)
    watcher_logger.addHandler(logging.StreamHandler(sys.stderr))
    # setup_logger sets INFO level and formatter for root logger stderr stream handlers,
    # so we don't need to assign those here
    setup_logger(None, LOGS_PATH)

    parser = argparse.ArgumentParser()

    subparsers = parser.add_subparsers(dest='action')
    subparsers.add_parser('save')
    check_subparser = subparsers.add_parser('check')
    check_subparser.add_argument('--pre', action='store_true', default=False)
    check_subparser.add_argument('--edition', required=False, default=None)

    args = parser.parse_args()

    logging.debug('Executing with arguments "%s"', str(args))

    if args.action == 'save':
        save_edition()
    elif args.action == 'check':
        new_edition = args.edition or CLEditions.get_cl_edition(skip_marker_check=True)
        if new_edition is None:
            logging.error("error: could not determine the current CloudLinux edition")
            logging.error(
                "Please, check the machine's registration state using rhn_check and force "
                "a re-registration if necessary. Contact CloudLinux support if the problem persists."
            )
            exit(1)
        check(new_edition, is_pre_check=args.pre)
    else:
        parser.print_help()
        exit(1)