import argparse

from rasa.cli.arguments.default_arguments import (
    add_domain_param,
    add_stories_param,
    add_out_param,
    add_nlu_data_param,
)


def set_visualize_stories_arguments(parser: argparse.ArgumentParser) -> None:
    """Sets the CLI arguments for `rasa data visualize."""
    add_domain_param(parser)
    add_stories_param(parser)

    add_out_param(
        parser,
        default="graph.html",
        help_text="Filename of the output path, e.g. 'graph.html'.",
    )

    parser.add_argument(
        "--max-history",
        default=2,
        type=int,
        help="Max history to consider when merging paths in the output graph.",
    )

    add_nlu_data_param(
        parser,
        default=None,
        help_text="File or folder containing your NLU data, "
        "used to insert example messages into the graph.",
    )