import csv
import uuid
from datetime import timedelta

from django.contrib import messages
from django.contrib.messages.api import get_messages
from django.contrib.messages.storage.base import Message
from django.test import Client, TestCase
from django.urls import reverse

from bda.models import Salle, Tirage
from gestioncof.models import (
    CalendarSubscription, Club, Event, Survey, SurveyAnswer
)
from gestioncof.tests.testcases import ViewTestCaseMixin

from .utils import create_user


class ExportMembersViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'cof.membres_export'
    url_expected = '/export/members'

    auth_user = 'staff'
    auth_forbidden = [None, 'user', 'member']

    def test(self):
        u1, u2 = self.users['member'], self.users['staff']
        u1.first_name = 'first'
        u1.last_name = 'last'
        u1.email = 'user@mail.net'
        u1.save()
        u1.profile.phone = '0123456789'
        u1.profile.departement = 'Dept'
        u1.profile.save()

        r = self.client.get(self.url)

        self.assertEqual(r.status_code, 200)
        data = list(csv.reader(r.content.decode('utf-8').split('\n')[:-1]))
        self.assertListEqual(data, [
            [
                str(u1.pk), 'member', 'first', 'last', 'user@mail.net',
                '0123456789', '1A', 'Dept', 'normalien',
            ],
            [str(u2.pk), 'staff', '', '', '', '', '1A', '', 'normalien'],
        ])


class MegaHelpers:
    def setUp(self):
        super().setUp()

        u1 = create_user('u1')
        u1.first_name = 'first'
        u1.last_name = 'last'
        u1.email = 'user@mail.net'
        u1.save()
        u1.profile.phone = '0123456789'
        u1.profile.departement = 'Dept'
        u1.profile.comments = 'profile.comments'
        u1.profile.save()

        u2 = create_user('u2')
        u2.profile.save()

        m = Event.objects.create(title='MEGA 2017')

        cf1 = m.commentfields.create(name='Commentaire')
        cf2 = m.commentfields.create(
            name='Comment Field 2', fieldtype='char',
        )

        option_type = m.options.create(name='Conscrit/Orga ?')
        choice_orga = option_type.choices.create(value='Orga')
        choice_conscrit = option_type.choices.create(value='Conscrit')

        mr1 = m.eventregistration_set.create(user=u1)
        mr1.options.add(choice_orga)
        mr1.comments.create(commentfield=cf1, content='Comment 1')
        mr1.comments.create(commentfield=cf2, content='Comment 2')

        mr2 = m.eventregistration_set.create(user=u2)
        mr2.options.add(choice_conscrit)

        self.u1 = u1
        self.u2 = u2
        self.m = m
        self.choice_orga = choice_orga
        self.choice_conscrit = choice_conscrit


class ExportMegaViewTests(MegaHelpers, ViewTestCaseMixin, TestCase):
    url_name = 'cof.mega_export'
    url_expected = '/export/mega'

    auth_user = 'staff'
    auth_forbidden = [None, 'user', 'member']

    def test(self):
        r = self.client.get(self.url)

        self.assertEqual(r.status_code, 200)
        self.assertListEqual(self.load_from_csv_response(r), [
            [
                'u1', 'first', 'last', 'user@mail.net', '0123456789',
                str(self.u1.pk), 'profile.comments', 'Comment 1---Comment 2',
            ],
            ['u2', '', '', '', '', str(self.u2.pk), '', ''],
        ])


class ExportMegaOrgasViewTests(MegaHelpers, ViewTestCaseMixin, TestCase):
    url_name = 'cof.mega_export_orgas'
    url_expected = '/export/mega/orgas'

    auth_user = 'staff'
    auth_forbidden = [None, 'user', 'member']

    def test(self):

        r = self.client.get(self.url)

        self.assertEqual(r.status_code, 200)
        self.assertListEqual(self.load_from_csv_response(r), [
            [
                'u1', 'first', 'last', 'user@mail.net', '0123456789',
                str(self.u1.pk), 'profile.comments', 'Comment 1---Comment 2',
            ],
        ])


class ExportMegaParticipantsViewTests(
        MegaHelpers, ViewTestCaseMixin, TestCase):
    url_name = 'cof.mega_export_participants'
    url_expected = '/export/mega/participants'

    auth_user = 'staff'
    auth_forbidden = [None, 'user', 'member']

    def test(self):
        r = self.client.get(self.url)

        self.assertEqual(r.status_code, 200)
        self.assertListEqual(self.load_from_csv_response(r), [
            ['u2', '', '', '', '', str(self.u2.pk), '', ''],
        ])


class ExportMegaRemarksViewTests(
        MegaHelpers, ViewTestCaseMixin, TestCase):
    url_name = 'cof.mega_export_remarks'
    url_expected = '/export/mega/avecremarques'

    auth_user = 'staff'
    auth_forbidden = [None, 'user', 'member']

    def test(self):
        r = self.client.get(self.url)

        self.assertEqual(r.status_code, 200)
        self.assertListEqual(self.load_from_csv_response(r), [
            [
                'u1', 'first', 'last', 'user@mail.net', '0123456789',
                str(self.u1.pk), 'profile.comments', 'Comment 1',
            ],
        ])


class ClubListViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'liste-clubs'
    url_expected = '/clubs/liste'

    auth_user = 'member'
    auth_forbidden = [None, 'user']

    def setUp(self):
        super().setUp()

        self.c1 = Club.objects.create(name='Club1')
        self.c2 = Club.objects.create(name='Club2')

        m = self.users['member']
        self.c1.membres.add(m)
        self.c1.respos.add(m)

    def test_as_member(self):
        r = self.client.get(self.url)

        self.assertEqual(r.status_code, 200)
        self.assertEqual(r.context['owned_clubs'].get(), self.c1)
        self.assertEqual(r.context['other_clubs'].get(), self.c2)

    def test_as_staff(self):
        u = self.users['staff']
        c = Client()
        c.force_login(u)

        r = c.get(self.url)

        self.assertEqual(r.status_code, 200)
        self.assertQuerysetEqual(
            r.context['owned_clubs'], map(repr, [self.c1, self.c2]),
            ordered=False,
        )


class ClubMembersViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'membres-club'

    auth_user = 'staff'
    auth_forbidden = [None, 'user', 'member']

    @property
    def url_kwargs(self):
        return {'name': self.c.name}

    @property
    def url_expected(self):
        return '/clubs/membres/{}'.format(self.c.name)

    def setUp(self):
        super().setUp()

        self.u1 = create_user('u1')
        self.u2 = create_user('u2')

        self.c = Club.objects.create(name='Club')
        self.c.membres.add(self.u1, self.u2)
        self.c.respos.add(self.u1)

    def test_as_staff(self):
        r = self.client.get(self.url)

        self.assertEqual(r.status_code, 200)
        self.assertEqual(r.context['members_no_respo'].get(), self.u2)

    def test_as_respo(self):
        u = self.users['user']
        self.c.respos.add(u)

        c = Client()
        c.force_login(u)
        r = c.get(self.url)

        self.assertEqual(r.status_code, 200)


class ClubChangeRespoViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'change-respo'

    auth_user = 'staff'
    auth_forbidden = [None, 'user', 'member']

    @property
    def url_kwargs(self):
        return {'club_name': self.c.name, 'user_id': self.users['user'].pk}

    @property
    def url_expected(self):
        return '/clubs/change_respo/{}/{}'.format(
            self.c.name, self.users['user'].pk,
        )

    def setUp(self):
        super().setUp()

        self.c = Club.objects.create(name='Club')

    def test(self):
        u = self.users['user']
        expected_redirect = reverse('membres-club', kwargs={
            'name': self.c.name,
        })
        self.c.membres.add(u)

        r = self.client.get(self.url)
        self.assertRedirects(r, expected_redirect)
        self.assertIn(u, self.c.respos.all())

        self.client.get(self.url)
        self.assertNotIn(u, self.c.respos.all())


class CalendarViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'calendar'
    url_expected = '/calendar/subscription'

    auth_user = 'member'
    auth_forbidden = [None, 'user']

    post_expected_message = Message(
        messages.SUCCESS, "Calendrier mis à jour avec succès.")

    def test_get(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_new(self):
        r = self.client.post(self.url, {
            'subscribe_to_events': True,
            'subscribe_to_my_shows': True,
            'other_shows': [],
        })

        self.assertEqual(r.status_code, 200)
        self.assertIn(self.post_expected_message, get_messages(r.wsgi_request))
        cs = self.users['member'].calendarsubscription
        self.assertTrue(cs.subscribe_to_events)
        self.assertTrue(cs.subscribe_to_my_shows)

    def test_post_edit(self):
        u = self.users['member']
        token = uuid.uuid4()
        cs = CalendarSubscription.objects.create(token=token, user=u)

        r = self.client.post(self.url, {
            'other_shows': [],
        })

        self.assertEqual(r.status_code, 200)
        self.assertIn(self.post_expected_message, get_messages(r.wsgi_request))
        cs.refresh_from_db()
        self.assertEqual(cs.token, token)
        self.assertFalse(cs.subscribe_to_events)
        self.assertFalse(cs.subscribe_to_my_shows)

    def test_post_other_shows(self):
        t = Tirage.objects.create(
            ouverture=self.now,
            fermeture=self.now,
            active=True,
        )
        l = Salle.objects.create()
        s = t.spectacle_set.create(
            date=self.now, price=3.5, slots=20, location=l, listing=True)

        r = self.client.post(self.url, {'other_shows': [str(s.pk)]})

        self.assertEqual(r.status_code, 200)


class CalendarICSViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'calendar.ics'

    auth_user = None
    auth_forbidden = []

    @property
    def url_kwargs(self):
        return {'token': self.token}

    @property
    def url_expected(self):
        return '/calendar/{}/calendar.ics'.format(self.token)

    def setUp(self):
        super().setUp()

        self.token = uuid.uuid4()

        self.t = Tirage.objects.create(
            ouverture=self.now,
            fermeture=self.now,
            active=True,
        )
        l = Salle.objects.create(name='Location')
        self.s1 = self.t.spectacle_set.create(
            price=1, slots=10, location=l, listing=True,
            title='Spectacle 1', date=self.now + timedelta(days=1),
        )
        self.s2 = self.t.spectacle_set.create(
            price=2, slots=20, location=l, listing=True,
            title='Spectacle 2', date=self.now + timedelta(days=2),
        )
        self.s3 = self.t.spectacle_set.create(
            price=3, slots=30, location=l, listing=True,
            title='Spectacle 3', date=self.now + timedelta(days=3),
        )

    def test(self):
        u = self.users['user']
        p = u.participant_set.create(tirage=self.t)
        p.attribution_set.create(spectacle=self.s1)

        self.cs = CalendarSubscription.objects.create(
            user=u, token=self.token,
            subscribe_to_my_shows=True, subscribe_to_events=True,
        )
        self.cs.other_shows.add(self.s2)

        r = self.client.get(self.url)

        def get_dt_from_ical(v):
            return v.dt

        self.assertCalEqual(r.content.decode('utf-8'), [
            {
                'summary': 'Spectacle 1',
                'dtstart': (get_dt_from_ical, (
                    (self.now + timedelta(days=1)).replace(microsecond=0)
                )),
                'dtend': (get_dt_from_ical, (
                    (self.now + timedelta(days=1, hours=2)).replace(
                        microsecond=0)
                )),
                'location': 'Location',
                'uid': 'show-{}-{}@example.com'.format(self.s1.pk, self.t.pk),
            },
            {
                'summary': 'Spectacle 2',
                'dtstart': (get_dt_from_ical, (
                    (self.now + timedelta(days=2)).replace(microsecond=0)
                )),
                'dtend': (get_dt_from_ical, (
                    (self.now + timedelta(days=2, hours=2)).replace(
                        microsecond=0)
                )),
                'location': 'Location',
                'uid': 'show-{}-{}@example.com'.format(self.s2.pk, self.t.pk),
            },
        ])


class EventViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'event.details'
    http_methods = ['GET', 'POST']

    auth_user = 'user'
    auth_forbidden = [None]

    post_expected_message = Message(messages.SUCCESS, (
        "Votre inscription a bien été enregistrée ! Vous pouvez cependant la "
        "modifier jusqu'à la fin des inscriptions."
    ))

    @property
    def url_kwargs(self):
        return {'event_id': self.e.pk}

    @property
    def url_expected(self):
        return '/event/{}'.format(self.e.pk)

    def setUp(self):
        super().setUp()

        self.e = Event.objects.create()

        self.ecf1 = self.e.commentfields.create(name='Comment Field 1')
        self.ecf2 = self.e.commentfields.create(
            name='Comment Field 2', fieldtype='char',
        )

        self.o1 = self.e.options.create(name='Option 1')
        self.o2 = self.e.options.create(name='Option 2', multi_choices=True)

        self.oc1 = self.o1.choices.create(value='O1 - Choice 1')
        self.oc2 = self.o1.choices.create(value='O1 - Choice 2')
        self.oc3 = self.o2.choices.create(value='O2 - Choice 1')
        self.oc4 = self.o2.choices.create(value='O2 - Choice 2')

    def test_get(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_new(self):
        r = self.client.post(self.url, {
            'option_{}'.format(self.o1.pk): [str(self.oc1.pk)],
            'option_{}'.format(self.o2.pk): [
                str(self.oc3.pk), str(self.oc4.pk),
            ],
        })

        self.assertEqual(r.status_code, 200)
        self.assertIn(self.post_expected_message, get_messages(r.wsgi_request))

        er = self.e.eventregistration_set.get(user=self.users['user'])
        self.assertQuerysetEqual(
            er.options.all(), map(repr, [self.oc1, self.oc3, self.oc4]),
            ordered=False,
        )
        # TODO: Make the view care about comments.
        # self.assertQuerysetEqual(
        #     er.comments.all(), map(repr, []),
        #     ordered=False,
        # )

    def test_post_edit(self):
        er = self.e.eventregistration_set.create(user=self.users['user'])
        er.options.add(self.oc1, self.oc3, self.oc4)
        er.comments.create(
            commentfield=self.ecf1, content='Comment 1',
        )

        r = self.client.post(self.url, {
            'option_{}'.format(self.o1.pk): [],
            'option_{}'.format(self.o2.pk): [str(self.oc3.pk)],
        })

        self.assertEqual(r.status_code, 200)
        self.assertIn(self.post_expected_message, get_messages(r.wsgi_request))

        er.refresh_from_db()
        self.assertQuerysetEqual(
            er.options.all(), map(repr, [self.oc3]),
            ordered=False,
        )
        # TODO: Make the view care about comments.
        # self.assertQuerysetEqual(
        #     er.comments.all(), map(repr, []),
        #     ordered=False,
        # )


class EventStatusViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'event.details.status'

    http_methods = ['GET', 'POST']

    auth_user = 'staff'
    auth_forbidden = [None, 'user', 'member']

    @property
    def url_kwargs(self):
        return {'event_id': self.e.pk}

    @property
    def url_expected(self):
        return '/event/{}/status'.format(self.e.pk)

    def setUp(self):
        super().setUp()

        self.e = Event.objects.create()

        self.cf1 = self.e.commentfields.create(name='Comment Field 1')
        self.cf2 = self.e.commentfields.create(
            name='Comment Field 2', fieldtype='char',
        )

        self.o1 = self.e.options.create(name='Option 1')
        self.o2 = self.e.options.create(name='Option 2', multi_choices=True)

        self.oc1 = self.o1.choices.create(value='O1 - Choice 1')
        self.oc2 = self.o1.choices.create(value='O1 - Choice 2')
        self.oc3 = self.o2.choices.create(value='O2 - Choice 1')
        self.oc4 = self.o2.choices.create(value='O2 - Choice 2')

        self.er1 = self.e.eventregistration_set.create(user=self.users['user'])
        self.er1.options.add(self.oc1)
        self.er2 = self.e.eventregistration_set.create(
            user=self.users['member'],
        )

    def _get_oc_filter_name(self, oc):
        return 'option_{}_choice_{}'.format(oc.event_option.pk, oc.pk)

    def _test_filters(self, filters, expected):
        r = self.client.post(self.url, {
            self._get_oc_filter_name(oc): v for oc, v in filters
        })

        self.assertEqual(r.status_code, 200)
        self.assertQuerysetEqual(
            r.context['user_choices'], map(repr, expected),
            ordered=False,
        )

    def test_filter_none(self):
        self._test_filters([(self.oc1, 'none')], [self.er1, self.er2])

    def test_filter_yes(self):
        self._test_filters([(self.oc1, 'yes')], [self.er1])

    def test_filter_no(self):
        self._test_filters([(self.oc1, 'no')], [self.er2])


class SurveyViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'survey.details'
    http_methods = ['GET', 'POST']

    auth_user = 'user'
    auth_forbidden = [None]

    post_expected_message = Message(messages.SUCCESS, (
        "Votre réponse a bien été enregistrée ! Vous pouvez cependant la "
        "modifier jusqu'à la fin du sondage."
    ))

    @property
    def url_kwargs(self):
        return {'survey_id': self.s.pk}

    @property
    def url_expected(self):
        return '/survey/{}'.format(self.s.pk)

    def setUp(self):
        super().setUp()

        self.s = Survey.objects.create(title='Title')

        self.q1 = self.s.questions.create(question='Question 1 ?')
        self.q2 = self.s.questions.create(
            question='Question 2 ?',
            multi_answers=True,
        )

        self.qa1 = self.q1.answers.create(answer='Q1 - Answer 1')
        self.qa2 = self.q1.answers.create(answer='Q1 - Answer 2')
        self.qa3 = self.q2.answers.create(answer='Q2 - Answer 1')
        self.qa4 = self.q2.answers.create(answer='Q2 - Answer 2')

    def test_get(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def test_post_new(self):
        r = self.client.post(self.url, {
            'question_{}'.format(self.q1.pk): [str(self.qa1.pk)],
            'question_{}'.format(self.q2.pk): [
                str(self.qa3.pk), str(self.qa4.pk),
            ],
        })

        self.assertEqual(r.status_code, 200)
        self.assertIn(self.post_expected_message, get_messages(r.wsgi_request))

        a = self.s.surveyanswer_set.get(user=self.users['user'])
        self.assertQuerysetEqual(
            a.answers.all(), map(repr, [self.qa1, self.qa3, self.qa4]),
            ordered=False,
        )

    def test_post_edit(self):
        a = self.s.surveyanswer_set.create(user=self.users['user'])
        a.answers.add(self.qa1, self.qa1, self.qa4)

        r = self.client.post(self.url, {
            'question_{}'.format(self.q1.pk): [],
            'question_{}'.format(self.q2.pk): [str(self.qa3.pk)],
        })

        self.assertEqual(r.status_code, 200)
        self.assertIn(self.post_expected_message, get_messages(r.wsgi_request))

        a.refresh_from_db()
        self.assertQuerysetEqual(
            a.answers.all(), map(repr, [self.qa3]),
            ordered=False,
        )

    def test_post_delete(self):
        a = self.s.surveyanswer_set.create(user=self.users['user'])
        a.answers.add(self.qa1, self.qa4)

        r = self.client.post(self.url, {'delete': '1'})

        self.assertEqual(r.status_code, 200)
        expected_message = Message(
            messages.SUCCESS, "Votre réponse a bien été supprimée")
        self.assertIn(expected_message, get_messages(r.wsgi_request))

        with self.assertRaises(SurveyAnswer.DoesNotExist):
            a.refresh_from_db()

    def test_forbidden_closed(self):
        self.s.survey_open = False
        self.s.save()

        r = self.client.get(self.url)

        self.assertNotEqual(r.status_code, 200)

    def test_forbidden_old(self):
        self.s.old = True
        self.s.save()

        r = self.client.get(self.url)

        self.assertNotEqual(r.status_code, 200)


class SurveyStatusViewTests(ViewTestCaseMixin, TestCase):
    url_name = 'survey.details.status'

    http_methods = ['GET', 'POST']

    auth_user = 'staff'
    auth_forbidden = [None, 'user', 'member']

    @property
    def url_kwargs(self):
        return {'survey_id': self.s.pk}

    @property
    def url_expected(self):
        return '/survey/{}/status'.format(self.s.pk)

    def setUp(self):
        super().setUp()

        self.s = Survey.objects.create(title='Title')

        self.q1 = self.s.questions.create(question='Question 1 ?')
        self.q2 = self.s.questions.create(
            question='Question 2 ?',
            multi_answers=True,
        )

        self.qa1 = self.q1.answers.create(answer='Q1 - Answer 1')
        self.qa2 = self.q1.answers.create(answer='Q1 - Answer 2')
        self.qa3 = self.q2.answers.create(answer='Q2 - Answer 1')
        self.qa4 = self.q2.answers.create(answer='Q2 - Answer 2')

        self.a1 = self.s.surveyanswer_set.create(user=self.users['user'])
        self.a1.answers.add(self.qa1)
        self.a2 = self.s.surveyanswer_set.create(user=self.users['member'])

    def test_get(self):
        r = self.client.get(self.url)
        self.assertEqual(r.status_code, 200)

    def _get_qa_filter_name(self, qa):
        return 'question_{}_answer_{}'.format(qa.survey_question.pk, qa.pk)

    def _test_filters(self, filters, expected):
        r = self.client.post(self.url, {
            self._get_qa_filter_name(qa): v for qa, v in filters
        })

        self.assertEqual(r.status_code, 200)
        self.assertQuerysetEqual(
            r.context['user_answers'], map(repr, expected),
            ordered=False,
        )

    def test_filter_none(self):
        self._test_filters([(self.qa1, 'none')], [self.a1, self.a2])

    def test_filter_yes(self):
        self._test_filters([(self.qa1, 'yes')], [self.a1])

    def test_filter_no(self):
        self._test_filters([(self.qa1, 'no')], [self.a2])