# Copyright Spack Project Developers. See COPYRIGHT file for details.
#
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
import argparse
import os

import spack.container
import spack.container.images
import spack.llnl.util.tty

description = "create a container build recipe from an environment"
section = "environments"
level = "long"


def setup_parser(subparser: argparse.ArgumentParser) -> None:
    subparser.add_argument(
        "--list-os",
        action="store_true",
        default=False,
        help="list all the OS that can be used in the bootstrap phase and exit",
    )
    subparser.add_argument(
        "--last-stage",
        choices=("bootstrap", "build", "final"),
        default="final",
        help="last stage in the container recipe",
    )


def containerize(parser, args):
    if args.list_os:
        possible_os = spack.container.images.all_bootstrap_os()
        msg = "The following operating systems can be used to bootstrap Spack:"
        msg += "\n{0}".format(" ".join(possible_os))
        spack.llnl.util.tty.msg(msg)
        return

    config_dir = args.env_dir or os.getcwd()
    config_file = os.path.abspath(os.path.join(config_dir, "spack.yaml"))
    if not os.path.exists(config_file):
        msg = "file not found: {0}"
        raise ValueError(msg.format(config_file))

    config = spack.container.validate(config_file)
    recipe = spack.container.recipe(config, last_phase=args.last_stage)
    print(recipe)
