#!/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 = {
    'image': {
        'RE_STABLE_VERSION': re.compile(RE_STABLE_VERSION_BASE),
        'STRIP_TEXT': ''
    },
    'tensorrtImage': {
        'RE_STABLE_VERSION': re.compile(rf'{RE_STABLE_VERSION_BASE}-tensorrt'),
        'STRIP_TEXT': '-tensorrt'
    },
}


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 == 'image':
            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)))
