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

import spack.cmd
import spack.config
import spack.environment as ev
import spack.store
from spack.cmd.common import arguments
from spack.graph import DAGWithDependencyTypes, SimpleDAG, graph_ascii, graph_dot, static_graph_dot
from spack.llnl.util import tty

description = "generate graphs of package dependency relationships"
section = "query"
level = "long"


def setup_parser(subparser: argparse.ArgumentParser) -> None:
    setattr(setup_parser, "parser", subparser)
    subparser.epilog = """
Outside of an environment, the command concretizes specs and graphs them, unless the
--installed option is given. In that case specs are matched from the current DB.

If an environment is active, specs are matched from the currently available concrete specs
in the lockfile.

"""
    method = subparser.add_mutually_exclusive_group()
    method.add_argument(
        "-a", "--ascii", action="store_true", help="draw graph as ascii to stdout (default)"
    )
    method.add_argument(
        "-d", "--dot", action="store_true", help="generate graph in dot format and print to stdout"
    )

    subparser.add_argument(
        "-s",
        "--static",
        action="store_true",
        help="graph static (possible) deps, don't concretize (implies ``--dot``)",
    )
    subparser.add_argument(
        "-c",
        "--color",
        action="store_true",
        help="use different colors for different dependency types",
    )

    subparser.add_argument(
        "-i", "--installed", action="store_true", help="graph specs from the DB"
    )

    arguments.add_common_arguments(subparser, ["deptype", "specs"])


def graph(parser, args):
    env = ev.active_environment()
    if args.installed and env:
        args.subparser.error("cannot use --installed with an active environment")

    if args.color and not args.dot:
        args.subparser.error("the --color option can be used only with --dot")

    if args.installed:
        if not args.specs:
            specs = spack.store.STORE.db.query()
        else:
            result = []
            for item in args.specs:
                result.extend(spack.store.STORE.db.query(item))
            specs = list(set(result))
    elif env:
        specs = env.concrete_roots()
        if args.specs:
            specs = env.all_matching_specs(*args.specs)

    else:
        specs = spack.cmd.parse_specs(args.specs, concretize=not args.static)

    if not specs:
        tty.die("no spec matching the query")

    if args.static:
        static_graph_dot(specs, depflag=args.deptype)
        return

    if args.dot:
        builder = SimpleDAG()
        if args.color:
            builder = DAGWithDependencyTypes()
        graph_dot(specs, builder=builder, depflag=args.deptype)
        return

    # ascii is default: user doesn't need to provide it explicitly
    debug = spack.config.get("config:debug")
    graph_ascii(specs[0], debug=debug, depflag=args.deptype)
    for spec in specs[1:]:
        print()  # extra line bt/w independent graphs
        graph_ascii(spec, debug=debug)
