# -*- coding: utf-8 -*-
import re

from django import forms
from django.template import loader, Context
from django.utils.translation import ugettext as _
from django.utils.translation import activate, deactivate

from .base import CrispyTestCase
from .forms import CheckboxesTestForm, TestForm
from crispy_forms.bootstrap import (
    PrependedAppendedText, AppendedText, PrependedText, InlineRadios,
    Tab, TabHolder, AccordionGroup, Accordion, Alert, InlineCheckboxes,
    FieldWithButtons, StrictButton
)
from crispy_forms.helper import FormHelper
from crispy_forms.layout import (
    Layout, HTML, Field, MultiWidgetField
)
from crispy_forms.utils import render_crispy_form


class TestLayoutObjects(CrispyTestCase):
    def test_multiwidget_field(self):
        template = loader.get_template_from_string(u"""
            {% load crispy_forms_tags %}
            {% crispy form %}
        """)

        test_form = TestForm()
        test_form.helper = FormHelper()
        test_form.helper.layout = Layout(
            MultiWidgetField(
                'datetime_field',
                attrs=(
                    {'rel': 'test_dateinput'},
                    {'rel': 'test_timeinput', 'style': 'width: 30px;', 'type': "hidden"}
                )
            )
        )

        c = Context({'form': test_form})

        html = template.render(c)

        self.assertEqual(html.count('class="dateinput'), 1)
        self.assertEqual(html.count('rel="test_dateinput"'), 1)
        self.assertEqual(html.count('rel="test_timeinput"'), 1)
        self.assertEqual(html.count('style="width: 30px;"'), 1)
        self.assertEqual(html.count('type="hidden"'), 1)

    def test_field_type_hidden(self):
        template = loader.get_template_from_string(u"""
            {% load crispy_forms_tags %}
            {% crispy test_form %}
        """)

        test_form = TestForm()
        test_form.helper = FormHelper()
        test_form.helper.layout = Layout(
            Field('email', type="hidden", data_test=12),
            Field('datetime_field'),
        )

        c = Context({
            'test_form': test_form,
        })
        html = template.render(c)

        # Check form parameters
        self.assertEqual(html.count('data-test="12"'), 1)
        self.assertEqual(html.count('name="email"'), 1)
        self.assertEqual(html.count('class="dateinput'), 1)
        self.assertEqual(html.count('class="timeinput'), 1)

    def test_field_wrapper_class(self):
        form = TestForm()
        form.helper = FormHelper()
        form.helper.layout = Layout(Field('email', wrapper_class="testing"))

        html = render_crispy_form(form)
        if self.current_template_pack == 'bootstrap':
            self.assertEqual(html.count('class="control-group testing"'), 1)
        elif self.current_template_pack == 'bootstrap3':
            self.assertEqual(html.count('class="form-group testing"'), 1)

    def test_html_with_carriage_returns(self):
        test_form = TestForm()
        test_form.helper = FormHelper()
        test_form.helper.layout = Layout(
            HTML("""
                if (a==b){
                    // some comment
                    a+1;
                    foo();
                }
            """)
        )
        html = render_crispy_form(test_form)

        if self.current_template_pack == 'uni_form':
            self.assertEqual(html.count('\n'), 22)
        else:
            self.assertEqual(html.count('\n'), 24)

    def test_i18n(self):
        activate('es')
        form = TestForm()
        form.helper = FormHelper()
        form.helper.layout = Layout(
            HTML(_("Enter a valid value."))
        )
        html = render_crispy_form(form)
        self.assertTrue(u"Introduzca un valor correcto" in html)
        deactivate()


class TestBootstrapLayoutObjects(CrispyTestCase):
    def test_custom_django_widget(self):
        class CustomRadioSelect(forms.RadioSelect):
            pass

        class CustomCheckboxSelectMultiple(forms.CheckboxSelectMultiple):
            pass

        # Make sure an inherited RadioSelect gets rendered as it
        form = CheckboxesTestForm()
        form.fields['inline_radios'].widget = CustomRadioSelect()
        form.helper = FormHelper()
        form.helper.layout = Layout('inline_radios')

        html = render_crispy_form(form)
        self.assertTrue('class="radio"' in html)

        # Make sure an inherited CheckboxSelectMultiple gets rendered as it
        form.fields['checkboxes'].widget = CustomCheckboxSelectMultiple()
        form.helper.layout = Layout('checkboxes')
        html = render_crispy_form(form)
        self.assertTrue('class="checkbox"' in html)

    def test_prepended_appended_text(self):
        test_form = TestForm()
        test_form.helper = FormHelper()
        test_form.helper.layout = Layout(
            PrependedAppendedText('email', '@', 'gmail.com'),
            AppendedText('password1', '#'),
            PrependedText('password2', '$'),
        )
        html = render_crispy_form(test_form)

        # Check form parameters
        if self.current_template_pack == 'bootstrap':
            self.assertEqual(html.count('<span class="add-on">@</span>'), 1)
            self.assertEqual(html.count('<span class="add-on">gmail.com</span>'), 1)
            self.assertEqual(html.count('<span class="add-on">#</span>'), 1)
            self.assertEqual(html.count('<span class="add-on">$</span>'), 1)

        if self.current_template_pack == 'bootstrap3':
            self.assertEqual(html.count('<span class="input-group-addon">@</span>'), 1)
            self.assertEqual(html.count('<span class="input-group-addon">gmail.com</span>'), 1)
            self.assertEqual(html.count('<span class="input-group-addon">#</span>'), 1)
            self.assertEqual(html.count('<span class="input-group-addon">$</span>'), 1)

    def test_inline_radios(self):
        test_form = CheckboxesTestForm()
        test_form.helper = FormHelper()
        test_form.helper.layout = Layout(
            InlineRadios('inline_radios')
        )
        html = render_crispy_form(test_form)

        if self.current_template_pack == 'bootstrap':
            self.assertEqual(html.count('radio inline"'), 2)
        elif self.current_template_pack == 'bootstrap3':
            self.assertEqual(html.count('radio-inline"'), 2)

    def test_accordion_and_accordiongroup(self):
        test_form = TestForm()
        test_form.helper = FormHelper()
        test_form.helper.layout = Layout(
            Accordion(
                AccordionGroup(
                    'one',
                    'first_name'
                ),
                AccordionGroup(
                    'two',
                    'password1',
                    'password2'
                )
            )
        )
        html = render_crispy_form(test_form)

        if self.current_template_pack == 'bootstrap':
            self.assertEqual(html.count('<div class="accordion"'), 1)
            self.assertEqual(html.count('<div class="accordion-group">'), 2)
            self.assertEqual(html.count('<div class="accordion-heading">'), 2)
        else:
            self.assertEqual(html.count('<div class="panel panel-default"'), 2)
            self.assertEqual(html.count('<div class="panel-group"'), 1)
            self.assertEqual(html.count('<div class="panel-heading">'), 2)

        self.assertEqual(html.count('<div id="one"'), 1)
        self.assertEqual(html.count('<div id="two"'), 1)
        self.assertEqual(html.count('name="first_name"'), 1)
        self.assertEqual(html.count('name="password1"'), 1)
        self.assertEqual(html.count('name="password2"'), 1)

    def test_alert(self):
        test_form = TestForm()
        test_form.helper = FormHelper()
        test_form.helper.layout = Layout(
            Alert(content='Testing...')
        )
        html = render_crispy_form(test_form)

        self.assertEqual(html.count('<div class="alert"'), 1)
        self.assertEqual(html.count('<button type="button" class="close"'), 1)
        self.assertEqual(html.count('Testing...'), 1)

    def test_alert_block(self):
        test_form = TestForm()
        test_form.helper = FormHelper()
        test_form.helper.layout = Layout(
            Alert(content='Testing...', block=True)
        )
        html = render_crispy_form(test_form)

        self.assertEqual(html.count('<div class="alert alert-block"'), 1)
        self.assertEqual(html.count('Testing...'), 1)

    def test_tab_and_tab_holder(self):
        test_form = TestForm()
        test_form.helper = FormHelper()
        test_form.helper.layout = Layout(
            TabHolder(
                Tab('one',
                    'first_name',
                    css_id="custom-name",
                    css_class="first-tab-class"
                ),
                Tab('two',
                    'password1',
                    'password2'
                )
            )
        )
        html = render_crispy_form(test_form)

        self.assertEqual(
            html.count(
                '<li class="tab-pane active"><a href="#custom-name" data-toggle="tab">One</a></li>'
            ),
            1
        )
        self.assertEqual(html.count('class="tab-pane first-tab-class active"'), 1)
        self.assertEqual(html.count('<li class="tab-pane'), 2)
        self.assertEqual(html.count('tab-pane'), 4)
        self.assertEqual(html.count('<div id="custom-name"'), 1)
        self.assertEqual(html.count('<div id="two"'), 1)
        self.assertEqual(html.count('name="first_name"'), 1)
        self.assertEqual(html.count('name="password1"'), 1)
        self.assertEqual(html.count('name="password2"'), 1)

    def test_tab_helper_reuse(self):
        # this is a proper form, according to the docs.
        # note that the helper is a class property here,
        # shared between all instances
        class TestForm(forms.Form):
            val1 = forms.CharField(required=False)
            val2 = forms.CharField(required=True)
            helper = FormHelper()
            helper.layout = Layout(
                TabHolder(
                    Tab('one', 'val1',),
                    Tab('two', 'val2',)
                )
            )

        # first render of form => everything is fine
        test_form = TestForm()
        html = render_crispy_form(test_form)

        # second render of form => first tab should be active,
        # but not duplicate class
        test_form = TestForm()
        html = render_crispy_form(test_form)
        self.assertEqual(html.count('class="tab-pane active active"'), 0)

        # render a new form, now with errors
        test_form = TestForm(data={'val1': 'foo'})
        html = render_crispy_form(test_form)
        # tab 1 should not be active
        self.assertEqual(html.count('<div id="one" \n    class="tab-pane active'), 0)
        # tab 2 should be active
        self.assertEqual(html.count('<div id="two" \n    class="tab-pane active'), 1)

    def test_radio_attrs(self):
        form = CheckboxesTestForm()
        form.fields['inline_radios'].widget.attrs = {'class': "first"}
        form.fields['checkboxes'].widget.attrs = {'class': "second"}
        html = render_crispy_form(form)
        self.assertTrue('class="first"' in html)
        self.assertTrue('class="second"' in html)

    def test_field_with_buttons(self):
        form = TestForm()
        form.helper = FormHelper()
        form.helper.layout = Layout(
            FieldWithButtons(
                Field('password1', css_class="span4"),
                StrictButton("Go!", css_id="go-button"),
                StrictButton("No!", css_class="extra"),
                StrictButton("Test", type="submit", name="whatever", value="something"),
                css_class="extra",
                autocomplete="off"
            )
        )
        html = render_crispy_form(form)

        form_group_class = 'control-group'
        if self.current_template_pack == 'bootstrap3':
            form_group_class = 'form-group'

        self.assertEqual(html.count('class="%s extra"' % form_group_class), 1)
        self.assertEqual(html.count('autocomplete="off"'), 1)
        self.assertEqual(html.count('class="span4'), 1)
        self.assertEqual(html.count('id="go-button"'), 1)
        self.assertEqual(html.count("Go!"), 1)
        self.assertEqual(html.count("No!"), 1)
        self.assertEqual(html.count('class="btn"'), 2)
        self.assertEqual(html.count('class="btn extra"'), 1)
        self.assertEqual(html.count('type="submit"'), 1)
        self.assertEqual(html.count('name="whatever"'), 1)
        self.assertEqual(html.count('value="something"'), 1)

        if self.current_template_pack == 'bootstrap':
            self.assertEqual(html.count('class="input-append"'), 1)
        elif self.current_template_pack == 'bootstrap3':
            self.assertEqual(html.count('class="input-group-btn'), 1)

        # Make sure white spaces between buttons are there in bootstrap
        self.assertEqual(len(re.findall(r'</button> <', html)), 3)

    def test_hidden_fields(self):
        form = TestForm()
        # All fields hidden
        for field in form.fields:
            form.fields[field].widget = forms.HiddenInput()

        form.helper = FormHelper()
        form.helper.layout = Layout(
            AppendedText('password1', 'foo'),
            PrependedText('password2', 'bar'),
            PrependedAppendedText('email', 'bar'),
            InlineCheckboxes('first_name'),
            InlineRadios('last_name'),
        )
        html = render_crispy_form(form)
        self.assertEqual(html.count("<input"), 5)
        self.assertEqual(html.count('type="hidden"'), 5)
        self.assertEqual(html.count('<label'), 0)

    def test_multiplecheckboxes(self):
        test_form = CheckboxesTestForm()
        html = render_crispy_form(test_form)

        self.assertEqual(html.count('checked="checked"'), 6)

        test_form.helper = FormHelper(test_form)
        test_form.helper[1].wrap(InlineCheckboxes, inline=True)
        html = render_crispy_form(test_form)

        if self.current_template_pack == 'bootstrap':
            self.assertEqual(html.count('checkbox inline"'), 3)
            self.assertEqual(html.count('inline"'), 3)
        elif self.current_template_pack == 'bootstrap3':
            self.assertEqual(html.count('checkbox-inline"'), 3)
            self.assertEqual(html.count('inline="True"'), 4)