# -*- coding: utf-8 -*-


# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; version 2 of the License.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU Library General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, see <https://gnu.org/licenses/>.


import argparse
import logging

from .tree import Tree
from .installer import Installer


def main(args=None):
    parser = argparse.ArgumentParser()
    subparser = parser.add_subparsers(help="Sub commands")

    treep = subparser.add_parser("tree", help="Compose OSTree repository")
    treep.set_defaults(_class=Tree, func="run")
    treep.add_argument(
        "--repo",
        metavar="PATH",
        required=True,
        help="where to put the OSTree repo (required)",
    )
    treep.add_argument(
        "--treefile",
        metavar="FILE",
        required=True,
        help="treefile for rpm-ostree (required)",
    )
    treep.add_argument(
        "--log-dir",
        metavar="DIR",
        required=True,
        help="where to log output and commitid (required). \
                             Note: commitid file will be written to this dir",
    )
    treep.add_argument(
        "--extra-config", metavar="FILE", help="JSON file contains extra configurations"
    )
    treep.add_argument(
        "--version",
        metavar="VERSION",
        help="version string to be added as versioning metadata",
    )
    treep.add_argument(
        "--update-summary", action="store_true", help="update summary metadata"
    )
    treep.add_argument(
        "--ostree-ref", metavar="PATH", help="override ref value from treefile"
    )
    treep.add_argument(
        "--force-new-commit",
        action="store_true",
        help="do not use rpm-ostree's built-in change detection",
    )
    treep.add_argument(
        "--unified-core",
        action="store_true",
        help="use unified core mode in rpm-ostree",
    )

    installerp = subparser.add_parser(
        "installer", help="Create an OSTree installer image"
    )
    installerp.set_defaults(_class=Installer, func="run")
    installerp.add_argument(
        "-p",
        "--product",
        metavar="PRODUCT",
        required=True,
        help="product name (required)",
    )
    installerp.add_argument(
        "-v",
        "--version",
        metavar="VERSION",
        required=True,
        help="version identifier (required)",
    )
    installerp.add_argument(
        "-r",
        "--release",
        metavar="RELEASE",
        required=True,
        help="release information (required)",
    )
    installerp.add_argument(
        "-s",
        "--source",
        metavar="REPOSITORY",
        required=True,
        action="append",
        help="source repository (required)",
    )
    installerp.add_argument(
        "-o",
        "--output",
        metavar="DIR",
        required=True,
        help="path to image output directory (required)",
    )
    installerp.add_argument("--log-dir", metavar="DIR", help="path to log directory")
    installerp.add_argument("--volid", metavar="VOLID", help="volume id")
    installerp.add_argument("--variant", metavar="VARIANT", help="variant name")
    installerp.add_argument("--rootfs-size", metavar="SIZE")
    installerp.add_argument("--nomacboot", action="store_true", default=False)
    installerp.add_argument("--noupgrade", action="store_true", default=False)
    installerp.add_argument("--isfinal", action="store_true", default=False)

    installerp.add_argument(
        "--installpkgs",
        metavar="PACKAGE",
        action="append",
        help="package glob to install before runtime-install.tmpl",
    )
    installerp.add_argument(
        "--add-template",
        metavar="FILE",
        action="append",
        help="Additional template for runtime image",
    )
    installerp.add_argument(
        "--add-template-var",
        metavar="ADD_TEMPLATE_VARS",
        action="append",
        help="Set variable for runtime image template",
    )
    installerp.add_argument(
        "--add-arch-template",
        metavar="FILE",
        action="append",
        help="Additional template for architecture-specific image",
    )
    installerp.add_argument(
        "--add-arch-template-var",
        metavar="ADD_ARCH_TEMPLATE_VARS",
        action="append",
        help="Set variable for architecture-specific image",
    )

    installerp.add_argument(
        "--extra-config", metavar="FILE", help="JSON file contains extra configurations"
    )

    args = parser.parse_args(args)

    logging.basicConfig(format="%(message)s", level=logging.DEBUG)

    _class = args._class()
    _class.set_args(args)
    func = getattr(_class, args.func)
    func()