1 files changed, 21 insertions, 0 deletions
diff --git a/src/Enmarcha.Experimental.ServiceViewer/Data/QueryExtensions/GtfsCalendarQueryExtensions.cs b/src/Enmarcha.Experimental.ServiceViewer/Data/QueryExtensions/GtfsCalendarQueryExtensions.cs
new file mode 100644
index 0000000..5de48ce
--- /dev/null
+++ b/src/Enmarcha.Experimental.ServiceViewer/Data/QueryExtensions/GtfsCalendarQueryExtensions.cs
@@ -0,0 +1,21 @@
+using Enmarcha.Experimental.ServiceViewer.Data.Gtfs;
+
+namespace Enmarcha.Experimental.ServiceViewer.Data.QueryExtensions;
+
+public static class GtfsCalendarQueryExtensions
+{
+    public static IQueryable<GtfsCalendar> WhereDayOfWeek(this IQueryable<GtfsCalendar> query, DayOfWeek dayOfWeek)
+    {
+        return dayOfWeek switch
+        {
+            DayOfWeek.Monday => query.Where(c => c.Monday),
+            DayOfWeek.Tuesday => query.Where(c => c.Tuesday),
+            DayOfWeek.Wednesday => query.Where(c => c.Wednesday),
+            DayOfWeek.Thursday => query.Where(c => c.Thursday),
+            DayOfWeek.Friday => query.Where(c => c.Friday),
+            DayOfWeek.Saturday => query.Where(c => c.Saturday),
+            DayOfWeek.Sunday => query.Where(c => c.Sunday),
+            _ => query
+        };
+    }
+}