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

pkg/updaters/privateinternetaccess/servers.go

@@ -0,0 +1,99 @@
+package privateinternetaccess
+
+import (
+	"context"
+	"fmt"
+	"sort"
+	"time"
+
+	"github.com/qdm12/gluetun/internal/models"
+	"github.com/qdm12/gluetun/internal/provider/common"
+)
+
+func (u *Updater) FetchServers(ctx context.Context, minServers int) (
+	servers []models.Server, err error,
+) {
+	nts := make(nameToServer)
+
+	noChangeCounter := 0
+	const maxNoChange = 10
+	const betweenDuration = 200 * time.Millisecond
+	const maxDuration = time.Minute
+
+	maxTimer := time.NewTimer(maxDuration)
+
+	for {
+		data, err := fetchAPI(ctx, u.client)
+		if err != nil {
+			return nil, err
+		}
+
+		change := addData(data.Regions, nts)
+
+		if !change {
+			noChangeCounter++
+			if noChangeCounter == maxNoChange {
+				break
+			}
+		} else {
+			noChangeCounter = 0
+		}
+
+		timer := time.NewTimer(betweenDuration)
+		maxTimeout := false
+		select {
+		case <-ctx.Done():
+			if !timer.Stop() {
+				<-timer.C
+			}
+			if !maxTimer.Stop() {
+				<-timer.C
+			}
+			return nil, ctx.Err()
+		case <-timer.C:
+		case <-maxTimer.C:
+			if !timer.Stop() {
+				<-timer.C
+			}
+			maxTimeout = true
+		}
+
+		if maxTimeout {
+			break
+		}
+	}
+
+	servers = nts.toServersSlice()
+
+	if len(servers) < minServers {
+		return nil, fmt.Errorf("%w: %d and expected at least %d",
+			common.ErrNotEnoughServers, len(servers), minServers)
+	}
+
+	sort.Sort(models.SortableServers(servers))
+
+	return servers, nil
+}
+
+func addData(regions []regionData, nts nameToServer) (change bool) {
+	for _, region := range regions {
+		if region.Offline {
+			continue
+		}
+		for _, server := range region.Servers.UDP {
+			const tcp, udp = false, true
+			if nts.add(server.CN, region.DNS, region.Name, tcp, udp, region.PortForward, server.IP) {
+				change = true
+			}
+		}
+
+		for _, server := range region.Servers.TCP {
+			const tcp, udp = true, false
+			if nts.add(server.CN, region.DNS, region.Name, tcp, udp, region.PortForward, server.IP) {
+				change = true
+			}
+		}
+	}
+
+	return change
+}