Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
de.schildbach.pte.AbstractHafasClientInterfaceProvider Maven / Gradle / Ivy
/*
* Copyright the original author or authors.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see types, final Location location,
final int maxDistance, final int maxLocations) throws IOException {
if (location.hasCoord())
return jsonLocGeoPos(types, location.coord, maxDistance, maxLocations);
else
throw new IllegalArgumentException("cannot handle: " + location);
}
@Override
public QueryDeparturesResult queryDepartures(final String stationId, final @Nullable Date time,
final int maxDepartures, final boolean equivs) throws IOException {
return jsonStationBoard(stationId, time, maxDepartures, equivs);
}
@Override
public SuggestLocationsResult suggestLocations(final CharSequence constraint) throws IOException {
return jsonLocMatch(constraint, null, 0);
}
@Override
public QueryTripsResult queryTrips(final Location from, final @Nullable Location via, final Location to,
final Date date, final boolean dep, final @Nullable TripOptions options) throws IOException {
return jsonTripSearch(from, via, to, date, dep, options != null ? options.products : null,
options != null ? options.walkSpeed : null, null);
}
@Override
public QueryTripsResult queryMoreTrips(final QueryTripsContext context, final boolean later) throws IOException {
final JsonContext jsonContext = (JsonContext) context;
return jsonTripSearch(jsonContext.from, jsonContext.via, jsonContext.to, jsonContext.date, jsonContext.dep,
jsonContext.products, jsonContext.walkSpeed, later ? jsonContext.laterContext : jsonContext.earlierContext);
}
protected final NearbyLocationsResult jsonLocGeoPos(final Set types, final Point coord,
int maxDistance, int maxLocations) throws IOException {
if (maxDistance == 0)
maxDistance = DEFAULT_MAX_DISTANCE;
if (maxLocations == 0)
maxLocations = DEFAULT_MAX_LOCATIONS;
final boolean getStations = types.contains(LocationType.STATION);
final boolean getPOIs = types.contains(LocationType.POI);
final String request = wrapJsonApiRequest("LocGeoPos", "{\"ring\":" //
+ "{\"cCrd\":{\"x\":" + coord.getLonAs1E6() + ",\"y\":" + coord.getLatAs1E6() + "}," //
+ "\"maxDist\":" + maxDistance + "}," //
+ "\"getStops\":" + getStations + "," //
+ "\"getPOIs\":" + getPOIs + "," //
+ "\"maxLoc\":" + maxLocations + "}", //
false);
final HttpUrl url = requestUrl(request);
final CharSequence page = httpClient.get(url, request, "application/json");
try {
final JSONObject head = new JSONObject(page.toString());
final String headErr = head.optString("err", null);
if (headErr != null && !"OK".equals(headErr)) {
final String headErrTxt = head.optString("errTxt");
throw new RuntimeException(headErr + " " + headErrTxt);
}
final JSONArray svcResList = head.getJSONArray("svcResL");
checkState(svcResList.length() == 2);
final ResultHeader header = parseServerInfo(svcResList.getJSONObject(0), head.getString("ver"));
final JSONObject svcRes = svcResList.getJSONObject(1);
checkState("LocGeoPos".equals(svcRes.getString("meth")));
final String err = svcRes.getString("err");
if (!"OK".equals(err)) {
final String errTxt = svcRes.optString("errTxt");
log.debug("Hafas error: {} {}", err, errTxt);
if ("FAIL".equals(err) && "HCI Service: request failed".equals(errTxt))
return new NearbyLocationsResult(header, NearbyLocationsResult.Status.SERVICE_DOWN);
if ("CGI_READ_FAILED".equals(err))
return new NearbyLocationsResult(header, NearbyLocationsResult.Status.SERVICE_DOWN);
throw new RuntimeException(err + " " + errTxt);
}
final JSONObject res = svcRes.getJSONObject("res");
final JSONObject common = res.getJSONObject("common");
/* final List remarks = */ parseRemList(common.getJSONArray("remL"));
final JSONArray locL = res.optJSONArray("locL");
final List locations;
if (locL != null) {
locations = parseLocList(locL);
// filter unwanted location types
for (Iterator i = locations.iterator(); i.hasNext();) {
final Location location = i.next();
if (!types.contains(location.type))
i.remove();
}
} else {
locations = Collections.emptyList();
}
return new NearbyLocationsResult(header, locations);
} catch (final JSONException x) {
throw new ParserException("cannot parse json: '" + page + "' on " + url, x);
}
}
protected final QueryDeparturesResult jsonStationBoard(final String stationId, final @Nullable Date time,
final int maxDepartures, final boolean equivs) throws IOException {
final Calendar c = new GregorianCalendar(timeZone);
c.setTime(time);
final CharSequence jsonDate = jsonDate(c);
final CharSequence jsonTime = jsonTime(c);
final CharSequence normalizedStationId = normalizeStationId(stationId);
final CharSequence stbFltrEquiv = Boolean.toString(!equivs);
final CharSequence maxJny = Integer.toString(maxDepartures != 0 ? maxDepartures : DEFAULT_MAX_DEPARTURES);
final String request = wrapJsonApiRequest("StationBoard", "{\"type\":\"DEP\"," //
+ "\"date\":\"" + jsonDate + "\"," //
+ "\"time\":\"" + jsonTime + "\"," //
+ "\"stbLoc\":{\"type\":\"S\"," + "\"state\":\"F\"," // F/M
+ "\"extId\":" + JSONObject.quote(normalizedStationId.toString()) + "}," //
+ (apiVersion.compareToIgnoreCase("1.19") < 0 ? "\"stbFltrEquiv\":" + stbFltrEquiv + "," : "") //
+ "\"maxJny\":" + maxJny + "}", false);
final HttpUrl url = requestUrl(request);
final CharSequence page = httpClient.get(url, request, "application/json");
try {
final JSONObject head = new JSONObject(page.toString());
final String headErr = head.optString("err", null);
if (headErr != null && !"OK".equals(headErr)) {
final String headErrTxt = head.optString("errTxt");
throw new RuntimeException(headErr + " " + headErrTxt);
}
final JSONArray svcResList = head.getJSONArray("svcResL");
checkState(svcResList.length() == 2);
final ResultHeader header = parseServerInfo(svcResList.getJSONObject(0), head.getString("ver"));
final QueryDeparturesResult result = new QueryDeparturesResult(header);
final JSONObject svcRes = svcResList.optJSONObject(1);
checkState("StationBoard".equals(svcRes.getString("meth")));
final String err = svcRes.getString("err");
if (!"OK".equals(err)) {
final String errTxt = svcRes.optString("errTxt");
log.debug("Hafas error: {} {}", err, errTxt);
if ("LOCATION".equals(err) && "HCI Service: location missing or invalid".equals(errTxt))
return new QueryDeparturesResult(header, QueryDeparturesResult.Status.INVALID_STATION);
if ("FAIL".equals(err) && "HCI Service: request failed".equals(errTxt))
return new QueryDeparturesResult(header, QueryDeparturesResult.Status.SERVICE_DOWN);
if ("PROBLEMS".equals(err) && "HCI Service: problems during service execution".equals(errTxt))
return new QueryDeparturesResult(header, QueryDeparturesResult.Status.SERVICE_DOWN);
if ("CGI_READ_FAILED".equals(err))
return new QueryDeparturesResult(header, QueryDeparturesResult.Status.SERVICE_DOWN);
throw new RuntimeException(err + " " + errTxt);
}
final JSONObject res = svcRes.getJSONObject("res");
final JSONObject common = res.getJSONObject("common");
final List remarks = parseRemList(common.getJSONArray("remL"));
final List operators = parseOpList(common.getJSONArray("opL"));
final List lines = parseProdList(common.getJSONArray("prodL"), operators);
final JSONArray locList = common.getJSONArray("locL");
final JSONArray jnyList = res.optJSONArray("jnyL");
if (jnyList != null) {
for (int iJny = 0; iJny < jnyList.length(); iJny++) {
final JSONObject jny = jnyList.getJSONObject(iJny);
final JSONObject stbStop = jny.getJSONObject("stbStop");
final boolean cancelled = stbStop.optBoolean("dCncl", false);
if (cancelled)
continue;
final String stbStopPlatformS = stbStop.optString("dPlatfS", null);
c.clear();
ParserUtils.parseIsoDate(c, jny.getString("date"));
final Date baseDate = c.getTime();
final Date plannedTime = parseJsonTime(c, baseDate, stbStop.getString("dTimeS"));
final Date predictedTime = parseJsonTime(c, baseDate, stbStop.optString("dTimeR", null));
final int dProdX = stbStop.optInt("dProdX", -1);
final Line line = dProdX != -1 ? lines.get(dProdX) : null;
final Location location = equivs ? parseLoc(locList, stbStop.getInt("locX"), null)
: new Location(LocationType.STATION, stationId);
final Position position = normalizePosition(stbStopPlatformS);
final String jnyDirTxt = jny.getString("dirTxt");
Location destination = null;
// if last entry in stopL happens to be our destination, use it
final JSONArray stopList = jny.optJSONArray("stopL");
if (stopList != null) {
final int lastStopIdx = stopList.getJSONObject(stopList.length() - 1).getInt("locX");
final String lastStopName = locList.getJSONObject(lastStopIdx).getString("name");
if (jnyDirTxt.equals(lastStopName))
destination = parseLoc(locList, lastStopIdx, null);
}
// otherwise split unidentified destination as if it was a station and use it
if (destination == null) {
final String[] splitJnyDirTxt = splitStationName(jnyDirTxt);
destination = new Location(LocationType.ANY, null, splitJnyDirTxt[0], splitJnyDirTxt[1]);
}
final JSONArray remList = jny.optJSONArray("remL");
String message = null;
if (remList != null) {
for (int iRem = 0; iRem < remList.length(); iRem++) {
final JSONObject rem = remList.getJSONObject(iRem);
final String[] remark = remarks.get(rem.getInt("remX"));
if ("l?".equals(remark[0]))
message = remark[1];
}
}
if (line != null) {
final Departure departure = new Departure(plannedTime, predictedTime, line, position,
destination, null, message);
StationDepartures stationDepartures = findStationDepartures(result.stationDepartures, location);
if (stationDepartures == null) {
stationDepartures = new StationDepartures(location, new ArrayList(8), null);
result.stationDepartures.add(stationDepartures);
}
stationDepartures.departures.add(departure);
}
}
}
// sort departures
for (final StationDepartures stationDepartures : result.stationDepartures)
Collections.sort(stationDepartures.departures, Departure.TIME_COMPARATOR);
return result;
} catch (final JSONException x) {
throw new ParserException("cannot parse json: '" + page + "' on " + url, x);
}
}
protected final SuggestLocationsResult jsonLocMatch(final CharSequence constraint,
final @Nullable Set types, int maxLocations) throws IOException {
checkNotNull(constraint);
if (maxLocations == 0)
maxLocations = DEFAULT_MAX_LOCATIONS;
final String type;
if (types == null || types.contains(LocationType.ANY)
|| types.containsAll(EnumSet.of(LocationType.STATION, LocationType.ADDRESS, LocationType.POI)))
type = "ALL";
else
type = Joiner.on("").skipNulls().join(types.contains(LocationType.STATION) ? "S" : null,
types.contains(LocationType.ADDRESS) ? "A" : null, types.contains(LocationType.POI) ? "P" : null);
final String loc = "{\"name\":" + JSONObject.quote(constraint + "?") + ",\"type\":\"" + type + "\"}";
final String request = wrapJsonApiRequest("LocMatch",
"{\"input\":{\"field\":\"S\",\"loc\":" + loc + ",\"maxLoc\":" + maxLocations + "}}", false);
final HttpUrl url = requestUrl(request);
final CharSequence page = httpClient.get(url, request, "application/json");
try {
final JSONObject head = new JSONObject(page.toString());
final String headErr = head.optString("err", null);
if (headErr != null && !"OK".equals(headErr)) {
final String headErrTxt = head.optString("errTxt");
throw new RuntimeException(headErr + " " + headErrTxt);
}
final JSONArray svcResList = head.getJSONArray("svcResL");
checkState(svcResList.length() == 2);
final ResultHeader header = parseServerInfo(svcResList.getJSONObject(0), head.getString("ver"));
final JSONObject svcRes = svcResList.optJSONObject(1);
checkState("LocMatch".equals(svcRes.getString("meth")));
final String err = svcRes.getString("err");
if (!"OK".equals(err)) {
final String errTxt = svcRes.optString("errTxt");
log.debug("Hafas error: {} {}", err, errTxt);
if ("FAIL".equals(err) && "HCI Service: request failed".equals(errTxt))
return new SuggestLocationsResult(header, SuggestLocationsResult.Status.SERVICE_DOWN);
if ("CGI_READ_FAILED".equals(err))
return new SuggestLocationsResult(header, SuggestLocationsResult.Status.SERVICE_DOWN);
throw new RuntimeException(err + " " + errTxt);
}
final JSONObject res = svcRes.getJSONObject("res");
final JSONObject common = res.getJSONObject("common");
/* final List remarks = */ parseRemList(common.getJSONArray("remL"));
final JSONObject match = res.getJSONObject("match");
final List locations = parseLocList(match.optJSONArray("locL"));
final List suggestedLocations = new ArrayList<>(locations.size());
for (final Location location : locations)
suggestedLocations.add(new SuggestedLocation(location));
// TODO weight
return new SuggestLocationsResult(header, suggestedLocations);
} catch (final JSONException x) {
throw new ParserException("cannot parse json: '" + page + "' on " + url, x);
}
}
private static final Joiner JOINER = Joiner.on(' ').skipNulls();
private Location jsonTripSearchIdentify(final Location location) throws IOException {
if (location.hasId())
return location;
if (location.hasName()) {
final SuggestLocationsResult result = jsonLocMatch(JOINER.join(location.place, location.name), null, 1);
if (result.status == SuggestLocationsResult.Status.OK) {
final List locations = result.getLocations();
if (!locations.isEmpty())
return locations.get(0);
}
}
if (location.hasCoord()) {
final NearbyLocationsResult result = jsonLocGeoPos(EnumSet.allOf(LocationType.class), location.coord, 0, 1);
if (result.status == NearbyLocationsResult.Status.OK) {
final List locations = result.locations;
if (!locations.isEmpty())
return locations.get(0);
}
}
return null;
}
protected final QueryTripsResult jsonTripSearch(Location from, @Nullable Location via, Location to, final Date time,
final boolean dep, final @Nullable Set products, final @Nullable WalkSpeed walkSpeed,
final @Nullable String moreContext) throws IOException {
from = jsonTripSearchIdentify(from);
if (from == null)
return new QueryTripsResult(new ResultHeader(network, SERVER_PRODUCT),
QueryTripsResult.Status.UNKNOWN_FROM);
if (via != null) {
via = jsonTripSearchIdentify(via);
if (via == null)
return new QueryTripsResult(new ResultHeader(network, SERVER_PRODUCT),
QueryTripsResult.Status.UNKNOWN_VIA);
}
to = jsonTripSearchIdentify(to);
if (to == null)
return new QueryTripsResult(new ResultHeader(network, SERVER_PRODUCT), QueryTripsResult.Status.UNKNOWN_TO);
final Calendar c = new GregorianCalendar(timeZone);
c.setTime(time);
final CharSequence outDate = jsonDate(c);
final CharSequence outTime = jsonTime(c);
final CharSequence outFrwd = Boolean.toString(dep);
final CharSequence jnyFltr = products != null ? productsString(products) : null;
final String meta = "foot_speed_" + (walkSpeed != null ? walkSpeed : WalkSpeed.NORMAL).name().toLowerCase();
final CharSequence jsonContext = moreContext != null ? "\"ctxScr\":" + JSONObject.quote(moreContext) + "," : "";
final String request = wrapJsonApiRequest("TripSearch", "{" //
+ jsonContext //
+ "\"depLocL\":[" + jsonLocation(from) + "]," //
+ "\"arrLocL\":[" + jsonLocation(to) + "]," //
+ (via != null ? "\"viaLocL\":[{\"loc\":" + jsonLocation(via) + "}]," : "") //
+ "\"outDate\":\"" + outDate + "\"," //
+ "\"outTime\":\"" + outTime + "\"," //
+ "\"outFrwd\":" + outFrwd + "," //
+ (jnyFltr != null
? "\"jnyFltrL\":[{\"value\":\"" + jnyFltr + "\",\"mode\":\"BIT\",\"type\":\"PROD\"}]," : "") //
+ "\"gisFltrL\":[{\"mode\":\"FB\",\"profile\":{\"type\":\"F\",\"linDistRouting\":false,\"maxdist\":2000},\"type\":\"M\",\"meta\":\""
+ meta + "\"}]," //
+ "\"getPolyline\":false,\"getPasslist\":true,\"getIST\":false,\"getEco\":false,\"extChgTime\":-1}", //
false);
final HttpUrl url = requestUrl(request);
final CharSequence page = httpClient.get(url, request, "application/json");
try {
final JSONObject head = new JSONObject(page.toString());
final String headErr = head.optString("err", null);
if (headErr != null && !"OK".equals(headErr)) {
final String headErrTxt = head.optString("errTxt");
throw new RuntimeException(headErr + " " + headErrTxt);
}
final JSONArray svcResList = head.getJSONArray("svcResL");
checkState(svcResList.length() == 2);
final ResultHeader header = parseServerInfo(svcResList.getJSONObject(0), head.getString("ver"));
final JSONObject svcRes = svcResList.optJSONObject(1);
checkState("TripSearch".equals(svcRes.getString("meth")));
final String err = svcRes.getString("err");
if (!"OK".equals(err)) {
final String errTxt = svcRes.optString("errTxt");
log.debug("Hafas error: {} {}", err, errTxt);
if ("H890".equals(err)) // No connections found.
return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS);
if ("H891".equals(err)) // No route found (try entering an intermediate station).
return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS);
if ("H892".equals(err)) // HAFAS Kernel: Request too complex (try entering less intermediate
// stations).
return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS);
if ("H895".equals(err)) // Departure/Arrival are too near.
return new QueryTripsResult(header, QueryTripsResult.Status.TOO_CLOSE);
if ("H9220".equals(err)) // Nearby to the given address stations could not be found.
return new QueryTripsResult(header, QueryTripsResult.Status.UNRESOLVABLE_ADDRESS);
if ("H886".equals(err)) // HAFAS Kernel: No connections found within the requested time
// interval.
return new QueryTripsResult(header, QueryTripsResult.Status.NO_TRIPS);
if ("H887".equals(err)) // HAFAS Kernel: Kernel computation time limit reached.
return new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN);
if ("H9240".equals(err)) // HAFAS Kernel: Internal error.
return new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN);
if ("H9360".equals(err)) // Date outside of the timetable period.
return new QueryTripsResult(header, QueryTripsResult.Status.INVALID_DATE);
if ("H9380".equals(err)) // Departure/Arrival/Intermediate or equivalent stations def'd more
// than once.
return new QueryTripsResult(header, QueryTripsResult.Status.TOO_CLOSE);
if ("FAIL".equals(err) && "HCI Service: request failed".equals(errTxt))
return new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN);
if ("PROBLEMS".equals(err) && "HCI Service: problems during service execution".equals(errTxt))
return new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN);
if ("LOCATION".equals(err) && "HCI Service: location missing or invalid".equals(errTxt))
return new QueryTripsResult(header, QueryTripsResult.Status.UNKNOWN_LOCATION);
if ("CGI_READ_FAILED".equals(err))
return new QueryTripsResult(header, QueryTripsResult.Status.SERVICE_DOWN);
throw new RuntimeException(err + " " + errTxt);
}
final JSONObject res = svcRes.getJSONObject("res");
final JSONObject common = res.getJSONObject("common");
final List remarks = parseRemList(common.getJSONArray("remL"));
final JSONArray locList = common.getJSONArray("locL");
final List operators = parseOpList(common.getJSONArray("opL"));
final List lines = parseProdList(common.getJSONArray("prodL"), operators);
final JSONArray outConList = res.optJSONArray("outConL");
final List trips = new ArrayList<>(outConList.length());
for (int iOutCon = 0; iOutCon < outConList.length(); iOutCon++) {
final JSONObject outCon = outConList.getJSONObject(iOutCon);
final Location tripFrom = parseLoc(locList, outCon.getJSONObject("dep").getInt("locX"), null);
final Location tripTo = parseLoc(locList, outCon.getJSONObject("arr").getInt("locX"), null);
c.clear();
ParserUtils.parseIsoDate(c, outCon.getString("date"));
final Date baseDate = c.getTime();
final JSONArray secList = outCon.optJSONArray("secL");
final List legs = new ArrayList<>(secList.length());
for (int iSec = 0; iSec < secList.length(); iSec++) {
final JSONObject sec = secList.getJSONObject(iSec);
final String secType = sec.getString("type");
final JSONObject secDep = sec.getJSONObject("dep");
final Stop departureStop = parseJsonStop(secDep, locList, c, baseDate);
final JSONObject secArr = sec.getJSONObject("arr");
final Stop arrivalStop = parseJsonStop(secArr, locList, c, baseDate);
final Trip.Leg leg;
if ("JNY".equals(secType)) {
final JSONObject jny = sec.getJSONObject("jny");
final Line line = lines.get(jny.getInt("prodX"));
final String dirTxt = jny.optString("dirTxt", null);
final Location destination;
if (dirTxt != null) {
final String[] splitDirTxt = splitStationName(dirTxt);
destination = new Location(LocationType.ANY, null, splitDirTxt[0], splitDirTxt[1]);
} else {
destination = null;
}
final JSONArray stopList = jny.optJSONArray("stopL");
final List intermediateStops;
if (stopList != null) {
checkState(stopList.length() >= 2);
intermediateStops = new ArrayList<>(stopList.length());
for (int iStop = 1; iStop < stopList.length() - 1; iStop++) {
final JSONObject stop = stopList.getJSONObject(iStop);
final Stop intermediateStop = parseJsonStop(stop, locList, c, baseDate);
intermediateStops.add(intermediateStop);
}
} else {
intermediateStops = null;
}
final JSONArray remList = jny.optJSONArray("remL");
String message = null;
if (remList != null) {
for (int iRem = 0; iRem < remList.length(); iRem++) {
final JSONObject rem = remList.getJSONObject(iRem);
final String[] remark = remarks.get(rem.getInt("remX"));
if ("l?".equals(remark[0]))
message = remark[1];
}
}
leg = new Trip.Public(line, destination, departureStop, arrivalStop, intermediateStops, null,
message);
} else if ("DEVI".equals(secType)) {
leg = new Trip.Individual(Trip.Individual.Type.TRANSFER, departureStop.location,
departureStop.getDepartureTime(), arrivalStop.location, arrivalStop.getArrivalTime(),
null, 0);
} else if ("WALK".equals(secType) || "TRSF".equals(secType)) {
final JSONObject gis = sec.getJSONObject("gis");
final int distance = gis.optInt("dist", 0);
leg = new Trip.Individual(Trip.Individual.Type.WALK, departureStop.location,
departureStop.getDepartureTime(), arrivalStop.location, arrivalStop.getArrivalTime(),
null, distance);
} else {
throw new IllegalStateException("cannot handle type: " + secType);
}
legs.add(leg);
}
final JSONObject trfRes = outCon.optJSONObject("trfRes");
final List fares = new LinkedList<>();
if (trfRes != null) {
final JSONArray fareSetList = trfRes.getJSONArray("fareSetL");
for (int iFareSet = 0; iFareSet < fareSetList.length(); iFareSet++) {
final JSONObject fareSet = fareSetList.getJSONObject(iFareSet);
final String fareSetName = fareSet.optString("name", null);
final String fareSetDescription = fareSet.optString("desc", null);
if (fareSetName != null || fareSetDescription != null) {
final JSONArray fareList = fareSet.getJSONArray("fareL");
for (int iFare = 0; iFare < fareList.length(); iFare++) {
final JSONObject jsonFare = fareList.getJSONObject(iFare);
final String name = jsonFare.getString("name");
final JSONArray ticketList = jsonFare.optJSONArray("ticketL");
if (ticketList != null) {
for (int iTicket = 0; iTicket < ticketList.length(); iTicket++) {
final JSONObject jsonTicket = ticketList.getJSONObject(iTicket);
final String ticketName = jsonTicket.getString("name");
final Currency currency = Currency.getInstance(jsonTicket.getString("cur"));
final float price = jsonTicket.getInt("prc") / 100f;
final Fare fare = parseJsonTripFare(name, fareSetDescription, ticketName,
currency, price);
if (fare != null)
fares.add(fare);
}
} else {
final Currency currency = Currency.getInstance(jsonFare.getString("cur"));
final float price = jsonFare.getInt("prc") / 100f;
final Fare fare = parseJsonTripFare(fareSetName, fareSetDescription, name, currency,
price);
if (fare != null)
fares.add(fare);
}
}
}
}
}
final Trip trip = new Trip(null, tripFrom, tripTo, legs, fares, null, null);
trips.add(trip);
}
final JsonContext context = new JsonContext(from, via, to, time, dep, products, walkSpeed,
res.optString("outCtxScrF"), res.optString("outCtxScrB"));
return new QueryTripsResult(header, null, from, null, to, context, trips);
} catch (final JSONException x) {
throw new ParserException("cannot parse json: '" + page + "' on " + url, x);
}
}
protected Fare parseJsonTripFare(final @Nullable String fareSetName, final @Nullable String fareSetDescription,
final String name, final Currency currency, final float price) {
if (name.endsWith("- Jahreskarte") || name.endsWith("- Monatskarte"))
return null;
if (name.startsWith("Vollpreis - "))
return new Fare(fareSetName, Fare.Type.ADULT, currency, price, name.substring(12), null);
if (name.startsWith("Kind - "))
return new Fare(fareSetName, Fare.Type.CHILD, currency, price, name.substring(7), null);
return null;
}
private String wrapJsonApiRequest(final String meth, final String req, final boolean formatted) {
return "{" //
+ (apiAuthorization != null ? "\"auth\":" + apiAuthorization + "," : "") //
+ "\"client\":" + checkNotNull(apiClient) + "," //
+ (apiExt != null ? "\"ext\":\"" + apiExt + "\"," : "") //
+ "\"ver\":\"" + checkNotNull(apiVersion) + "\",\"lang\":\"eng\"," //
+ "\"svcReqL\":[" //
+ "{\"meth\":\"ServerInfo\",\"req\":{\"getServerDateTime\":true,\"getTimeTablePeriod\":false}}," //
+ "{\"meth\":\"" + meth + "\",\"cfg\":{\"polyEnc\":\"GPA\"},\"req\":" + req + "}" //
+ "]," //
+ "\"formatted\":" + formatted + "}";
}
private HttpUrl requestUrl(final String body) {
final HttpUrl.Builder url = apiBase.newBuilder().addPathSegment(apiEndpoint);
if (requestChecksumSalt != null) {
final HashCode checksum = MD5.newHasher().putString(body, Charsets.UTF_8).putBytes(requestChecksumSalt)
.hash();
url.addQueryParameter("checksum", checksum.toString());
}
if (requestMicMacSalt != null) {
final HashCode mic = MD5.newHasher().putString(body, Charsets.UTF_8).hash();
url.addQueryParameter("mic", HEX.encode(mic.asBytes()));
final HashCode mac = MD5.newHasher().putString(HEX.encode(mic.asBytes()), Charsets.UTF_8)
.putBytes(requestMicMacSalt).hash();
url.addQueryParameter("mac", HEX.encode(mac.asBytes()));
}
return url.build();
}
private String jsonLocation(final Location location) {
if (location.type == LocationType.STATION && location.hasId())
return "{\"type\":\"S\",\"extId\":" + JSONObject.quote(location.id) + "}";
else if (location.type == LocationType.ADDRESS && location.hasId())
return "{\"type\":\"A\",\"lid\":" + JSONObject.quote(location.id) + "}";
else if (location.type == LocationType.POI && location.hasId())
return "{\"type\":\"P\",\"lid\":" + JSONObject.quote(location.id) + "}";
else
throw new IllegalArgumentException("cannot handle: " + location);
}
private CharSequence jsonDate(final Calendar time) {
final int year = time.get(Calendar.YEAR);
final int month = time.get(Calendar.MONTH) + 1;
final int day = time.get(Calendar.DAY_OF_MONTH);
return String.format(Locale.ENGLISH, "%04d%02d%02d", year, month, day);
}
private CharSequence jsonTime(final Calendar time) {
final int hour = time.get(Calendar.HOUR_OF_DAY);
final int minute = time.get(Calendar.MINUTE);
return String.format(Locale.ENGLISH, "%02d%02d00", hour, minute);
}
private ResultHeader parseServerInfo(final JSONObject serverInfo, final String serverVersion) throws JSONException {
checkState("ServerInfo".equals(serverInfo.getString("meth")));
final String err = serverInfo.optString("err", null);
if (err != null && !"OK".equals(err)) {
final String errTxt = serverInfo.optString("errTxt");
log.info("ServerInfo error: {} {}, ignoring", err, errTxt);
return new ResultHeader(network, SERVER_PRODUCT, serverVersion, null, 0, null);
}
final JSONObject res = serverInfo.getJSONObject("res");
final Calendar c = new GregorianCalendar(timeZone);
ParserUtils.parseIsoDate(c, res.getString("sD"));
c.setTime(parseJsonTime(c, c.getTime(), res.getString("sT")));
return new ResultHeader(network, SERVER_PRODUCT, serverVersion, null, c.getTimeInMillis(), null);
}
private static final Pattern P_JSON_TIME = Pattern.compile("(\\d{2})?(\\d{2})(\\d{2})(\\d{2})");
private final Date parseJsonTime(final Calendar calendar, final Date baseDate, final CharSequence str) {
if (str == null)
return null;
final Matcher m = P_JSON_TIME.matcher(str);
if (m.matches()) {
calendar.setTime(baseDate);
if (m.group(1) != null)
calendar.add(Calendar.DAY_OF_YEAR, Integer.parseInt(m.group(1)));
calendar.set(Calendar.HOUR_OF_DAY, Integer.parseInt(m.group(2)));
calendar.set(Calendar.MINUTE, Integer.parseInt(m.group(3)));
calendar.set(Calendar.SECOND, Integer.parseInt(m.group(4)));
return calendar.getTime();
}
throw new RuntimeException("cannot parse: '" + str + "'");
}
private Stop parseJsonStop(final JSONObject json, final JSONArray locList, final Calendar c, final Date baseDate)
throws JSONException {
final Location location = parseLoc(locList, json.getInt("locX"), null);
final boolean arrivalCancelled = json.optBoolean("aCncl", false);
final Date plannedArrivalTime = parseJsonTime(c, baseDate, json.optString("aTimeS", null));
final Date predictedArrivalTime = parseJsonTime(c, baseDate, json.optString("aTimeR", null));
final Position plannedArrivalPosition = normalizePosition(json.optString("aPlatfS", null));
final Position predictedArrivalPosition = normalizePosition(json.optString("aPlatfR", null));
final boolean departureCancelled = json.optBoolean("dCncl", false);
final Date plannedDepartureTime = parseJsonTime(c, baseDate, json.optString("dTimeS", null));
final Date predictedDepartureTime = parseJsonTime(c, baseDate, json.optString("dTimeR", null));
final Position plannedDeparturePosition = normalizePosition(json.optString("dPlatfS", null));
final Position predictedDeparturePosition = normalizePosition(json.optString("dPlatfR", null));
return new Stop(location, plannedArrivalTime, predictedArrivalTime, plannedArrivalPosition,
predictedArrivalPosition, arrivalCancelled, plannedDepartureTime, predictedDepartureTime,
plannedDeparturePosition, predictedDeparturePosition, departureCancelled);
}
private List parseRemList(final JSONArray remList) throws JSONException {
final List remarks = new ArrayList<>(remList.length());
for (int i = 0; i < remList.length(); i++) {
final JSONObject rem = remList.getJSONObject(i);
final String code = rem.optString("code", null);
final String txtS = rem.optString("txtS", null);
final String txtN = rem.optString("txtN", null);
remarks.add(new String[] { code, txtS != null ? txtS : txtN });
}
return remarks;
}
private List parseLocList(final JSONArray locList) throws JSONException {
final List locations = new ArrayList<>(locList.length());
for (int iLoc = 0; iLoc < locList.length(); iLoc++)
locations.add(parseLoc(locList, iLoc, null));
return locations;
}
private Location parseLoc(final JSONArray locList, final int locListIndex,
@Nullable Set previousLocListIndexes) throws JSONException {
final JSONObject loc = locList.getJSONObject(locListIndex);
final String type = loc.getString("type");
final LocationType locationType;
final String id;
final String[] placeAndName;
final Set products;
if ("S".equals(type)) {
final int mMastLocX = loc.optInt("mMastLocX", -1);
if (mMastLocX != -1) {
if (previousLocListIndexes == null)
previousLocListIndexes = new HashSet<>();
if (!previousLocListIndexes.contains(mMastLocX)) {
previousLocListIndexes.add(locListIndex);
return parseLoc(locList, mMastLocX, previousLocListIndexes);
}
}
locationType = LocationType.STATION;
id = normalizeStationId(loc.getString("extId"));
placeAndName = splitStationName(loc.getString("name"));
final int pCls = loc.optInt("pCls", -1);
products = pCls != -1 ? intToProducts(pCls) : null;
} else if ("P".equals(type)) {
locationType = LocationType.POI;
id = loc.getString("lid");
placeAndName = splitPOI(loc.getString("name"));
products = null;
} else if ("A".equals(type)) {
locationType = LocationType.ADDRESS;
id = loc.getString("lid");
placeAndName = splitAddress(loc.getString("name"));
products = null;
} else {
throw new RuntimeException("Unknown type " + type + ": " + loc);
}
final JSONObject crd = loc.optJSONObject("crd");
if (crd != null)
return new Location(locationType, id, Point.from1E6(crd.getInt("y"), crd.getInt("x")), placeAndName[0],
placeAndName[1], products);
else
return new Location(LocationType.STATION, id, null, placeAndName[0], placeAndName[1], products);
}
private List parseOpList(final JSONArray opList) throws JSONException {
final List operators = new ArrayList<>(opList.length());
for (int i = 0; i < opList.length(); i++) {
final JSONObject op = opList.getJSONObject(i);
final String operator = op.getString("name");
operators.add(operator);
}
return operators;
}
private List parseProdList(final JSONArray prodList, final List operators) throws JSONException {
final int prodListLen = prodList.length();
final List lines = new ArrayList<>(prodListLen);
for (int iProd = 0; iProd < prodListLen; iProd++) {
final JSONObject prod = prodList.getJSONObject(iProd);
final String name = Strings.emptyToNull(prod.getString("name"));
final String nameS = prod.optString("nameS", null);
final String number = prod.optString("number", null);
final int oprIndex = prod.optInt("oprX", -1);
final String operator = oprIndex != -1 ? operators.get(oprIndex) : null;
final int cls = prod.optInt("cls", -1);
final Product product = cls != -1 ? intToProduct(cls) : null;
lines.add(newLine(operator, product, name, nameS, number));
}
return lines;
}
protected Line newLine(final String operator, final Product product, final @Nullable String name,
final @Nullable String shortName, final @Nullable String number) {
final String longName;
if (name != null)
longName = name + (number != null && !name.endsWith(number) ? " (" + number + ")" : "");
else if (shortName != null)
longName = shortName + (number != null && !shortName.endsWith(number) ? " (" + number + ")" : "");
else
longName = number;
if (product == Product.BUS || product == Product.TRAM) {
// For bus and tram, prefer a slightly shorter label without the product prefix
final String label;
if (shortName != null)
label = shortName;
else if (number != null && name != null && name.endsWith(number))
label = number;
else
label = name;
return new Line(null, operator, product, label, longName, lineStyle(operator, product, label));
} else {
// Otherwise the longer label is fine
return new Line(null, operator, product, name, longName, lineStyle(operator, product, name));
}
}
@SuppressWarnings("serial")
public static class JsonContext implements QueryTripsContext {
public final Location from, via, to;
public final Date date;
public final boolean dep;
public final Set products;
public final WalkSpeed walkSpeed;
public final String laterContext, earlierContext;
public JsonContext(final Location from, final @Nullable Location via, final Location to, final Date date,
final boolean dep, final Set products, final WalkSpeed walkSpeed, final String laterContext,
final String earlierContext) {
this.from = from;
this.via = via;
this.to = to;
this.date = date;
this.dep = dep;
this.products = products;
this.walkSpeed = walkSpeed;
this.laterContext = laterContext;
this.earlierContext = earlierContext;
}
@Override
public boolean canQueryLater() {
return laterContext != null;
}
@Override
public boolean canQueryEarlier() {
return earlierContext != null;
}
}
}
