#!/usr/bin/env python3

from ansible.module_utils.basic import AnsibleModule
from caldav import DAVClient
import vobject

def connect_calendar(url, user=None, password=None):
    # Verbindung zum Kalender herstellen
    client = DAVClient(url, username=user, password=password) if user else DAVClient(url)
    principal = client.principal()
    calendars = principal.calendars()
    
    if not calendars:
        raise Exception(f"Kein Kalender unter {url} gefunden")
    
    return calendars[0]

def get_event_uid_map(calendar):
    # Mapping der Ereignis-UIDs zu den Events erstellen
    uid_map = {}
    for event in calendar.events():
        try:
            cal = vobject.readOne(event.data)
            uid = cal.vevent.uid.value
            uid_map[uid] = event
        except Exception as e:
            continue  # Fehler bei einem Event überspringen
    return uid_map

def sync_calendars(source_cal, target_cal):
    # Kalender synchronisieren
    src_events = get_event_uid_map(source_cal)
    tgt_events = get_event_uid_map(target_cal)

    changed = False
    added, updated, removed = [], [], []

    # Neue und aktualisierte Ereignisse hinzufügen
    for uid, src_event in src_events.items():
        src_data = src_event.data
        if uid not in tgt_events:
            target_cal.add_event(src_data)
            added.append(uid)
            changed = True
        elif src_data != tgt_events[uid].data:
            tgt_events[uid].delete()
            target_cal.add_event(src_data)
            updated.append(uid)
            changed = True

    # Entfernte Ereignisse löschen
    for uid in set(tgt_events.keys()) - set(src_events.keys()):
        tgt_events[uid].delete()
        removed.append(uid)
        changed = True

    return changed, added, updated, removed

def run_module():
    # Definieren der Moduleingaben
    module_args = dict(
        source_url=dict(type='str', required=True),
        source_user=dict(type='str', required=False, default=None),
        source_password=dict(type='str', required=False, no_log=True, default=None),
        target_url=dict(type='str', required=True),
        target_user=dict(type='str', required=False, default=None),
        target_password=dict(type='str', required=False, no_log=True, default=None)
    )

    result = dict(changed=False)

    module = AnsibleModule(argument_spec=module_args, supports_check_mode=False)

    try:
        # Verbindung zu den Kalendern herstellen
        source_cal = connect_calendar(module.params['source_url'], module.params['source_user'], module.params['source_password'])
        target_cal = connect_calendar(module.params['target_url'], module.params['target_user'], module.params['target_password'])

        # Kalender synchronisieren
        changed, added, updated, removed = sync_calendars(source_cal, target_cal)

        result.update({
            "changed": changed,
            "added": added,
            "updated": updated,
            "removed": removed,
        })

    except Exception as e:
        module.fail_json(msg=str(e), **result)

    module.exit_json(**result)

if __name__ == '__main__':
    run_module()