codeql/java/kotlin-extractor/kotlin_plugin_versions.py

import platform
import re
import subprocess
import sys

def is_windows():
    '''Whether we appear to be running on Windows'''
    if platform.system() == 'Windows':
        return True
    if platform.system().startswith('CYGWIN'):
        return True
    return False

def version_tuple_to_string(version):
    return f'{version[0]}.{version[1]}.{version[2]}'

def version_string_to_tuple(version):
    m = re.match(r'([0-9]+)\.([0-9]+)\.([0-9]+)', version)
    return tuple([int(m.group(i)) for i in range(1, 4)])

many_versions = [ '1.4.32', '1.5.31', '1.6.10', '1.6.20' ]

many_versions_tuples = [version_string_to_tuple(v) for v in many_versions]

def get_single_version(fakeVersionOutput = None):
    # TODO: `shell=True` is a workaround to get CI working on Windows. It breaks the build on Linux.
    versionOutput = subprocess.run(['kotlinc', '-version'], stdout=subprocess.PIPE, stderr=subprocess.PIPE, universal_newlines=True, shell=is_windows()).stderr if fakeVersionOutput is None else fakeVersionOutput
    m = re.match(r'.* kotlinc-jvm ([0-9]+\.[0-9]+\.[0-9]+) .*', versionOutput)
    if m is None:
        raise Exception('Cannot detect version of kotlinc (got ' + str(versionOutput) + ')')
    current_version = version_string_to_tuple(m.group(1))
    matching_minor_versions = [ version for version in many_versions_tuples if version[0:2] == current_version[0:2] ]
    if len(matching_minor_versions) == 0:
        raise Exception(f'Cannot find a matching minor version for kotlinc version {current_version} (got {versionOutput}; know about {str(many_versions)})')

    matching_minor_versions.sort()

    for version in matching_minor_versions:
        if version >= current_version:
            return version_tuple_to_string(version)

    return version_tuple_to_string(matching_minor_versions[-1])

    raise Exception(f'No suitable kotlinc version found for {current_version} (got {versionOutput}; know about {str(many_versions)})')

def get_latest_url():
    version = many_versions[-1]
    url = 'https://github.com/JetBrains/kotlin/releases/download/v' + version + '/kotlin-compiler-' + version + '.zip'
    return url

if __name__ == "__main__":
    args = sys.argv
    if len(args) < 2:
        raise Exception("Bad arguments")
    command = args[1]
    if command == 'latest-url':
        print(get_latest_url())
    elif command == 'single-version':
        print(get_single_version(*args[2:]))
    else:
        raise Exception("Unknown command: " + command)