implement location load indicator in gateway selection cell

2 files changed, 12 insertions, 9 deletions

diff --git a/app/src/main/java/se/leap/bitmaskclient/base/fragments/GatewaySelectionFragment.java b/app/src/main/java/se/leap/bitmaskclient/base/fragments/GatewaySelectionFragment.java
index d1222cd7..e4d8ca8b 100644
--- a/app/src/main/java/se/leap/bitmaskclient/base/fragments/GatewaySelectionFragment.java
+++ b/app/src/main/java/se/leap/bitmaskclient/base/fragments/GatewaySelectionFragment.java
@@ -46,6 +46,7 @@ import se.leap.bitmaskclient.base.MainActivity;
 import se.leap.bitmaskclient.base.models.Location;
 import se.leap.bitmaskclient.base.utils.PreferenceHelper;
 import se.leap.bitmaskclient.base.views.IconSwitchEntry;
+import se.leap.bitmaskclient.base.views.LocationIndicator;
 import se.leap.bitmaskclient.eip.EipCommand;
 import se.leap.bitmaskclient.eip.EipStatus;
 import se.leap.bitmaskclient.eip.GatewaysManager;
@@ -189,7 +190,7 @@ public class GatewaySelectionFragment extends Fragment implements SharedPreferen
 
         static class ViewHolder extends RecyclerView.ViewHolder {
             public AppCompatTextView locationLabel;
-            public AppCompatTextView qualityLabel;
+            public LocationIndicator locationIndicator;
             public AppCompatImageView checkedIcon;
             public View layout;
 
@@ -197,7 +198,7 @@ public class GatewaySelectionFragment extends Fragment implements SharedPreferen
                 super(v);
                 layout = v;
                 locationLabel = (AppCompatTextView) v.findViewById(R.id.location);
-                qualityLabel = (AppCompatTextView) v.findViewById(R.id.quality);
+                locationIndicator = (LocationIndicator) v.findViewById(R.id.quality);
                 checkedIcon = (AppCompatImageView) v.findViewById(R.id.checked_icon);
             }
         }
@@ -263,7 +264,7 @@ public class GatewaySelectionFragment extends Fragment implements SharedPreferen
             DrawableCompat.setTint(checkIcon, ContextCompat.getColor(holder.layout.getContext(), R.color.colorSuccess));
             holder.checkedIcon.setImageDrawable(checkIcon);
             holder.checkedIcon.setVisibility(location.selected ? VISIBLE : INVISIBLE);
-            holder.qualityLabel.setText(getQualityString(location.averageLoad));
+            holder.locationIndicator.setLoad(GatewaysManager.Load.getLoadByValue(location.averageLoad));
             if (location.selected) {
                 selectedLocation = location;
             }
diff --git a/app/src/main/java/se/leap/bitmaskclient/eip/GatewaysManager.java b/app/src/main/java/se/leap/bitmaskclient/eip/GatewaysManager.java
index b311315c..24e9c323 100644
--- a/app/src/main/java/se/leap/bitmaskclient/eip/GatewaysManager.java
+++ b/app/src/main/java/se/leap/bitmaskclient/eip/GatewaysManager.java
@@ -130,9 +130,15 @@ public class GatewaysManager {
         while ((gateway = select(n, null)) != null) {
             if (!locationNames.containsKey(gateway.getName())) {
                 locationNames.put(gateway.getName(), locations.size());
+                // fake values for now
+                Random rand = new Random();
+                double averageLoad = rand.nextDouble(); //location.averageLoad;
+
+
                 Location location = new Location(
                         gateway.getName(),
-                        gateway.getFullness(),
+                        averageLoad
+                        /*gateway.getFullness()*/,
                         1,
                         gateway.getName().equals(selectedCity));
                 locations.add(location);
@@ -153,11 +159,7 @@ public class GatewaysManager {
         List <Location> locations = getGatewayLocations();
         for (Location location : locations) {
             if (location.name.equals(name)) {
-
-                // fake values for now
-                Random rand = new Random();
-                double averageLoad = rand.nextDouble(); //location.averageLoad;
-                return Load.getLoadByValue(averageLoad);
+                return Load.getLoadByValue(location.averageLoad);
             }
         }