#    Copyright (C) 2019  Alban Gruin
#
#    celcatsanitizer is free software: you can redistribute it and/or modify
#    it under the terms of the GNU Affero General Public License as published
#    by the Free Software Foundation, either version 3 of the License, or
#    (at your option) any later version.
#
#    celcatsanitizer is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU Affero General Public License for more details.
#
#    You should have received a copy of the GNU Affero General Public License
#    along with celcatsanitizer.  If not, see <http://www.gnu.org/licenses/>.

from datetime import date, datetime, timedelta
from html import unescape

from django.utils import timezone

import requests

from ...models import Course, Group, Module, Room
from ...utils import get_current_week, get_week
from .abstractparser import AbstractParser
from .ups2018 import GROUP_PREFIXES


class Parser(AbstractParser):
    def __get_event(self, event, year, week):
        if event["allDay"]:
            return

        begin = timezone.make_aware(
            datetime.strptime(event["start"], "%Y-%m-%dT%H:%M:%S")
        )
        end = timezone.make_aware(
            datetime.strptime(event["end"], "%Y-%m-%dT%H:%M:%S")
        )

        if year is not None and week is not None:
            event_year, event_week, _ = begin.isocalendar()
            if event_year != year or event_week != week:
                return

        data = [unescape(st.strip())
                for st in event["description"].split("<br />")]
        groups = []
        rooms = []

        course = Course.objects.create(
            source=self.source, begin=begin, end=end,
            celcat_id=event["id"]
        )

        max_i = len(data)

        if event.get("eventCategory") is not None and \
           len(event.get("eventCategory", "")) > 0:
            course.type = event["eventCategory"]
            max_i -= 1

        if event.get("module", "") is not None and \
           len(event.get("module", "")) > 0:
            module, _ = Module.objects.get_or_create(name=event["module"])
            course.module = module

        i = 0
        while i < max_i and not data[i].startswith(GROUP_PREFIXES):
            rooms.append(data[i])
            i += 1
        course.rooms.add(*Room.objects.filter(name__in=rooms))

        if len(rooms) != course.rooms.count():
            print(rooms, course.rooms)

        while i < max_i and data[i].startswith(GROUP_PREFIXES):
            group, _ = Group.objects.get_or_create(source=self.source,
                                                   celcat_name=data[i])
            groups.append(group)
            i += 1
        course.groups.add(*groups)

        if i < max_i and course.module is not None and \
           data[i].startswith(course.module.name):
            course.name = data[i]
            i += 1

        course.notes = "\n".join(data[i:max_i]).strip()
        if "other" in data[i]:
            print("Warning: \"other\" in notes")

        return course

    def get_events(self, today, year=None, week=None):
        for event in self.events:
            course = self.__get_event(event, year, week)
            if course is not None:
                yield course

    def get_update_date(self):
        return

    def get_weeks(self):
        # FIXME: détection automatique à partir des événements présents
        beginning, _ = get_week(*get_current_week())
        self.weeks = {"1": beginning}

        return self.weeks

    def get_source(self):
        start = date.today()
        end = start + timedelta(days=365)

        req = requests.post(self.source.url,
                            headers={"User-Agent": self.user_agent},
                            data={"calView": "month",
                                  "resType": 103,
                                  "federationIds[]": self.source.metadata,
                                  "start": start.strftime("%Y-%m-%d"),
                                  "end": end.strftime("%Y-%m-%d")})
        req.encoding = "uft8"
        req.raise_for_status()

        self.events = req.json()
        return self.events