Compare commits
17 commits
169e128ca1
...
422b4727c0
Author | SHA1 | Date | |
---|---|---|---|
422b4727c0 | |||
9e680e3bb2 | |||
8421b49834 | |||
d4ad04b0f3 | |||
1abb5155e3 | |||
5785e4a98b | |||
dd273558e4 | |||
2465a90305 | |||
b66b8f79b1 | |||
1d9db5b491 | |||
de5d85cff4 | |||
44a1be5659 | |||
e4de0347ad | |||
0b5a8bf487 | |||
0bcaf40ea3 | |||
8ef53a64ad | |||
145f7b0ee1 |
34 changed files with 1489 additions and 764 deletions
2
.vscode/launch.json
vendored
2
.vscode/launch.json
vendored
|
@ -6,7 +6,7 @@
|
|||
"type": "coreclr",
|
||||
"request": "launch",
|
||||
"preLaunchTask": "buildConsoleTest",
|
||||
"program": "${workspaceFolder}/ConsoleTest/bin/Debug/net6.0/ConsoleTest.dll",
|
||||
"program": "${workspaceFolder}/ConsoleTest/bin/Debug/net7.0/ConsoleTest.dll",
|
||||
"args": [],
|
||||
"cwd": "${workspaceFolder}",
|
||||
"stopAtEntry": false,
|
||||
|
|
|
@ -6,7 +6,7 @@
|
|||
|
||||
<PropertyGroup>
|
||||
<OutputType>Exe</OutputType>
|
||||
<TargetFramework>net6.0</TargetFramework>
|
||||
<TargetFrameworks>net6.0;net7.0;net8.0</TargetFrameworks>
|
||||
</PropertyGroup>
|
||||
|
||||
</Project>
|
||||
|
|
|
@ -1,4 +1,5 @@
|
|||
using System;
|
||||
using System.Linq;
|
||||
using System.Text.Json;
|
||||
using System.Threading.Tasks;
|
||||
using InfoferScraper;
|
||||
|
@ -7,6 +8,7 @@ using InfoferScraper.Scrapers;
|
|||
while (true) {
|
||||
Console.WriteLine("1. Scrape Train");
|
||||
Console.WriteLine("2. Scrape Station");
|
||||
Console.WriteLine("3. Scrape Itineraries");
|
||||
Console.WriteLine("0. Exit");
|
||||
|
||||
var input = Console.ReadLine()?.Trim();
|
||||
|
@ -17,6 +19,9 @@ while (true) {
|
|||
case "2":
|
||||
await PrintStation();
|
||||
break;
|
||||
case "3":
|
||||
await ScrapeItineraries();
|
||||
break;
|
||||
case null:
|
||||
case "0":
|
||||
goto INPUT_LOOP_BREAK;
|
||||
|
@ -35,7 +40,7 @@ async Task PrintTrain() {
|
|||
|
||||
Console.WriteLine(
|
||||
JsonSerializer.Serialize(
|
||||
await TrainScraper.Scrape(trainNumber),
|
||||
await new TrainScraper().Scrape(trainNumber),
|
||||
new JsonSerializerOptions {
|
||||
PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
|
||||
WriteIndented = true,
|
||||
|
@ -53,7 +58,7 @@ async Task PrintStation() {
|
|||
|
||||
Console.WriteLine(
|
||||
JsonSerializer.Serialize(
|
||||
await StationScraper.Scrape(stationName),
|
||||
await new StationScraper().Scrape(stationName),
|
||||
new JsonSerializerOptions {
|
||||
PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
|
||||
WriteIndented = true,
|
||||
|
@ -61,3 +66,30 @@ async Task PrintStation() {
|
|||
)
|
||||
);
|
||||
}
|
||||
async Task ScrapeItineraries() {
|
||||
Console.Write("From station: ");
|
||||
var from = Console.ReadLine();
|
||||
Console.Write("To station: ");
|
||||
var to = Console.ReadLine();
|
||||
|
||||
if (from == null || to == null) return;
|
||||
|
||||
var data = await new RouteScraper().Scrape(from, to);
|
||||
|
||||
Console.WriteLine($"{data.Count} itineraries:");
|
||||
Console.WriteLine();
|
||||
|
||||
void PrintArrDepLine(DateTimeOffset date, string station) {
|
||||
Console.WriteLine($"{date:HH:mm} {station}");
|
||||
}
|
||||
|
||||
foreach (var itinerary in data) {
|
||||
foreach (var train in itinerary.Trains) {
|
||||
PrintArrDepLine(train.DepartureDate, train.From);
|
||||
Console.WriteLine($" {train.TrainRank,-4} {train.TrainNumber,-5} ({train.Operator}), {train.Km,3} km via {string.Join(", ", train.IntermediateStops)}");
|
||||
PrintArrDepLine(train.ArrivalDate, train.To);
|
||||
}
|
||||
|
||||
Console.WriteLine();
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,5 +1,5 @@
|
|||
# https://hub.docker.com/_/microsoft-dotnet
|
||||
FROM mcr.microsoft.com/dotnet/sdk:6.0 AS build
|
||||
FROM mcr.microsoft.com/dotnet/sdk:7.0 AS build
|
||||
WORKDIR /source
|
||||
|
||||
# copy csproj and restore as distinct layers
|
||||
|
@ -14,10 +14,10 @@ COPY server/. ./server/
|
|||
COPY scraper/. ./scraper/
|
||||
COPY ConsoleTest/. ./ConsoleTest/
|
||||
WORKDIR /source/server
|
||||
RUN dotnet publish -c release -o /app --no-restore
|
||||
RUN dotnet publish -f net7.0 -c release -o /app --no-restore
|
||||
|
||||
# final stage/image
|
||||
FROM mcr.microsoft.com/dotnet/aspnet:6.0
|
||||
FROM mcr.microsoft.com/dotnet/aspnet:7.0
|
||||
WORKDIR /app
|
||||
COPY --from=build /app ./
|
||||
ENV INSIDE_DOCKER=true
|
||||
|
|
|
@ -5,8 +5,6 @@ services:
|
|||
image: new_infofer_scraper
|
||||
build: .
|
||||
ports:
|
||||
- ${PORT:-5000}:80
|
||||
- ${PORT:-5001}:80
|
||||
environment:
|
||||
- DB_DIR=/data
|
||||
volumes:
|
||||
- ./data:/data
|
||||
DB_DIR: /data
|
||||
|
|
|
@ -2,13 +2,14 @@
|
|||
|
||||
<PropertyGroup>
|
||||
<Nullable>enable</Nullable>
|
||||
<TargetFramework>net6.0</TargetFramework>
|
||||
<TargetFrameworks>net6.0;net7.0;net8.0</TargetFrameworks>
|
||||
</PropertyGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<PackageReference Include="AngleSharp" Version="0.16.0" />
|
||||
<PackageReference Include="Flurl" Version="3.0.2" />
|
||||
<PackageReference Include="Jetbrains.Annotations" Version="2021.2.0" />
|
||||
<PackageReference Include="Newtonsoft.Json" Version="13.0.3" />
|
||||
<PackageReference Include="NodaTime" Version="3.0.5" />
|
||||
</ItemGroup>
|
||||
|
||||
|
|
62
scraper/src/Models/Itinerary.cs
Normal file
62
scraper/src/Models/Itinerary.cs
Normal file
|
@ -0,0 +1,62 @@
|
|||
using System;
|
||||
using System.Collections.Generic;
|
||||
|
||||
namespace scraper.Models.Itinerary;
|
||||
|
||||
#region Interfaces
|
||||
|
||||
public interface IItinerary {
|
||||
public IReadOnlyList<IItineraryTrain> Trains { get; }
|
||||
}
|
||||
|
||||
public interface IItineraryTrain {
|
||||
public string From { get; }
|
||||
public string To { get; }
|
||||
public IReadOnlyList<string> IntermediateStops { get; }
|
||||
public DateTimeOffset DepartureDate { get; }
|
||||
public DateTimeOffset ArrivalDate { get; }
|
||||
public int Km { get; }
|
||||
public string Operator { get; }
|
||||
public string TrainRank { get; }
|
||||
public string TrainNumber { get; }
|
||||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region Implementations
|
||||
|
||||
internal record Itinerary : IItinerary {
|
||||
private List<IItineraryTrain> ModifyableTrains { get; set; } = new();
|
||||
|
||||
public IReadOnlyList<IItineraryTrain> Trains => ModifyableTrains;
|
||||
|
||||
internal void AddTrain(IItineraryTrain train) {
|
||||
ModifyableTrains.Add(train);
|
||||
}
|
||||
|
||||
internal void AddTrain(Action<ItineraryTrain> configurator) {
|
||||
ItineraryTrain newTrain = new();
|
||||
configurator(newTrain);
|
||||
AddTrain(newTrain);
|
||||
}
|
||||
}
|
||||
|
||||
internal record ItineraryTrain : IItineraryTrain {
|
||||
private List<string> ModifyableIntermediateStops { get; set; } = new();
|
||||
|
||||
public string From { get; internal set; } = "";
|
||||
public string To { get; internal set; } = "";
|
||||
public IReadOnlyList<string> IntermediateStops => ModifyableIntermediateStops;
|
||||
public DateTimeOffset DepartureDate { get; internal set; } = new();
|
||||
public DateTimeOffset ArrivalDate { get; internal set; } = new();
|
||||
public int Km { get; internal set; } = 0;
|
||||
public string Operator { get; internal set; } = "";
|
||||
public string TrainRank { get; internal set; } = "";
|
||||
public string TrainNumber { get; internal set; } = "";
|
||||
|
||||
internal void AddIntermediateStop(string stop) {
|
||||
ModifyableIntermediateStops.Add(stop);
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
|
@ -37,9 +37,6 @@ namespace InfoferScraper.Models.Station {
|
|||
}
|
||||
|
||||
public interface IStationStatus : IStatus {
|
||||
new int Delay { get; }
|
||||
new bool Real { get; }
|
||||
public bool Cancelled { get; }
|
||||
public string? Platform { get; }
|
||||
}
|
||||
|
||||
|
|
|
@ -6,10 +6,13 @@ namespace InfoferScraper.Models.Status {
|
|||
/// Determines whether delay was actually reported or is an approximation
|
||||
/// </summary>
|
||||
public bool Real { get; }
|
||||
|
||||
public bool Cancelled { get; }
|
||||
}
|
||||
|
||||
internal record Status : IStatus {
|
||||
public int Delay { get; set; }
|
||||
public bool Real { get; set; }
|
||||
public bool Cancelled { get; set; }
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,9 +1,9 @@
|
|||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Text.Json;
|
||||
using System.Text.Json.Serialization;
|
||||
using InfoferScraper.Models.Status;
|
||||
using InfoferScraper.Models.Train.JsonConverters;
|
||||
using Newtonsoft.Json;
|
||||
using Newtonsoft.Json.Converters;
|
||||
using Newtonsoft.Json.Serialization;
|
||||
|
||||
namespace InfoferScraper.Models.Train {
|
||||
#region Interfaces
|
||||
|
@ -44,6 +44,7 @@ namespace InfoferScraper.Models.Train {
|
|||
|
||||
public interface ITrainStopDescription {
|
||||
public string Name { get; }
|
||||
public string LinkName { get; }
|
||||
public int Km { get; }
|
||||
|
||||
/// <summary>
|
||||
|
@ -88,14 +89,14 @@ namespace InfoferScraper.Models.Train {
|
|||
|
||||
#endregion
|
||||
|
||||
[JsonConverter(typeof(StatusKindConverter))]
|
||||
[JsonConverter(typeof(StringEnumConverter), typeof(CamelCaseNamingStrategy))]
|
||||
public enum StatusKind {
|
||||
Passing,
|
||||
Arrival,
|
||||
Departure,
|
||||
}
|
||||
|
||||
[JsonConverter(typeof(NoteKindConverter))]
|
||||
[JsonConverter(typeof(StringEnumConverter), typeof(CamelCaseNamingStrategy))]
|
||||
public enum NoteKind {
|
||||
TrainNumberChange,
|
||||
DetachingWagons,
|
||||
|
@ -170,6 +171,7 @@ namespace InfoferScraper.Models.Train {
|
|||
internal record TrainStopDescription : ITrainStopDescription {
|
||||
private List<ITrainStopNote> ModifyableNotes { get; } = new();
|
||||
public string Name { get; set; } = "";
|
||||
public string LinkName { get; set; } = "";
|
||||
public int Km { get; set; }
|
||||
public int? StoppingTime { get; set; }
|
||||
public string? Platform { get; set; }
|
||||
|
@ -241,76 +243,4 @@ namespace InfoferScraper.Models.Train {
|
|||
}
|
||||
|
||||
#endregion
|
||||
|
||||
#region JSON Converters
|
||||
|
||||
namespace JsonConverters {
|
||||
internal class StatusKindConverter : JsonConverterFactory {
|
||||
public override bool CanConvert(Type typeToConvert) {
|
||||
return typeToConvert == typeof(StatusKind);
|
||||
}
|
||||
|
||||
public override JsonConverter? CreateConverter(Type typeToConvert, JsonSerializerOptions options) {
|
||||
return new Converter();
|
||||
}
|
||||
|
||||
private class Converter : JsonConverter<StatusKind> {
|
||||
public override StatusKind Read(
|
||||
ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options
|
||||
) {
|
||||
return reader.GetString() switch {
|
||||
"arrival" => StatusKind.Arrival,
|
||||
"departure" => StatusKind.Departure,
|
||||
"passing" => StatusKind.Passing,
|
||||
_ => throw new NotImplementedException()
|
||||
};
|
||||
}
|
||||
|
||||
public override void Write(Utf8JsonWriter writer, StatusKind value, JsonSerializerOptions options) {
|
||||
writer.WriteStringValue(value switch {
|
||||
StatusKind.Passing => "passing",
|
||||
StatusKind.Arrival => "arrival",
|
||||
StatusKind.Departure => "departure",
|
||||
_ => throw new NotImplementedException()
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
internal class NoteKindConverter : JsonConverterFactory {
|
||||
public override bool CanConvert(Type typeToConvert) {
|
||||
return typeToConvert == typeof(NoteKind);
|
||||
}
|
||||
|
||||
public override JsonConverter? CreateConverter(Type typeToConvert, JsonSerializerOptions options) {
|
||||
return new Converter();
|
||||
}
|
||||
|
||||
private class Converter : JsonConverter<NoteKind> {
|
||||
public override NoteKind Read(
|
||||
ref Utf8JsonReader reader, Type typeToConvert, JsonSerializerOptions options
|
||||
) {
|
||||
return reader.GetString() switch {
|
||||
"departsAs" => NoteKind.DepartsAs,
|
||||
"trainNumberChange" => NoteKind.TrainNumberChange,
|
||||
"receivingWagons" => NoteKind.ReceivingWagons,
|
||||
"detachingWagons" => NoteKind.DetachingWagons,
|
||||
_ => throw new NotImplementedException()
|
||||
};
|
||||
}
|
||||
|
||||
public override void Write(Utf8JsonWriter writer, NoteKind value, JsonSerializerOptions options) {
|
||||
writer.WriteStringValue(value switch {
|
||||
NoteKind.DepartsAs => "departsAs",
|
||||
NoteKind.TrainNumberChange => "trainNumberChange",
|
||||
NoteKind.DetachingWagons => "detachingWagons",
|
||||
NoteKind.ReceivingWagons => "receivingWagons",
|
||||
_ => throw new NotImplementedException()
|
||||
});
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
#endregion
|
||||
}
|
||||
|
|
220
scraper/src/Scrapers/Route.cs
Normal file
220
scraper/src/Scrapers/Route.cs
Normal file
|
@ -0,0 +1,220 @@
|
|||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Linq;
|
||||
using System.Net;
|
||||
using System.Net.Http;
|
||||
using System.Text.RegularExpressions;
|
||||
using System.Threading.Tasks;
|
||||
using AngleSharp;
|
||||
using AngleSharp.Dom;
|
||||
using AngleSharp.Html.Dom;
|
||||
using Flurl;
|
||||
using InfoferScraper.Models.Train;
|
||||
using NodaTime;
|
||||
using NodaTime.Extensions;
|
||||
using scraper.Models.Itinerary;
|
||||
|
||||
namespace InfoferScraper.Scrapers;
|
||||
|
||||
public class RouteScraper {
|
||||
private const string BaseUrl = "https://mersultrenurilor.infofer.ro/ro-RO/";
|
||||
private static readonly DateTimeZone BucharestTz = DateTimeZoneProviders.Tzdb["Europe/Bucharest"];
|
||||
|
||||
private static readonly Regex KmTrainRankNoRegex = new(@"^([0-9]+)\skm\scu\s([A-Z-]+)\s([0-9]+)$");
|
||||
private static readonly Regex OperatorRegex = new(@$"^Operat\sde\s([{Utils.RoLetters}\s]+)$");
|
||||
private static readonly Regex DepArrRegex = new(@"^(Ple|Sos)\s([0-9]+)\s([a-z]+)\.?\s([0-9]+):([0-9]+)$");
|
||||
|
||||
private static readonly Dictionary<string, int> Months = new Dictionary<string, int>() {
|
||||
["ian"] = 1,
|
||||
["feb"] = 2,
|
||||
["mar"] = 3,
|
||||
["apr"] = 4,
|
||||
["mai"] = 5,
|
||||
["iun"] = 6,
|
||||
["iul"] = 7,
|
||||
["aug"] = 8,
|
||||
["sep"] = 9,
|
||||
["oct"] = 10,
|
||||
["noi"] = 11,
|
||||
["dec"] = 12,
|
||||
};
|
||||
|
||||
private readonly CookieContainer cookieContainer = new();
|
||||
|
||||
private readonly HttpClient httpClient;
|
||||
|
||||
public RouteScraper(HttpClientHandler? httpClientHandler = null) {
|
||||
if (httpClientHandler == null) {
|
||||
httpClientHandler = new HttpClientHandler {
|
||||
CookieContainer = cookieContainer,
|
||||
UseCookies = true,
|
||||
};
|
||||
}
|
||||
else {
|
||||
httpClientHandler.CookieContainer = cookieContainer;
|
||||
httpClientHandler.UseCookies = true;
|
||||
}
|
||||
httpClient = new HttpClient(httpClientHandler) {
|
||||
BaseAddress = new Uri(BaseUrl),
|
||||
DefaultRequestVersion = new Version(2, 0),
|
||||
};
|
||||
}
|
||||
|
||||
public async Task<List<IItinerary>?> Scrape(string from, string to, DateTimeOffset? dateOverride = null) {
|
||||
var dateOverrideInstant = dateOverride?.ToInstant().InZone(BucharestTz);
|
||||
dateOverride = dateOverrideInstant?.ToDateTimeOffset();
|
||||
TrainScrapeResult result = new();
|
||||
|
||||
var asConfig = Configuration.Default;
|
||||
var asContext = BrowsingContext.New(asConfig);
|
||||
|
||||
var firstUrl = "Rute-trenuri"
|
||||
.AppendPathSegment(from)
|
||||
.AppendPathSegment(to);
|
||||
if (dateOverride != null) {
|
||||
firstUrl = firstUrl.SetQueryParam("DepartureDate", $"{dateOverride:d.MM.yyyy}");
|
||||
}
|
||||
firstUrl = firstUrl.SetQueryParam("OrderingTypeId", "0");
|
||||
firstUrl = firstUrl.SetQueryParam("TimeSelectionId", "0");
|
||||
firstUrl = firstUrl.SetQueryParam("MinutesInDay", "0");
|
||||
firstUrl = firstUrl.SetQueryParam("ConnectionsTypeId", "1");
|
||||
firstUrl = firstUrl.SetQueryParam("BetweenTrainsMinimumMinutes", "5");
|
||||
firstUrl = firstUrl.SetQueryParam("ChangeStationName", "");
|
||||
|
||||
var firstResponse = await httpClient.GetStringAsync(firstUrl);
|
||||
var firstDocument = await asContext.OpenAsync(req => req.Content(firstResponse));
|
||||
var firstForm = firstDocument.GetElementById("form-search")!;
|
||||
|
||||
var firstResult = firstForm
|
||||
.QuerySelectorAll<IHtmlInputElement>("input")
|
||||
.Where(elem => elem.Name != null)
|
||||
.ToDictionary(elem => elem.Name!, elem => elem.Value);
|
||||
|
||||
var secondUrl = "".AppendPathSegments("Itineraries", "GetItineraries");
|
||||
var secondResponse = await httpClient.PostAsync(
|
||||
secondUrl,
|
||||
#pragma warning disable CS8620
|
||||
new FormUrlEncodedContent(firstResult)
|
||||
#pragma warning restore CS8620
|
||||
);
|
||||
var secondResponseContent = await secondResponse.Content.ReadAsStringAsync();
|
||||
var secondDocument = await asContext.OpenAsync(
|
||||
req => req.Content(secondResponseContent)
|
||||
);
|
||||
|
||||
var (itineraryInfoDiv, _) = secondDocument
|
||||
.QuerySelectorAll("body > div");
|
||||
|
||||
if (itineraryInfoDiv == null) {
|
||||
return null;
|
||||
}
|
||||
|
||||
var itinerariesLi = secondDocument
|
||||
.QuerySelectorAll("body > ul > li");
|
||||
var itineraries = new List<IItinerary>();
|
||||
foreach (var itineraryLi in itinerariesLi) {
|
||||
var itinerary = new Itinerary();
|
||||
|
||||
var cardDivs = itineraryLi.QuerySelectorAll(":scope > div > div > div > div");
|
||||
var detailsDivs = cardDivs.Last()
|
||||
.QuerySelectorAll(":scope > div > div")[1]
|
||||
.QuerySelectorAll(":scope > div");
|
||||
var trainItineraryAndDetailsLis = detailsDivs[0]
|
||||
.QuerySelectorAll(":scope > ul > li");
|
||||
var stations = new List<string>();
|
||||
var details = new List<ItineraryTrain>();
|
||||
foreach (var (idx, li) in trainItineraryAndDetailsLis.Select((li, idx) => (idx, li))) {
|
||||
if (idx % 2 == 0) {
|
||||
// Station
|
||||
stations.Add(
|
||||
li
|
||||
.QuerySelectorAll(":scope > div > div > div > div")[1]
|
||||
.Text()
|
||||
.WithCollapsedSpaces()
|
||||
);
|
||||
}
|
||||
else {
|
||||
var now = LocalDateTime.FromDateTime(DateTime.Now);
|
||||
// Detail
|
||||
var detailColumns = li.QuerySelectorAll(":scope > div > div");
|
||||
var leftSideDivs = detailColumns[0].QuerySelectorAll(":scope > div");
|
||||
|
||||
var departureDateText = leftSideDivs[0]
|
||||
.QuerySelectorAll(":scope > div")[1]
|
||||
.Text()
|
||||
.WithCollapsedSpaces();
|
||||
var departureDateMatch = DepArrRegex.Match(departureDateText);
|
||||
var departureDate = new LocalDateTime(
|
||||
now.Year,
|
||||
Months[departureDateMatch.Groups[3].Value],
|
||||
int.Parse(departureDateMatch.Groups[2].Value),
|
||||
int.Parse(departureDateMatch.Groups[4].Value),
|
||||
int.Parse(departureDateMatch.Groups[5].Value),
|
||||
0
|
||||
);
|
||||
if (departureDate < now.PlusDays(-1)) {
|
||||
departureDate = departureDate.PlusYears(1);
|
||||
}
|
||||
|
||||
var arrivalDateText = leftSideDivs[3]
|
||||
.QuerySelectorAll(":scope > div")[1]
|
||||
.Text()
|
||||
.WithCollapsedSpaces();
|
||||
var arrivalDateMatch = DepArrRegex.Match(arrivalDateText);
|
||||
var arrivalDate = new LocalDateTime(
|
||||
now.Year,
|
||||
Months[arrivalDateMatch.Groups[3].Value],
|
||||
int.Parse(arrivalDateMatch.Groups[2].Value),
|
||||
int.Parse(arrivalDateMatch.Groups[4].Value),
|
||||
int.Parse(arrivalDateMatch.Groups[5].Value),
|
||||
0
|
||||
);
|
||||
if (arrivalDate < now.PlusDays(-1)) {
|
||||
arrivalDate = arrivalDate.PlusYears(1);
|
||||
}
|
||||
|
||||
var rightSideDivs = detailColumns[1].QuerySelectorAll(":scope > div > div");
|
||||
var kmRankNumberText = rightSideDivs[0]
|
||||
.QuerySelectorAll(":scope > div > div")[0]
|
||||
.Text()
|
||||
.WithCollapsedSpaces();
|
||||
var kmRankNumberMatch = KmTrainRankNoRegex.Match(kmRankNumberText);
|
||||
|
||||
var operatorText = rightSideDivs[0]
|
||||
.QuerySelectorAll(":scope > div > div")[1]
|
||||
.Text()
|
||||
.WithCollapsedSpaces();
|
||||
var operatorMatch = OperatorRegex.Match(operatorText);
|
||||
|
||||
var train = new ItineraryTrain {
|
||||
ArrivalDate = BucharestTz.AtLeniently(arrivalDate).ToDateTimeOffset(),
|
||||
DepartureDate = BucharestTz.AtLeniently(departureDate).ToDateTimeOffset(),
|
||||
Km = int.Parse(kmRankNumberMatch.Groups[1].Value),
|
||||
TrainRank = kmRankNumberMatch.Groups[2].Value,
|
||||
TrainNumber = kmRankNumberMatch.Groups[3].Value,
|
||||
Operator = operatorMatch.Groups[1].Value,
|
||||
};
|
||||
|
||||
foreach (var div in leftSideDivs[2]
|
||||
.QuerySelectorAll(":scope > div")
|
||||
.Where((_, i) => i % 2 != 0)) {
|
||||
var text = div.Text().WithCollapsedSpaces();
|
||||
if (text == "Nu sunt stații intermediare.") continue;
|
||||
train.AddIntermediateStop(div.Text().WithCollapsedSpaces());
|
||||
}
|
||||
|
||||
details.Add(train);
|
||||
}
|
||||
}
|
||||
foreach (var ((iFrom, iTo), detail) in stations.Zip(stations.Skip(1)).Zip(details)) {
|
||||
detail.From = iFrom;
|
||||
detail.To = iTo;
|
||||
itinerary.AddTrain(detail);
|
||||
}
|
||||
|
||||
itineraries.Add(itinerary);
|
||||
}
|
||||
|
||||
return itineraries;
|
||||
}
|
||||
}
|
|
@ -14,7 +14,7 @@ using NodaTime;
|
|||
using NodaTime.Extensions;
|
||||
|
||||
namespace InfoferScraper.Scrapers {
|
||||
public static class StationScraper {
|
||||
public class StationScraper {
|
||||
private static readonly Regex StationInfoRegex = new($@"^([{Utils.RoLetters}.0-9 ]+)\sîn\s([0-9.]+)$");
|
||||
|
||||
private static readonly Regex StoppingTimeRegex = new(
|
||||
|
@ -33,17 +33,28 @@ namespace InfoferScraper.Scrapers {
|
|||
|
||||
private const string BaseUrl = "https://mersultrenurilor.infofer.ro/ro-RO/";
|
||||
|
||||
private static readonly CookieContainer CookieContainer = new();
|
||||
private readonly CookieContainer cookieContainer = new();
|
||||
|
||||
private static readonly HttpClient HttpClient = new(new HttpClientHandler {
|
||||
CookieContainer = CookieContainer,
|
||||
UseCookies = true,
|
||||
}) {
|
||||
BaseAddress = new Uri(BaseUrl),
|
||||
DefaultRequestVersion = new Version(2, 0),
|
||||
};
|
||||
private readonly HttpClient httpClient;
|
||||
|
||||
public static async Task<IStationScrapeResult> Scrape(string stationName, DateTimeOffset? date = null) {
|
||||
public StationScraper(HttpClientHandler? httpClientHandler = null) {
|
||||
if (httpClientHandler == null) {
|
||||
httpClientHandler = new HttpClientHandler {
|
||||
CookieContainer = cookieContainer,
|
||||
UseCookies = true,
|
||||
};
|
||||
}
|
||||
else {
|
||||
httpClientHandler.CookieContainer = cookieContainer;
|
||||
httpClientHandler.UseCookies = true;
|
||||
}
|
||||
httpClient = new HttpClient(httpClientHandler) {
|
||||
BaseAddress = new Uri(BaseUrl),
|
||||
DefaultRequestVersion = new Version(2, 0),
|
||||
};
|
||||
}
|
||||
|
||||
public async Task<IStationScrapeResult> Scrape(string stationName, DateTimeOffset? date = null) {
|
||||
var dateInstant = date?.ToInstant().InZone(BucharestTz);
|
||||
date = dateInstant?.ToDateTimeOffset();
|
||||
|
||||
|
@ -59,7 +70,7 @@ namespace InfoferScraper.Scrapers {
|
|||
if (date != null) {
|
||||
firstUrl = firstUrl.SetQueryParam("Date", $"{date:d.MM.yyyy}");
|
||||
}
|
||||
var firstResponse = await HttpClient.GetStringAsync(firstUrl);
|
||||
var firstResponse = await httpClient.GetStringAsync(firstUrl);
|
||||
var firstDocument = await asContext.OpenAsync(req => req.Content(firstResponse));
|
||||
var firstForm = firstDocument.GetElementById("form-search")!;
|
||||
|
||||
|
@ -69,7 +80,7 @@ namespace InfoferScraper.Scrapers {
|
|||
.ToDictionary(elem => elem.Name!, elem => elem.Value);
|
||||
|
||||
var secondUrl = "".AppendPathSegments("Stations", "StationsResult");
|
||||
var secondResponse = await HttpClient.PostAsync(
|
||||
var secondResponse = await httpClient.PostAsync(
|
||||
secondUrl,
|
||||
#pragma warning disable CS8620
|
||||
new FormUrlEncodedContent(firstResult)
|
||||
|
@ -167,9 +178,9 @@ namespace InfoferScraper.Scrapers {
|
|||
.Text()
|
||||
.WithCollapsedSpaces();
|
||||
foreach (var station in routeDiv.QuerySelectorAll(":scope > div > div")[1]
|
||||
.Text()
|
||||
.WithCollapsedSpaces()
|
||||
.Split(" - ")) {
|
||||
.Text()
|
||||
.WithCollapsedSpaces()
|
||||
.Split(" - ")) {
|
||||
arrDep.ModifyableTrain.AddRouteStation(station);
|
||||
}
|
||||
|
||||
|
|
|
@ -15,7 +15,7 @@ using NodaTime.Extensions;
|
|||
using scraper.Exceptions;
|
||||
|
||||
namespace InfoferScraper.Scrapers {
|
||||
public static class TrainScraper {
|
||||
public class TrainScraper {
|
||||
private const string BaseUrl = "https://mersultrenurilor.infofer.ro/ro-RO/";
|
||||
private static readonly Regex TrainInfoRegex = new(@"^([A-Z-]+)\s([0-9]+)\sîn\s([0-9.]+)$");
|
||||
private static readonly Regex OperatorRegex = new(@"^Operat\sde\s(.+)$");
|
||||
|
@ -51,16 +51,28 @@ namespace InfoferScraper.Scrapers {
|
|||
|
||||
private static readonly DateTimeZone BucharestTz = DateTimeZoneProviders.Tzdb["Europe/Bucharest"];
|
||||
|
||||
private static readonly CookieContainer CookieContainer = new();
|
||||
private static readonly HttpClient HttpClient = new(new HttpClientHandler {
|
||||
CookieContainer = CookieContainer,
|
||||
UseCookies = true,
|
||||
}) {
|
||||
BaseAddress = new Uri(BaseUrl),
|
||||
DefaultRequestVersion = new Version(2, 0),
|
||||
};
|
||||
private readonly CookieContainer cookieContainer = new();
|
||||
private readonly HttpClient httpClient;
|
||||
|
||||
public static async Task<ITrainScrapeResult?> Scrape(string trainNumber, DateTimeOffset? dateOverride = null) {
|
||||
public TrainScraper(HttpClientHandler? httpClientHandler = null)
|
||||
{
|
||||
if (httpClientHandler == null) {
|
||||
httpClientHandler = new HttpClientHandler {
|
||||
CookieContainer = cookieContainer,
|
||||
UseCookies = true,
|
||||
};
|
||||
}
|
||||
else {
|
||||
httpClientHandler.CookieContainer = cookieContainer;
|
||||
httpClientHandler.UseCookies = true;
|
||||
}
|
||||
httpClient = new HttpClient(httpClientHandler) {
|
||||
BaseAddress = new Uri(BaseUrl),
|
||||
DefaultRequestVersion = new Version(2, 0),
|
||||
};
|
||||
}
|
||||
|
||||
public async Task<ITrainScrapeResult?> Scrape(string trainNumber, DateTimeOffset? dateOverride = null) {
|
||||
var dateOverrideInstant = dateOverride?.ToInstant().InZone(BucharestTz);
|
||||
dateOverride = dateOverrideInstant?.ToDateTimeOffset();
|
||||
TrainScrapeResult result = new();
|
||||
|
@ -73,7 +85,7 @@ namespace InfoferScraper.Scrapers {
|
|||
if (dateOverride != null) {
|
||||
firstUrl = firstUrl.SetQueryParam("Date", $"{dateOverride:d.MM.yyyy}");
|
||||
}
|
||||
var firstResponse = await HttpClient.GetStringAsync(firstUrl);
|
||||
var firstResponse = await httpClient.GetStringAsync(firstUrl);
|
||||
var firstDocument = await asContext.OpenAsync(req => req.Content(firstResponse));
|
||||
var firstForm = firstDocument.GetElementById("form-search")!;
|
||||
|
||||
|
@ -83,7 +95,7 @@ namespace InfoferScraper.Scrapers {
|
|||
.ToDictionary(elem => elem.Name!, elem => elem.Value);
|
||||
|
||||
var secondUrl = "".AppendPathSegments("Trains", "TrainsResult");
|
||||
var secondResponse = await HttpClient.PostAsync(
|
||||
var secondResponse = await httpClient.PostAsync(
|
||||
secondUrl,
|
||||
#pragma warning disable CS8620
|
||||
new FormUrlEncodedContent(firstResult)
|
||||
|
@ -153,6 +165,11 @@ namespace InfoferScraper.Scrapers {
|
|||
.QuerySelectorAll(":scope > div")[0]
|
||||
.Text()
|
||||
.WithCollapsedSpaces();
|
||||
stopDescription.LinkName = new Flurl.Url(stopDetails
|
||||
.QuerySelectorAll(":scope > div")[0]
|
||||
.QuerySelector(":scope a")
|
||||
.Attributes["href"]
|
||||
.Value).PathSegments.Last();
|
||||
var scrapedKm = stopDetails
|
||||
.QuerySelectorAll(":scope > div")[1]
|
||||
.Text()
|
||||
|
@ -196,7 +213,12 @@ namespace InfoferScraper.Scrapers {
|
|||
statusElement.Text().WithCollapsedSpaces(replaceWith: " ")
|
||||
).Groups as IEnumerable<Group>).Skip(1).Select(group => group.Value);
|
||||
arrDep.MakeStatus(status => {
|
||||
status.Delay = string.IsNullOrEmpty(onTime) ? int.Parse(delay) : 0;
|
||||
if (string.IsNullOrEmpty(onTime) && delay == null) {
|
||||
status.Cancelled = true;
|
||||
}
|
||||
else {
|
||||
status.Delay = string.IsNullOrEmpty(onTime) ? int.Parse(delay) : 0;
|
||||
}
|
||||
status.Real = string.IsNullOrEmpty(approx);
|
||||
});
|
||||
}
|
||||
|
|
|
@ -1,5 +1,6 @@
|
|||
using System.Collections.Generic;
|
||||
using Microsoft.AspNetCore.Mvc;
|
||||
using Server.Models.Database;
|
||||
using Server.Services.Interfaces;
|
||||
|
||||
namespace Server.Controllers.V2;
|
||||
|
@ -15,7 +16,7 @@ public class StationsController : Controller {
|
|||
}
|
||||
|
||||
[HttpGet("")]
|
||||
public ActionResult<IEnumerable<IStationRecord>> ListStations() {
|
||||
public ActionResult<IEnumerable<StationListing>> ListStations() {
|
||||
return Ok(Database.Stations);
|
||||
}
|
||||
}
|
||||
|
|
|
@ -1,5 +1,6 @@
|
|||
using System.Collections.Generic;
|
||||
using Microsoft.AspNetCore.Mvc;
|
||||
using Server.Models.Database;
|
||||
using Server.Services.Interfaces;
|
||||
|
||||
namespace Server.Controllers.V2;
|
||||
|
@ -15,7 +16,7 @@ public class TrainsController : Controller {
|
|||
}
|
||||
|
||||
[HttpGet("")]
|
||||
public ActionResult<IEnumerable<ITrainRecord>> ListTrains() {
|
||||
public ActionResult<IEnumerable<TrainListing>> ListTrains() {
|
||||
return Ok(Database.Trains);
|
||||
}
|
||||
}
|
||||
|
|
40
server/Controllers/V3/ItinerariesController.cs
Normal file
40
server/Controllers/V3/ItinerariesController.cs
Normal file
|
@ -0,0 +1,40 @@
|
|||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Threading.Tasks;
|
||||
using Microsoft.AspNetCore.Http;
|
||||
using Microsoft.AspNetCore.Mvc;
|
||||
using scraper.Models.Itinerary;
|
||||
using Server.Services.Interfaces;
|
||||
|
||||
namespace Server.Controllers.V3;
|
||||
|
||||
[ApiController]
|
||||
[ApiExplorerSettings(GroupName = "v3")]
|
||||
[Route("/v3/[controller]")]
|
||||
public class ItinerariesController : Controller {
|
||||
private IDataManager DataManager { get; }
|
||||
private IDatabase Database { get; }
|
||||
|
||||
public ItinerariesController(IDataManager dataManager, IDatabase database) {
|
||||
this.DataManager = dataManager;
|
||||
this.Database = database;
|
||||
}
|
||||
|
||||
|
||||
[HttpGet("")]
|
||||
[ProducesResponseType(typeof(IEnumerable<IItinerary>), StatusCodes.Status200OK)]
|
||||
[ProducesResponseType(StatusCodes.Status404NotFound)]
|
||||
public async Task<ActionResult<IEnumerable<IItinerary>>> FindItineraries(
|
||||
[FromQuery] string from,
|
||||
[FromQuery] string to,
|
||||
[FromQuery] DateTimeOffset? date
|
||||
) {
|
||||
var itineraries = await DataManager.FetchItineraries(from, to, date);
|
||||
|
||||
if (itineraries == null) {
|
||||
return NotFound();
|
||||
}
|
||||
|
||||
return Ok(itineraries);
|
||||
}
|
||||
}
|
|
@ -4,6 +4,7 @@ using System.Threading.Tasks;
|
|||
using InfoferScraper.Models.Station;
|
||||
using Microsoft.AspNetCore.Http;
|
||||
using Microsoft.AspNetCore.Mvc;
|
||||
using Server.Models.Database;
|
||||
using Server.Services.Interfaces;
|
||||
|
||||
namespace Server.Controllers.V3;
|
||||
|
@ -21,7 +22,7 @@ public class StationsController : Controller {
|
|||
}
|
||||
|
||||
[HttpGet("")]
|
||||
public ActionResult<IEnumerable<IStationRecord>> ListStations() {
|
||||
public ActionResult<IEnumerable<StationListing>> ListStations() {
|
||||
return Ok(Database.Stations);
|
||||
}
|
||||
|
||||
|
|
|
@ -5,6 +5,7 @@ using InfoferScraper.Models.Train;
|
|||
using Microsoft.AspNetCore.Http;
|
||||
using Microsoft.AspNetCore.Mvc;
|
||||
using scraper.Exceptions;
|
||||
using Server.Models.Database;
|
||||
using Server.Services.Interfaces;
|
||||
|
||||
namespace Server.Controllers.V3;
|
||||
|
@ -22,7 +23,7 @@ public class TrainsController : Controller {
|
|||
}
|
||||
|
||||
[HttpGet("")]
|
||||
public ActionResult<IEnumerable<ITrainRecord>> ListTrains() {
|
||||
public ActionResult<IEnumerable<TrainListing>> ListTrains() {
|
||||
return Ok(Database.Trains);
|
||||
}
|
||||
|
||||
|
|
5
server/Models/Database/MongoSettings.cs
Normal file
5
server/Models/Database/MongoSettings.cs
Normal file
|
@ -0,0 +1,5 @@
|
|||
namespace Server.Models.Database;
|
||||
|
||||
public record MongoSettings(string ConnectionString, string DatabaseName) {
|
||||
public MongoSettings() : this("", "") { }
|
||||
}
|
17
server/Models/Database/StationAlias.cs
Normal file
17
server/Models/Database/StationAlias.cs
Normal file
|
@ -0,0 +1,17 @@
|
|||
using MongoDB.Bson;
|
||||
using MongoDB.Bson.Serialization.Attributes;
|
||||
using Newtonsoft.Json;
|
||||
|
||||
namespace Server.Models.Database;
|
||||
|
||||
public record StationAlias(
|
||||
[property: BsonId]
|
||||
[property: BsonRepresentation(BsonType.ObjectId)]
|
||||
[property: JsonProperty(NullValueHandling = NullValueHandling.Ignore)]
|
||||
string? Id,
|
||||
string Name,
|
||||
[property: BsonRepresentation(BsonType.ObjectId)]
|
||||
string? ListingId
|
||||
) {
|
||||
public StationAlias() : this(null, "", null) { }
|
||||
}
|
18
server/Models/Database/StationListing.cs
Normal file
18
server/Models/Database/StationListing.cs
Normal file
|
@ -0,0 +1,18 @@
|
|||
using System.Collections.Generic;
|
||||
using MongoDB.Bson;
|
||||
using MongoDB.Bson.Serialization.Attributes;
|
||||
using Newtonsoft.Json;
|
||||
|
||||
namespace Server.Models.Database;
|
||||
|
||||
public record StationListing(
|
||||
[property: BsonId]
|
||||
[property: BsonRepresentation(BsonType.ObjectId)]
|
||||
[property: JsonProperty(NullValueHandling = NullValueHandling.Ignore)]
|
||||
string? Id,
|
||||
string Name,
|
||||
List<string> StoppedAtBy
|
||||
) {
|
||||
public StationListing() : this(null, "", new()) { }
|
||||
public StationListing(string name, List<string> stoppedAtBy) : this(null, name, stoppedAtBy) { }
|
||||
}
|
20
server/Models/Database/TrainListing.cs
Normal file
20
server/Models/Database/TrainListing.cs
Normal file
|
@ -0,0 +1,20 @@
|
|||
using MongoDB.Bson;
|
||||
using MongoDB.Bson.Serialization.Attributes;
|
||||
using Newtonsoft.Json;
|
||||
|
||||
namespace Server.Models.Database;
|
||||
|
||||
public record TrainListing(
|
||||
[property: BsonId]
|
||||
[property: BsonRepresentation(BsonType.ObjectId)]
|
||||
[property: JsonProperty(NullValueHandling = NullValueHandling.Ignore)]
|
||||
string? Id,
|
||||
string Rank,
|
||||
string Number,
|
||||
string Company,
|
||||
[property: BsonRepresentation(BsonType.ObjectId)]
|
||||
string? LatestDescription
|
||||
) {
|
||||
public TrainListing() : this(null, "", "", "", null) { }
|
||||
public TrainListing(string rank, string number, string company) : this(null, rank, number, company, null) { }
|
||||
}
|
9
server/Models/ProxySettings.cs
Normal file
9
server/Models/ProxySettings.cs
Normal file
|
@ -0,0 +1,9 @@
|
|||
namespace Server.Models;
|
||||
|
||||
public record ProxySettings(string Url, ProxyCredentials? Credentials = null) {
|
||||
public ProxySettings() : this("") { }
|
||||
}
|
||||
|
||||
public record ProxyCredentials(string Username, string Password) {
|
||||
public ProxyCredentials() : this("", "") { }
|
||||
}
|
|
@ -1,47 +1,94 @@
|
|||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Diagnostics;
|
||||
using System.Net;
|
||||
using System.Net.Http;
|
||||
using System.Threading.Tasks;
|
||||
using InfoferScraper.Models.Train;
|
||||
using InfoferScraper;
|
||||
using InfoferScraper.Models.Station;
|
||||
using InfoferScraper.Models.Train;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using scraper.Models.Itinerary;
|
||||
using Server.Models;
|
||||
using Server.Services.Interfaces;
|
||||
using Server.Utils;
|
||||
using InfoferScraper;
|
||||
|
||||
namespace Server.Services.Implementations {
|
||||
public class DataManager : IDataManager {
|
||||
private ILogger<DataManager> Logger { get; }
|
||||
private IDatabase Database { get; }
|
||||
|
||||
private NodaTime.IDateTimeZoneProvider TzProvider { get; }
|
||||
private NodaTime.DateTimeZone CfrTimeZone => TzProvider["Europe/Bucharest"];
|
||||
|
||||
public DataManager(NodaTime.IDateTimeZoneProvider tzProvider, IDatabase database) {
|
||||
public DataManager(NodaTime.IDateTimeZoneProvider tzProvider, IDatabase database, ILogger<DataManager> logger, ProxySettings? proxySettings) {
|
||||
this.TzProvider = tzProvider;
|
||||
this.Database = database;
|
||||
this.Logger = logger;
|
||||
|
||||
HttpClientHandler httpClientHandler = new (){
|
||||
UseProxy = proxySettings != null,
|
||||
Proxy = proxySettings == null ? null : new WebProxy(proxySettings.Url),
|
||||
DefaultProxyCredentials = proxySettings?.Credentials == null ? null : new NetworkCredential(proxySettings.Credentials.Username, proxySettings.Credentials.Password),
|
||||
};
|
||||
InfoferScraper.Scrapers.StationScraper stationScraper = new(httpClientHandler);
|
||||
InfoferScraper.Scrapers.TrainScraper trainScraper = new(httpClientHandler);
|
||||
InfoferScraper.Scrapers.RouteScraper routeScraper = new(httpClientHandler);
|
||||
|
||||
stationCache = new(async (t) => {
|
||||
var (stationName, date) = t;
|
||||
Logger.LogDebug("Fetching station {StationName} for date {Date}", stationName, date);
|
||||
var zonedDate = new NodaTime.LocalDate(date.Year, date.Month, date.Day).AtStartOfDayInZone(CfrTimeZone);
|
||||
|
||||
var station = await InfoferScraper.Scrapers.StationScraper.Scrape(stationName, zonedDate.ToDateTimeOffset());
|
||||
var station = await stationScraper.Scrape(stationName, zonedDate.ToDateTimeOffset());
|
||||
if (station != null) {
|
||||
await Database.OnStationData(station);
|
||||
_ = Task.Run(async () => {
|
||||
var watch = Stopwatch.StartNew();
|
||||
await Database.OnStationData(station);
|
||||
var ms = watch.ElapsedMilliseconds;
|
||||
Logger.LogInformation("OnStationData timing: {StationDataMs} ms", ms);
|
||||
});
|
||||
}
|
||||
return station;
|
||||
}, TimeSpan.FromMinutes(1));
|
||||
trainCache = new(async (t) => {
|
||||
var (trainNumber, date) = t;
|
||||
Logger.LogDebug("Fetching train {TrainNumber} for date {Date}", trainNumber, date);
|
||||
var zonedDate = new NodaTime.LocalDate(date.Year, date.Month, date.Day).AtStartOfDayInZone(CfrTimeZone);
|
||||
|
||||
var train = await InfoferScraper.Scrapers.TrainScraper.Scrape(trainNumber, zonedDate.ToDateTimeOffset());
|
||||
var train = await trainScraper.Scrape(trainNumber, zonedDate.ToDateTimeOffset());
|
||||
if (train != null) {
|
||||
await Database.OnTrainData(train);
|
||||
_ = Task.Run(async () => {
|
||||
var watch = Stopwatch.StartNew();
|
||||
await Database.OnTrainData(train);
|
||||
var ms = watch.ElapsedMilliseconds;
|
||||
Logger.LogInformation("OnTrainData timing: {StationDataMs} ms", ms);
|
||||
});
|
||||
}
|
||||
return train;
|
||||
}, TimeSpan.FromSeconds(30));
|
||||
itinerariesCache = new(async (t) => {
|
||||
var (from, to, date) = t;
|
||||
Logger.LogDebug("Fetching itinerary from {From} to {To} for date {Date}", from, to, date);
|
||||
var zonedDate = new NodaTime.LocalDate(date.Year, date.Month, date.Day).AtStartOfDayInZone(CfrTimeZone);
|
||||
|
||||
var itineraries = await routeScraper.Scrape(from, to, zonedDate.ToDateTimeOffset());
|
||||
if (itineraries != null) {
|
||||
_ = Task.Run(async () => {
|
||||
var watch = Stopwatch.StartNew();
|
||||
await Database.OnItineraries(itineraries);
|
||||
var ms = watch.ElapsedMilliseconds;
|
||||
Logger.LogInformation("OnItineraries timing: {StationDataMs} ms", ms);
|
||||
});
|
||||
}
|
||||
|
||||
return itineraries;
|
||||
}, TimeSpan.FromMinutes(1));
|
||||
}
|
||||
|
||||
private readonly AsyncCache<(string, DateOnly), IStationScrapeResult?> stationCache;
|
||||
private readonly AsyncCache<(string, DateOnly), ITrainScrapeResult?> trainCache;
|
||||
private readonly AsyncCache<(string, string, DateOnly), IReadOnlyList<IItinerary>?> itinerariesCache;
|
||||
|
||||
public Task<IStationScrapeResult?> FetchStation(string stationName, DateTimeOffset date) {
|
||||
var cfrDateTime = new NodaTime.ZonedDateTime(NodaTime.Instant.FromDateTimeOffset(date), CfrTimeZone);
|
||||
|
@ -56,5 +103,12 @@ namespace Server.Services.Implementations {
|
|||
|
||||
return trainCache.GetItem((trainNumber, cfrDate));
|
||||
}
|
||||
|
||||
public async Task<IReadOnlyList<IItinerary>?> FetchItineraries(string from, string to, DateTimeOffset? date = null) {
|
||||
var cfrDateTime = new NodaTime.ZonedDateTime(NodaTime.Instant.FromDateTimeOffset(date ?? DateTimeOffset.Now), CfrTimeZone);
|
||||
var cfrDate = new DateOnly(cfrDateTime.Year, cfrDateTime.Month, cfrDateTime.Day);
|
||||
|
||||
return await itinerariesCache.GetItem((from, to, cfrDate));
|
||||
}
|
||||
}
|
||||
}
|
||||
|
|
|
@ -2,187 +2,347 @@ using System;
|
|||
using System.Collections.Generic;
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using System.Text.Json;
|
||||
using System.Text.Json.Nodes;
|
||||
using System.Text.Json.Serialization;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
using InfoferScraper.Models.Station;
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Microsoft.Extensions.Options;
|
||||
using MongoDB.Bson;
|
||||
using MongoDB.Bson.Serialization.Attributes;
|
||||
using MongoDB.Driver;
|
||||
using Newtonsoft.Json;
|
||||
using Newtonsoft.Json.Linq;
|
||||
using Newtonsoft.Json.Serialization;
|
||||
using scraper.Models.Itinerary;
|
||||
using Server.Models.Database;
|
||||
using Server.Utils;
|
||||
|
||||
namespace Server.Services.Implementations;
|
||||
|
||||
public class Database : Server.Services.Interfaces.IDatabase {
|
||||
private static readonly JsonSerializerOptions serializerOptions = new() {
|
||||
PropertyNamingPolicy = JsonNamingPolicy.CamelCase,
|
||||
private static readonly JsonSerializerSettings jsonSerializerSettings = new() {
|
||||
ContractResolver = new DefaultContractResolver {
|
||||
NamingStrategy = new CamelCaseNamingStrategy(),
|
||||
},
|
||||
};
|
||||
|
||||
private ILogger<Database> Logger { get; }
|
||||
|
||||
private bool shouldCommitOnEveryChange = true;
|
||||
private bool dbDataDirty = false;
|
||||
private bool stationsDirty = false;
|
||||
private bool trainsDirty = false;
|
||||
public DbRecord DbData { get; private set; } = new(3);
|
||||
|
||||
public DbRecord DbData { get; private set; } = new(2);
|
||||
private List<StationRecord> stations = new();
|
||||
private List<TrainRecord> trains = new();
|
||||
|
||||
public IReadOnlyList<Server.Services.Interfaces.IStationRecord> Stations => stations;
|
||||
public IReadOnlyList<Server.Services.Interfaces.ITrainRecord> Trains => trains;
|
||||
public IReadOnlyList<StationListing> Stations => stationListingsCollection
|
||||
.Aggregate(PipelineDefinition<StationListing, StationListing>.Create(
|
||||
"{ $addFields: { stoppedAtCount: { $size: \"$stoppedAtBy\" } } }",
|
||||
"{ $sort: { stoppedAtCount: -1 } }",
|
||||
"{ $unset: \"stoppedAtCount\" }"
|
||||
))
|
||||
.ToList();
|
||||
public IReadOnlyList<TrainListing> Trains => trainListingsCollection.FindSync(_ => true).ToList();
|
||||
public IReadOnlyList<StationAlias> StationAliases => stationAliasCollection.FindSync(_ => true).ToList();
|
||||
|
||||
private static readonly string DbDir = Environment.GetEnvironmentVariable("DB_DIR") ?? Path.Join(Environment.CurrentDirectory, "db");
|
||||
private static readonly string DbFile = Path.Join(DbDir, "db.json");
|
||||
private static readonly string StationsFile = Path.Join(DbDir, "stations.json");
|
||||
private static readonly string TrainsFile = Path.Join(DbDir, "trains.json");
|
||||
|
||||
public IDisposable MakeDbTransaction() {
|
||||
shouldCommitOnEveryChange = false;
|
||||
return new Server.Utils.ActionDisposable(() => {
|
||||
if (dbDataDirty) File.WriteAllText(DbFile, JsonSerializer.Serialize(DbData, serializerOptions));
|
||||
if (stationsDirty) {
|
||||
stations.Sort((s1, s2) => s2.StoppedAtBy.Count.CompareTo(s1.StoppedAtBy.Count));
|
||||
File.WriteAllText(StationsFile, JsonSerializer.Serialize(stations, serializerOptions));
|
||||
}
|
||||
if (trainsDirty) File.WriteAllText(TrainsFile, JsonSerializer.Serialize(trains, serializerOptions));
|
||||
dbDataDirty = stationsDirty = trainsDirty = false;
|
||||
shouldCommitOnEveryChange = true;
|
||||
});
|
||||
}
|
||||
private readonly IMongoDatabase db;
|
||||
private readonly IMongoCollection<DbRecord> dbRecordCollection;
|
||||
private readonly IMongoCollection<TrainListing> trainListingsCollection;
|
||||
private readonly IMongoCollection<StationListing> stationListingsCollection;
|
||||
private readonly IMongoCollection<StationAlias> stationAliasCollection;
|
||||
private readonly AsyncThrottle throttle;
|
||||
|
||||
public Database(ILogger<Database> logger) {
|
||||
private readonly Dictionary<string, string> trainObjectIds = new();
|
||||
private readonly Dictionary<string, string> stationObjectIds = new();
|
||||
|
||||
public Database(ILogger<Database> logger, IOptions<MongoSettings> mongoSettings) {
|
||||
Logger = logger;
|
||||
|
||||
if (!Directory.Exists(DbDir)) {
|
||||
Logger.LogDebug("Creating directory: {DbDir}", DbDir);
|
||||
Directory.CreateDirectory(DbDir);
|
||||
}
|
||||
var settings = MongoClientSettings.FromConnectionString(mongoSettings.Value.ConnectionString);
|
||||
settings.ServerApi = new(ServerApiVersion.V1);
|
||||
settings.MaxConnectionPoolSize = 10000;
|
||||
MongoClient mongoClient = new(settings);
|
||||
Logger.LogDebug("Created monogClient");
|
||||
throttle = new(mongoClient.Settings.MaxConnectionPoolSize / 2);
|
||||
db = mongoClient.GetDatabase(mongoSettings.Value.DatabaseName) ?? throw new NullReferenceException("Unable to get Mongo database");
|
||||
Logger.LogDebug("Created db");
|
||||
dbRecordCollection = db.GetCollection<DbRecord>("db");
|
||||
trainListingsCollection = db.GetCollection<TrainListing>("trainListings");
|
||||
stationListingsCollection = db.GetCollection<StationListing>("stationListings");
|
||||
stationAliasCollection = db.GetCollection<StationAlias>("stationAliases");
|
||||
|
||||
Migration();
|
||||
|
||||
if (File.Exists(DbFile)) {
|
||||
DbData = JsonSerializer.Deserialize<DbRecord>(File.ReadAllText(DbFile), serializerOptions)!;
|
||||
}
|
||||
else {
|
||||
File.WriteAllText(DbFile, JsonSerializer.Serialize(DbData, serializerOptions));
|
||||
}
|
||||
|
||||
if (File.Exists(StationsFile)) {
|
||||
stations = JsonSerializer.Deserialize<List<StationRecord>>(File.ReadAllText(StationsFile), serializerOptions)!;
|
||||
}
|
||||
|
||||
if (File.Exists(TrainsFile)) {
|
||||
trains = JsonSerializer.Deserialize<List<TrainRecord>>(File.ReadAllText(TrainsFile), serializerOptions)!;
|
||||
}
|
||||
Task.Run(async () => await Initialize());
|
||||
}
|
||||
|
||||
private void Migration() {
|
||||
if (!File.Exists(DbFile)) {
|
||||
// using var _ = Logger.BeginScope("Migrating DB version 1 -> 2");
|
||||
if (!File.Exists(DbFile) && File.Exists(TrainsFile)) {
|
||||
Logger.LogInformation("Migrating DB version 1 -> 2");
|
||||
if (File.Exists(StationsFile)) {
|
||||
Logger.LogDebug("Converting StationsFile");
|
||||
var oldStations = JsonNode.Parse(File.ReadAllText(StationsFile));
|
||||
var oldStations = JToken.Parse(File.ReadAllText(StationsFile));
|
||||
List<StationListing> stations = new();
|
||||
if (oldStations != null) {
|
||||
Logger.LogDebug("Found {StationsCount} stations", oldStations.AsArray().Count);
|
||||
foreach (var station in oldStations.AsArray()) {
|
||||
Logger.LogDebug("Found {StationsCount} stations", oldStations.Children().Count());
|
||||
foreach (var station in oldStations.Children()) {
|
||||
if (station == null) continue;
|
||||
station["stoppedAtBy"] = new JsonArray(station["stoppedAtBy"]!.AsArray().Select(num => (JsonNode)(num!).ToString()!).ToArray());
|
||||
station["stoppedAtBy"] = new JArray(station["stoppedAtBy"]!.Children().Select(num => (JToken)(num!).ToString()!).ToArray());
|
||||
}
|
||||
stations = JsonSerializer.Deserialize<List<StationRecord>>(oldStations, serializerOptions)!;
|
||||
stations = oldStations.ToObject<List<StationListing>>(JsonSerializer.Create(jsonSerializerSettings))!;
|
||||
}
|
||||
Logger.LogDebug("Rewriting StationsFile");
|
||||
File.WriteAllText(StationsFile, JsonSerializer.Serialize(stations, serializerOptions));
|
||||
File.WriteAllText(StationsFile, JsonConvert.SerializeObject(stations, jsonSerializerSettings));
|
||||
}
|
||||
if (File.Exists(TrainsFile)) {
|
||||
Logger.LogDebug("Converting TrainsFile");
|
||||
var oldTrains = JsonNode.Parse(File.ReadAllText(TrainsFile));
|
||||
var oldTrains = JToken.Parse(File.ReadAllText(TrainsFile));
|
||||
List<TrainListing> trains = new();
|
||||
if (oldTrains != null) {
|
||||
Logger.LogDebug("Found {TrainsCount} trains", oldTrains.AsArray().Count);
|
||||
foreach (var train in oldTrains.AsArray()) {
|
||||
Logger.LogDebug("Found {TrainsCount} trains", oldTrains.Children().Count());
|
||||
foreach (var train in oldTrains.Children()) {
|
||||
if (train == null) continue;
|
||||
train["number"] = train["numberString"];
|
||||
train.AsObject().Remove("numberString");
|
||||
train["numberString"]?.Remove();
|
||||
}
|
||||
trains = JsonSerializer.Deserialize<List<TrainRecord>>(oldTrains, serializerOptions)!;
|
||||
trains = oldTrains.ToObject<List<TrainListing>>(JsonSerializer.Create(jsonSerializerSettings))!;
|
||||
}
|
||||
Logger.LogDebug("Rewriting TrainsFile");
|
||||
File.WriteAllText(TrainsFile, JsonSerializer.Serialize(trains, serializerOptions));
|
||||
File.WriteAllText(TrainsFile, JsonConvert.SerializeObject(trains, jsonSerializerSettings));
|
||||
}
|
||||
DbData = new(2);
|
||||
File.WriteAllText(DbFile, JsonSerializer.Serialize(DbData, serializerOptions));
|
||||
File.WriteAllText(DbFile, JsonConvert.SerializeObject(DbData, jsonSerializerSettings));
|
||||
Migration();
|
||||
}
|
||||
else {
|
||||
var oldDbData = JsonNode.Parse(File.ReadAllText(DbFile));
|
||||
else if (File.Exists(DbFile)) {
|
||||
var oldDbData = JToken.Parse(File.ReadAllText(DbFile));
|
||||
if (((int?)oldDbData?["version"]) == 2) {
|
||||
Logger.LogInformation("DB Version: 2; noop");
|
||||
Logger.LogInformation("Migrating DB version 2 -> 3 (transition from fs+JSON to MongoDB)");
|
||||
|
||||
if (File.Exists(StationsFile)) {
|
||||
Logger.LogDebug("Converting StationsFile");
|
||||
var stations = JsonConvert.DeserializeObject<List<StationListing>>(File.ReadAllText(StationsFile));
|
||||
stationListingsCollection.InsertMany(stations);
|
||||
File.Delete(StationsFile);
|
||||
}
|
||||
|
||||
if (File.Exists(TrainsFile)) {
|
||||
Logger.LogDebug("Converting TrainsFile");
|
||||
var trains = JsonConvert.DeserializeObject<List<TrainListing>>(File.ReadAllText(TrainsFile));
|
||||
trainListingsCollection.InsertMany(trains);
|
||||
File.Delete(TrainsFile);
|
||||
}
|
||||
|
||||
File.Delete(DbFile);
|
||||
try {
|
||||
Directory.Delete(DbDir);
|
||||
}
|
||||
catch (Exception) {
|
||||
// Deleting of the directory is optional; may not be allowed in Docker or similar
|
||||
}
|
||||
|
||||
var x = dbRecordCollection.FindSync(_ => true).ToList()!;
|
||||
if (x.Count != 0) {
|
||||
Logger.LogWarning("db collection contained data when migrating to V3");
|
||||
using (var _ = Logger.BeginScope("Already existing data:")) {
|
||||
foreach (var dbRecord in x) {
|
||||
Logger.LogInformation("Id: {Id}, Version: {Version}", dbRecord.Id, dbRecord.Version);
|
||||
}
|
||||
}
|
||||
Logger.LogInformation("Backing up existing data");
|
||||
var backupDbRecordCollection = db.GetCollection<DbRecord>("db-backup");
|
||||
backupDbRecordCollection.InsertMany(x);
|
||||
Logger.LogDebug("Removing existing data");
|
||||
dbRecordCollection.DeleteMany(_ => true);
|
||||
}
|
||||
dbRecordCollection.InsertOne(new(3));
|
||||
Migration();
|
||||
}
|
||||
else {
|
||||
throw new Exception("Unexpected Database version");
|
||||
throw new("Unexpected Database version, only DB Version 2 uses DbFile");
|
||||
}
|
||||
}
|
||||
else {
|
||||
var datas = dbRecordCollection.FindSync(_ => true).ToList();
|
||||
if (datas.Count == 0) {
|
||||
Logger.LogInformation("No db record found, new database");
|
||||
dbRecordCollection.InsertOne(DbData);
|
||||
}
|
||||
else {
|
||||
DbData = datas[0];
|
||||
}
|
||||
if (DbData.Version == 3) {
|
||||
Logger.LogInformation("Using MongoDB Database Version 3; noop");
|
||||
}
|
||||
else {
|
||||
throw new($"Unexpected Database version: {DbData.Version}");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
private async Task Initialize() {
|
||||
await foreach (var entry in await stationAliasCollection.FindAsync(_ => true)) {
|
||||
if (entry?.ListingId is null) continue;
|
||||
stationObjectIds.Add(entry.Name, entry.ListingId);
|
||||
}
|
||||
}
|
||||
|
||||
private readonly SemaphoreSlim insertTrainLock = new (1, 1);
|
||||
public async Task<string> FoundTrain(string rank, string number, string company) {
|
||||
number = string.Join("", number.TakeWhile(c => '0' <= c && c <= '9'));
|
||||
if (!trains.Where(train => train.Number == number).Any()) {
|
||||
Logger.LogDebug("Found train {Rank} {Number} from {Company}", rank, number, company);
|
||||
trains.Add(new(number, rank, company));
|
||||
if (shouldCommitOnEveryChange) {
|
||||
await File.WriteAllTextAsync(TrainsFile, JsonSerializer.Serialize(trains, serializerOptions));
|
||||
number = string.Join("", number.TakeWhile(c => c is >= '0' and <= '9'));
|
||||
// If there is a matching ObjectId, then it's already in the database
|
||||
if (trainObjectIds.ContainsKey(number)) return number;
|
||||
await insertTrainLock.WaitAsync();
|
||||
try {
|
||||
var possibleTrains = await (await throttle.MakeRequest(() => trainListingsCollection.FindAsync(
|
||||
Builders<TrainListing>.Filter.Eq("number", number)
|
||||
))).ToListAsync();
|
||||
if (possibleTrains.Count == 0) {
|
||||
Logger.LogDebug("Found train {Rank} {Number} from {Company}", rank, number, company);
|
||||
TrainListing listing = new(number: number, rank: rank, company: company);
|
||||
await throttle.MakeRequest(() => trainListingsCollection.InsertOneAsync(listing));
|
||||
if (listing.Id != null) {
|
||||
trainObjectIds[number] = listing.Id;
|
||||
}
|
||||
}
|
||||
else {
|
||||
trainsDirty = true;
|
||||
foreach (var possibleTrain in possibleTrains) {
|
||||
trainObjectIds[possibleTrain.Number] = possibleTrain.Id!;
|
||||
}
|
||||
}
|
||||
}
|
||||
finally {
|
||||
insertTrainLock.Release();
|
||||
}
|
||||
|
||||
return number;
|
||||
}
|
||||
|
||||
private readonly SemaphoreSlim insertStationLock = new (1, 1);
|
||||
|
||||
public async Task FoundStation(string name) {
|
||||
if (!stations.Where(station => station.Name == name).Any()) {
|
||||
// if (!await throttle.MakeRequest(() => stationListingsCollection.Find(Builders<StationListing>.Filter.Eq("name", name)).AnyAsync())) {
|
||||
// Logger.LogDebug("Found station {StationName}", name);
|
||||
// await throttle.MakeRequest(() => stationListingsCollection.InsertOneAsync(new(name, new())));
|
||||
|
||||
// }
|
||||
// If there is a matching ObjectId, then it's already in the database
|
||||
if (stationObjectIds.ContainsKey(name)) return;
|
||||
|
||||
await insertStationLock.WaitAsync();
|
||||
UpdateResult update;
|
||||
try {
|
||||
update = await stationListingsCollection.UpdateOneAsync(
|
||||
Builders<StationListing>.Filter.Eq("name", name),
|
||||
Builders<StationListing>.Update.Combine(
|
||||
Builders<StationListing>.Update.SetOnInsert("name", name),
|
||||
Builders<StationListing>.Update.SetOnInsert("stoppedAtBy", new List<string>())
|
||||
),
|
||||
new UpdateOptions {
|
||||
IsUpsert = true,
|
||||
}
|
||||
);
|
||||
if (update.IsAcknowledged && update.ModifiedCount > 0) {
|
||||
var listingId = update.UpsertedId.AsObjectId.ToString();
|
||||
stationObjectIds[name] = listingId;
|
||||
await stationAliasCollection.UpdateOneAsync(
|
||||
Builders<StationAlias>.Filter.Eq("name", name),
|
||||
Builders<StationAlias>.Update.Combine(
|
||||
Builders<StationAlias>.Update.SetOnInsert("name", name),
|
||||
Builders<StationAlias>.Update.SetOnInsert("listingId", listingId)
|
||||
),
|
||||
new UpdateOptions { IsUpsert = true }
|
||||
);
|
||||
}
|
||||
}
|
||||
finally {
|
||||
insertStationLock.Release();
|
||||
}
|
||||
|
||||
if (update.IsAcknowledged && update.MatchedCount == 0) {
|
||||
Logger.LogDebug("Found station {StationName}", name);
|
||||
stations.Add(new(name, new()));
|
||||
if (shouldCommitOnEveryChange) {
|
||||
await File.WriteAllTextAsync(StationsFile, JsonSerializer.Serialize(stations, serializerOptions));
|
||||
}
|
||||
else {
|
||||
stationsDirty = true;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public async Task FoundTrainAtStation(string stationName, string trainNumber) {
|
||||
trainNumber = string.Join("", trainNumber.TakeWhile(c => '0' <= c && c <= '9'));
|
||||
await FoundStation(stationName);
|
||||
var dirty = false;
|
||||
for (var i = 0; i < stations.Count; i++) {
|
||||
if (stations[i].Name == stationName) {
|
||||
if (!stations[i].StoppedAtBy.Contains(trainNumber)) {
|
||||
Logger.LogDebug("Found train {TrainNumber} at station {StationName}", trainNumber, stationName);
|
||||
stations[i].ActualStoppedAtBy.Add(trainNumber);
|
||||
dirty = true;
|
||||
}
|
||||
break;
|
||||
}
|
||||
public async Task FoundStations(IEnumerable<string> names) {
|
||||
var unknownStations = names.ToList();
|
||||
if (unknownStations.All(s => stationObjectIds.ContainsKey(s))) {
|
||||
return;
|
||||
}
|
||||
if (dirty) {
|
||||
if (shouldCommitOnEveryChange) {
|
||||
stations.Sort((s1, s2) => s2.StoppedAtBy.Count.CompareTo(s1.StoppedAtBy.Count));
|
||||
await File.WriteAllTextAsync(StationsFile, JsonSerializer.Serialize(stations, serializerOptions));
|
||||
}
|
||||
else {
|
||||
stationsDirty = true;
|
||||
}
|
||||
|
||||
unknownStations.RemoveAll(s => stationObjectIds.ContainsKey(s));
|
||||
var existingStations = await (await stationListingsCollection.FindAsync(
|
||||
Builders<StationListing>.Filter.StringIn("name", unknownStations.Select((n) => new StringOrRegularExpression(n)))
|
||||
)).ToListAsync();
|
||||
foreach (var existingStation in existingStations) {
|
||||
stationObjectIds[existingStation.Name] = existingStation.Id!;
|
||||
}
|
||||
|
||||
unknownStations.RemoveAll(s => existingStations.Select(st => st.Name).Contains(s));
|
||||
if (unknownStations.Count == 0) return;
|
||||
var unknownStationListings = unknownStations.Select((s) => new StationListing(s, new())).ToList();
|
||||
await stationListingsCollection.InsertManyAsync(unknownStationListings);
|
||||
foreach (var listing in unknownStationListings) {
|
||||
stationObjectIds[listing.Name] = listing.Id!;
|
||||
}
|
||||
Logger.LogDebug("Found stations {StationNames}", unknownStations);
|
||||
}
|
||||
|
||||
public async Task FoundTrainAtStation(string stationName, string trainNumber) {
|
||||
trainNumber = string.Join("", trainNumber.TakeWhile(c => c is >= '0' and <= '9'));
|
||||
await FoundStation(stationName);
|
||||
UpdateResult updateResult;
|
||||
if (stationObjectIds.ContainsKey(stationName)) {
|
||||
updateResult = await throttle.MakeRequest(() => stationListingsCollection.UpdateOneAsync(
|
||||
Builders<StationListing>.Filter.Eq("_id", ObjectId.Parse(stationObjectIds[stationName])),
|
||||
Builders<StationListing>.Update.AddToSet("stoppedAtBy", trainNumber)
|
||||
));
|
||||
}
|
||||
else {
|
||||
updateResult = await throttle.MakeRequest(() => stationListingsCollection.UpdateOneAsync(
|
||||
Builders<StationListing>.Filter.Eq("name", stationName),
|
||||
Builders<StationListing>.Update.AddToSet("stoppedAtBy", trainNumber)
|
||||
));
|
||||
}
|
||||
if (updateResult.IsAcknowledged && updateResult.ModifiedCount > 0) {
|
||||
Logger.LogDebug("Found train {TrainNumber} at station {StationName}", trainNumber, stationName);
|
||||
}
|
||||
}
|
||||
|
||||
public async Task FoundTrainAtStations(IEnumerable<string> stationNames, string trainNumber) {
|
||||
trainNumber = string.Join("", trainNumber.TakeWhile(c => c is >= '0' and <= '9'));
|
||||
var enumerable = stationNames as string[] ?? stationNames.ToArray();
|
||||
await FoundStations(enumerable);
|
||||
var objectIds = enumerable
|
||||
.Select<string, ObjectId?>((stationName) => stationObjectIds.ContainsKey(stationName) ? ObjectId.Parse(stationObjectIds[stationName]) : null)
|
||||
.ToList();
|
||||
UpdateResult updateResult;
|
||||
if (!objectIds.Any((id) => id is null)) {
|
||||
updateResult = await throttle.MakeRequest(() => stationListingsCollection.UpdateManyAsync(
|
||||
Builders<StationListing>.Filter.In("_id", objectIds),
|
||||
Builders<StationListing>.Update.AddToSet("stoppedAtBy", trainNumber)
|
||||
));
|
||||
}
|
||||
else {
|
||||
updateResult = await throttle.MakeRequest(() => stationListingsCollection.UpdateManyAsync(
|
||||
Builders<StationListing>.Filter.StringIn("name", enumerable.Select(sn => new StringOrRegularExpression(sn))),
|
||||
Builders<StationListing>.Update.AddToSet("stoppedAtBy", trainNumber)
|
||||
));
|
||||
}
|
||||
if (updateResult.IsAcknowledged && updateResult.ModifiedCount > 0) {
|
||||
Logger.LogDebug("Found train {TrainNumber} at stations {StationNames}", trainNumber, stationNames);
|
||||
}
|
||||
}
|
||||
|
||||
public async Task OnTrainData(InfoferScraper.Models.Train.ITrainScrapeResult trainData) {
|
||||
using var _ = MakeDbTransaction();
|
||||
var trainNumber = await FoundTrain(trainData.Rank, trainData.Number, trainData.Operator);
|
||||
foreach (var group in trainData.Groups) {
|
||||
foreach (var station in group.Stations) {
|
||||
await FoundTrainAtStation(station.Name, trainNumber);
|
||||
}
|
||||
}
|
||||
await FoundTrainAtStations(
|
||||
trainData.Groups
|
||||
.SelectMany(g => g.Stations)
|
||||
.Select(trainStop => trainStop.Name)
|
||||
.Distinct(),
|
||||
trainNumber
|
||||
);
|
||||
}
|
||||
|
||||
public async Task OnStationData(InfoferScraper.Models.Station.IStationScrapeResult stationData) {
|
||||
|
@ -191,48 +351,40 @@ public class Database : Server.Services.Interfaces.IDatabase {
|
|||
async Task ProcessTrain(InfoferScraper.Models.Station.IStationArrDep train) {
|
||||
var trainNumber = train.Train.Number;
|
||||
trainNumber = await FoundTrain(train.Train.Rank, trainNumber, train.Train.Operator);
|
||||
await FoundTrainAtStation(stationName, trainNumber);
|
||||
if (train.Train.Route.Count != 0) {
|
||||
foreach (var station in train.Train.Route) {
|
||||
await FoundTrainAtStation(station, trainNumber);
|
||||
}
|
||||
}
|
||||
await FoundTrainAtStations(Enumerable.Repeat(stationName, 1).Concat(train.Train.Route).Distinct(), trainNumber);
|
||||
}
|
||||
|
||||
using var _ = MakeDbTransaction();
|
||||
|
||||
List<IStationArrDep> arrdep = new();
|
||||
if (stationData.Arrivals != null) {
|
||||
foreach (var train in stationData.Arrivals) {
|
||||
await ProcessTrain(train);
|
||||
}
|
||||
arrdep.AddRange(stationData.Arrivals);
|
||||
}
|
||||
if (stationData.Departures != null) {
|
||||
foreach (var train in stationData.Departures) {
|
||||
await ProcessTrain(train);
|
||||
arrdep.AddRange(stationData.Departures);
|
||||
}
|
||||
|
||||
foreach (var train in arrdep.DistinctBy((t) => t.Train.Number)) {
|
||||
await ProcessTrain(train);
|
||||
}
|
||||
}
|
||||
|
||||
public async Task OnItineraries(IReadOnlyList<IItinerary> itineraries) {
|
||||
foreach (var itinerary in itineraries) {
|
||||
foreach (var train in itinerary.Trains) {
|
||||
await FoundTrainAtStations(
|
||||
train.IntermediateStops.Concat(new[] { train.From, train.To }),
|
||||
train.TrainNumber
|
||||
);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public record DbRecord(int Version);
|
||||
|
||||
public record StationRecord : Server.Services.Interfaces.IStationRecord {
|
||||
[JsonPropertyName("stoppedAtBy")]
|
||||
public List<string> ActualStoppedAtBy { get; init; }
|
||||
|
||||
public string Name { get; init; }
|
||||
[JsonIgnore]
|
||||
public IReadOnlyList<string> StoppedAtBy => ActualStoppedAtBy;
|
||||
|
||||
public StationRecord() {
|
||||
Name = "";
|
||||
ActualStoppedAtBy = new();
|
||||
}
|
||||
|
||||
public StationRecord(string name, List<string> stoppedAtBy) {
|
||||
Name = name;
|
||||
ActualStoppedAtBy = stoppedAtBy;
|
||||
}
|
||||
public record DbRecord(
|
||||
[property: BsonId]
|
||||
[property: BsonRepresentation(BsonType.ObjectId)]
|
||||
[property: JsonProperty(NullValueHandling = NullValueHandling.Ignore)]
|
||||
string? Id,
|
||||
int Version
|
||||
) {
|
||||
public DbRecord(int version) : this(null, version) { }
|
||||
}
|
||||
|
||||
public record TrainRecord(string Number, string Rank, string Company) : Server.Services.Interfaces.ITrainRecord;
|
||||
|
|
|
@ -1,11 +1,14 @@
|
|||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Threading.Tasks;
|
||||
using InfoferScraper.Models.Train;
|
||||
using InfoferScraper.Models.Station;
|
||||
using scraper.Models.Itinerary;
|
||||
|
||||
namespace Server.Services.Interfaces;
|
||||
|
||||
public interface IDataManager {
|
||||
public Task<IStationScrapeResult?> FetchStation(string stationName, DateTimeOffset date);
|
||||
public Task<ITrainScrapeResult?> FetchTrain(string trainNumber, DateTimeOffset date);
|
||||
public Task<IReadOnlyList<IItinerary>?> FetchItineraries(string from, string to, DateTimeOffset? date = null);
|
||||
}
|
||||
|
|
|
@ -2,27 +2,19 @@ using System.Collections.Generic;
|
|||
using System.Threading.Tasks;
|
||||
using InfoferScraper.Models.Train;
|
||||
using InfoferScraper.Models.Station;
|
||||
using scraper.Models.Itinerary;
|
||||
using Server.Models.Database;
|
||||
|
||||
namespace Server.Services.Interfaces;
|
||||
|
||||
public interface IDatabase {
|
||||
public IReadOnlyList<IStationRecord> Stations { get; }
|
||||
public IReadOnlyList<ITrainRecord> Trains { get; }
|
||||
public IReadOnlyList<StationListing> Stations { get; }
|
||||
public IReadOnlyList<TrainListing> Trains { get; }
|
||||
|
||||
public Task<string> FoundTrain(string rank, string number, string company);
|
||||
public Task FoundStation(string name);
|
||||
public Task FoundTrainAtStation(string stationName, string trainName);
|
||||
public Task OnTrainData(ITrainScrapeResult trainData);
|
||||
public Task OnStationData(IStationScrapeResult stationData);
|
||||
}
|
||||
|
||||
public interface IStationRecord {
|
||||
public string Name { get; }
|
||||
public IReadOnlyList<string> StoppedAtBy { get; }
|
||||
}
|
||||
|
||||
public interface ITrainRecord {
|
||||
public string Rank { get; }
|
||||
public string Number { get; }
|
||||
public string Company { get; }
|
||||
public Task OnItineraries(IReadOnlyList<IItinerary> itineraries);
|
||||
}
|
||||
|
|
|
@ -1,13 +1,17 @@
|
|||
using System;
|
||||
using System.Net;
|
||||
using System.Text.Json;
|
||||
using Microsoft.AspNetCore.Builder;
|
||||
using Microsoft.AspNetCore.Hosting;
|
||||
using Microsoft.AspNetCore.Http;
|
||||
using Microsoft.AspNetCore.HttpOverrides;
|
||||
using Microsoft.Extensions.Configuration;
|
||||
using Microsoft.Extensions.DependencyInjection;
|
||||
using Microsoft.Extensions.Hosting;
|
||||
using Microsoft.OpenApi.Models;
|
||||
using MongoDB.Bson.Serialization.Conventions;
|
||||
using Newtonsoft.Json.Serialization;
|
||||
using Server.Models;
|
||||
using Server.Models.Database;
|
||||
using Server.Services.Implementations;
|
||||
using Server.Services.Interfaces;
|
||||
|
||||
|
@ -26,12 +30,33 @@ namespace Server {
|
|||
options.KnownProxies.Add(Dns.GetHostAddresses("host.docker.internal")[0]);
|
||||
});
|
||||
}
|
||||
|
||||
services.Configure<ProxySettings>(Configuration.GetSection("Proxy"));
|
||||
services.Configure<MongoSettings>(Configuration.GetSection("TrainDataMongo"));
|
||||
var conventionPack = new ConventionPack { new CamelCaseElementNameConvention() };
|
||||
ConventionRegistry.Register("camelCase", conventionPack, _ => true);
|
||||
services.AddSingleton<IDataManager, DataManager>();
|
||||
services.AddSingleton<IDatabase, Database>();
|
||||
services.AddSingleton<NodaTime.IDateTimeZoneProvider>(NodaTime.DateTimeZoneProviders.Tzdb);
|
||||
services.AddSingleton(NodaTime.DateTimeZoneProviders.Tzdb);
|
||||
|
||||
services.AddSingleton<IFileStorage>((serviceProvider) => {
|
||||
var conf = serviceProvider.GetRequiredService<IConfiguration>();
|
||||
var section = conf.GetSection("FileStorage");
|
||||
switch (section["Type"]) {
|
||||
case "local": {
|
||||
var dir = section["Directory"];
|
||||
return new LocalFileStorage(dir!);
|
||||
}
|
||||
default:
|
||||
throw new Exception("Unable to configure FileStorage");
|
||||
}
|
||||
});
|
||||
|
||||
services.AddControllers()
|
||||
.AddJsonOptions(options => {
|
||||
options.JsonSerializerOptions.PropertyNamingPolicy = JsonNamingPolicy.CamelCase;
|
||||
.AddNewtonsoftJson(options => {
|
||||
options.SerializerSettings.ContractResolver = new DefaultContractResolver {
|
||||
NamingStrategy = new CamelCaseNamingStrategy(),
|
||||
};
|
||||
});
|
||||
services.AddSwaggerGen(c => {
|
||||
c.SwaggerDoc("v1", new OpenApiInfo { Title = "InfoTren Scraper", Version = "v1" });
|
||||
|
@ -57,6 +82,30 @@ namespace Server {
|
|||
c.SwaggerEndpoint("/swagger/v1/swagger.json", "InfoTren Scraper v1");
|
||||
});
|
||||
|
||||
app.MapWhen(x => x.Request.Path.StartsWithSegments("/rapidoc"), appBuilder => {
|
||||
appBuilder.Run(async context => {
|
||||
context.Response.ContentType = "text/html";
|
||||
|
||||
await context.Response.WriteAsync(
|
||||
"""
|
||||
<!doctype html> <!-- Important: must specify -->
|
||||
<html>
|
||||
<head>
|
||||
<meta charset="utf-8"> <!-- Important: rapi-doc uses utf8 characters -->
|
||||
<script type="module" src="https://unpkg.com/rapidoc/dist/rapidoc-min.js"></script>
|
||||
</head>
|
||||
<body>
|
||||
<rapi-doc
|
||||
spec-url="/swagger/v3/swagger.json"
|
||||
theme = "dark"
|
||||
> </rapi-doc>
|
||||
</body>
|
||||
</html>
|
||||
"""
|
||||
);
|
||||
});
|
||||
});
|
||||
|
||||
// app.UseHttpsRedirection();
|
||||
|
||||
app.UseRouting();
|
||||
|
|
38
server/Utils/AsyncThrottle.cs
Normal file
38
server/Utils/AsyncThrottle.cs
Normal file
|
@ -0,0 +1,38 @@
|
|||
using System;
|
||||
using System.Threading;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace Server.Utils;
|
||||
|
||||
// Inspired from: https://stackoverflow.com/a/57517920
|
||||
public class AsyncThrottle {
|
||||
private readonly SemaphoreSlim openConnectionSemaphore;
|
||||
|
||||
public AsyncThrottle(int limit) {
|
||||
openConnectionSemaphore = new(limit, limit);
|
||||
}
|
||||
|
||||
public async Task<T> MakeRequest<T>(Task<T> task) => await MakeRequest(() => task);
|
||||
public async Task<T> MakeRequest<T>(Func<Task<T>> taskCreator) {
|
||||
await openConnectionSemaphore.WaitAsync();
|
||||
try {
|
||||
var result = await taskCreator();
|
||||
return result;
|
||||
}
|
||||
finally {
|
||||
openConnectionSemaphore.Release();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public async Task MakeRequest(Task task) => await MakeRequest(() => task);
|
||||
public async Task MakeRequest(Func<Task> taskCreator) {
|
||||
await openConnectionSemaphore.WaitAsync();
|
||||
try {
|
||||
await taskCreator();
|
||||
}
|
||||
finally {
|
||||
openConnectionSemaphore.Release();
|
||||
}
|
||||
}
|
||||
}
|
7
server/Utils/Constants.cs
Normal file
7
server/Utils/Constants.cs
Normal file
|
@ -0,0 +1,7 @@
|
|||
using NodaTime;
|
||||
|
||||
namespace Server.Utils;
|
||||
|
||||
public static class Constants {
|
||||
public static readonly DateTimeZone BucharestTz = DateTimeZoneProviders.Tzdb["Europe/Bucharest"];
|
||||
}
|
33
server/Utils/IAsyncCusorAsyncAdapter.cs
Normal file
33
server/Utils/IAsyncCusorAsyncAdapter.cs
Normal file
|
@ -0,0 +1,33 @@
|
|||
using System.Collections.Generic;
|
||||
using System.Threading.Tasks;
|
||||
using MongoDB.Driver;
|
||||
|
||||
namespace Server.Utils;
|
||||
|
||||
public record IAsyncCusorAsyncEnumerator<T>(IAsyncCursor<T> Cursor) {
|
||||
private IEnumerator<T>? enumerator = null;
|
||||
|
||||
public T Current => enumerator!.Current;
|
||||
|
||||
public async Task<bool> MoveNextAsync() {
|
||||
bool result;
|
||||
if (enumerator != null) {
|
||||
result = enumerator.MoveNext();
|
||||
if (result) return true;
|
||||
}
|
||||
|
||||
result = await Cursor.MoveNextAsync();
|
||||
if (result) {
|
||||
enumerator = Cursor.Current.GetEnumerator();
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
||||
|
||||
public static class IAsyncCursorExtensions {
|
||||
public static IAsyncCusorAsyncEnumerator<T> GetAsyncEnumerator<T>(this IAsyncCursor<T> cursor) {
|
||||
return new(cursor);
|
||||
}
|
||||
}
|
|
@ -5,5 +5,9 @@
|
|||
"Microsoft": "Warning",
|
||||
"Microsoft.Hosting.Lifetime": "Information"
|
||||
}
|
||||
}
|
||||
},
|
||||
"TrainDataMongo": {
|
||||
"ConnectionString": "mongodb://localhost:27017",
|
||||
"DatabaseName": "NewInfoferScraper"
|
||||
},
|
||||
}
|
||||
|
|
|
@ -9,5 +9,9 @@
|
|||
"Microsoft.Hosting.Lifetime": "Information"
|
||||
}
|
||||
},
|
||||
"TrainDataMongo": {
|
||||
"ConnectionString": "mongodb://mongo:27017",
|
||||
"DatabaseName": "NewInfoferScraper"
|
||||
},
|
||||
"AllowedHosts": "*"
|
||||
}
|
||||
|
|
|
@ -4,12 +4,16 @@
|
|||
<Nullable>enable</Nullable>
|
||||
<AssemblyName>Server</AssemblyName>
|
||||
<RootNamespace>Server</RootNamespace>
|
||||
<TargetFramework>net6.0</TargetFramework>
|
||||
<LangVersion>11</LangVersion>
|
||||
<TargetFrameworks>net6.0;net7.0;net8.0</TargetFrameworks>
|
||||
</PropertyGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<PackageReference Include="Microsoft.AspNetCore.Mvc.NewtonsoftJson" Version="6.0.21" />
|
||||
<PackageReference Include="Microsoft.Data.Sqlite" Version="6.0.1" />
|
||||
<PackageReference Include="Microsoft.EntityFrameworkCore.Sqlite" Version="5.0.13" />
|
||||
<PackageReference Include="MongoDB.Analyzer" Version="1.1.0" />
|
||||
<PackageReference Include="MongoDB.Driver" Version="2.19.1" />
|
||||
<PackageReference Include="Nanoid" Version="2.1.0" />
|
||||
<PackageReference Include="Swashbuckle.AspNetCore" Version="5.6.3" />
|
||||
</ItemGroup>
|
||||
|
@ -18,8 +22,4 @@
|
|||
<ProjectReference Include="..\scraper\scraper.csproj" />
|
||||
</ItemGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<Folder Include="Utils" />
|
||||
</ItemGroup>
|
||||
|
||||
</Project>
|
||||
|
|
Loading…
Add table
Reference in a new issue