import json
from pprint import pprint

def lambda_handler(event, context):
    pprint(event)
    
    uri = event['uri']
    image_name, tag_name = uri.split(':')
    
    name_parts = image_name.split('/')
    if len(name_parts) > 1:
        registry_name = name_parts[0]
        image_name = "/".join(name_parts[1:])
    else:
        registry_name = 'dockerhub'

    with open('public_registry_properties.json', 'r') as f:
        registry_props = json.load(f)

    if registry_props.get(registry_name):
        props = registry_props[registry_name]
        ecr_repo_name = "/".join([props['namespace'], image_name])
    else:
        # unrecognized registry, pass it through
        props = { "namespace": None, "pull_through": False }
        ecr_repo_name = "/".join(name_parts)

    return {
        "image": {
            "source_uri": uri,
            "target_image": ":".join([ecr_repo_name, tag_name]),
            "ecr_repository": ecr_repo_name,
            "tag": tag_name,
            "pull_through_supported": str(props["pull_through"]).lower(),
        }
    }