#!/opt/cloudlinux/venv/bin/python3 -bb
import argparse
import subprocess

# The following imports might cause an ImportError if there's no alt-python27-cllib in the CL venv.
# However, cllib is always installed during the installation of the CloudLinux OS, so that is very unlikely.
# alt-python27-cllib is not a direct dependency of the rhn-client-tools package, because during the installation
# of the CloudLinux OS, rhn-client-tools must be installed before alt-python27-cllib.
from clcommon.lib.cledition import CLEditions
from clcommon.utils import is_ubuntu


def _run_edition_change_check(edition: str):
    # TODO: remove the check after resolving CLOS-3105
    if not is_ubuntu():
        # run interactively
        subprocess.run(['cloudlinux-edition-watcher', 'check', '--pre', '--edition', edition])


def main(jwt_token, transition_allowed: bool):
    """
    You add your other actions here, but don't forget to handle errors.
    @param transition_allowed:
        True means that we expect transition to start and it won't
        affect any system actions like another yum transaction.
    """
    if transition_allowed:
        edition = CLEditions.get_from_jwt(token=jwt_token)
        _run_edition_change_check(edition)


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('--new-token',
                        help='New jwt token value that will be later saved to file')

    parser.add_argument('--allow-transition', action='store_true', default=False)

    args = parser.parse_args()

    main(args.new_token, transition_allowed=args.allow_transition)