Files
gluetun/internal/updater/updater.go
Quentin McGaw 7d824a5179 chore(settings): refactor settings processing (#756)
- Better settings tree structure logged using `qdm12/gotree`
- Read settings from environment variables, then files, then secret files
- Settings methods to default them, merge them and override them
- `DNS_PLAINTEXT_ADDRESS` default changed to `127.0.0.1` to use DoT. Warning added if set to something else.
- `HTTPPROXY_LISTENING_ADDRESS` instead of `HTTPPROXY_PORT` (with retro-compatibility)
2022-01-06 06:40:23 -05:00

119 lines
4.0 KiB
Go

// Package updater implements update mechanisms for each VPN provider servers.
package updater
import (
"context"
"net/http"
"strings"
"time"
"github.com/qdm12/gluetun/internal/configuration/settings"
"github.com/qdm12/gluetun/internal/constants"
"github.com/qdm12/gluetun/internal/models"
"github.com/qdm12/gluetun/internal/updater/resolver"
"github.com/qdm12/gluetun/internal/updater/unzip"
)
type Updater interface {
UpdateServers(ctx context.Context) (allServers models.AllServers, err error)
}
type updater struct {
// configuration
options settings.Updater
// state
servers models.AllServers
// Functions for tests
logger Logger
timeNow func() time.Time
presolver resolver.Parallel
client *http.Client
unzipper unzip.Unzipper
}
func New(settings settings.Updater, httpClient *http.Client,
currentServers models.AllServers, logger Logger) Updater {
unzipper := unzip.New(httpClient)
return &updater{
logger: logger,
timeNow: time.Now,
presolver: resolver.NewParallelResolver(settings.DNSAddress.String()),
client: httpClient,
unzipper: unzipper,
options: settings,
servers: currentServers,
}
}
type updateFunc func(ctx context.Context) (err error)
func (u *updater) UpdateServers(ctx context.Context) (allServers models.AllServers, err error) {
for _, provider := range u.options.Providers {
u.logger.Info("updating " + strings.Title(provider) + " servers...")
updateProvider := u.getUpdateFunction(provider)
// TODO support servers offering only TCP or only UDP
// for NordVPN and PureVPN
err = updateProvider(ctx)
if err != nil {
if ctxErr := ctx.Err(); ctxErr != nil {
return allServers, ctxErr
}
u.logger.Error(err.Error())
}
}
return u.servers, nil
}
func (u *updater) getUpdateFunction(provider string) (updateFunction updateFunc) {
switch provider {
case constants.Custom:
panic("cannot update custom provider")
case constants.Cyberghost:
return func(ctx context.Context) (err error) { return u.updateCyberghost(ctx) }
case constants.Expressvpn:
return func(ctx context.Context) (err error) { return u.updateExpressvpn(ctx) }
case constants.Fastestvpn:
return func(ctx context.Context) (err error) { return u.updateFastestvpn(ctx) }
case constants.HideMyAss:
return func(ctx context.Context) (err error) { return u.updateHideMyAss(ctx) }
case constants.Ipvanish:
return func(ctx context.Context) (err error) { return u.updateIpvanish(ctx) }
case constants.Ivpn:
return func(ctx context.Context) (err error) { return u.updateIvpn(ctx) }
case constants.Mullvad:
return func(ctx context.Context) (err error) { return u.updateMullvad(ctx) }
case constants.Nordvpn:
return func(ctx context.Context) (err error) { return u.updateNordvpn(ctx) }
case constants.Perfectprivacy:
return func(ctx context.Context) (err error) { return u.updatePerfectprivacy(ctx) }
case constants.Privado:
return func(ctx context.Context) (err error) { return u.updatePrivado(ctx) }
case constants.PrivateInternetAccess:
return func(ctx context.Context) (err error) { return u.updatePIA(ctx) }
case constants.Privatevpn:
return func(ctx context.Context) (err error) { return u.updatePrivatevpn(ctx) }
case constants.Protonvpn:
return func(ctx context.Context) (err error) { return u.updateProtonvpn(ctx) }
case constants.Purevpn:
return func(ctx context.Context) (err error) { return u.updatePurevpn(ctx) }
case constants.Surfshark:
return func(ctx context.Context) (err error) { return u.updateSurfshark(ctx) }
case constants.Torguard:
return func(ctx context.Context) (err error) { return u.updateTorguard(ctx) }
case constants.VPNUnlimited:
return func(ctx context.Context) (err error) { return u.updateVPNUnlimited(ctx) }
case constants.Vyprvpn:
return func(ctx context.Context) (err error) { return u.updateVyprvpn(ctx) }
case constants.Wevpn:
return func(ctx context.Context) (err error) { return u.updateWevpn(ctx) }
case constants.Windscribe:
return func(ctx context.Context) (err error) { return u.updateWindscribe(ctx) }
default:
panic("provider " + provider + " is unknown")
}
}