Source code for pepys_import.resolvers.command_line_input

from prompt_toolkit import prompt
from prompt_toolkit.completion import FuzzyCompleter, WordCompleter
from prompt_toolkit.validation import Validator

from pepys_import.utils.text_formatting_utils import format_menu


def get_fuzzy_completer(completer_list):
    return FuzzyCompleter(WordCompleter(completer_list), pattern="^[a-zA-Z0-9_ ]*")


def is_valid(option):
    return option in [str(1), str(2), ".", "?", "HELP"]


def create_menu(title, choices, cancel="import", completer=None, validate_method=None):
    """
    A basic function which creates a menu with title and choices.

    :param cancel:
    :param title: Heading text
    :type title: String
    :param choices: Options to choose
    :type choices: List of strings
    :param completer: Optional argument that shows possible options while typing.
    :type completer: :class:`prompt_toolkit.completion.FuzzyWordCompleter`
    :param validate_method: Possible validator function
    :type validate_method: Function
    :return: Entered choice
    :rtype: String
    """
    validator = None
    if validate_method is not None:
        validator = Validator.from_callable(
            validate_method,
            error_message="You didn't select a valid option",
            move_cursor_to_end=True,
        )

    title_, choices_ = "\n" + title + "\n", ""
    for index, choice in enumerate(choices, 1):
        choices_ += f"   {str(index)}) {choice}\n"
    choices_ += f"   .) Cancel {cancel}\n > "
    choice = prompt(format_menu(title_, choices_), completer=completer, validator=validator)

    return choice