truenas/library/ix-dev/community/minecraft/upgrade_strategy

#!/usr/bin/python3
import json
import re
import sys

from catalog_update.upgrade_strategy import semantic_versioning

RE_STABLE_VERSION_BASE = r'\d+\.\d+\.\d+'
ENUMS = {
    'j8Image': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java8'),
        'STRIP_TEXT': '-java8'
    },
    'j8gceImage': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java8-graalvm-ce'),
        'STRIP_TEXT': '-java8-graalvm-ce'
    },
    'j8j9Image': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java8-openj9'),
        'STRIP_TEXT': '-java8-openj9'
    },
    'j8jdkImage': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java8-jdk'),
        'STRIP_TEXT': '-java8-jdk'
    },
    'j11Image': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java11'),
        'STRIP_TEXT': '-java11'
    },
    'j11j9Image': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java11-openj9'),
        'STRIP_TEXT': '-java11-openj9'
    },
    'j11jdkImage': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java11-jdk'),
        'STRIP_TEXT': '-java11-jdk'
    },
    'j17Image': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java17'),
        'STRIP_TEXT': '-java17'
    },
    'j17j9Image': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java17-openj9'),
        'STRIP_TEXT': '-java17-openj9'
    },
    'j17jdkImage': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java17-jdk'),
        'STRIP_TEXT': '-java17-jdk'
    },
    'j17gceImage': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java17-graalvm-ce'),
        'STRIP_TEXT': '-java17-graalvm-ce'
    },
    'j17alpineImage': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java17-alpine'),
        'STRIP_TEXT': '-java17-alpine'
    },
    'j19Image': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java19'),
        'STRIP_TEXT': '-java19'
    },
    'j21Image': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java21'),
        'STRIP_TEXT': '-java21'
    },
    'j21gImage': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java21-graalvm'),
        'STRIP_TEXT': '-java21-graalvm'
    },
    'j21alpineImage': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-java21-alpine'),
        'STRIP_TEXT': '-java21-alpine'
    }
}


def newer_mapping(image_tags):
    output = {
        "tags": {},
        "app_version": ""
    }

    for key in image_tags.keys():
        STRIP_TEXT = ENUMS[key].get('STRIP_TEXT', None) if key in ENUMS else None
        RE_STABLE_VERSION = ENUMS[key].get('RE_STABLE_VERSION', None) if key in ENUMS else None

        if (STRIP_TEXT is None) or (RE_STABLE_VERSION is None):
            continue

        tags = {t.strip(STRIP_TEXT): t for t in image_tags[key] if RE_STABLE_VERSION.fullmatch(t)}
        version = semantic_versioning(list(tags))

        if not version:
            continue

        if key == 'j8Image':
            output['app_version'] = version

        output['tags'][key] = tags[version]

    return output


if __name__ == '__main__':
    try:
        versions_json = json.loads(sys.stdin.read())
    except ValueError:
        raise ValueError('Invalid json specified')

    print(json.dumps(newer_mapping(versions_json)))