#!/usr/bin/python3

# Copyright (C) 2009-2011, Benjamin Drung <bdrung@debian.org>
#
# Permission to use, copy, modify, and/or distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
# WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
# MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
# ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
# WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
# ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
# OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.

# pylint: disable=invalid-name
# pylint: enable=invalid-name

"""provides information about Ubuntu's distributions"""

import argparse
import os
import sys

from distro_info import UbuntuDistroInfo, convert_date


def parse_args() -> argparse.Namespace:
    script_name = os.path.basename(sys.argv[0])
    usage = f"{script_name} [options]"
    epilog = f"See {script_name}(1) for more info."
    parser = argparse.ArgumentParser(usage=usage, epilog=epilog)

    parser.add_argument(
        "--date",
        dest="date",
        default=None,
        help="date for calculating the version (default: today).",
    )
    parser.add_argument(
        "-a", "--all", dest="all", action="store_true", help="list all known versions"
    )
    parser.add_argument(
        "-d", "--devel", dest="devel", action="store_true", help="latest development version"
    )
    parser.add_argument(
        "-s", "--stable", dest="stable", action="store_true", help="latest stable version"
    )
    parser.add_argument(
        "--lts", dest="lts", action="store_true", help="latest long term support (LTS) version"
    )
    parser.add_argument(
        "--supported",
        dest="supported",
        action="store_true",
        help="list of all supported versions (including development)",
    )
    parser.add_argument(
        "--unsupported",
        dest="unsupported",
        action="store_true",
        help="list of all unsupported stable versions",
    )

    args = parser.parse_args()

    versions = [args.all, args.devel, args.lts, args.stable, args.supported, args.unsupported]
    if len([x for x in versions if x]) != 1:
        parser.error(
            "You have to select exactly one of --all, --devel, --lts, "
            "--stable, --supported, --unsupported."
        )

    if args.date is not None:
        try:
            args.date = convert_date(args.date)
        except ValueError:
            parser.error("Option --date needs to be a date in ISO 8601 format.")

    return args


def main() -> None:
    args = parse_args()
    if args.all:
        for distro in UbuntuDistroInfo().all:
            sys.stdout.write(distro + "\n")
    elif args.devel:
        sys.stdout.write(UbuntuDistroInfo().devel(args.date) + "\n")
    elif args.lts:
        sys.stdout.write(UbuntuDistroInfo().lts(args.date) + "\n")
    elif args.stable:
        sys.stdout.write(UbuntuDistroInfo().stable(args.date) + "\n")
    elif args.supported:
        for distro in UbuntuDistroInfo().supported(args.date):
            sys.stdout.write(distro + "\n")
    elif args.unsupported:
        for distro in UbuntuDistroInfo().unsupported(args.date):
            sys.stdout.write(distro + "\n")


if __name__ == "__main__":
    main()
