gluetun/internal/configuration/settings/validation/servers.go

package validation

import (
	"sort"

	"github.com/qdm12/gluetun-servers/pkg/models"
)

func sortedInsert(ss []string, s string) []string {
	i := sort.SearchStrings(ss, s)
	ss = append(ss, "")
	copy(ss[i+1:], ss[i:])
	ss[i] = s
	return ss
}

func ExtractCountries(servers []models.Server) (values []string) {
	seen := make(map[string]struct{}, len(servers))
	values = make([]string, 0, len(servers))
	for _, server := range servers {
		value := server.Country
		if value == "" {
			continue
		}
		_, alreadySeen := seen[value]
		if alreadySeen {
			continue
		}
		seen[value] = struct{}{}

		values = sortedInsert(values, value)
	}
	return values
}

func ExtractCategories(servers []models.Server) (values []string) {
	seen := make(map[string]struct{}, len(servers))
	values = make([]string, 0, len(servers))
	for _, server := range servers {
		categories := server.Categories
		if len(categories) == 0 {
			continue
		}

		for _, value := range categories {
			_, alreadySeen := seen[value]
			if alreadySeen {
				continue
			}
			seen[value] = struct{}{}

			values = sortedInsert(values, value)
		}
	}
	return values
}

func ExtractRegions(servers []models.Server) (values []string) {
	seen := make(map[string]struct{}, len(servers))
	values = make([]string, 0, len(servers))
	for _, server := range servers {
		value := server.Region
		if value == "" {
			continue
		}
		_, alreadySeen := seen[value]
		if alreadySeen {
			continue
		}
		seen[value] = struct{}{}

		values = sortedInsert(values, value)
	}
	return values
}

func ExtractCities(servers []models.Server) (values []string) {
	seen := make(map[string]struct{}, len(servers))
	values = make([]string, 0, len(servers))
	for _, server := range servers {
		value := server.City
		if value == "" {
			continue
		}
		_, alreadySeen := seen[value]
		if alreadySeen {
			continue
		}
		seen[value] = struct{}{}

		values = sortedInsert(values, value)
	}
	return values
}

func ExtractISPs(servers []models.Server) (values []string) {
	seen := make(map[string]struct{}, len(servers))
	values = make([]string, 0, len(servers))
	for _, server := range servers {
		value := server.ISP
		if value == "" {
			continue
		}
		_, alreadySeen := seen[value]
		if alreadySeen {
			continue
		}
		seen[value] = struct{}{}

		values = sortedInsert(values, value)
	}
	return values
}

func ExtractServerNames(servers []models.Server) (values []string) {
	seen := make(map[string]struct{}, len(servers))
	values = make([]string, 0, len(servers))
	for _, server := range servers {
		value := server.ServerName
		if value == "" {
			continue
		}
		_, alreadySeen := seen[value]
		if alreadySeen {
			continue
		}
		seen[value] = struct{}{}

		values = sortedInsert(values, value)
	}
	return values
}

func ExtractHostnames(servers []models.Server) (values []string) {
	seen := make(map[string]struct{}, len(servers))
	values = make([]string, 0, len(servers))
	for _, server := range servers {
		value := server.Hostname
		if value == "" {
			continue
		}
		_, alreadySeen := seen[value]
		if alreadySeen {
			continue
		}
		seen[value] = struct{}{}

		values = sortedInsert(values, value)
	}
	return values
}