48 lines
1004 B
Go
48 lines
1004 B
Go
package updater
|
|
|
|
import (
|
|
"net/netip"
|
|
|
|
"github.com/qdm12/gluetun/internal/constants/vpn"
|
|
"github.com/qdm12/gluetun/internal/models"
|
|
)
|
|
|
|
type cityToServer map[string]models.Server
|
|
|
|
func (cts cityToServer) add(city string, ips []netip.Addr) {
|
|
server, ok := cts[city]
|
|
if !ok {
|
|
server.VPN = vpn.OpenVPN
|
|
server.City = city
|
|
server.IPs = ips
|
|
server.TCP = true
|
|
server.UDP = true
|
|
} else {
|
|
// Do not insert duplicate IP addresses
|
|
existingIPs := make(map[string]struct{}, len(server.IPs))
|
|
for _, ip := range server.IPs {
|
|
existingIPs[ip.String()] = struct{}{}
|
|
}
|
|
|
|
for _, ip := range ips {
|
|
ipString := ip.String()
|
|
_, ok := existingIPs[ipString]
|
|
if ok {
|
|
continue
|
|
}
|
|
existingIPs[ipString] = struct{}{}
|
|
server.IPs = append(server.IPs, ip)
|
|
}
|
|
}
|
|
|
|
cts[city] = server
|
|
}
|
|
|
|
func (cts cityToServer) toServersSlice() (servers []models.Server) {
|
|
servers = make([]models.Server, 0, len(cts))
|
|
for _, server := range cts {
|
|
servers = append(servers, server)
|
|
}
|
|
return servers
|
|
}
|