2 files changed, 38 insertions, 0 deletions
diff --git a/pkg/vpn/bonafide/bonafide.go b/pkg/vpn/bonafide/bonafide.go
index a9a7d85..cff5fc2 100644
--- a/pkg/vpn/bonafide/bonafide.go
+++ b/pkg/vpn/bonafide/bonafide.go
@@ -278,6 +278,10 @@ func (b *Bonafide) ListLocationFullness(transport string) map[string]float64 {
 	return b.gateways.listLocationFullness(transport)
 }
 
+func (b *Bonafide) ListLocationLabels(transport string) map[string][]string {
+	return b.gateways.listLocationLabels(transport)
+}
+
 func (b *Bonafide) SetManualGateway(label string) {
 	b.gateways.setUserChoice(label)
 }
@@ -286,6 +290,13 @@ func (b *Bonafide) SetAutomaticGateway() {
 	b.gateways.setAutomaticChoice()
 }
 
+func (b *Bonafide) GetBestLocation(transport string) string {
+	if b.gateways == nil {
+		return ""
+	}
+	return b.gateways.getBestLocation(transport, b.tzOffsetHours)
+}
+
 func (b *Bonafide) IsManualLocation() bool {
 	if b.gateways == nil {
 		return false
diff --git a/pkg/vpn/bonafide/gateways.go b/pkg/vpn/bonafide/gateways.go
index 53ab320..4299bb2 100644
--- a/pkg/vpn/bonafide/gateways.go
+++ b/pkg/vpn/bonafide/gateways.go
@@ -113,6 +113,20 @@ func (p *gatewayPool) listLocationFullness(transport string) map[string]float64
 	return cm
 }
 
+/* returns a map of location: labels for the ui to use */
+func (p *gatewayPool) listLocationLabels(transport string) map[string][]string {
+	cm := make(map[string][]string)
+	locations := p.getLocations()
+	if len(locations) == 0 {
+		return cm
+	}
+	for _, loc := range locations {
+		current := p.locations[loc]
+		cm[loc] = []string{current.Name, current.CountryCode}
+	}
+	return cm
+}
+
 /* this method should only be used if we have no usable menshen list. */
 func (p *gatewayPool) getRandomGatewaysByLocation(location, transport string) ([]Gateway, error) {
 	if !p.isValidLocation(location) {
@@ -274,6 +288,19 @@ func (p *gatewayPool) getBest(transport string, tz, max int) ([]Gateway, error)
 	}
 }
 
+/* returns the location for the first recommended gateway */
+func (p *gatewayPool) getBestLocation(transport string, tz int) string {
+	best, err := p.getBest(transport, tz, 1)
+	if err != nil {
+		return ""
+	}
+	if len(best) != 1 {
+		return ""
+	}
+	return best[0].Location
+
+}
+
 func (p *gatewayPool) getAll(transport string, tz int) ([]Gateway, error) {
 	if len(p.recommended) != 0 {
 		return p.getGatewaysFromMenshen(transport, 999)