#!/usr/bin/python3

import os
import sys
import yaml


def migrate(values):
    values["network"]["web_port"] = {
        "port_number": values["network"]["web_port"],
        "bind_mode": "published",
        "host_ips": [],
    }
    extra_ports = []

    for port in values["network"].get("additional_ports", []):
        extra_ports.append(
            {
                "bind_mode": "published",
                "port_number": port,
                "container_port": port,
                "protocol": "tcp",
                "host_ips": [],
            }
        )
        extra_ports.append(
            {
                "bind_mode": "published",
                "port_number": port,
                "container_port": port,
                "protocol": "udp",
                "host_ips": [],
            }
        )
    values["network"]["additional_ports"] = extra_ports

    return values


if __name__ == "__main__":
    if len(sys.argv) != 2:
        exit(1)

    if os.path.exists(sys.argv[1]):
        with open(sys.argv[1], "r") as f:
            print(yaml.dump(migrate(yaml.safe_load(f.read()))))