[feat] restrict locations to selected transport

2 files changed, 4 insertions, 0 deletions

diff --git a/pkg/backend/api.go b/pkg/backend/api.go
index 53800b3..96f3072 100644
--- a/pkg/backend/api.go
+++ b/pkg/backend/api.go
@@ -81,6 +81,7 @@ func UseAutomaticGateway() {
 
 func SetTransport(label string) {
 	ctx.bm.SetTransport(label)
+	go trigger(OnStatusChanged)
 }
 
 func GetTransport() *C.char {
diff --git a/pkg/vpn/bonafide/gateways.go b/pkg/vpn/bonafide/gateways.go
index a0d41fb..53ab320 100644
--- a/pkg/vpn/bonafide/gateways.go
+++ b/pkg/vpn/bonafide/gateways.go
@@ -97,6 +97,9 @@ func (p *gatewayPool) listLocationFullness(transport string) map[string]float64
 	}
 	if len(p.recommended) != 0 {
 		for _, gw := range p.recommended {
+			if gw.gateway.Transport != transport {
+				continue
+			}
 			if _, ok := cm[gw.gateway.Location]; ok {
 				continue
 			}