# -*- coding: utf-8 -*-
from copy import copy

import pytest

from questionary import Choice
from questionary import Separator
from tests.utils import KeyInputs
from tests.utils import feed_cli_with_input
from tests.utils import patched_prompt


def test_legacy_name():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bar", "bazz"]}
    text = KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("list", message, text, **kwargs)
    assert result == "foo"


def test_select_first_choice():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bar", "bazz"]}
    text = KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "foo"


def test_select_first_choice_with_token_title():
    message = "Foo message"
    kwargs = {
        "choices": [
            Choice(title=[("class:text", "foo")]),
            Choice(title=[("class:text", "bar")]),
            Choice(title=[("class:text", "bazz")]),
        ]
    }
    text = KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "foo"


def test_select_second_choice():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bar", "bazz"]}
    text = KeyInputs.DOWN + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bar"


def test_select_third_choice():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bar", "bazz"]}
    text = KeyInputs.DOWN + KeyInputs.DOWN + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_select_third_choice_using_shortcuts_and_arrows():
    message = "Foo message"
    kwargs = {
        "choices": ["foo", "bar", "bazz"],
        "use_shortcuts": True,
    }
    text = KeyInputs.TWO + KeyInputs.DOWN + KeyInputs.ENTER + "\r"
    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_select_second_choice_using_j_k():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bar", "bazz"]}
    text = "jjk" + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bar"


def test_select_second_choice_using_emacs_keys():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bar", "bazz"]}
    text = (
        KeyInputs.CONTROLN
        + KeyInputs.CONTROLN
        + KeyInputs.CONTROLP
        + KeyInputs.ENTER
        + "\r"
    )

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bar"


def test_select_with_instruction():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bar", "bazz"], "instruction": "sample instruction"}
    text = KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "foo"


def test_cycle_to_first_choice():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bar", "bazz"]}
    text = KeyInputs.DOWN + KeyInputs.DOWN + KeyInputs.DOWN + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "foo"


def test_cycle_backwards():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bar", "bazz"]}
    text = KeyInputs.UP + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_cycle_backwards_using_k():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bar", "bazz"]}
    text = "k" + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_separator_down():
    message = "Foo message"
    kwargs = {"choices": ["foo", Separator(), "bazz"]}
    text = KeyInputs.DOWN + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_separator_up():
    message = "Foo message"
    kwargs = {"choices": ["foo", Separator(), "bazz", Separator("--END--")]}
    text = KeyInputs.UP + KeyInputs.UP + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "foo"


def test_select_random_input():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bazz"]}
    text = "some random input" + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "foo"


def test_select_ctr_c():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bazz"]}
    text = KeyInputs.CONTROLC

    with pytest.raises(KeyboardInterrupt):
        feed_cli_with_input("select", message, text, **kwargs)


def test_select_empty_choices():
    message = "Foo message"
    kwargs = {"choices": []}
    text = KeyInputs.ENTER + "\r"

    with pytest.raises(ValueError):
        feed_cli_with_input("select", message, text, **kwargs)


def test_disallow_shortcut_key():
    message = "Foo message"
    kwargs = {
        "choices": ["foo", "bar", Choice("bazz", shortcut_key=False)],
        "use_shortcuts": True,
    }
    text = KeyInputs.THREE + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "foo"


def test_allow_shortcut_key_with_True():
    message = "Foo message"
    kwargs = {
        "choices": ["foo", "bar", Choice("bazz", shortcut_key=True)],
        "use_shortcuts": True,
    }
    text = KeyInputs.THREE + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_auto_shortcut_key_stable_in_loop():
    message = "Foo message"
    choices = [
        Choice("foo"),
        Choice("bar"),
    ]
    kwargs = {
        "choices": choices,
        "use_shortcuts": True,
    }
    text = "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "foo"
    result_shortcut_keys = [copy(c.shortcut_key) for c in choices]
    result2, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result2 == "foo"
    result2_shortcut_keys = [copy(c.shortcut_key) for c in choices]
    assert (
        result_shortcut_keys == result2_shortcut_keys
    ), "Shortcut keys changed across two runs of 'select'"


def test_select_initial_choice_with_value():
    message = "Foo message"
    choice = Choice(title="bazz", value="bar")
    kwargs = {"choices": ["foo", choice], "default": "bar"}
    text = KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bar"


def test_select_initial_choice():
    message = "Foo message"
    choice = Choice("bazz")
    kwargs = {"choices": ["foo", choice], "default": choice}
    text = KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_select_initial_choice_string():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bazz"], "default": "bazz"}
    text = KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_select_initial_choice_duplicate():
    message = "Foo message"
    choice = Choice("foo")
    kwargs = {"choices": ["foo", choice, "bazz"], "default": choice}
    text = KeyInputs.DOWN + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_select_initial_choice_not_selectable():
    message = "Foo message"
    separator = Separator()
    kwargs = {"choices": ["foo", "bazz", separator], "default": separator}
    text = KeyInputs.ENTER + "\r"

    with pytest.raises(ValueError):
        feed_cli_with_input("select", message, text, **kwargs)


def test_select_initial_choice_non_existant():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bazz"], "default": "bar"}
    text = KeyInputs.ENTER + "\r"

    with pytest.raises(ValueError):
        feed_cli_with_input("select", message, text, **kwargs)


def test_no_invalid_parameters_are_forwarded():
    # the `validate_while_typing` parameter is an additional parameter that
    # gets forwarded to the `PromptSession`. this checks that the parameter
    # isn't forwarded to a method that does not expect it
    patched_prompt(
        [
            {
                "type": "select",
                "name": "theme",
                "message": "What do you want to do?",
                "choices": [
                    "Order a pizza",
                    "Make a reservation",
                ],
            }
        ],
        text=KeyInputs.ENTER + "\r",
        validate_while_typing=False,
    )


def test_select_arrow_keys():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bazz"], "use_arrow_keys": True}
    text = KeyInputs.DOWN + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_fail_for_unreachable_choice():
    message = "Foo message"
    kwargs = {
        "choices": ["foo", "bar", Choice("bazz", shortcut_key=False)],
        "use_shortcuts": True,
        "use_arrow_keys": False,
    }
    text = KeyInputs.THREE + "\r"

    with pytest.raises(RuntimeError):
        feed_cli_with_input("select", message, text, **kwargs)


def test_fail_on_no_method_to_move_selection():
    message = "Foo message"
    kwargs = {
        "choices": ["foo", Choice("bar", disabled="bad"), "bazz"],
        "use_shortcuts": False,
        "use_arrow_keys": False,
        "use_jk_keys": False,
        "use_emacs_keys": False,
    }
    text = KeyInputs.ENTER + "\r"

    with pytest.raises(ValueError):
        feed_cli_with_input("select", message, text, **kwargs)


def test_jk_and_shortcut_conflict_fails():
    message = "Foo message"
    kwargs = {
        "choices": ["foo", Choice("bar", shortcut_key="j"), "bazz"],
        "use_shortcuts": True,
        "use_arrow_keys": True,
        "use_jk_keys": True,
    }
    text = KeyInputs.ENTER + "\r"

    with pytest.raises(ValueError):
        feed_cli_with_input("select", message, text, **kwargs)


def test_jk_and_shortcut_conflict_avoided_by_disabling_ij_keys():
    message = "Foo message"
    kwargs = {
        "choices": ["foo", Choice("bar", shortcut_key="j"), "bazz"],
        "use_shortcuts": True,
        "use_arrow_keys": True,
        "use_jk_keys": False,
    }
    text = KeyInputs.ENTER + "\r"

    feed_cli_with_input("select", message, text, **kwargs)


def test_select_shortcuts():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bazz"], "use_shortcuts": True}
    text = "2" + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_select_no_arrow_keys():
    message = "Foo message"
    kwargs = {
        "choices": ["foo", "bazz"],
        "use_arrow_keys": False,
        "use_shortcuts": True,
    }
    text = KeyInputs.DOWN + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "foo"


def test_select_no_shortcuts():
    message = "Foo message"
    kwargs = {
        "choices": ["foo", "bazz"],
        "use_arrow_keys": True,
        "use_shortcuts": False,
    }
    text = "2" + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "foo"


def test_select_default_has_arrow_keys():
    message = "Foo message"
    kwargs = {"choices": ["foo", "bazz"]}
    text = KeyInputs.DOWN + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input("select", message, text, **kwargs)
    assert result == "bazz"


def test_select_filter_with_jk_movement_exception():
    message = "Foo message"
    kwargs = {
        "choices": ["foo", "bazz"],
        "use_arrow_keys": True,
        "use_shortcuts": False,
    }
    text = "2" + KeyInputs.ENTER + "\r"
    with pytest.raises(ValueError):
        feed_cli_with_input("select", message, text, use_search_filter=True, **kwargs)


def test_filter_prefix_one_letter():
    message = "Foo message"
    kwargs = {"choices": ["abc", "def", "ghi", "jkl"]}
    text = "g" + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input(
        "select",
        message,
        text,
        use_search_filter=True,
        use_jk_keys=False,
        **kwargs,
    )
    assert result == "ghi"


def test_filter_prefix_multiple_letters():
    message = "Foo message"
    kwargs = {"choices": ["abc", "def", "ghi", "jkl", "jag", "jja"]}
    text = "j" + "j" + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input(
        "select",
        message,
        text,
        use_search_filter=True,
        use_jk_keys=False,
        **kwargs,
    )
    assert result == "jja"


def test_select_filter_handle_backspace():
    message = "Foo message"
    kwargs = {"choices": ["abc", "def", "ghi", "jkl", "jag", "jja"]}
    text = "j" + "j" + KeyInputs.BACK + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input(
        "select",
        message,
        text,
        use_search_filter=True,
        use_jk_keys=False,
        **kwargs,
    )
    assert result == "jkl"

    message = "Foo message"
    kwargs = {"choices": ["abc", "def", "ghi", "jkl", "jag", "jja"]}
    text = (
        "j"
        + "j"
        + KeyInputs.BACK
        + KeyInputs.BACK
        + KeyInputs.BACK
        + KeyInputs.BACK
        + KeyInputs.ENTER
        + "\r"
    )

    result, cli = feed_cli_with_input(
        "select",
        message,
        text,
        use_search_filter=True,
        use_jk_keys=False,
        **kwargs,
    )
    assert result == "abc"


def test_select_goes_back_to_top_after_filtering():
    message = "Foo message"
    kwargs = {"choices": ["abc", "def", "ghi", "jkl", "jag", "jja"]}
    text = KeyInputs.DOWN + KeyInputs.DOWN + "j" + KeyInputs.ENTER + "\r"

    result, cli = feed_cli_with_input(
        "select",
        message,
        text,
        use_search_filter=True,
        use_jk_keys=False,
        **kwargs,
    )
    assert result == "jkl"