import os
import sys
import json
import requests
import time


def get_latest_version():
    response = None
    time_to_sleep = 2
    for _ in range(10):
        response = requests.get(
            'https://storage.googleapis.com/kubernetes-release/release/stable.txt')
        if response.status_code == 200:
            break
        print('Failed to obtain latest version info, retrying.')
        time.sleep(time_to_sleep)
        time_to_sleep *= 2
    return response.content.decode('utf-8')


version_arg = sys.argv[1]
installed_version_info = None
PASSED = True

try:
    print('kubectl version --client -o json')
    installed_version_info = json.load(
        os.popen('kubectl version --client -o json'))
    print(
        f'installed version: {installed_version_info["clientVersion"]["gitVersion"]}')
except Exception as ex:
    sys.exit('kubectl not installed')

try:
    installed_version = installed_version_info['clientVersion']['gitVersion']
    # NOT Match
    if version_arg[0] == '!':
        undesired_version = version_arg[1:]
        print(f'undesired version: {undesired_version}')
        if installed_version == undesired_version:
            print(
                f'installed version ({installed_version}) matches undesire {undesired_version} - FAIL')
            PASSED = False
    # Exact Match
    else:
        if version_arg == 'latest':
            print('checking latest version')
            desired_version = get_latest_version()
        else:
            desired_version = version_arg

        print(f'desired version: {desired_version}')
        if installed_version != desired_version:
            print(
                f'installed version ({installed_version}) does not match desired ({desired_version}) - FAIL')
            PASSED = False
except Exception as ex:
    print(f'Exception: {ex}')
    pass

if not PASSED:
    sys.exit('Setting up of '+version_arg+' kubectl failed')
print('Test passed')
sys.exit()