dists

#!/usr/bin/env python3

import os
import sys
import itertools

from distro_info import DebianDistroInfo, UbuntuDistroInfo


def get_sorted_dists(distro_functions):
    dist_objects = itertools.chain(
        *(func(result="object") for func in distro_functions)
    )
    unique_dists = {
        dist.series for dist in sorted(dist_objects, key=lambda x: x.created)
    }
    return sorted(unique_dists)


def get_debian_dists():
    debian = DebianDistroInfo()

    oldstable = debian.codename("oldstable")
    stable = debian.codename("stable")
    testing = debian.codename("testing")

    dists = [oldstable, stable, testing, "unstable", "experimental"]

    dists.extend(
        [
            f"{release}-{suffix}"
            for release in [oldstable, stable, testing]
            for suffix in ["backports", "proposed-updates", "updates"]
        ]
    )

    dists.extend([f"{release}-backports-sloppy" for release in [oldstable, stable]])
    return ",".join(sorted(dists))


def get_ubuntu_dists():
    ubuntu_info = UbuntuDistroInfo()

    sorted_ubuntu_dists = get_sorted_dists(
        [ubuntu_info.supported]
    )  # , ubuntu_info.supported_esm])
    dists = []
    for release in sorted_ubuntu_dists:
        dists.extend(
            [
                f"{release}-{suffix}"
                for suffix in ["proposed", "updates", "backports", "security"]
            ]
        )
        dists.append(release)
    return ",".join(dists)


def main():
    if len(sys.argv) != 2 or sys.argv[1] not in ["debian", "ubuntu", "ubuntu-ports"]:
        script = os.path.basename(sys.argv[0])
        print(f"Usage: {script} [debian|ubuntu|ubuntu-ports]", file=sys.stderr)
        sys.exit(1)

    if sys.argv[1] == "debian":
        print(get_debian_dists())
        sys.exit(0)

    print(get_ubuntu_dists())


if __name__ == "__main__":
    main()