import yaml

from varken import VERSION

org = 'boerderij'
project = 'varken'
namespace = f"{org}/{project}"

yaml_arr = []
tags = ['latest', VERSION]

# Docker image, arch, variant, os
arch_list = [('arm', 'arm', 'v6', 'linux'),
             ('armhf', 'arm', 'v7', 'linux'),
             ('arm64', 'arm64', 'v8', 'linux'),
             ('amd64', 'amd64', None, 'linux')]

for tag in tags:
    yaml_doc = {
        'image': f'{namespace}:{tag}',
        'manifests': []
    }
    for arch in arch_list:
        info = {
            'image': f"{namespace}:{tag}-{arch[0]}",
            'platform': {
                'architecture': arch[1],
                'os': arch[3]
            }
        }
        if arch[2]:
                info['platform']['variant'] = arch[2]
        yaml_doc['manifests'].append(info)
    yaml_arr.append(yaml_doc)

with open(f".manifest.yaml", 'w') as file:
    yaml.dump_all(yaml_arr, file, default_flow_style=False)