Calendarr/backend/caldav_client.py

import logging
import uuid
from datetime import date, datetime, timedelta, timezone
from typing import Dict, List, Optional

import caldav
from icalendar import Calendar, Event, vDatetime, vRecur

logger = logging.getLogger(__name__)

PALETTE = ["#4285f4", "#ea4335", "#fbbc04", "#34a853", "#ff6d00", "#46bdc6", "#8e24aa"]


def _client(url: str, username: str, password: str) -> caldav.DAVClient:
    return caldav.DAVClient(url=url, username=username, password=password)


def fetch_calendars(url: str, username: str, password: str) -> List[Dict]:
    """Return list of {url, name, color} dicts for all calendars in the account."""
    try:
        client = _client(url, username, password)
        principal = client.principal()
        calendars = principal.calendars()
    except Exception as e:
        raise ValueError(f"Cannot connect to CalDAV server: {e}") from e

    result = []
    for idx, cal in enumerate(calendars):
        try:
            try:
                name = cal.get_properties([caldav.elements.dav.DisplayName()])[
                    "{DAV:}displayname"
                ]
            except Exception:
                name = str(cal.url).rstrip("/").split("/")[-1] or "Calendar"

            color = None
            try:
                props = cal.get_properties(
                    [caldav.elements.cdav.CalendarColor()]
                )
                raw = props.get("{http://apple.com/ns/ical/}calendar-color", "")
                if raw and len(raw) >= 7:
                    color = raw[:7]
            except Exception:
                pass

            result.append(
                {
                    "url": str(cal.url),
                    "name": name or "Calendar",
                    "color": color or PALETTE[idx % len(PALETTE)],
                }
            )
        except Exception as exc:
            logger.warning("Could not inspect calendar %s: %s", cal.url, exc)
            result.append(
                {
                    "url": str(cal.url),
                    "name": "Calendar",
                    "color": PALETTE[idx % len(PALETTE)],
                }
            )
    return result


def fetch_events(
    url: str,
    username: str,
    password: str,
    calendar_url: str,
    start: datetime,
    end: datetime,
) -> List[Dict]:
    """Fetch events from a calendar within [start, end]."""
    try:
        client = _client(url, username, password)
        cal = client.calendar(url=calendar_url)
        resources = cal.date_search(start=start, end=end, expand=True)
    except Exception as exc:
        logger.error("Error fetching events from %s: %s", calendar_url, exc)
        return []

    events = []
    for resource in resources:
        try:
            parsed = _parse_ics(resource.data, str(resource.url))
            if parsed:
                events.extend(parsed)
        except Exception as exc:
            logger.warning("Could not parse event %s: %s", resource.url, exc)
    return events


def _parse_ics(raw: str, event_url: str) -> List[Dict]:
    """Parse raw ICS data and return a list of event dicts."""
    cal = Calendar.from_ical(raw)
    results = []
    for component in cal.walk():
        if component.name != "VEVENT":
            continue
        try:
            uid = str(component.get("UID", uuid.uuid4()))
            title = str(component.get("SUMMARY", "Untitled"))
            location = str(component.get("LOCATION", "") or "")
            description = str(component.get("DESCRIPTION", "") or "")
            color = str(component.get("X-CALENDARR-COLOR", "") or "")
            rrule_prop = component.get("RRULE")
            rrule_str = rrule_prop.to_ical().decode("utf-8") if rrule_prop else None
            recurrence_id = component.get("RECURRENCE-ID")
            is_recurring = rrule_str is not None or recurrence_id is not None

            dtstart_prop = component.get("DTSTART")
            dtend_prop = component.get("DTEND")
            duration_prop = component.get("DURATION")

            if dtstart_prop is None:
                continue

            dtstart = dtstart_prop.dt
            all_day = isinstance(dtstart, date) and not isinstance(dtstart, datetime)

            if all_day:
                if dtend_prop:
                    dtend = dtend_prop.dt
                elif duration_prop:
                    dtend = dtstart + duration_prop.dt
                else:
                    dtend = dtstart + timedelta(days=1)
                start_str = dtstart.isoformat()
                end_str = dtend.isoformat() if isinstance(dtend, date) else dtstart.isoformat()
            else:
                if dtstart.tzinfo is None:
                    dtstart = dtstart.replace(tzinfo=timezone.utc)
                if dtend_prop:
                    dtend = dtend_prop.dt
                    if isinstance(dtend, date) and not isinstance(dtend, datetime):
                        dtend = datetime.combine(dtend, datetime.min.time()).replace(
                            tzinfo=timezone.utc
                        )
                    elif dtend.tzinfo is None:
                        dtend = dtend.replace(tzinfo=timezone.utc)
                elif duration_prop:
                    dtend = dtstart + duration_prop.dt
                else:
                    dtend = dtstart + timedelta(hours=1)
                start_str = dtstart.isoformat()
                end_str = dtend.isoformat()

            results.append(
                {
                    "id": uid,
                    "url": event_url,
                    "title": title,
                    "start": start_str,
                    "end": end_str,
                    "allDay": all_day,
                    "location": location,
                    "description": description,
                    "color": color or None,
                    "rrule": rrule_str,
                    "recurring": is_recurring,
                }
            )
        except Exception as exc:
            logger.warning("Skipping malformed VEVENT: %s", exc)
    return results


def create_event(
    url: str,
    username: str,
    password: str,
    calendar_url: str,
    data: Dict,
) -> str:
    client = _client(url, username, password)
    cal_obj = client.calendar(url=calendar_url)

    cal = Calendar()
    cal.add("prodid", "-//Calendarr//EN")
    cal.add("version", "2.0")

    event = Event()
    uid = str(uuid.uuid4())
    event.add("uid", uid)
    event.add("summary", data.get("title", "New Event"))
    event.add("dtstamp", datetime.now(timezone.utc))

    if data.get("allDay"):
        start = date.fromisoformat(data["start"][:10])
        end_raw = data.get("end", data["start"])[:10]
        end = date.fromisoformat(end_raw)
        if end <= start:
            end = start + timedelta(days=1)
        event.add("dtstart", start)
        event.add("dtend", end)
    else:
        start = _parse_dt(data["start"])
        end = _parse_dt(data["end"])
        event.add("dtstart", start)
        event.add("dtend", end)

    if data.get("location"):
        event.add("location", data["location"])
    if data.get("description"):
        event.add("description", data["description"])
    if data.get("color"):
        event.add("x-calendarr-color", data["color"])
    if data.get("rrule"):
        event.add("rrule", _parse_rrule_str(data["rrule"]))

    cal.add_component(event)
    cal_obj.save_event(cal.to_ical().decode("utf-8"))
    return uid


def update_event(
    url: str, username: str, password: str, event_url: str, data: Dict,
    calendar_url: str = None,
):
    client = _client(url, username, password)
    if calendar_url:
        cal_obj = client.calendar(url=calendar_url)
        resource = caldav.Event(client=client, url=event_url, parent=cal_obj)
    else:
        resource = caldav.Event(client=client, url=event_url)
    resource.load()
    raw = resource.data

    cal = Calendar.from_ical(raw)
    new_cal = Calendar()
    for key, val in cal.items():
        new_cal.add(key, val)

    for component in cal.walk():
        if component.name != "VEVENT":
            if component.name != "VCALENDAR":
                new_cal.add_component(component)
            continue

        if "title" in data or "summary" in data:
            if "SUMMARY" in component:
                del component["SUMMARY"]
            component.add("summary", data.get("title", data.get("summary", "")))

        if "start" in data:
            if "DTSTART" in component:
                del component["DTSTART"]
            if data.get("allDay"):
                component.add("dtstart", date.fromisoformat(data["start"][:10]))
            else:
                component.add("dtstart", _parse_dt(data["start"]))

        if "end" in data:
            if "DTEND" in component:
                del component["DTEND"]
            if data.get("allDay"):
                component.add("dtend", date.fromisoformat(data["end"][:10]))
            else:
                component.add("dtend", _parse_dt(data["end"]))

        if "location" in data:
            if "LOCATION" in component:
                del component["LOCATION"]
            component.add("location", data["location"])
        if "description" in data:
            if "DESCRIPTION" in component:
                del component["DESCRIPTION"]
            component.add("description", data["description"])
        if "color" in data:
            if "X-CALENDARR-COLOR" in component:
                del component["X-CALENDARR-COLOR"]
            component.add("x-calendarr-color", data["color"])
        if "rrule" in data:
            if "RRULE" in component:
                del component["RRULE"]
            if data["rrule"]:
                component.add("rrule", _parse_rrule_str(data["rrule"]))

        if "exdate" in data and data["exdate"]:
            # Parse YYYYMMDD string into a proper EXDATE
            exdate_str = data["exdate"]
            # Determine if event uses dates or datetimes
            dtstart_prop = component.get("DTSTART")
            if dtstart_prop and isinstance(dtstart_prop.dt, date) and not isinstance(dtstart_prop.dt, datetime):
                exdate_val = date(int(exdate_str[:4]), int(exdate_str[4:6]), int(exdate_str[6:8]))
            else:
                exdate_val = datetime(int(exdate_str[:4]), int(exdate_str[4:6]), int(exdate_str[6:8]), tzinfo=timezone.utc)
            component.add("exdate", [exdate_val])

        new_cal.add_component(component)

    resource.data = new_cal.to_ical().decode("utf-8")
    resource.save()


def delete_event(url: str, username: str, password: str, event_url: str,
                  calendar_url: str = None):
    client = _client(url, username, password)
    if calendar_url:
        cal_obj = client.calendar(url=calendar_url)
        resource = caldav.Event(client=client, url=event_url, parent=cal_obj)
    else:
        resource = caldav.Event(client=client, url=event_url)
    resource.delete()


def _parse_rrule_str(rrule_str: str) -> vRecur:
    """Parse an RRULE string like 'FREQ=WEEKLY;BYDAY=MO,WE' into a vRecur."""
    params = {}
    for part in rrule_str.split(";"):
        if "=" not in part:
            continue
        key, val = part.split("=", 1)
        if "," in val:
            params[key] = val.split(",")
        else:
            params[key] = val
    return vRecur(params)


def _parse_dt(s: str) -> datetime:
    s = s.replace("Z", "+00:00")
    dt = datetime.fromisoformat(s)
    if dt.tzinfo is None:
        dt = dt.replace(tzinfo=timezone.utc)
    return dt