import pytest

from django import forms

from crispy_forms.bootstrap import AppendedText
from crispy_forms.exceptions import DynamicError
from crispy_forms.helper import FormHelper, FormHelpersException
from crispy_forms.layout import HTML, Div, Field, Fieldset, Layout, MultiField
from crispy_forms.tests.forms import SampleForm

from .conftest import only_uni_form


def test_wrap_all_fields():
    helper = FormHelper()
    layout = Layout("email", "password1", "password2",)
    helper.layout = layout

    helper.all().wrap(Field, css_class="test-class")
    for field in layout.fields:
        assert isinstance(field, Field)
        assert field.attrs["class"] == "test-class"

    assert layout[0][0] == "email"
    assert layout[1][0] == "password1"
    assert layout[2][0] == "password2"


def test_wrap_selected_fields():
    helper = FormHelper()
    layout = Layout("email", "password1", "password2",)
    helper.layout = layout

    helper[1:3].wrap(Field, css_class="test-class")
    assert not isinstance(layout.fields[0], Field)
    assert isinstance(layout.fields[1], Field)
    assert isinstance(layout.fields[2], Field)

    helper[0].wrap(Fieldset, "legend", css_class="test-class")
    assert isinstance(layout[0], Fieldset)
    assert layout[0].legend == "legend"
    assert layout[0][0] == "email"


def test_wrap_together_with_slices():
    helper = FormHelper()
    layout = Layout("email", "password1", "password2",)
    helper.layout = layout
    helper[1:3].wrap_together(Field, css_class="test-class")
    assert layout.fields[0] == "email"
    assert isinstance(layout.fields[1], Field)
    assert layout.fields[1][0] == "password1"
    assert layout.fields[1][1] == "password2"

    layout = Layout(Div("email"), "password1", "password2",)
    helper.layout = layout
    helper[0:3].wrap_together(Field, css_class="test-class")
    assert isinstance(layout.fields[0], Field)
    assert isinstance(layout.fields[0][0], Div)
    assert layout.fields[0][0][0] == "email"
    assert layout.fields[0][1] == "password1"
    assert layout.fields[0][2] == "password2"

    layout = Layout("email", "password1", "password2",)
    helper.layout = layout
    helper[0].wrap_together(Field, css_class="test-class")
    assert isinstance(layout.fields[0], Field)
    assert layout.fields[1] == "password1"
    assert layout.fields[2] == "password2"

    layout = Layout("email", "password1", "password2",)
    helper.layout = layout
    helper[0].wrap_together(Fieldset, "legend", css_class="test-class")
    assert isinstance(layout.fields[0], Fieldset)
    assert layout.fields[0].legend == "legend"
    assert layout.fields[1] == "password1"
    assert layout.fields[2] == "password2"


def test_wrap_together_partial_slices():
    helper = FormHelper()
    layout = Layout("email", "password1", "password2",)
    helper.layout = layout

    helper[:2].wrap_together(Field, css_class="test-class")
    assert isinstance(layout.fields[0], Field)
    assert layout.fields[1] == "password2"
    assert layout.fields[0][0] == "email"
    assert layout.fields[0][1] == "password1"

    helper = FormHelper()
    layout = Layout("email", "password1", "password2",)
    helper.layout = layout

    helper[1:].wrap_together(Field, css_class="test-class")
    assert layout.fields[0] == "email"
    assert isinstance(layout.fields[1], Field)
    assert layout.fields[1][0] == "password1"
    assert layout.fields[1][1] == "password2"


def test_update_attributes():
    helper = FormHelper()
    helper.layout = Layout("email", Field("password1"), "password2",)
    helper["password1"].update_attributes(readonly=True)
    assert "readonly" in helper.layout[1].attrs


def test_update_attributes_and_wrap_once():
    helper = FormHelper()
    layout = Layout("email", Field("password1"), "password2",)
    helper.layout = layout
    helper.filter(Field).update_attributes(readonly=True)
    assert isinstance(layout[1], Field)
    assert layout[1].attrs == {"readonly": True}

    layout = Layout("email", Div(Field("password1")), "password2",)
    helper.layout = layout
    helper.filter(Field, max_level=2).update_attributes(readonly=True)
    assert isinstance(layout[1][0], Field)
    assert layout[1][0].attrs == {"readonly": True}

    layout = Layout("email", Div(Field("password1")), "password2",)
    helper.layout = layout

    helper.filter(str, greedy=True).wrap_once(Field)
    helper.filter(Field, greedy=True).update_attributes(readonly=True)

    assert isinstance(layout[0], Field)
    assert isinstance(layout[1][0], Field)
    assert isinstance(layout[1][0][0], str)
    assert isinstance(layout[2], Field)
    assert layout[1][0].attrs == {"readonly": True}
    assert layout[0].attrs == {"readonly": True}
    assert layout[2].attrs == {"readonly": True}


def test_get_layout_objects():
    layout_1 = Layout(Div())
    assert layout_1.get_layout_objects(Div) == [[[0], "div"]]

    layout_2 = Layout(Div(Div(Div("email")), Div("password1"), "password2"))
    assert layout_2.get_layout_objects(Div) == [[[0], "div"]]
    assert layout_2.get_layout_objects(Div, max_level=1) == [[[0], "div"], [[0, 0], "div"], [[0, 1], "div"]]
    assert layout_2.get_layout_objects(Div, max_level=2) == [
        [[0], "div"],
        [[0, 0], "div"],
        [[0, 0, 0], "div"],
        [[0, 1], "div"],
    ]

    layout_3 = Layout("email", Div("password1"), "password2",)
    assert layout_3.get_layout_objects(str, max_level=2) == [[[0], "email"], [[1, 0], "password1"], [[2], "password2"]]

    layout_4 = Layout(Div(Div("field_name"), "field_name2",), Div("password"), "extra_field")
    assert layout_4.get_layout_objects(Div) == [[[0], "div"], [[1], "div"]]
    assert layout_4.get_layout_objects(Div, max_level=1) == [[[0], "div"], [[0, 0], "div"], [[1], "div"]]


def test_filter_and_wrap():
    helper = FormHelper()
    layout = Layout("email", Div("password1"), "password2",)
    helper.layout = layout

    helper.filter(str).wrap(Field, css_class="test-class")
    assert isinstance(layout.fields[0], Field)
    assert isinstance(layout.fields[1], Div)
    assert isinstance(layout.fields[2], Field)
    assert layout[2][0] == "password2"

    # Wrapping a div in a div
    helper.filter(Div).wrap(Div, css_class="test-class")
    assert isinstance(layout.fields[1], Div)
    assert isinstance(layout.fields[1].fields[0], Div)
    assert layout[1][0][0] == "password1"


def test_filter_and_wrap_side_effects():
    helper = FormHelper()
    layout = Layout(Div("extra_field", Div("password1"),),)
    helper.layout = layout
    with pytest.raises(DynamicError):
        helper.filter(Div, max_level=2).wrap(Div, css_class="test-class")


def test_get_field_names():
    layout_1 = Div("field_name")
    assert layout_1.get_field_names() == [[[0], "field_name"]]

    layout_2 = Div(Div("field_name"))
    assert layout_2.get_field_names() == [[[0, 0], "field_name"]]

    layout_3 = Div(Div("field_name"), "password")
    assert layout_3.get_field_names() == [[[0, 0], "field_name"], [[1], "password"]]

    layout_4 = Div(Div(Div("field_name"), "field_name2",), Div("password"), "extra_field")
    assert layout_4.get_field_names() == [
        [[0, 0, 0], "field_name"],
        [[0, 1], "field_name2"],
        [[1, 0], "password"],
        [[2], "extra_field"],
    ]

    layout_5 = Div(Div("field_name", "field_name2",), "extra_field")
    assert layout_5.get_field_names() == [
        [[0, 0], "field_name"],
        [[0, 1], "field_name2"],
        [[1], "extra_field"],
    ]


def test_layout_get_field_names():
    layout_1 = Layout(Div("field_name"), "password")
    assert layout_1.get_field_names() == [
        [[0, 0], "field_name"],
        [[1], "password"],
    ]

    layout_2 = Layout(Div("field_name"), "password", Fieldset("legend", "extra_field"))
    assert layout_2.get_field_names() == [
        [[0, 0], "field_name"],
        [[1], "password"],
        [[2, 0], "extra_field"],
    ]

    layout_3 = Layout(Div(Div(Div("email")), Div("password1"), "password2"))
    assert layout_3.get_field_names() == [
        [[0, 0, 0, 0], "email"],
        [[0, 1, 0], "password1"],
        [[0, 2], "password2"],
    ]


def test_filter_by_widget(advanced_layout):
    form = SampleForm()
    form.helper = FormHelper(form)
    form.helper.layout = advanced_layout
    assert form.helper.filter_by_widget(forms.PasswordInput).slice == [
        [[0, 1, 0, 0], "password1"],
        [[0, 4, 0], "password2"],
    ]


def test_exclude_by_widget(advanced_layout):
    form = SampleForm()
    form.helper = FormHelper(form)
    form.helper.layout = advanced_layout
    assert form.helper.exclude_by_widget(forms.PasswordInput).slice == [
        [[0, 0, 0, 0], "email"],
        [[0, 3, 0], "first_name"],
        [[1], "last_name"],
    ]


def test_exclude_by_widget_and_wrap(advanced_layout):
    form = SampleForm()
    form.helper = FormHelper(form)
    form.helper.layout = advanced_layout
    form.helper.exclude_by_widget(forms.PasswordInput).wrap(Field, css_class="hero")
    # Check wrapped fields
    assert isinstance(form.helper.layout[0][0][0][0], Field)
    assert isinstance(form.helper.layout[0][3][0], Field)
    assert isinstance(form.helper.layout[1], Field)
    # Check others stay the same
    assert isinstance(form.helper.layout[0][3][1], HTML)
    assert isinstance(form.helper.layout[0][1][0][0], str)
    assert isinstance(form.helper.layout[0][4][0], str)


def test_all_without_layout():
    form = SampleForm()
    form.helper = FormHelper()
    with pytest.raises(FormHelpersException):
        form.helper.all().wrap(Div)


def test_filter_by_widget_without_form(advanced_layout):
    form = SampleForm()
    form.helper = FormHelper()
    form.helper.layout = advanced_layout
    with pytest.raises(FormHelpersException):
        form.helper.filter_by_widget(forms.PasswordInput)


def test_formhelper__getitem__():
    helper = FormHelper()
    layout = Layout(Div("email"), "password1",)
    helper.layout = layout
    helper["email"].wrap(Field, css_class="hero")
    assert isinstance(layout[0][0], Field)
    assert layout[0][0][0] == "email"

    helper = FormHelper()
    helper.layout = Layout("password1")
    helper["password1"].wrap(AppendedText, "extra")
    assert isinstance(helper.layout[0], AppendedText)
    assert helper.layout[0][0] == "password1"
    assert helper.layout[0].text == "extra"


def test_formhelper__setitem__():
    helper = FormHelper()
    layout = Layout("first_field", Div("email"))
    helper.layout = layout
    helper[0] = "replaced"
    assert layout[0] == "replaced"


def test_formhelper__delitem__and__len__():
    helper = FormHelper()
    layout = Layout("first_field", Div("email"))
    helper.layout = layout
    del helper[0]
    assert len(helper) == 1


def test__delitem__and__len__layout_object():
    layout = Layout("first_field", Div("email"))
    del layout[0]
    assert len(layout) == 1


def test__getitem__layout_object():
    layout = Layout(Div(Div(Div("email")), Div("password1"), "password2"))
    assert isinstance(layout[0], Div)
    assert isinstance(layout[0][0], Div)
    assert isinstance(layout[0][0][0], Div)
    assert isinstance(layout[0][1], Div)
    assert isinstance(layout[0][1][0], str)
    assert isinstance(layout[0][2], str)


def test__getattr__append_layout_object():
    layout = Layout(Div("email"))
    layout.append("password1")
    assert isinstance(layout[0], Div)
    assert isinstance(layout[0][0], str)
    assert isinstance(layout[1], str)


def test__setitem__layout_object():
    layout = Layout(Div("email"))
    layout[0][0] = "password1"
    assert isinstance(layout[0], Div)
    assert layout[0][0] == "password1"


@only_uni_form
def test_filter():
    helper = FormHelper()
    helper.layout = Layout(Div(MultiField("field_name"), "field_name2",), Div("password"), "extra_field")
    assert helper.filter(Div, MultiField).slice == [[[0], "div"], [[1], "div"]]
    assert helper.filter(Div, MultiField, max_level=1).slice == [[[0], "div"], [[0, 0], "multifield"], [[1], "div"]]
    assert helper.filter(MultiField, max_level=1).slice == [[[0, 0], "multifield"]]
