mirror of
https://github.com/tonytins/amtkstat.git
synced 2026-10-24 13:42:51 -04:00
Finished wiring up loadTrainStatus()
Did a little code formatting after that cause Xcode is weird
This commit is contained in:
parent
3eb2018a8e
commit
f20d49f075
5 changed files with 62 additions and 45 deletions
|
|
@ -9,56 +9,60 @@ enum AmtrakError: Error {
|
|||
|
||||
struct AmtrakClient {
|
||||
var decoder = JSONDecoder()
|
||||
|
||||
|
||||
let stationBaseURL = "https://api.amtraker.com/v3/stations"
|
||||
let trainsBaseURL = "https://api.amtraker.com/v3/trains"
|
||||
let trainsBaseURL = "https://api.amtraker.com/v3/trains"
|
||||
let staleBaseURL = "https://api.amtraker.com/v3/stale"
|
||||
|
||||
|
||||
func fetchTrain(number trainid: String) async throws -> TrainResponse {
|
||||
try await get("\(trainsBaseURL)/\(trainid)")
|
||||
}
|
||||
|
||||
|
||||
func fetchAllTrains() async throws -> TrainResponse {
|
||||
try await get(trainsBaseURL)
|
||||
}
|
||||
|
||||
|
||||
func fetchAllTrainStatus(num trainNum: String) async throws -> [Train] {
|
||||
let reponse = try await fetchTrain(number: trainNum)
|
||||
return reponse.values.flatMap(\.self)
|
||||
}
|
||||
|
||||
func fetchStation(code stationid: String) async throws -> StationResponse {
|
||||
try await get("\(stationBaseURL)/\(stationid)")
|
||||
}
|
||||
|
||||
|
||||
func fetchAllStations() async throws -> StationResponse {
|
||||
try await get(stationBaseURL)
|
||||
}
|
||||
|
||||
|
||||
func trainNumber(fromTrainId trainId: String) -> String {
|
||||
trainId.split(separator: "-").first.map(String.init) ?? trainId
|
||||
}
|
||||
|
||||
|
||||
func fetchStationTrainNumbers(stationCode code: String) async throws -> [String] {
|
||||
let reponse = try await fetchStation(code: code)
|
||||
|
||||
|
||||
guard let meta = reponse[code] else {
|
||||
throw AmtrakError.stationNotFound(code)
|
||||
}
|
||||
|
||||
|
||||
return Array(Set(meta.trains.map(trainNumber(fromTrainId:))))
|
||||
}
|
||||
|
||||
|
||||
func get<T: Decodable>(_ urlString: String) async throws -> T {
|
||||
|
||||
guard let url = URL(string: urlString) else {
|
||||
throw AmtrakError.invalidURL(urlString)
|
||||
}
|
||||
|
||||
|
||||
let (data, reponse) = try await URLSession.shared.data(from: url)
|
||||
guard let http = reponse as? HTTPURLResponse else {
|
||||
throw AmtrakError.invalidResponse
|
||||
}
|
||||
|
||||
guard (200..<300).contains(http.statusCode) else {
|
||||
|
||||
guard (200 ..< 300).contains(http.statusCode) else {
|
||||
throw AmtrakError.httpStatus(http.statusCode)
|
||||
}
|
||||
|
||||
|
||||
return try decoder.decode(T.self, from: data)
|
||||
}
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue