From 8dce62c1ab0dfcc2bacd70f832558f29a3b3bfa9 Mon Sep 17 00:00:00 2001
From: Antonio El Khoury <antonio.elkhoury@gmail.com>
Date: Sun, 1 Feb 2015 22:28:39 +0100
Subject: [PATCH] Navitia: Fix query context builder.

---
 .../pte/AbstractNavitiaProvider.java          | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/enabler/src/de/schildbach/pte/AbstractNavitiaProvider.java b/enabler/src/de/schildbach/pte/AbstractNavitiaProvider.java
index 5f42a932..e18dcb4d 100644
--- a/enabler/src/de/schildbach/pte/AbstractNavitiaProvider.java
+++ b/enabler/src/de/schildbach/pte/AbstractNavitiaProvider.java
@@ -1077,11 +1077,22 @@ public abstract class AbstractNavitiaProvider extends AbstractNetworkProvider
 					else
 					{
 						// Fill context.
+						String prevQueryUri = null;
+						String nextQueryUri = null;
 						final JSONArray links = head.getJSONArray("links");
-						final JSONObject prev = links.getJSONObject(0);
-						final String prevQueryUri = prev.getString("href");
-						final JSONObject next = links.getJSONObject(1);
-						final String nextQueryUri = next.getString("href");
+						for (int i = 0; i < links.length(); ++i)
+						{
+							final JSONObject link = links.getJSONObject(i);
+							final String type = link.getString("type");
+							if (type.equals("prev"))
+							{
+								prevQueryUri = link.getString("href");
+							}
+							else if (type.equals("next"))
+							{
+								nextQueryUri = link.getString("href");
+							}
+						}
 
 						final QueryTripsResult result = new QueryTripsResult(resultHeader, queryUri.toString(), from, null, to, new Context(from, to,
 								prevQueryUri, nextQueryUri), new LinkedList<Trip>());