#! /home/ggentile/perso/projects/python/Envs/dorfsvald-pim/bin/python3

import os
import datetime
import calendar

from yaml import load
from icalendar.parser import Contentlines, Contentline
from dateutil.rrule import rrule, FREQNAMES
import caldav


class Task:
    def __init__(self, ref_date, name, data):
        self.ref_date = ref_date
        self.name = name
        data['start']['freq'] = FREQNAMES.index(data['start']['freq'])
        self.date_begin = rrule(count=1, dtstart=ref_date, **data['start'])[0]
        self.date_end = self.date_begin + datetime.timedelta(microseconds=-1, **data['duration'])

        self.title = data['title']
        self.body = data['body']
        self.priority = data['priority']

    def to_ical(self):
        current_dtstamp = datetime.datetime.now().strftime('%Y%m%dT%H%M%SZ')
        uid = "DV-Taskmanager-{}-{}".format(self.name, self.date_end.strftime('%Y%m%d'))
        c = Contentlines([
            Contentline('BEGIN:VCALENDAR'),
            Contentline('VERSION:2.0'),
            Contentline('PRODID:-//DorfsvaldNet//Pim Atutoask Client//FR'),
            Contentline('BEGIN:VTODO'),
            Contentline('DTSTAMP:%s' % current_dtstamp),
            Contentline('UID:%s' % uid),
            Contentline('CREATED:%s' % current_dtstamp),
            Contentline('LAST-MODIFIED:%s' % current_dtstamp),
            Contentline('SEQUENCE:%s' % '4'),
            Contentline('DESCRIPTION:%s' % self.body),
            Contentline('SUMMARY:%s' % self.title),
            Contentline('PRIORITY:%s' % self.priority),
            Contentline('DUE;VALUE=DATE:%s' % self.date_end.strftime('%Y%m%d')),
            Contentline('DTSTART;VALUE=DATE:%s' % self.date_begin.strftime('%Y%m%d')),
            Contentline('PERCENT-COMPLETE:0'),
            Contentline('STATUS:NEEDS-ACTION'),
            Contentline('END:VTODO'),
            Contentline('END:VCALENDAR')
        ])

        return c.to_ical()


class Client:
    def __init__(self, url, username, password, calendar_name):
        self.client = caldav.DAVClient(url, username=username, password=password)
        self.principal = self.client.principal()
        self.calendar = None
        for c in self.principal.calendars():
            if c.name == calendar_name:
                self.calendar = c

        if self.calendar is None:
             raise LookupError('No calendar named "{}"'.format(calendar_name))

    def add_event(self, event):
        self.calendar.add_event(task.to_ical())


# conf = load('./configuration.yml')

ref_date = datetime.date.today() + datetime.timedelta(days=2)
with open('./tasks.yml', 'r') as content_file:
    tasks_conf = load(content_file.read())

task_list = []
for task_name, task_data in tasks_conf.items():
    t = Task(ref_date, task_name, task_data)
    if t.date_begin.date() == ref_date:
        task_list.append(t)

username="ggentile"
password="ID0t'8h6"
url = "https://cloud.dorfsvald.net/webdav/ggentile/"
calendar = "calendar"


client = Client(url, username, password, calendar)
for task in task_list:
    client.add_event(task)

print('process finished: {} tasks created'.format(len(task_list)))