chore(updater): move updater packages to pkg/updaters/<name>

pkg/updaters/fastestvpn/hosttoserver.go

@@ -0,0 +1,93 @@
+package fastestvpn
+
+import (
+	"net/netip"
+
+	"github.com/qdm12/gluetun/internal/constants/vpn"
+	"github.com/qdm12/gluetun/internal/models"
+)
+
+type hostToServerData map[string]serverData
+
+type serverData struct {
+	openvpn    bool
+	wireguard  bool
+	country    string
+	city       string
+	openvpnUDP bool
+	openvpnTCP bool
+	ips        []netip.Addr
+}
+
+func (hts hostToServerData) add(host, vpnType, country, city string, tcp, udp bool) {
+	serverData, ok := hts[host]
+	switch vpnType {
+	case vpn.OpenVPN:
+		serverData.openvpn = true
+		serverData.openvpnTCP = serverData.openvpnTCP || tcp
+		serverData.openvpnUDP = serverData.openvpnUDP || udp
+	case vpn.Wireguard:
+		serverData.wireguard = true
+	default:
+		panic("protocol not supported")
+	}
+
+	if !ok {
+		serverData.country = country
+		serverData.city = city
+	} else if city != "" {
+		// some servers are listed without the city although
+		// they are also listed with the city described, so update
+		// the city field.
+		serverData.city = city
+	}
+
+	hts[host] = serverData
+}
+
+func (hts hostToServerData) toHostsSlice() (hosts []string) {
+	hosts = make([]string, 0, len(hts))
+	for host := range hts {
+		hosts = append(hosts, host)
+	}
+	return hosts
+}
+
+func (hts hostToServerData) adaptWithIPs(hostToIPs map[string][]netip.Addr) {
+	for host, serverData := range hts {
+		ips := hostToIPs[host]
+		if len(ips) == 0 {
+			delete(hts, host)
+			continue
+		}
+		serverData.ips = ips
+		hts[host] = serverData
+	}
+}
+
+func (hts hostToServerData) toServersSlice() (servers []models.Server) {
+	servers = make([]models.Server, 0, 2*len(hts)) //nolint:mnd
+	for hostname, serverData := range hts {
+		baseServer := models.Server{
+			Hostname: hostname,
+			Country:  serverData.country,
+			City:     serverData.city,
+			IPs:      serverData.ips,
+		}
+		if serverData.openvpn {
+			openvpnServer := baseServer
+			openvpnServer.VPN = vpn.OpenVPN
+			openvpnServer.TCP = serverData.openvpnTCP
+			openvpnServer.UDP = serverData.openvpnUDP
+			servers = append(servers, openvpnServer)
+		}
+		if serverData.wireguard {
+			wireguardServer := baseServer
+			wireguardServer.VPN = vpn.Wireguard
+			const wireguardPublicKey = "658QxufMbjOTmB61Z7f+c7Rjg7oqWLnepTalqBERjF0="
+			wireguardServer.WgPubKey = wireguardPublicKey
+			servers = append(servers, wireguardServer)
+		}
+	}
+	return servers
+}