chore(surfshark): add package internal/provider/surshark/server

- Merge `internal/models/location.go` and `internal/constants/surfshark.go` into `internal/provider/surfshark/servers/locationdata.go`
This commit is contained in:
Quentin McGaw
2022-05-07 19:12:29 +00:00
parent 306de8feda
commit 1b2bcf901a
8 changed files with 30 additions and 36 deletions

View File

@@ -3,15 +3,14 @@ package settings
import (
"strings"
"github.com/qdm12/gluetun/internal/constants"
"github.com/qdm12/gluetun/internal/models"
"github.com/qdm12/gluetun/internal/provider/surfshark/servers"
)
func surfsharkRetroRegion(selection ServerSelection) (
updatedSelection ServerSelection) {
locationData := constants.SurfsharkLocationData()
locationData := servers.LocationData()
retroToLocation := make(map[string]models.SurfsharkLocationData, len(locationData))
retroToLocation := make(map[string]servers.ServerLocation, len(locationData))
for _, data := range locationData {
if data.RetroLoc == "" {
continue

View File

@@ -1,12 +1,12 @@
package validation
import (
"github.com/qdm12/gluetun/internal/constants"
"github.com/qdm12/gluetun/internal/provider/surfshark/servers"
)
// TODO remove in v4.
func SurfsharkRetroLocChoices() (choices []string) {
locationData := constants.SurfsharkLocationData()
locationData := servers.LocationData()
choices = make([]string, 0, len(locationData))
seen := make(map[string]struct{}, len(locationData))
for _, data := range locationData {

View File

@@ -1,12 +0,0 @@
package models
// SurfsharkLocationData is required to keep location data on Surfshark
// servers that are not obtained through their API.
type SurfsharkLocationData struct {
Region string
Country string
City string
RetroLoc string // TODO remove in v4
Hostname string
MultiHop bool
}

View File

@@ -1,13 +1,20 @@
package constants
package servers
import (
"github.com/qdm12/gluetun/internal/models"
)
// LocationData is required to keep location data on Surfshark
// servers that are not obtained through their API.
type ServerLocation struct {
Region string
Country string
City string
RetroLoc string // TODO remove in v4
Hostname string
MultiHop bool
}
// TODO remove retroRegion and servers from API in v4.
func SurfsharkLocationData() (data []models.SurfsharkLocationData) {
func LocationData() (data []ServerLocation) {
//nolint:lll
return []models.SurfsharkLocationData{
return []ServerLocation{
{Region: "Asia Pacific", Country: "Australia", City: "Adelaide", RetroLoc: "Australia Adelaide", Hostname: "au-adl.prod.surfshark.com", MultiHop: false},
{Region: "Asia Pacific", Country: "Australia", City: "Brisbane", RetroLoc: "Australia Brisbane", Hostname: "au-bne.prod.surfshark.com", MultiHop: false},
{Region: "Asia Pacific", Country: "Australia", City: "Melbourne", RetroLoc: "Australia Melbourne", Hostname: "au-mel.prod.surfshark.com", MultiHop: false},

View File

@@ -7,7 +7,7 @@ import (
"fmt"
"net/http"
"github.com/qdm12/gluetun/internal/constants"
"github.com/qdm12/gluetun/internal/provider/surfshark/servers"
)
// Note: no multi-hop and some servers are missing from their API.
@@ -18,7 +18,7 @@ func addServersFromAPI(ctx context.Context, client *http.Client,
return err
}
locationData := constants.SurfsharkLocationData()
locationData := servers.LocationData()
hostToLocation := hostToLocation(locationData)
const tcp, udp = true, true

View File

@@ -4,15 +4,15 @@ import (
"errors"
"fmt"
"github.com/qdm12/gluetun/internal/models"
"github.com/qdm12/gluetun/internal/provider/surfshark/servers"
)
var (
errHostnameNotFound = errors.New("hostname not found in hostname to location mapping")
)
func getHostInformation(host string, hostnameToLocation map[string]models.SurfsharkLocationData) (
data models.SurfsharkLocationData, err error) {
func getHostInformation(host string, hostnameToLocation map[string]servers.ServerLocation) (
data servers.ServerLocation, err error) {
locationData, ok := hostnameToLocation[host]
if !ok {
return locationData, fmt.Errorf("%w: %s", errHostnameNotFound, host)
@@ -21,9 +21,9 @@ func getHostInformation(host string, hostnameToLocation map[string]models.Surfsh
return locationData, nil
}
func hostToLocation(locationData []models.SurfsharkLocationData) (
hostToLocation map[string]models.SurfsharkLocationData) {
hostToLocation = make(map[string]models.SurfsharkLocationData, len(locationData))
func hostToLocation(locationData []servers.ServerLocation) (
hostToLocation map[string]servers.ServerLocation) {
hostToLocation = make(map[string]servers.ServerLocation, len(locationData))
for _, data := range locationData {
hostToLocation[data.Hostname] = data
}

View File

@@ -1,12 +1,12 @@
package surfshark
import (
"github.com/qdm12/gluetun/internal/constants"
"github.com/qdm12/gluetun/internal/provider/surfshark/servers"
)
// getRemainingServers finds extra servers not found in the API or in the ZIP file.
func getRemainingServers(hts hostToServer) {
locationData := constants.SurfsharkLocationData()
locationData := servers.LocationData()
hostnameToLocationLeft := hostToLocation(locationData)
for _, hostnameDone := range hts.toHostsSlice() {
delete(hostnameToLocationLeft, hostnameDone)

View File

@@ -4,7 +4,7 @@ import (
"context"
"strings"
"github.com/qdm12/gluetun/internal/constants"
"github.com/qdm12/gluetun/internal/provider/surfshark/servers"
"github.com/qdm12/gluetun/internal/updater/openvpn"
"github.com/qdm12/gluetun/internal/updater/unzip"
)
@@ -24,7 +24,7 @@ func addOpenVPNServersFromZip(ctx context.Context,
hostnamesDoneSet[hostname] = struct{}{}
}
locationData := constants.SurfsharkLocationData()
locationData := servers.LocationData()
hostToLocation := hostToLocation(locationData)
for fileName, content := range contents {