Filter calendar to BenSimaShared and Kate only

1 files changed, 7 insertions, 2 deletions

diff --git a/Omni/Agent/Tools/Calendar.hs b/Omni/Agent/Tools/Calendar.hs
index fbf7aae..900785d 100644
--- a/Omni/Agent/Tools/Calendar.hs
+++ b/Omni/Agent/Tools/Calendar.hs
@@ -59,11 +59,15 @@ test =
           Right cals -> (not (null cals) || null cals) Test.@=? True
     ]
 
+defaultCalendars :: [String]
+defaultCalendars = ["BenSimaShared", "Kate"]
+
 listEvents :: Text -> IO (Either Text Text)
 listEvents range = do
   let rangeArg = if Text.null range then "today 7d" else Text.unpack range
+      calArgs = concatMap (\c -> ["-a", c]) defaultCalendars
   result <-
-    try <| readProcessWithExitCode "khal" ["list", rangeArg, "-o"] ""
+    try <| readProcessWithExitCode "khal" (["list"] <> calArgs <> [rangeArg, "-o"]) ""
   case result of
     Left (e :: SomeException) ->
       pure (Left ("khal error: " <> tshow e))
@@ -94,8 +98,9 @@ addEvent calendarName eventSpec location alarm description = do
 
 searchEvents :: Text -> IO (Either Text Text)
 searchEvents query = do
+  let calArgs = concatMap (\c -> ["-a", c]) defaultCalendars
   result <-
-    try <| readProcessWithExitCode "khal" ["search", Text.unpack query] ""
+    try <| readProcessWithExitCode "khal" (["search"] <> calArgs <> [Text.unpack query]) ""
   case result of
     Left (e :: SomeException) ->
       pure (Left ("khal error: " <> tshow e))