diff options
Diffstat (limited to 'src/Costasdev.Busurbano.Backend/Controllers/StopsTile.cs')
| -rw-r--r-- | src/Costasdev.Busurbano.Backend/Controllers/StopsTile.cs | 198 |
1 files changed, 198 insertions, 0 deletions
diff --git a/src/Costasdev.Busurbano.Backend/Controllers/StopsTile.cs b/src/Costasdev.Busurbano.Backend/Controllers/StopsTile.cs new file mode 100644 index 0000000..56f836e --- /dev/null +++ b/src/Costasdev.Busurbano.Backend/Controllers/StopsTile.cs @@ -0,0 +1,198 @@ +using Costasdev.Busurbano.Backend.GraphClient; +using Costasdev.Busurbano.Backend.GraphClient.App; +using NetTopologySuite.IO.VectorTiles; +using NetTopologySuite.IO.VectorTiles.Mapbox; + +using Microsoft.AspNetCore.Mvc; + +using Microsoft.Extensions.Caching.Memory; +using NetTopologySuite.Features; +using System.Text.Json; +using Costasdev.Busurbano.Backend.Helpers; + +[ApiController] +[Route("api/tiles")] +public class TileController : ControllerBase +{ + private readonly ILogger<TileController> _logger; + private readonly IMemoryCache _cache; + private readonly HttpClient _httpClient; + + public TileController( + ILogger<TileController> logger, + IMemoryCache cache, + HttpClient httpClient + ) + { + _logger = logger; + _cache = cache; + _httpClient = httpClient; + } + + /* + vitrasa:20223: # Castrelos (Pavillón) - Final U1 + hide: true +vitrasa:20146: # García Barbón 7 - final líneas A y 18A + hide: true +vitrasa:20220: # (Samil) COIA-SAMIL - Final L15A + hide: true +vitrasa:20001: # (Samil) Samil por Beiramar - Final L15B + hide: true +vitrasa:20002: # (Samil) Samil por Torrecedeira - Final L15C + hide: true +vitrasa:20144: # (Samil) Samil por Coia - Final C3D+C3i + hide: true +vitrasa:20145: # (Samil) Samil por Bouzas - Final C3D+C3i + hide: true + */ + private static readonly string[] HiddenStops = + [ + "vitrasa:20223", + "vitrasa:20146", + "vitrasa:20220", + "vitrasa:20001", + "vitrasa:20002", + "vitrasa:20144", + "vitrasa:20145" + ]; + + [HttpGet("stops/{z}/{x}/{y}")] + public async Task<IActionResult> GetTrafficTile(int z, int x, int y) + { + var cacheHit = _cache.TryGetValue($"stops-tile-{z}-{x}-{y}", out byte[]? cachedTile); + if (cacheHit && cachedTile != null) + { + Response.Headers.Append("X-Cache-Hit", "true"); + return File(cachedTile, "application/x-protobuf"); + } + + // Calculate bounding box in EPSG:4326 + double n = Math.Pow(2, z); + double lonMin = x / n * 360.0 - 180.0; + double lonMax = (x + 1) / n * 360.0 - 180.0; + + double latMaxRad = Math.Atan(Math.Sinh(Math.PI * (1 - 2 * y / n))); + double latMax = latMaxRad * 180.0 / Math.PI; + + double latMinRad = Math.Atan(Math.Sinh(Math.PI * (1 - 2 * (y + 1) / n))); + double latMin = latMinRad * 180.0 / Math.PI; + + var requestContent = StopTileRequestContent.Query(new StopTileRequestContent.Bbox(lonMin, latMin, lonMax, latMax)); + var request = new HttpRequestMessage(HttpMethod.Post, "http://100.67.54.115:3957/otp/gtfs/v1"); + request.Content = JsonContent.Create(new GraphClientRequest + { + Query = requestContent + }); + + var response = await _httpClient.SendAsync(request); + var responseBody = await response.Content.ReadFromJsonAsync<GraphClientResponse<StopTileResponse>>(); + + if (responseBody == null || !responseBody.IsSuccess) + { + _logger.LogError("Error fetching stop data: {StatusCode}", response.StatusCode); + _logger.LogError("Sent request: {RequestContent}", await request.Content.ReadAsStringAsync()); + _logger.LogError("Response body: {ResponseBody}", await response.Content.ReadAsStringAsync()); + return StatusCode(500, "Error fetching stop data"); + } + + var tileDef = new NetTopologySuite.IO.VectorTiles.Tiles.Tile(x, y, z); + VectorTile vt = new() { TileId = tileDef.Id }; + var lyr = new Layer { Name = "stops" }; + + responseBody.Data?.StopsByBbox?.ForEach(stop => + { + var idParts = stop.GtfsId.Split(':', 2); + string codeWithinFeed = stop.Code ?? string.Empty; + + // TODO: Refactor this, maybe do it client-side or smth + if (idParts[0] == "vitrasa") + { + var digits = new string(codeWithinFeed.Where(char.IsDigit).ToArray()); + if (int.TryParse(digits, out int code)) + { + codeWithinFeed = code.ToString(); + } + } + + if (HiddenStops.Contains($"{idParts[0]}:{codeWithinFeed}")) + { + return; + } + + var fallbackColours = GetFallbackColourForFeed(idParts[0]); + + Feature feature = new() + { + Geometry = new NetTopologySuite.Geometries.Point(stop.Lon, stop.Lat), + Attributes = new AttributesTable + { + // The ID will be used to request the arrivals + { "id", stop.GtfsId }, + // The feed is the first part of the GTFS ID, corresponding to the feed where the info comes from, used for icons probably + { "feed", idParts[0] }, + // The public identifier, usually feed:code or feed:id, recognisable by users and in other systems + { "code", $"{idParts[0]}:{codeWithinFeed}" }, + // The name of the stop + { "name", stop.Name }, + // Routes + { "routes", JsonSerializer.Serialize(stop.Routes?.OrderBy( + r => r.ShortName, + Comparer<string?>.Create(SortingHelper.SortRouteShortNames) + ).Select(r => { + var colour = r.Color ?? fallbackColours.Color; + string textColour; + + if (r.Color is null) // None is present, use fallback + { + textColour = fallbackColours.TextColor; + } + else if (r.TextColor is null || r.TextColor.EndsWith("000000")) + { + // Text colour not provided, or default-black; check the better contrasting + textColour = ContrastHelper.GetBestTextColour(colour); + } + else + { + // Use provided text colour + textColour = r.TextColor; + } + + return new { + shortName = r.ShortName, + colour, + textColour + }; + })) } + } + }; + + lyr.Features.Add(feature); + }); + + vt.Layers.Add(lyr); + + using var ms = new MemoryStream(); + vt.Write(ms, minLinealExtent: 1, minPolygonalExtent: 2); + + _cache.Set($"stops-tile-{z}-{x}-{y}", ms.ToArray(), TimeSpan.FromMinutes(15)); + Response.Headers.Append("X-Cache-Hit", "false"); + + return File(ms.ToArray(), "application/x-protobuf"); + } + + private static (string Color, string TextColor) GetFallbackColourForFeed(string feed) + { + return feed switch + { + "vitrasa" => ("#95D516", "#000000"), + "santiago" => ("#508096", "#FFFFFF"), + "coruna" => ("#E61C29", "#FFFFFF"), + "xunta" => ("#007BC4", "#FFFFFF"), + "renfe" => ("#870164", "#FFFFFF"), + "feve" => ("#EE3D32", "#FFFFFF"), + _ => ("#000000", "#FFFFFF"), + + }; + } + +} |