Don't expose the full gateway out of the vpn module

We just care about locations and fullness.

2 files changed, 7 insertions, 12 deletions

diff --git a/pkg/vpn/bonafide/bonafide.go b/pkg/vpn/bonafide/bonafide.go
index 321b47f..e0d9c9c 100644
--- a/pkg/vpn/bonafide/bonafide.go
+++ b/pkg/vpn/bonafide/bonafide.go
@@ -224,13 +224,8 @@ func (b *Bonafide) GetAllGateways(transport string) ([]Gateway, error) {
 	return gws, err
 }
 
-func (b *Bonafide) PickGatewayForCities(transport string) (map[string]Gateway, error) {
-	return b.gateways.pickGatewayForCities(transport), nil
-}
-
-func (b *Bonafide) GetGatewayDetails(host string) (Gateway, error) {
-	gw, err := b.gateways.getGatewayByHost(host)
-	return gw, err
+func (b *Bonafide) ListLocationFullness(transport string) map[string]float64 {
+	return b.gateways.listLocationFullness(transport)
 }
 
 func (b *Bonafide) SetManualGateway(label string) {
diff --git a/pkg/vpn/bonafide/gateways.go b/pkg/vpn/bonafide/gateways.go
index 2d94c63..5474a2f 100644
--- a/pkg/vpn/bonafide/gateways.go
+++ b/pkg/vpn/bonafide/gateways.go
@@ -34,7 +34,7 @@ type Gateway struct {
 	Protocols    []string
 	Options      map[string]string
 	Transport    string
-	Fullness     float32
+	Fullness     float64
 }
 
 /* gatewayDistance is used in the timezone distance fallback */
@@ -96,13 +96,13 @@ func (p *gatewayPool) isValidCity(city string) bool {
 	return valid
 }
 
-/* returns a map of city: gateway for the ui to use */
-func (p *gatewayPool) pickGatewayForCities(transport string) map[string]Gateway {
+/* returns a map of location: fullness for the ui to use */
+func (p *gatewayPool) listLocationFullness(transport string) map[string]float64 {
 	cities := p.getCities()
-	cm := make(map[string]Gateway)
+	cm := make(map[string]float64)
 	for _, city := range cities {
 		gw, _ := p.getRandomGatewayByCity(city, transport)
-		cm[city] = gw
+		cm[city] = gw.Fullness
 	}
 	return cm
 }