✨ Filter calendars

Author: oschrenk

Sources/Cli/Calendars.swift

@@ -20,10 +20,51 @@ struct Calendars: ParsableCommand {
     valueName: "f"
   )) var format: CalendarFormat = .json
 
+  @Option(help: ArgumentHelp(
+    "Select calendars <v>. A comma separated list of calendar UUIDs",
+    valueName: "v"
+  )) var selectCalendars: [String] = []
+
+  @Option(help: ArgumentHelp(
+    "Ignore calendars <v>. A comma separated list of calendar UUIDs",
+    valueName: "v"
+  )) var ignoreCalendars: [String] = []
+
+  @Option(help: ArgumentHelp(
+    "Select calendar sources <s>. A comma separated list of calendar sources",
+    valueName: "s"
+  )) var selectCalendarSources: [String] = []
+
+  @Option(help: ArgumentHelp(
+    "Ignore calendar sources <s>. A comma separated list of calendar sources",
+    valueName: "s"
+  )) var ignoreCalendarSources: [String] = []
+
+  @Option(help: ArgumentHelp(
+    "Select calendar types <v>. A comma separated list of calendar types. " +
+      "Available: [local|caldav|exchange|subscription|birthday]",
+    valueName: "v"
+  )) var selectCalendarTypes: [EKCalendarType] = []
+
+  @Option(help: ArgumentHelp(
+    "Ignore calendar types <v>. A comma separated list of calendar types. " +
+      "Available: [local|caldav|exchange|subscription|birthday]",
+    valueName: "v"
+  )) var ignoreCalendarTypes: [EKCalendarType] = []
+
   mutating func run() {
     Log.setDebug(debug)
 
-    let calendars = EventStore().calendars()
+    let calendarFilter = CalendarFilter.build(
+      selectCalendars: selectCalendars,
+      ignoreCalendars: ignoreCalendars,
+      selectCalendarSources: selectCalendarSources,
+      ignoreCalendarSources: ignoreCalendarSources,
+      selectCalendarTypes: selectCalendarTypes,
+      ignoreCalendarTypes: ignoreCalendarTypes
+    )
+
+    let calendars = EventStore().calendars(filter: calendarFilter)
 
     switch format {
     case .json:

Sources/EventStore.swift

@@ -31,10 +31,15 @@ struct EventStore {
     return eventStore.calendars(for: EKEntityType.event)
   }
 
-  func calendars() -> [PlanCalendar] {
-    return fetchCalendars().map { calendar in
-      calendar.asCal()
-    }
+  /// Returns a list of calendars
+  ///
+  /// - Parameters:
+  ///     - filter: A filter to select certain calendars
+  func calendars(filter: (PlanCalendar) -> Bool) -> [PlanCalendar] {
+    return CalendarSelector.build(
+      calendars: fetchCalendars(),
+      filter: filter
+    ).map { $0.asCal() }
   }
 
   /// Returns a list of events