CRSim/CRSim.ScreenSimulator/ViewModels/MetroPlatformScreenViewModel.cs

using CommunityToolkit.Mvvm.ComponentModel;
using CRSim.Core.Abstractions;
using CRSim.Core.Models;
using CRSim.ScreenSimulator.Abstractions;
using CRSim.ScreenSimulator.Models;

namespace CRSim.ScreenSimulator.ViewModels
{
    public partial class MetroPlatformScreenViewModel : ObservableObject, IScreenViewModel
    {
        public ITimeService TimeService { get; set; }
        public readonly Settings _settings;
        [ObservableProperty]
        private DateTime _currentTime = new();
        [ObservableProperty]
        private TrainInfo _firstTrain = new();
        [ObservableProperty]
        private TrainInfo _secondTrain = new();
        [ObservableProperty]
        private string _text;
        [ObservableProperty]
        private Uri _video;
        [ObservableProperty]
        public int _location;
        public System.Windows.Threading.Dispatcher UIDispatcher { get; set; }
        public List<TrainInfo> TrainInfos { get; set; } = [];
        public MetroPlatformScreenViewModel(ITimeService timeService, ISettingsService settingsService)
        {
            TimeService = timeService;
            _settings = settingsService.GetSettings();
            timeService.OneSecondElapsed += RefreshDisplay;
        }
        private void RefreshDisplay(object? sender, EventArgs e)
        {
            CurrentTime = TimeService.GetDateTimeNow();
            List<TrainInfo> itemsToRemove = [.. TrainInfos.Where(info => info.DepartureTime < TimeService.GetDateTimeNow())];
            foreach (var item in itemsToRemove)
            {
                TrainInfos.Remove(item);
            }
            FirstTrain = TrainInfos.Count > 0 ? TrainInfos[0] : new();
            SecondTrain = TrainInfos.Count > 1 ? TrainInfos[1] : new();

            OnPropertyChanged(nameof(FirstTrain));
            OnPropertyChanged(nameof(SecondTrain));
        }
        public void LoadData(Station station, TicketCheck? _ticketCheck, string platform)
        {
            var trains = station.TrainStops;
            foreach (var trainNumber in trains)
            {
                if (trainNumber != null && trainNumber.DepartureTime != null && trainNumber.Platform == platform)
                {
                    var now = TimeService.GetDateTimeNow();
                    var today = now.Date;
                    var scheduleOffset = TrainStatus.GetScheduleOffset(trainNumber.Status);
                    if (_settings.LoadTodayOnly && today.Add((trainNumber.DepartureTime ?? trainNumber.ArrivalTime)!.Value).Add(scheduleOffset) < now)
                    {
                        continue;
                    }
                    DateTime? AdjustTime(TimeSpan? time) =>
                        time.HasValue ? (today.Add(time.Value).Add(scheduleOffset) > now ? today.Add(time.Value) : today.Add(time.Value).AddDays(1)) : null;

                    TrainInfos.Add(new TrainInfo
                    {
                        TrainNumber = trainNumber.Number,
                        Terminal = trainNumber.Terminal,
                        ArrivalTime = AdjustTime(trainNumber.ArrivalTime),
                        DepartureTime = AdjustTime(trainNumber.DepartureTime),
                        State = trainNumber.Status
                    });
                }
            }
            TrainInfos = [.. TrainInfos.OrderBy(x => x.ArrivalTime ?? x.DepartureTime)];
        }

    }
}