Assimilate PIA v4 as PIA in code

internal/updater/pia.go

@@ -0,0 +1,97 @@
+package updater
+
+import (
+	"bytes"
+	"context"
+	"encoding/json"
+	"fmt"
+	"net"
+	"net/http"
+	"sort"
+	"strings"
+
+	"github.com/qdm12/gluetun/internal/models"
+)
+
+func (u *updater) updatePIA(ctx context.Context) (err error) {
+	const url = "https://serverlist.piaservers.net/vpninfo/servers/v4"
+	b, status, err := u.client.Get(ctx, url)
+	if err != nil {
+		return err
+	}
+	if status != http.StatusOK {
+		return fmt.Errorf("HTTP status code %d: %s", status, strings.ReplaceAll(string(b), "\n", ""))
+	}
+
+	// remove key/signature at the bottom
+	i := bytes.IndexRune(b, '\n')
+	b = b[:i]
+
+	var data struct {
+		Regions []struct {
+			Name        string `json:"name"`
+			PortForward bool   `json:"port_forward"`
+			Servers     struct {
+				UDP []struct {
+					IP net.IP `json:"ip"`
+					CN string `json:"cn"`
+				} `json:"ovpnudp"`
+				TCP []struct {
+					IP net.IP `json:"ip"`
+					CN string `json:"cn"`
+				} `json:"ovpntcp"`
+			} `json:"servers"`
+		} `json:"regions"`
+	}
+	if err := json.Unmarshal(b, &data); err != nil {
+		return err
+	}
+	servers := make([]models.PIAServer, 0, len(data.Regions))
+	for _, region := range data.Regions {
+		server := models.PIAServer{
+			Region:      region.Name,
+			PortForward: region.PortForward,
+		}
+		for _, udpServer := range region.Servers.UDP {
+			if len(server.OpenvpnUDP.CN) > 0 && server.OpenvpnUDP.CN != udpServer.CN {
+				return fmt.Errorf("CN is different for UDP for region %q: %q and %q",
+					region.Name, server.OpenvpnUDP.CN, udpServer.CN)
+			}
+			if udpServer.IP != nil {
+				server.OpenvpnUDP.IPs = append(server.OpenvpnUDP.IPs, udpServer.IP)
+			}
+			server.OpenvpnUDP.CN = udpServer.CN
+		}
+		for _, tcpServer := range region.Servers.TCP {
+			if len(server.OpenvpnTCP.CN) > 0 && server.OpenvpnTCP.CN != tcpServer.CN {
+				return fmt.Errorf("CN is different for TCP for region %q: %q and %q",
+					region.Name, server.OpenvpnTCP.CN, tcpServer.CN)
+			}
+			if tcpServer.IP != nil {
+				server.OpenvpnTCP.IPs = append(server.OpenvpnTCP.IPs, tcpServer.IP)
+			}
+			server.OpenvpnTCP.CN = tcpServer.CN
+		}
+		servers = append(servers, server)
+	}
+	sort.Slice(servers, func(i, j int) bool {
+		return servers[i].Region < servers[j].Region
+	})
+	if u.options.Stdout {
+		u.println(stringifyPIAServers(servers))
+	}
+	u.servers.Pia.Timestamp = u.timeNow().Unix()
+	u.servers.Pia.Servers = servers
+	return nil
+}
+
+func stringifyPIAServers(servers []models.PIAServer) (s string) {
+	s = "func PIAServers() []models.PIAServer {\n"
+	s += "	return []models.PIAServer{\n"
+	for _, server := range servers {
+		s += "		" + server.String() + ",\n"
+	}
+	s += "	}\n"
+	s += "}"
+	return s
+}