path: root/src/Costasdev.Busurbano.Backend/Services/OtpService.cs

using System.Globalization;
using Costasdev.Busurbano.Backend.Configuration;
using Costasdev.Busurbano.Backend.Helpers;
using Costasdev.Busurbano.Backend.Types.Otp;
using Costasdev.Busurbano.Backend.Types.Planner;
using Costasdev.Busurbano.Sources.OpenTripPlannerGql.Queries;
using Microsoft.Extensions.Caching.Memory;
using Microsoft.Extensions.Options;

namespace Costasdev.Busurbano.Backend.Services;

public class OtpService
{
    private readonly HttpClient _httpClient;
    private readonly AppConfiguration _config;
    private readonly IMemoryCache _cache;
    private readonly ILogger<OtpService> _logger;
    private readonly FareService _fareService;
    private readonly LineFormatterService _lineFormatter;
    private readonly FeedService _feedService;

    public OtpService(HttpClient httpClient, IOptions<AppConfiguration> config, IMemoryCache cache, ILogger<OtpService> logger, FareService fareService, LineFormatterService lineFormatter, FeedService feedService)
    {
        _httpClient = httpClient;
        _config = config.Value;
        _cache = cache;
        _logger = logger;
        _fareService = fareService;
        _lineFormatter = lineFormatter;
        _feedService = feedService;
    }

    public async Task<List<PlannerSearchResult>> GetAutocompleteAsync(string query)
    {
        if (string.IsNullOrWhiteSpace(query)) return new List<PlannerSearchResult>();

        var cacheKey = $"otp_autocomplete_{query.ToLowerInvariant()}";
        if (_cache.TryGetValue(cacheKey, out List<PlannerSearchResult>? cachedResults) && cachedResults != null)
        {
            return cachedResults;
        }

        try
        {
            // https://planificador-rutas-api.vigo.org/v1/autocomplete?text=XXXX&layers=venue,street,address&lang=es
            var url = $"{_config.OtpGeocodingBaseUrl}/autocomplete?text={Uri.EscapeDataString(query)}&layers=venue,address&lang=es";
            var response = await _httpClient.GetFromJsonAsync<OtpGeocodeResponse>(url);

            var results = response?.Features.Select(f => new PlannerSearchResult
            {
                Name = f.Properties?.Name,
                Label = $"{f.Properties?.PostalCode} {f.Properties?.LocalAdmin}, {f.Properties?.Region}",
                Layer = f.Properties?.Layer,
                Lat = f.Geometry?.Coordinates.Count > 1 ? f.Geometry.Coordinates[1] : 0,
                Lon = f.Geometry?.Coordinates.Count > 0 ? f.Geometry.Coordinates[0] : 0
            }).ToList() ?? new List<PlannerSearchResult>();

            _cache.Set(cacheKey, results, TimeSpan.FromMinutes(30)); // Cache for 30 mins
            return results;
        }
        catch (Exception ex)
        {
            _logger.LogError(ex, "Error fetching autocomplete results");
            return new List<PlannerSearchResult>();
        }
    }

    public async Task<PlannerSearchResult?> GetReverseGeocodeAsync(double lat, double lon)
    {
        var cacheKey = $"otp_reverse_{lat:F5}_{lon:F5}";
        if (_cache.TryGetValue(cacheKey, out PlannerSearchResult? cachedResult) && cachedResult != null)
        {
            return cachedResult;
        }

        try
        {
            // https://planificador-rutas-api.vigo.org/v1/reverse?point.lat=LAT&point.lon=LON&lang=es
            var url = $"{_config.OtpGeocodingBaseUrl}/reverse?point.lat={lat.ToString(CultureInfo.InvariantCulture)}&point.lon={lon.ToString(CultureInfo.InvariantCulture)}&lang=es";
            var response = await _httpClient.GetFromJsonAsync<OtpGeocodeResponse>(url);

            var feature = response?.Features.FirstOrDefault();
            if (feature == null) return null;

            var result = new PlannerSearchResult
            {
                Name = feature.Properties?.Name,
                Label = $"{feature.Properties?.PostalCode} {feature.Properties?.LocalAdmin}, {feature.Properties?.Region}",
                Layer = feature.Properties?.Layer,
                Lat = feature.Geometry?.Coordinates.Count > 1 ? feature.Geometry.Coordinates[1] : 0,
                Lon = feature.Geometry?.Coordinates.Count > 0 ? feature.Geometry.Coordinates[0] : 0
            };

            _cache.Set(cacheKey, result, TimeSpan.FromMinutes(60)); // Cache for 1 hour
            return result;
        }
        catch (Exception ex)
        {
            _logger.LogError(ex, "Error fetching reverse geocode results");
            return null;
        }
    }

    private Leg MapLeg(OtpLeg otpLeg)
    {
        return new Leg
        {
            Mode = otpLeg.Mode,
            RouteName = otpLeg.Route,
            RouteShortName = otpLeg.RouteShortName,
            RouteLongName = otpLeg.RouteLongName,
            Headsign = otpLeg.Headsign,
            AgencyName = otpLeg.AgencyName,
            RouteColor = otpLeg.RouteColor,
            RouteTextColor = otpLeg.RouteTextColor,
            From = MapPlace(otpLeg.From),
            To = MapPlace(otpLeg.To),
            StartTime = DateTimeOffset.FromUnixTimeMilliseconds(otpLeg.StartTime).UtcDateTime,
            EndTime = DateTimeOffset.FromUnixTimeMilliseconds(otpLeg.EndTime).UtcDateTime,
            DistanceMeters = otpLeg.Distance,
            Geometry = DecodePolyline(otpLeg.LegGeometry?.Points),
            Steps = otpLeg.Steps.Select(MapStep).ToList(),
            IntermediateStops = otpLeg.IntermediateStops.Select(MapPlace).Where(p => p != null).Cast<PlannerPlace>().ToList()
        };
    }

    private PlannerPlace? MapPlace(OtpPlace? otpPlace)
    {
        if (otpPlace == null) return null;
        var feedId = otpPlace.StopId?.Split(':')[0] ?? "unknown";
        return new PlannerPlace
        {
            Name = _feedService.NormalizeStopName(feedId, otpPlace.Name),
            Lat = otpPlace.Lat,
            Lon = otpPlace.Lon,
            StopId = otpPlace.StopId, // Use string directly
            StopCode = _feedService.NormalizeStopCode(feedId, otpPlace.StopCode ?? string.Empty)
        };
    }

    private Step MapStep(OtpWalkStep otpStep)
    {
        return new Step
        {
            DistanceMeters = otpStep.Distance,
            RelativeDirection = otpStep.RelativeDirection,
            AbsoluteDirection = otpStep.AbsoluteDirection,
            StreetName = otpStep.StreetName,
            Lat = otpStep.Lat,
            Lon = otpStep.Lon
        };
    }

    private PlannerGeometry? DecodePolyline(string? encodedPoints)
    {
        if (string.IsNullOrEmpty(encodedPoints)) return null;

        var coordinates = Decode(encodedPoints);
        return new PlannerGeometry
        {
            Coordinates = coordinates.Select(c => new List<double> { c.Lon, c.Lat }).ToList()
        };
    }

    // Polyline decoding algorithm
    private static List<(double Lat, double Lon)> Decode(string encodedPoints)
    {
        if (string.IsNullOrEmpty(encodedPoints))
            return new List<(double, double)>();

        var poly = new List<(double, double)>();
        char[] polylineChars = encodedPoints.ToCharArray();
        int index = 0;

        int currentLat = 0;
        int currentLng = 0;
        int next5bits;
        int sum;
        int shifter;

        while (index < polylineChars.Length)
        {
            // calculate next latitude
            sum = 0;
            shifter = 0;
            do
            {
                next5bits = (int)polylineChars[index++] - 63;
                sum |= (next5bits & 31) << shifter;
                shifter += 5;
            } while (next5bits >= 32 && index < polylineChars.Length);

            if (index >= polylineChars.Length)
                break;

            currentLat += (sum & 1) == 1 ? ~(sum >> 1) : (sum >> 1);

            // calculate next longitude
            sum = 0;
            shifter = 0;
            do
            {
                next5bits = (int)polylineChars[index++] - 63;
                sum |= (next5bits & 31) << shifter;
                shifter += 5;
            } while (next5bits >= 32 && index < polylineChars.Length);

            currentLng += (sum & 1) == 1 ? ~(sum >> 1) : (sum >> 1);

            poly.Add((Convert.ToDouble(currentLat) / 100000.0, Convert.ToDouble(currentLng) / 100000.0));
        }

        return poly;
    }

    public RoutePlan MapPlanResponse(PlanConnectionResponse response)
    {
        var itineraries = response.PlanConnection.Edges
            .Select(e => MapItinerary(e.Node))
            .ToList();

        return new RoutePlan
        {
            Itineraries = itineraries
        };
    }

    private Itinerary MapItinerary(PlanConnectionResponse.Node node)
    {
        var legs = node.Legs.Select(MapLeg).ToList();
        var fares = _fareService.CalculateFare(legs);

        return new Itinerary
        {
            DurationSeconds = node.DurationSeconds,
            StartTime = DateTime.Parse(node.Start8601, null, DateTimeStyles.RoundtripKind),
            EndTime = DateTime.Parse(node.End8601, null, DateTimeStyles.RoundtripKind),
            WalkDistanceMeters = node.WalkDistance,
            WalkTimeSeconds = node.WalkSeconds,
            TransitTimeSeconds = node.DurationSeconds - node.WalkSeconds - node.WaitingSeconds,
            WaitingTimeSeconds = node.WaitingSeconds,
            Legs = legs,
            CashFareEuro = fares.CashFareEuro,
            CardFareEuro = fares.CardFareEuro
        };
    }

    private Leg MapLeg(PlanConnectionResponse.Leg leg)
    {
        var feedId = leg.From.Stop?.GtfsId?.Split(':')[0] ?? "unknown";
        var shortName = _feedService.NormalizeRouteShortName(feedId, leg.Route?.ShortName ?? string.Empty);
        var headsign = leg.Headsign;

        if (feedId == "vitrasa")
        {
            headsign = headsign?.Replace("*", "");
            if (headsign == "FORA DE SERVIZO.G.B.")
            {
                headsign = "García Barbón, 7 (fora de servizo)";
            }

            switch (shortName)
            {
                case "A" when headsign != null && headsign.StartsWith("\"1\""):
                    shortName = "A1";
                    headsign = headsign.Replace("\"1\"", "");
                    break;
                case "6":
                    headsign = headsign?.Replace("\"", "");
                    break;
                case "FUT":
                    if (headsign == "CASTELAO-CAMELIAS-G.BARBÓN.M.GARRIDO")
                    {
                        shortName = "MAR";
                        headsign = "MARCADOR ⚽: CASTELAO-CAMELIAS-G.BARBÓN.M.GARRIDO";
                    }
                    else if (headsign == "P. ESPAÑA-T.VIGO-S.BADÍA")
                    {
                        shortName = "RIO";
                        headsign = "RÍO ⚽: P. ESPAÑA-T.VIGO-S.BADÍA";
                    }
                    else if (headsign == "NAVIA-BOUZAS-URZAIZ-G. ESPINO")
                    {
                        shortName = "GOL";
                        headsign = "GOL ⚽: NAVIA-BOUZAS-URZAIZ-G. ESPINO";
                    }
                    break;
            }
        }

        var color = leg.Route?.Color;
        var textColor = leg.Route?.TextColor;

        if (string.IsNullOrEmpty(color) || color == "FFFFFF")
        {
            var (fallbackColor, fallbackTextColor) = _feedService.GetFallbackColourForFeed(feedId);
            color = fallbackColor.Replace("#", "");
            textColor = fallbackTextColor.Replace("#", "");
        }
        else if (string.IsNullOrEmpty(textColor) || textColor == "000000")
        {
            textColor = ContrastHelper.GetBestTextColour(color).Replace("#", "");
        }

        return new Leg
        {
            Mode = leg.Mode,
            FeedId = feedId,
            RouteId = leg.Route?.GtfsId,
            RouteName = leg.Route?.LongName,
            RouteShortName = shortName,
            RouteLongName = leg.Route?.LongName,
            Headsign = headsign,
            AgencyName = leg.Route?.Agency?.Name,
            RouteColor = color,
            RouteTextColor = textColor,
            From = MapPlace(leg.From),
            To = MapPlace(leg.To),
            StartTime = DateTime.Parse(leg.Start.ScheduledTime8601, null, DateTimeStyles.RoundtripKind),
            EndTime = DateTime.Parse(leg.End.ScheduledTime8601, null, DateTimeStyles.RoundtripKind),
            DistanceMeters = leg.Distance,
            Geometry = DecodePolyline(leg.LegGeometry?.Points),
            Steps = leg.Steps.Select(MapStep).ToList(),
            IntermediateStops = leg.StopCalls.Select(sc => MapPlace(sc.StopLocation)).ToList()
        };
    }

    private PlannerPlace MapPlace(PlanConnectionResponse.LegPosition pos)
    {
        var feedId = pos.Stop?.GtfsId?.Split(':')[0] ?? "unknown";
        return new PlannerPlace
        {
            Name = _feedService.NormalizeStopName(feedId, pos.Name),
            Lat = pos.Latitude,
            Lon = pos.Longitude,
            StopId = pos.Stop?.GtfsId,
            StopCode = _feedService.NormalizeStopCode(feedId, pos.Stop?.Code ?? string.Empty),
            ZoneId = pos.Stop?.ZoneId
        };
    }

    private PlannerPlace MapPlace(PlanConnectionResponse.StopLocation stop)
    {
        var feedId = stop.GtfsId?.Split(':')[0] ?? "unknown";
        return new PlannerPlace
        {
            Name = _feedService.NormalizeStopName(feedId, stop.Name),
            Lat = stop.Latitude,
            Lon = stop.Longitude,
            StopId = stop.GtfsId,
            StopCode = _feedService.NormalizeStopCode(feedId, stop.Code ?? string.Empty)
        };
    }

    private Step MapStep(PlanConnectionResponse.Step step)
    {
        return new Step
        {
            DistanceMeters = step.Distance,
            RelativeDirection = step.RelativeDirection,
            AbsoluteDirection = step.AbsoluteDirection,
            StreetName = step.StreetName,
            Lat = step.Latitude,
            Lon = step.Longitude
        };
    }
}