Calendar stuff.

Wino.Mail.WinUI/Controls/Calendar/DayColumnControl.cs

@@ -1,6 +1,9 @@
-using System;
+using System;
+using System.Collections.Specialized;
+using System.Linq;
 using Microsoft.UI.Xaml;
 using Microsoft.UI.Xaml.Controls;
+using Wino.Core.Domain.Collections;
 using Wino.Core.Domain.Models.Calendar;
 
 namespace Wino.Calendar.Controls;
@@ -10,7 +13,6 @@ public partial class DayColumnControl : Control
     private const string PART_HeaderDateDayText = nameof(PART_HeaderDateDayText);
     private const string PART_IsTodayBorder = nameof(PART_IsTodayBorder);
     private const string PART_ColumnHeaderText = nameof(PART_ColumnHeaderText);
-
     private const string PART_AllDayItemsControl = nameof(PART_AllDayItemsControl);
 
     private const string TodayState = nameof(TodayState);
@@ -20,6 +22,7 @@ public partial class DayColumnControl : Control
     private TextBlock ColumnHeaderText;
     private Border IsTodayBorder;
     private ItemsControl AllDayItemsControl;
+    private CalendarEventCollection _boundEventsCollection;
 
     public CalendarDayModel DayModel
     {
@@ -27,11 +30,16 @@ public partial class DayColumnControl : Control
         set { SetValue(DayModelProperty, value); }
     }
 
-    public static readonly DependencyProperty DayModelProperty = DependencyProperty.Register(nameof(DayModel), typeof(CalendarDayModel), typeof(DayColumnControl), new PropertyMetadata(null, new PropertyChangedCallback(OnRenderingPropertiesChanged)));
+    public static readonly DependencyProperty DayModelProperty = DependencyProperty.Register(
+        nameof(DayModel),
+        typeof(CalendarDayModel),
+        typeof(DayColumnControl),
+        new PropertyMetadata(null, new PropertyChangedCallback(OnRenderingPropertiesChanged)));
 
     public DayColumnControl()
     {
         DefaultStyleKey = typeof(DayColumnControl);
+        Unloaded += OnUnloaded;
     }
 
     protected override void OnApplyTemplate()
@@ -43,6 +51,7 @@ public partial class DayColumnControl : Control
         IsTodayBorder = GetTemplateChild(PART_IsTodayBorder) as Border;
         AllDayItemsControl = GetTemplateChild(PART_AllDayItemsControl) as ItemsControl;
 
+        RegisterEventsCollectionHandlers();
         UpdateValues();
     }
 
@@ -50,15 +59,78 @@ public partial class DayColumnControl : Control
     {
         if (control is DayColumnControl columnControl)
         {
+            columnControl.RegisterEventsCollectionHandlers();
             columnControl.UpdateValues();
         }
     }
 
+    private void OnUnloaded(object sender, RoutedEventArgs e)
+    {
+        DeregisterEventsCollectionHandlers();
+    }
+
+    private bool IsMonthlyTemplate() => ColumnHeaderText == null;
+
+    private void RegisterEventsCollectionHandlers()
+    {
+        var nextCollection = DayModel?.EventsCollection;
+        if (ReferenceEquals(_boundEventsCollection, nextCollection))
+            return;
+
+        DeregisterEventsCollectionHandlers();
+
+        _boundEventsCollection = nextCollection;
+        if (_boundEventsCollection == null)
+            return;
+
+        ((INotifyCollectionChanged)_boundEventsCollection.AllDayEvents).CollectionChanged += EventsCollectionChanged;
+        ((INotifyCollectionChanged)_boundEventsCollection.RegularEvents).CollectionChanged += EventsCollectionChanged;
+    }
+
+    private void DeregisterEventsCollectionHandlers()
+    {
+        if (_boundEventsCollection == null)
+            return;
+
+        ((INotifyCollectionChanged)_boundEventsCollection.AllDayEvents).CollectionChanged -= EventsCollectionChanged;
+        ((INotifyCollectionChanged)_boundEventsCollection.RegularEvents).CollectionChanged -= EventsCollectionChanged;
+        _boundEventsCollection = null;
+    }
+
+    private void EventsCollectionChanged(object sender, NotifyCollectionChangedEventArgs e)
+    {
+        UpdateEventItemsSource();
+    }
+
+    private void UpdateEventItemsSource()
+    {
+        if (AllDayItemsControl == null || DayModel == null) return;
+
+        if (IsMonthlyTemplate())
+        {
+            // Month cells should show all events for the day, not only all-day/multi-day.
+            var monthlyItems = DayModel.EventsCollection.AllDayEvents
+                .Concat(DayModel.EventsCollection.RegularEvents)
+                .GroupBy(a => a.Id)
+                .Select(g => g.First())
+                .OrderBy(a => a.StartDate)
+                .ToList();
+
+            AllDayItemsControl.ItemsSource = monthlyItems;
+            return;
+        }
+
+        AllDayItemsControl.ItemsSource = DayModel.EventsCollection.AllDayEvents;
+    }
+
     private void UpdateValues()
     {
-        if (HeaderDateDayText == null || IsTodayBorder == null || DayModel == null) return;
+        if (DayModel == null) return;
 
-        HeaderDateDayText.Text = DayModel.RepresentingDate.Day.ToString();
+        if (HeaderDateDayText != null)
+        {
+            HeaderDateDayText.Text = DayModel.RepresentingDate.Day.ToString();
+        }
 
         // Monthly template does not use it.
         if (ColumnHeaderText != null)
@@ -66,8 +138,9 @@ public partial class DayColumnControl : Control
             ColumnHeaderText.Text = DayModel.RepresentingDate.ToString("dddd", DayModel.CalendarRenderOptions.CalendarSettings.CultureInfo);
         }
 
-        AllDayItemsControl.ItemsSource = DayModel.EventsCollection.AllDayEvents;
+        UpdateEventItemsSource();
 
+        if (IsTodayBorder == null) return;
         bool isToday = DayModel.RepresentingDate.Date == DateTime.Now.Date;
 
         VisualStateManager.GoToState(this, isToday ? TodayState : NotTodayState, false);