RejsePlanenToMqttBackgroundService.cs

// See https://aka.ms/new-console-template for more information
using HomeAssistantDiscoveryNet;
using Microsoft.Extensions.Options;
using MQTTnet;
using NodaTime;
using System.Text.Json;
using System.Text.Json.Serialization;
using ToMqttNet;

namespace Rejseplanen2Mqtt.Client;

public class RejsePlanenToMqttBackgroundService(
    ILogger<RejsePlanenToMqttBackgroundService> logger,
    IOptions<RejseplanenToMqttOptions> options,
    RejseplanenClient rejseplanenClient,
    MqttConnectionService mqtt) : BackgroundService
{
    private readonly ILogger<RejsePlanenToMqttBackgroundService> _logger = logger;
    private readonly RejseplanenToMqttOptions options = options.Value;
    private readonly RejseplanenClient _rejseplanenClient = rejseplanenClient;
    private readonly MqttConnectionService _mqtt = mqtt;

    protected override async Task ExecuteAsync(CancellationToken stoppingToken)
    {
        foreach (var trip in options.TripsToPublish)
        {
            var discoveryDoc = new MqttSensorDiscoveryConfig()
            {
                UniqueId = "rejseplanen_" + trip.Name.ToLower().Replace(" ", "_"),
                Name = trip.Name,
                UnitOfMeasurement = HomeAssistantUnits.TIME_MINUTES.Value,
                ValueTemplate = "{{ value_json.value }}",
                StateTopic = _mqtt.MqttOptions.NodeId + "/status/trips/" + trip.Name.ToLower().Replace(" ", "_"),
                JsonAttributesTopic = _mqtt.MqttOptions.NodeId + "/status/trips/" + trip.Name.ToLower().Replace(" ", "_"),
                JsonAttributesTemplate = "{{ value_json.attributes | to_json }}",
            };

            await _mqtt.PublishDiscoveryDocument(discoveryDoc);
        }

        while (!stoppingToken.IsCancellationRequested)
        {
            logger.LogInformation("Publishing trips...");
            foreach (var tripToPublish in options.TripsToPublish)
            {
                var localTime = SystemClock.Instance.GetCurrentInstant().InZone(DateTimeZoneProviders.Tzdb.GetZoneOrNull("Europe/Copenhagen")!).LocalDateTime;
                var requestOptions = new TripRequestOptions
                {
                    OriginId = tripToPublish.OriginId,
                    DestId = tripToPublish.DestId
                };

                if (tripToPublish.Time != null)
                {
                    requestOptions.Time = RejseplanenClient.TimePattern.Parse(tripToPublish.Time).Value;

                    if (requestOptions.Time.Value.PlusHours(3) < localTime.TimeOfDay) // We're far enough away from the request time that we should switch to the next day
                    {
                        requestOptions.Date = localTime.Date.PlusDays(1);
                    }
                }

                var response = await _rejseplanenClient.TripAsync(requestOptions);

                var result = new TripMqttStatusUpdate
                {
                    Value = -1,
                    Attributes = new()
                    {
                        Timestamp = SystemClock.Instance.GetCurrentInstant().ToUnixTimeSeconds(),
                    }
                };


                foreach (var directTrip in response.Where(x => !x.Cancelled && x.Legs.Count == 1)) // Only show direct trips
                {
                    var firstLeg = directTrip.Legs.First();
                    var tripStart = firstLeg.Origin;

                    var nextDepatureTime = RejseplanenClient.TimePattern.Parse(tripStart.RealtimeTime ?? tripStart.Time).Value;
                    var nextDepatureDate = RejseplanenClient.DatePattern.Parse(tripStart.RealtimeDate ?? tripStart.Date).Value;

                    var nextDepature = nextDepatureTime.On(nextDepatureDate);

                    var timeToNext = Period.Between(localTime, nextDepature, PeriodUnits.Minutes);
                    result.Attributes.Trips.Add(new TripMqttStatusUpdateAttributesTripInfo
                    {
                        DueIn = timeToNext.Minutes,
                        DueAt = (tripStart.RealtimeDate ?? tripStart.Date) + " " + (tripStart.RealtimeTime ?? tripStart.Time),
                        ScheduledAt = tripStart.Date + " " + tripStart.Time,
                        Route = firstLeg.Name,
                        Track = tripStart.RealtimeTrack ?? tripStart.Track,
                        Type = firstLeg.Type
                    });
                }

                result.Value = result.Attributes.Trips.Min(x => x.DueIn);
                await _mqtt.PublishAsync(new MqttApplicationMessageBuilder()
                    .WithTopic(_mqtt.MqttOptions.NodeId + "/status/trips/" + tripToPublish.Name.ToLower().Replace(" ", "_"))
                    .WithPayload(JsonSerializer.Serialize(result, RejseplanenJsonContext.Default.TripMqttStatusUpdate))
                    .Build());

            }
            await Task.Delay(TimeSpan.FromSeconds(options.UpdateIntervalSeconds), stoppingToken);
        }
    }
}

public class TripMqttStatusUpdate
{
    [JsonPropertyName("value")]
    public long Value { get; set; }
    [JsonPropertyName("attributes")]
    public TripMqttStatusUpdateAttributes Attributes { get; set; } = null!;
}

public class TripMqttStatusUpdateAttributes
{
    [JsonPropertyName("ts")]
    public long Timestamp { get; set; }

    [JsonPropertyName("trips")]
    public List<TripMqttStatusUpdateAttributesTripInfo> Trips { get; set; } = [];
}

public class TripMqttStatusUpdateAttributesTripInfo
{

    [JsonPropertyName("type")]
    public string Type { get; set; } = null!;

    [JsonPropertyName("route")]
    public string Route { get; set; } = null!;

    [JsonPropertyName("track")]
    public string? Track { get; set; }

    [JsonPropertyName("due_in")]
    public long DueIn { get; set; }

    [JsonPropertyName("due_at")]
    public string DueAt { get; set; } = null!;

    [JsonPropertyName("scheduled_at")]
    public string ScheduledAt { get; set; } = null!;
}

public class TripToPublish
{
    public string Name { get; set; } = null!;
    public string OriginId { get; set; } = null!;
    public string DestId { get; set; } = null!;
    public string? Time { get; set; } = null!;
}