from gcsa.free_busy import FreeBusy, TimeRange
from gcsa.serializers.base_serializer import BaseSerializer


class FreeBusySerializer(BaseSerializer):
    type_ = FreeBusy

    def __init__(self, free_busy):
        super().__init__(free_busy)

    @staticmethod
    def _to_json(free_busy: FreeBusy):
        """Isn't used as free busy data is read-only"""
        free_busy_json = {
            'calendars': {
                c: {
                    'busy': [
                        {
                            'start': start.isoformat(),
                            'end': end.isoformat(),
                        }
                        for start, end in free_busy.calendars.get(c, [])
                    ],
                    'errors': free_busy.calendars_errors.get(c, [])
                }
                for c in {**free_busy.calendars, **free_busy.calendars_errors}
            },
            'groups': {
                g: {
                    'calendars': free_busy.groups.get(g, []),
                    'errors': free_busy.groups_errors.get(g, [])
                }
                for g in {**free_busy.groups, **free_busy.groups_errors}
            },
            'timeMin': free_busy.time_min.isoformat(),
            'timeMax': free_busy.time_max.isoformat(),

        }
        return free_busy_json

    @staticmethod
    def _to_object(json_):
        time_min = FreeBusySerializer._get_datetime_from_string(json_['timeMin'])
        time_max = FreeBusySerializer._get_datetime_from_string(json_['timeMax'])
        groups_json = json_.get('groups')
        calendars_json = json_.get("calendars")

        if groups_json:
            groups = {gn: g['calendars'] for gn, g in groups_json.items() if g.get('calendars')}
            groups_errors = {gn: g['errors'] for gn, g in groups_json.items() if g.get('errors')}
        else:
            groups = {}
            groups_errors = {}

        if calendars_json:
            calendars = {
                cn: list(map(FreeBusySerializer._make_time_range, c['busy']))
                for cn, c in calendars_json.items() if c.get('busy') and not c.get('errors')
            }
            calendars_errors = {
                cn: c['errors']
                for cn, c in calendars_json.items() if c.get('errors')
            }
        else:
            calendars = {}
            calendars_errors = {}

        return FreeBusy(
            time_min=time_min,
            time_max=time_max,
            groups=groups,
            calendars=calendars,
            groups_errors=groups_errors,
            calendars_errors=calendars_errors
        )

    @staticmethod
    def _make_time_range(tp):
        return TimeRange(
            start=FreeBusySerializer._get_datetime_from_string(tp['start']),
            end=FreeBusySerializer._get_datetime_from_string(tp['end'])
        )