import os
import json
import datetime
import logging

from gi.repository import GLib

from gajim.plugins import GajimPlugin

from gajim.common import configpaths
from gajim.common import app
from gajim.common import ged

# Since Gajim 1.1.0 _() has to be imported
try:
    from gajim.common.i18n import _
except ImportError:
    pass

log = logging.getLogger('gajim.plugin_system.birthday')

TITLE = _('%s has birthday today')
TEXT = _('Send him a message')


class BirthDayPlugin(GajimPlugin):
    def init(self):
        self.config_dialog = None
        self.description = ('Birthday reminder plugin')
        self.events_handlers = {
            'vcard-received': (ged.GUI2, self._vcard_received)}

        self.timeout_id = None
        self._timeout_id_start = None

        self.showed_accounts = []

        self._birthdays = {}
        self._load_birthdays()

    def activate(self):
        self._timeout_id_start = GLib.timeout_add_seconds(
            5, self._check_birthdays_at_start)
        self._timeout_id = GLib.timeout_add_seconds(
            86400, self._check_birthdays)

    def deactivate(self):
        if self._timeout_id is not None:
            GLib.source_remove(self.timeout_id)
        if self._timeout_id_start is not None:
            GLib.source_remove(self._timeout_id_start)

    def _load_birthdays(self):
        path = os.path.join(configpaths.get('MY_DATA'), 'birthdays.json')
        if os.path.exists(path):
            with open(path, 'r', encoding='utf-8') as file:
                content = file.read()
                if content:
                    self._birthdays = json.loads(content)

    def _store_birthdays(self):
        path = os.path.join(configpaths.get('MY_DATA'), 'birthdays.json')
        with open(path, 'w', encoding='utf-8') as file:
            json.dump(self._birthdays, file)

    def _vcard_received(self, event):
        birthday = event.vcard_dict.get('BDAY')
        if not birthday:
            if event.jid in self._birthdays:
                del self._birthdays[event.jid]
                log.info('Received empty birthday: %s', event.jid)
        else:
            try:
                year, month, day = birthday.split('-')
                year = int(year)
                month = int(month)
                day = int(day)
            except Exception:
                log.warning('Invalid date: %s', birthday)
                if event.jid in self._birthdays:
                    del self._birthdays[event.jid]
            else:
                self._birthdays[event.jid] = (year, month, day)
                log.info('Received birthday: %s %s',
                         event.jid, (year, month, day))
        self._store_birthdays()

    def _check_birthdays_at_start(self):
        self._check_birthdays()

    def _check_birthdays(self):
        log.info('Check birthdays...')
        today = datetime.date.today()
        for jid, birthdate in self._birthdays.items():
            year, month, day = birthdate
            if today.month == month and today.day == day:
                account, contact = self._find_contact(jid)
                if contact is None:
                    if jid in self._birthdays:
                        del self._birthdays[jid]
                        self._store_birthdays()
                        continue
                else:
                    log.info('Issue notification for %s', jid)
                    nick = contact.get_shown_name() or jid
                    app.notification.popup(
                        'reminder',
                        jid,
                        account,
                        icon_name='trophy-gold',
                        title=TITLE % GLib.markup_escape_text(nick),
                        text=TEXT)

        return True

    def _find_contact(self, jid):
        accounts = app.contacts.get_accounts()
        for account in accounts:
            contact = app.contacts.get_contacts(account, jid)
            if contact is not None:
                return account, contact[0]