report.py 35 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823
  1. # This file is part of Radicale - CalDAV and CardDAV server
  2. # Copyright © 2008 Nicolas Kandel
  3. # Copyright © 2008 Pascal Halter
  4. # Copyright © 2008-2017 Guillaume Ayoub
  5. # Copyright © 2017-2021 Unrud <unrud@outlook.com>
  6. # Copyright © 2024-2024 Pieter Hijma <pieterhijma@users.noreply.github.com>
  7. # Copyright © 2024-2024 Ray <ray@react0r.com>
  8. # Copyright © 2024-2025 Georgiy <metallerok@gmail.com>
  9. # Copyright © 2024-2025 Peter Bieringer <pb@bieringer.de>
  10. # Copyright © 2025-2025 David Greaves <david@dgreaves.com>
  11. #
  12. # This library is free software: you can redistribute it and/or modify
  13. # it under the terms of the GNU General Public License as published by
  14. # the Free Software Foundation, either version 3 of the License, or
  15. # (at your option) any later version.
  16. #
  17. # This library is distributed in the hope that it will be useful,
  18. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  19. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  20. # GNU General Public License for more details.
  21. #
  22. # You should have received a copy of the GNU General Public License
  23. # along with Radicale. If not, see <http://www.gnu.org/licenses/>.
  24. import contextlib
  25. import copy
  26. import datetime
  27. import posixpath
  28. import socket
  29. import xml.etree.ElementTree as ET
  30. from http import client
  31. from typing import (Callable, Iterable, Iterator, List, Optional, Sequence,
  32. Tuple, Union)
  33. from urllib.parse import unquote, urlparse
  34. import vobject
  35. import vobject.base
  36. from vobject.base import ContentLine
  37. import radicale.item as radicale_item
  38. from radicale import httputils, pathutils, storage, types, xmlutils
  39. from radicale.app.base import Access, ApplicationBase
  40. from radicale.item import filter as radicale_filter
  41. from radicale.log import logger
  42. DT_FORMAT_TIMESTAMP: str = '%Y%m%dT%H%M%SZ'
  43. DT_FORMAT_DATE: str = '%Y%m%d'
  44. def free_busy_report(base_prefix: str, path: str, xml_request: Optional[ET.Element],
  45. collection: storage.BaseCollection, encoding: str,
  46. unlock_storage_fn: Callable[[], None],
  47. max_occurrence: int
  48. ) -> Tuple[int, Union[ET.Element, str]]:
  49. # NOTE: this function returns both an Element and a string because
  50. # free-busy reports are an edge-case on the return type according
  51. # to the spec.
  52. multistatus = ET.Element(xmlutils.make_clark("D:multistatus"))
  53. if xml_request is None:
  54. return client.MULTI_STATUS, multistatus
  55. root = xml_request
  56. if (root.tag == xmlutils.make_clark("C:free-busy-query") and
  57. collection.tag != "VCALENDAR"):
  58. logger.warning("Invalid REPORT method %r on %r requested",
  59. xmlutils.make_human_tag(root.tag), path)
  60. return client.FORBIDDEN, xmlutils.webdav_error("D:supported-report")
  61. time_range_element = root.find(xmlutils.make_clark("C:time-range"))
  62. assert isinstance(time_range_element, ET.Element)
  63. # Build a single filter from the free busy query for retrieval
  64. # TODO: filter for VFREEBUSY in additional to VEVENT but
  65. # test_filter doesn't support that yet.
  66. vevent_cf_element = ET.Element(xmlutils.make_clark("C:comp-filter"),
  67. attrib={'name': 'VEVENT'})
  68. vevent_cf_element.append(time_range_element)
  69. vcalendar_cf_element = ET.Element(xmlutils.make_clark("C:comp-filter"),
  70. attrib={'name': 'VCALENDAR'})
  71. vcalendar_cf_element.append(vevent_cf_element)
  72. filter_element = ET.Element(xmlutils.make_clark("C:filter"))
  73. filter_element.append(vcalendar_cf_element)
  74. filters = (filter_element,)
  75. # First pull from storage
  76. retrieved_items = list(collection.get_filtered(filters))
  77. # !!! Don't access storage after this !!!
  78. unlock_storage_fn()
  79. cal = vobject.iCalendar()
  80. collection_tag = collection.tag
  81. while retrieved_items:
  82. # Second filtering before evaluating occurrences.
  83. # ``item.vobject_item`` might be accessed during filtering.
  84. # Don't keep reference to ``item``, because VObject requires a lot of
  85. # memory.
  86. item, filter_matched = retrieved_items.pop(0)
  87. if not filter_matched:
  88. try:
  89. if not test_filter(collection_tag, item, filter_element):
  90. continue
  91. except ValueError as e:
  92. raise ValueError("Failed to free-busy filter item %r from %r: %s" %
  93. (item.href, collection.path, e)) from e
  94. except Exception as e:
  95. raise RuntimeError("Failed to free-busy filter item %r from %r: %s" %
  96. (item.href, collection.path, e)) from e
  97. fbtype = None
  98. if item.component_name == 'VEVENT':
  99. transp = getattr(item.vobject_item.vevent, 'transp', None)
  100. if transp and transp.value != 'OPAQUE':
  101. continue
  102. status = getattr(item.vobject_item.vevent, 'status', None)
  103. if not status or status.value == 'CONFIRMED':
  104. fbtype = 'BUSY'
  105. elif status.value == 'CANCELLED':
  106. fbtype = 'FREE'
  107. elif status.value == 'TENTATIVE':
  108. fbtype = 'BUSY-TENTATIVE'
  109. else:
  110. # Could do fbtype = status.value for x-name, I prefer this
  111. fbtype = 'BUSY'
  112. # TODO: coalesce overlapping periods
  113. if max_occurrence > 0:
  114. n_occurrences = max_occurrence+1
  115. else:
  116. n_occurrences = 0
  117. occurrences = radicale_filter.time_range_fill(item.vobject_item,
  118. time_range_element,
  119. "VEVENT",
  120. n=n_occurrences)
  121. if len(occurrences) >= max_occurrence:
  122. raise ValueError("FREEBUSY occurrences limit of {} hit"
  123. .format(max_occurrence))
  124. for occurrence in occurrences:
  125. vfb = cal.add('vfreebusy')
  126. vfb.add('dtstamp').value = item.vobject_item.vevent.dtstamp.value
  127. vfb.add('dtstart').value, vfb.add('dtend').value = occurrence
  128. if fbtype:
  129. vfb.add('fbtype').value = fbtype
  130. return (client.OK, cal.serialize())
  131. def xml_report(base_prefix: str, path: str, xml_request: Optional[ET.Element],
  132. collection: storage.BaseCollection, encoding: str,
  133. unlock_storage_fn: Callable[[], None],
  134. max_occurrence: int = 0,
  135. ) -> Tuple[int, ET.Element]:
  136. """Read and answer REPORT requests that return XML.
  137. Read rfc3253-3.6 for info.
  138. """
  139. multistatus = ET.Element(xmlutils.make_clark("D:multistatus"))
  140. if xml_request is None:
  141. return client.MULTI_STATUS, multistatus
  142. root = xml_request
  143. if root.tag in (xmlutils.make_clark("D:principal-search-property-set"),
  144. xmlutils.make_clark("D:principal-property-search"),
  145. xmlutils.make_clark("D:expand-property")):
  146. # We don't support searching for principals or indirect retrieving of
  147. # properties, just return an empty result.
  148. # InfCloud asks for expand-property reports (even if we don't announce
  149. # support for them) and stops working if an error code is returned.
  150. logger.warning("Unsupported REPORT method %r on %r requested",
  151. xmlutils.make_human_tag(root.tag), path)
  152. return client.MULTI_STATUS, multistatus
  153. if (root.tag == xmlutils.make_clark("C:calendar-multiget") and
  154. collection.tag != "VCALENDAR" or
  155. root.tag == xmlutils.make_clark("CR:addressbook-multiget") and
  156. collection.tag != "VADDRESSBOOK" or
  157. root.tag == xmlutils.make_clark("D:sync-collection") and
  158. collection.tag not in ("VADDRESSBOOK", "VCALENDAR")):
  159. logger.warning("Invalid REPORT method %r on %r requested",
  160. xmlutils.make_human_tag(root.tag), path)
  161. return client.FORBIDDEN, xmlutils.webdav_error("D:supported-report")
  162. props: Union[ET.Element, List]
  163. if root.find(xmlutils.make_clark("D:prop")) is not None:
  164. props = root.find(xmlutils.make_clark("D:prop")) # type: ignore[assignment]
  165. else:
  166. props = []
  167. hreferences: Iterable[str]
  168. if root.tag in (
  169. xmlutils.make_clark("C:calendar-multiget"),
  170. xmlutils.make_clark("CR:addressbook-multiget")):
  171. # Read rfc4791-7.9 for info
  172. hreferences = set()
  173. for href_element in root.findall(xmlutils.make_clark("D:href")):
  174. temp_url_path = urlparse(href_element.text).path
  175. assert isinstance(temp_url_path, str)
  176. href_path = pathutils.sanitize_path(unquote(temp_url_path))
  177. if (href_path + "/").startswith(base_prefix + "/"):
  178. hreferences.add(href_path[len(base_prefix):])
  179. else:
  180. logger.warning("Skipping invalid path %r in REPORT request on "
  181. "%r", href_path, path)
  182. elif root.tag == xmlutils.make_clark("D:sync-collection"):
  183. old_sync_token_element = root.find(
  184. xmlutils.make_clark("D:sync-token"))
  185. old_sync_token = ""
  186. if old_sync_token_element is not None and old_sync_token_element.text:
  187. old_sync_token = old_sync_token_element.text.strip()
  188. logger.debug("Client provided sync token: %r", old_sync_token)
  189. try:
  190. sync_token, names = collection.sync(old_sync_token)
  191. except ValueError as e:
  192. # Invalid sync token
  193. logger.warning("Client provided invalid sync token %r: %s",
  194. old_sync_token, e, exc_info=True)
  195. # client.CONFLICT doesn't work with some clients (e.g. InfCloud)
  196. return (client.FORBIDDEN,
  197. xmlutils.webdav_error("D:valid-sync-token"))
  198. hreferences = (pathutils.unstrip_path(
  199. posixpath.join(collection.path, n)) for n in names)
  200. # Append current sync token to response
  201. sync_token_element = ET.Element(xmlutils.make_clark("D:sync-token"))
  202. sync_token_element.text = sync_token
  203. multistatus.append(sync_token_element)
  204. else:
  205. hreferences = (path,)
  206. filters = (
  207. root.findall(xmlutils.make_clark("C:filter")) +
  208. root.findall(xmlutils.make_clark("CR:filter")))
  209. expand = root.find(".//" + xmlutils.make_clark("C:expand"))
  210. # if we have expand prop we use "filter (except time range) -> expand -> filter (only time range)" approach
  211. time_range_element = None
  212. main_filters = []
  213. for filter_ in filters:
  214. # extract time-range filter for processing after main filters
  215. # for expand request
  216. filter_copy = copy.deepcopy(filter_)
  217. if expand is not None:
  218. for comp_filter in filter_copy.findall(".//" + xmlutils.make_clark("C:comp-filter")):
  219. if comp_filter.get("name", "").upper() == "VCALENDAR":
  220. continue
  221. time_range_element = comp_filter.find(xmlutils.make_clark("C:time-range"))
  222. if time_range_element is not None:
  223. comp_filter.remove(time_range_element)
  224. main_filters.append(filter_copy)
  225. # Retrieve everything required for finishing the request.
  226. retrieved_items = list(retrieve_items(
  227. base_prefix, path, collection, hreferences, main_filters, multistatus))
  228. collection_tag = collection.tag
  229. # !!! Don't access storage after this !!!
  230. unlock_storage_fn()
  231. n_vevents = 0
  232. while retrieved_items:
  233. # ``item.vobject_item`` might be accessed during filtering.
  234. # Don't keep reference to ``item``, because VObject requires a lot of
  235. # memory.
  236. item, filters_matched = retrieved_items.pop(0)
  237. if filters and not filters_matched:
  238. try:
  239. if not all(test_filter(collection_tag, item, filter_)
  240. for filter_ in main_filters):
  241. continue
  242. except ValueError as e:
  243. raise ValueError("Failed to filter item %r from %r: %s" %
  244. (item.href, collection.path, e)) from e
  245. except Exception as e:
  246. raise RuntimeError("Failed to filter item %r from %r: %s" %
  247. (item.href, collection.path, e)) from e
  248. found_props = []
  249. not_found_props = []
  250. item_etag: str = ""
  251. for prop in props:
  252. element = ET.Element(prop.tag)
  253. if prop.tag == xmlutils.make_clark("D:getetag"):
  254. if expand is not None:
  255. item_etag = item.etag
  256. else:
  257. element.text = item.etag
  258. found_props.append(element)
  259. elif prop.tag == xmlutils.make_clark("D:getcontenttype"):
  260. element.text = xmlutils.get_content_type(item, encoding)
  261. found_props.append(element)
  262. elif prop.tag in (
  263. xmlutils.make_clark("C:calendar-data"),
  264. xmlutils.make_clark("CR:address-data")):
  265. element.text = item.serialize()
  266. if (expand is not None) and item.component_name == 'VEVENT':
  267. starts = expand.get('start')
  268. ends = expand.get('end')
  269. if (starts is None) or (ends is None):
  270. return client.FORBIDDEN, \
  271. xmlutils.webdav_error("C:expand")
  272. start = datetime.datetime.strptime(
  273. starts, DT_FORMAT_TIMESTAMP
  274. ).replace(tzinfo=datetime.timezone.utc)
  275. end = datetime.datetime.strptime(
  276. ends, DT_FORMAT_TIMESTAMP
  277. ).replace(tzinfo=datetime.timezone.utc)
  278. time_range_start = None
  279. time_range_end = None
  280. if time_range_element is not None:
  281. time_range_start, time_range_end = radicale_filter.parse_time_range(time_range_element)
  282. (expanded_element, n_vev) = _expand(
  283. element=element, item=copy.copy(item),
  284. start=start, end=end,
  285. time_range_start=time_range_start, time_range_end=time_range_end,
  286. max_occurrence=max_occurrence,
  287. )
  288. if n_vev == 0:
  289. logger.debug("No VEVENTs found after expansion for %r, skipping", item.href)
  290. continue
  291. n_vevents += n_vev
  292. found_props.append(expanded_element)
  293. else:
  294. found_props.append(element)
  295. if hasattr(item.vobject_item, "vevent_list"):
  296. n_vevents += len(item.vobject_item.vevent_list)
  297. # Avoid DoS with too many events
  298. if max_occurrence and n_vevents > max_occurrence:
  299. raise ValueError("REPORT occurrences limit of {} hit"
  300. .format(max_occurrence))
  301. else:
  302. not_found_props.append(element)
  303. assert item.href
  304. uri = pathutils.unstrip_path(
  305. posixpath.join(collection.path, item.href))
  306. if found_props or not_found_props:
  307. multistatus.append(xml_item_response(
  308. base_prefix, uri, found_props=found_props,
  309. not_found_props=not_found_props, found_item=True, item_etag=item_etag))
  310. return client.MULTI_STATUS, multistatus
  311. def _expand(
  312. element: ET.Element,
  313. item: radicale_item.Item,
  314. start: datetime.datetime,
  315. end: datetime.datetime,
  316. time_range_start: Optional[datetime.datetime] = None,
  317. time_range_end: Optional[datetime.datetime] = None,
  318. max_occurrence: int = 0,
  319. ) -> Tuple[ET.Element, int]:
  320. vevent_component: vobject.base.Component = copy.copy(item.vobject_item)
  321. logger.info("Expanding event %s", item.href)
  322. logger.debug(f"Expand range: {start} to {end}")
  323. logger.debug(f"Time range: {time_range_start} to {time_range_end}")
  324. # Split the vevents included in the component into one that contains the
  325. # recurrence information and others that contain a recurrence id to
  326. # override instances.
  327. base_vevent, vevents_overridden = _split_overridden_vevents(vevent_component)
  328. dt_format = DT_FORMAT_TIMESTAMP
  329. all_day_event = False
  330. if type(base_vevent.dtstart.value) is datetime.date:
  331. # If an event comes to us with a dtstart specified as a date
  332. # then in the response we return the date, not datetime
  333. dt_format = DT_FORMAT_DATE
  334. all_day_event = True
  335. # In case of dates, we need to remove timezone information since
  336. # rruleset.between computes with datetimes without timezone information
  337. start = start.replace(tzinfo=None)
  338. end = end.replace(tzinfo=None)
  339. if time_range_start is not None and time_range_end is not None:
  340. time_range_start = time_range_start.replace(tzinfo=None)
  341. time_range_end = time_range_end.replace(tzinfo=None)
  342. for vevent in vevents_overridden:
  343. _strip_single_event(vevent, dt_format)
  344. duration = None
  345. if hasattr(base_vevent, "dtend"):
  346. duration = base_vevent.dtend.value - base_vevent.dtstart.value
  347. elif hasattr(base_vevent, "duration"):
  348. try:
  349. duration = base_vevent.duration.value
  350. if duration.total_seconds() <= 0:
  351. logger.warning("Invalid DURATION: %s", duration)
  352. duration = None
  353. except (AttributeError, TypeError) as e:
  354. logger.warning("Failed to parse DURATION: %s", e)
  355. duration = None
  356. # Generate EXDATE to remove from expansion range
  357. exdates_set: set[datetime.datetime] = set()
  358. if hasattr(base_vevent, 'exdate'):
  359. exdates = base_vevent.exdate.value
  360. if not isinstance(exdates, list):
  361. exdates = [exdates]
  362. exdates_set = {
  363. exdate.astimezone(datetime.timezone.utc) if isinstance(exdate, datetime.datetime)
  364. else datetime.datetime.fromordinal(exdate.toordinal()).replace(tzinfo=None)
  365. for exdate in exdates
  366. }
  367. logger.debug("EXDATE values: %s", exdates_set)
  368. events_for_filtering = vevents_overridden
  369. rruleset = None
  370. if hasattr(base_vevent, 'rrule'):
  371. rruleset = base_vevent.getrruleset()
  372. else:
  373. # if event does not have rrule, only include base event
  374. events_for_filtering = [base_vevent]
  375. filtered_vevents = []
  376. if rruleset:
  377. # This function uses datetimes internally without timezone info for dates
  378. # A vobject rruleset is for the event dtstart.
  379. # Expanded over a given time range this will not include
  380. # events which started before the time range but are still
  381. # ongoing at the start of the range
  382. # To accomodate this, reduce the start time by the duration of
  383. # the event. If this introduces an extra reccurence point then
  384. # that event should be included as it is still ongoing. If no
  385. # extra point is generated then it was a no-op.
  386. rstart = start - duration if duration and duration.total_seconds() > 0 else start
  387. recurrences = rruleset.between(rstart, end, inc=True, count=max_occurrence)
  388. if max_occurrence and len(recurrences) >= max_occurrence:
  389. # this shouldn't be > and if it's == then assume a limit
  390. # was hit and ignore that maybe some would be filtered out
  391. # by EXDATE etc. This is anti-DoS, not precise limits
  392. raise ValueError("REPORT occurrences limit of {} hit"
  393. .format(max_occurrence))
  394. _strip_component(vevent_component)
  395. _strip_single_event(base_vevent, dt_format)
  396. i_overridden = 0
  397. for recurrence_dt in recurrences:
  398. recurrence_utc = recurrence_dt if all_day_event else recurrence_dt.astimezone(datetime.timezone.utc)
  399. logger.debug("Processing recurrence: %s (all_day_event: %s)", recurrence_utc, all_day_event)
  400. # Apply time-range filter
  401. if time_range_start is not None and time_range_end is not None:
  402. dtstart = recurrence_utc
  403. dtend = dtstart + duration if duration else dtstart
  404. # Start includes the time, end does not
  405. if not (dtstart <= time_range_end and dtend > time_range_start):
  406. logger.debug("Recurrence %s filtered out by time-range", recurrence_utc)
  407. continue
  408. # Check exdate
  409. if recurrence_utc in exdates_set:
  410. logger.debug("Recurrence %s excluded by EXDATE", recurrence_utc)
  411. continue
  412. # Check for overridden instances
  413. i_overridden, vevent = _find_overridden(i_overridden, vevents_overridden, recurrence_utc, dt_format)
  414. if not vevent:
  415. # Create new instance from recurrence
  416. vevent = copy.deepcopy(base_vevent)
  417. # For all day events, the system timezone may influence the
  418. # results, so use recurrence_dt
  419. recurrence_id = recurrence_dt if all_day_event else recurrence_utc
  420. logger.debug("Creating new VEVENT with RECURRENCE-ID: %s", recurrence_id)
  421. vevent.recurrence_id = ContentLine(
  422. name='RECURRENCE-ID',
  423. value=recurrence_id, params={}
  424. )
  425. _convert_to_utc(vevent, 'recurrence_id', dt_format)
  426. suffix = ''
  427. if (dt_format == DT_FORMAT_DATE):
  428. suffix = ';VALUE=DATE'
  429. else:
  430. suffix = ''
  431. vevent.dtstart = ContentLine(
  432. name='DTSTART' + suffix,
  433. value=recurrence_id.strftime(dt_format), params={}
  434. )
  435. # if there is a DTEND, override it. Duration does not need changing
  436. if hasattr(vevent, "dtend"):
  437. vevent.dtend = ContentLine(
  438. name='DTEND' + suffix,
  439. value=(recurrence_id + duration).strftime(dt_format), params={}
  440. )
  441. filtered_vevents.append(vevent)
  442. # Filter overridden and non-recurring events
  443. if time_range_start is not None and time_range_end is not None:
  444. for vevent in events_for_filtering:
  445. dtstart = vevent.dtstart.value
  446. # Handle string values for DTSTART/DTEND
  447. if isinstance(dtstart, str):
  448. try:
  449. dtstart = datetime.datetime.strptime(dtstart, dt_format)
  450. if all_day_event:
  451. dtstart = dtstart.date()
  452. except ValueError as e:
  453. logger.warning("Invalid DTSTART format: %s, error: %s", dtstart, e)
  454. continue
  455. dtend = dtstart + duration if duration else dtstart
  456. logger.debug(
  457. "Filtering VEVENT with DTSTART: %s (type: %s), DTEND: %s (type: %s)",
  458. dtstart, type(dtstart), dtend, type(dtend))
  459. # Convert to datetime for comparison
  460. if all_day_event and isinstance(dtstart, datetime.date) and not isinstance(dtstart, datetime.datetime):
  461. dtstart = datetime.datetime.fromordinal(dtstart.toordinal()).replace(tzinfo=None)
  462. dtend = datetime.datetime.fromordinal(dtend.toordinal()).replace(tzinfo=None)
  463. elif not all_day_event and isinstance(dtstart, datetime.datetime) \
  464. and isinstance(dtend, datetime.datetime):
  465. dtstart = dtstart.replace(tzinfo=datetime.timezone.utc)
  466. dtend = dtend.replace(tzinfo=datetime.timezone.utc)
  467. else:
  468. logger.warning("Unexpected DTSTART/DTEND type: dtstart=%s, dtend=%s", type(dtstart), type(dtend))
  469. continue
  470. if dtstart < time_range_end and dtend > time_range_start:
  471. if vevent not in filtered_vevents: # Avoid duplicates
  472. logger.debug("VEVENT passed time-range filter: %s", dtstart)
  473. filtered_vevents.append(vevent)
  474. else:
  475. logger.debug("VEVENT filtered out: %s", dtstart)
  476. # Rebuild component
  477. if not filtered_vevents:
  478. element.text = ""
  479. return element, 0
  480. else:
  481. vevent_component.vevent_list = filtered_vevents
  482. logger.debug("lbt: vevent_component %s", vevent_component)
  483. element.text = vevent_component.serialize()
  484. return element, len(filtered_vevents)
  485. def _convert_timezone(vevent: vobject.icalendar.RecurringComponent,
  486. name_prop: str,
  487. name_content_line: str):
  488. prop = getattr(vevent, name_prop, None)
  489. if prop:
  490. if type(prop.value) is datetime.date:
  491. date_time = datetime.datetime.fromordinal(
  492. prop.value.toordinal()
  493. ).replace(tzinfo=datetime.timezone.utc)
  494. else:
  495. date_time = prop.value.astimezone(datetime.timezone.utc)
  496. setattr(vevent, name_prop, ContentLine(name=name_content_line, value=date_time, params=[]))
  497. def _convert_to_utc(vevent: vobject.icalendar.RecurringComponent,
  498. name_prop: str,
  499. dt_format: str):
  500. prop = getattr(vevent, name_prop, None)
  501. if prop:
  502. setattr(vevent, name_prop, ContentLine(name=prop.name, value=prop.value.strftime(dt_format), params=[]))
  503. def _strip_single_event(vevent: vobject.icalendar.RecurringComponent, dt_format: str) -> None:
  504. _convert_timezone(vevent, 'dtstart', 'DTSTART')
  505. _convert_timezone(vevent, 'dtend', 'DTEND')
  506. _convert_timezone(vevent, 'recurrence_id', 'RECURRENCE-ID')
  507. # There is something strange behaviour during serialization native datetime, so converting manually
  508. _convert_to_utc(vevent, 'dtstart', dt_format)
  509. _convert_to_utc(vevent, 'dtend', dt_format)
  510. _convert_to_utc(vevent, 'recurrence_id', dt_format)
  511. try:
  512. delattr(vevent, 'rrule')
  513. delattr(vevent, 'exdate')
  514. delattr(vevent, 'exrule')
  515. delattr(vevent, 'rdate')
  516. except AttributeError:
  517. pass
  518. def _strip_component(vevent: vobject.base.Component) -> None:
  519. timezones_to_remove = []
  520. for component in vevent.components():
  521. if component.name == 'VTIMEZONE':
  522. timezones_to_remove.append(component)
  523. for timezone in timezones_to_remove:
  524. vevent.remove(timezone)
  525. def _split_overridden_vevents(
  526. component: vobject.base.Component,
  527. ) -> Tuple[
  528. vobject.icalendar.RecurringComponent,
  529. List[vobject.icalendar.RecurringComponent]
  530. ]:
  531. vevent_recurrence = None
  532. vevents_overridden = []
  533. for vevent in component.vevent_list:
  534. if hasattr(vevent, 'recurrence_id'):
  535. vevents_overridden += [vevent]
  536. elif vevent_recurrence:
  537. raise ValueError(
  538. f"component with UID {vevent.uid} "
  539. f"has more than one vevent with recurrence information"
  540. )
  541. else:
  542. vevent_recurrence = vevent
  543. if vevent_recurrence:
  544. return (
  545. vevent_recurrence, sorted(
  546. vevents_overridden,
  547. key=lambda vevent: vevent.recurrence_id.value
  548. )
  549. )
  550. else:
  551. raise ValueError(
  552. f"component with UID {vevent.uid} "
  553. f"does not have a vevent without a recurrence_id"
  554. )
  555. def _find_overridden(
  556. start: int,
  557. vevents: List[vobject.icalendar.RecurringComponent],
  558. dt: datetime.datetime,
  559. dt_format: str
  560. ) -> Tuple[int, Optional[vobject.icalendar.RecurringComponent]]:
  561. for i in range(start, len(vevents)):
  562. dt_event = datetime.datetime.strptime(
  563. vevents[i].recurrence_id.value,
  564. dt_format
  565. ).replace(tzinfo=datetime.timezone.utc)
  566. if dt_event == dt:
  567. return (i + 1, vevents[i])
  568. return (start, None)
  569. def xml_item_response(base_prefix: str, href: str,
  570. found_props: Sequence[ET.Element] = (),
  571. not_found_props: Sequence[ET.Element] = (),
  572. found_item: bool = True, item_etag: str = "") -> ET.Element:
  573. response = ET.Element(xmlutils.make_clark("D:response"))
  574. href_element = ET.Element(xmlutils.make_clark("D:href"))
  575. href_element.text = xmlutils.make_href(base_prefix, href)
  576. response.append(href_element)
  577. if found_item:
  578. for code, props in ((200, found_props), (404, not_found_props)):
  579. if props:
  580. propstat = ET.Element(xmlutils.make_clark("D:propstat"))
  581. status = ET.Element(xmlutils.make_clark("D:status"))
  582. status.text = xmlutils.make_response(code)
  583. prop_element = ET.Element(xmlutils.make_clark("D:prop"))
  584. if (item_etag != "") and (code == 200):
  585. prop_etag = ET.Element(xmlutils.make_clark("D:getetag"))
  586. prop_etag.text = item_etag
  587. prop_element.append(prop_etag)
  588. for prop in props:
  589. prop_element.append(prop)
  590. propstat.append(prop_element)
  591. propstat.append(status)
  592. response.append(propstat)
  593. else:
  594. status = ET.Element(xmlutils.make_clark("D:status"))
  595. status.text = xmlutils.make_response(404)
  596. response.append(status)
  597. return response
  598. def retrieve_items(
  599. base_prefix: str, path: str, collection: storage.BaseCollection,
  600. hreferences: Iterable[str], filters: Sequence[ET.Element],
  601. multistatus: ET.Element) -> Iterator[Tuple[radicale_item.Item, bool]]:
  602. """Retrieves all items that are referenced in ``hreferences`` from
  603. ``collection`` and adds 404 responses for missing and invalid items
  604. to ``multistatus``."""
  605. collection_requested = False
  606. def get_names() -> Iterator[str]:
  607. """Extracts all names from references in ``hreferences`` and adds
  608. 404 responses for invalid references to ``multistatus``.
  609. If the whole collections is referenced ``collection_requested``
  610. gets set to ``True``."""
  611. nonlocal collection_requested
  612. for hreference in hreferences:
  613. try:
  614. name = pathutils.name_from_path(hreference, collection)
  615. except ValueError as e:
  616. logger.warning("Skipping invalid path %r in REPORT request on "
  617. "%r: %s", hreference, path, e)
  618. response = xml_item_response(base_prefix, hreference,
  619. found_item=False)
  620. multistatus.append(response)
  621. continue
  622. if name:
  623. # Reference is an item
  624. yield name
  625. else:
  626. # Reference is a collection
  627. collection_requested = True
  628. for name, item in collection.get_multi(get_names()):
  629. if not item:
  630. uri = pathutils.unstrip_path(posixpath.join(collection.path, name))
  631. response = xml_item_response(base_prefix, uri, found_item=False)
  632. multistatus.append(response)
  633. else:
  634. yield item, False
  635. if collection_requested:
  636. yield from collection.get_filtered(filters)
  637. def test_filter(collection_tag: str, item: radicale_item.Item,
  638. filter_: ET.Element) -> bool:
  639. """Match an item against a filter."""
  640. if (collection_tag == "VCALENDAR" and
  641. filter_.tag != xmlutils.make_clark("C:%s" % filter_)):
  642. if len(filter_) == 0:
  643. return True
  644. if len(filter_) > 1:
  645. raise ValueError("Filter with %d children" % len(filter_))
  646. if filter_[0].tag != xmlutils.make_clark("C:comp-filter"):
  647. raise ValueError("Unexpected %r in filter" % filter_[0].tag)
  648. return radicale_filter.comp_match(item, filter_[0])
  649. if (collection_tag == "VADDRESSBOOK" and
  650. filter_.tag != xmlutils.make_clark("CR:%s" % filter_)):
  651. for child in filter_:
  652. if child.tag != xmlutils.make_clark("CR:prop-filter"):
  653. raise ValueError("Unexpected %r in filter" % child.tag)
  654. test = filter_.get("test", "anyof")
  655. if test == "anyof":
  656. return any(radicale_filter.prop_match(item.vobject_item, f, "CR")
  657. for f in filter_)
  658. if test == "allof":
  659. return all(radicale_filter.prop_match(item.vobject_item, f, "CR")
  660. for f in filter_)
  661. raise ValueError("Unsupported filter test: %r" % test)
  662. raise ValueError("Unsupported filter %r for %r" %
  663. (filter_.tag, collection_tag))
  664. class ApplicationPartReport(ApplicationBase):
  665. def do_REPORT(self, environ: types.WSGIEnviron, base_prefix: str,
  666. path: str, user: str) -> types.WSGIResponse:
  667. """Manage REPORT request."""
  668. access = Access(self._rights, user, path)
  669. if not access.check("r"):
  670. return httputils.NOT_ALLOWED
  671. try:
  672. xml_content = self._read_xml_request_body(environ)
  673. except RuntimeError as e:
  674. logger.warning("Bad REPORT request on %r: %s", path, e,
  675. exc_info=True)
  676. return httputils.BAD_REQUEST
  677. except socket.timeout:
  678. logger.debug("Client timed out", exc_info=True)
  679. return httputils.REQUEST_TIMEOUT
  680. with contextlib.ExitStack() as lock_stack:
  681. lock_stack.enter_context(self._storage.acquire_lock("r", user))
  682. item = next(iter(self._storage.discover(path)), None)
  683. if not item:
  684. return httputils.NOT_FOUND
  685. if not access.check("r", item):
  686. return httputils.NOT_ALLOWED
  687. if isinstance(item, storage.BaseCollection):
  688. collection = item
  689. else:
  690. assert item.collection is not None
  691. collection = item.collection
  692. max_occurrence = self.configuration.get("reporting", "max_freebusy_occurrence")
  693. if xml_content is not None and \
  694. xml_content.tag == xmlutils.make_clark("C:free-busy-query"):
  695. try:
  696. status, body = free_busy_report(
  697. base_prefix, path, xml_content, collection, self._encoding,
  698. lock_stack.close, max_occurrence)
  699. except ValueError as e:
  700. logger.warning(
  701. "Bad REPORT request on %r: %s", path, e, exc_info=True)
  702. return httputils.BAD_REQUEST
  703. headers = {"Content-Type": "text/calendar; charset=%s" % self._encoding}
  704. return status, headers, str(body)
  705. else:
  706. try:
  707. status, xml_answer = xml_report(
  708. base_prefix, path, xml_content, collection, self._encoding,
  709. lock_stack.close, max_occurrence)
  710. except ValueError as e:
  711. logger.warning(
  712. "Bad REPORT request on %r: %s", path, e, exc_info=True)
  713. return httputils.BAD_REQUEST
  714. headers = {"Content-Type": "text/xml; charset=%s" % self._encoding}
  715. return status, headers, self._xml_response(xml_answer)