add support for selecting a country via CLI flag

.github/workflows/go.yml

@@ -9,6 +9,8 @@ jobs:
     runs-on: ubuntu-latest
     steps:
     - uses: actions/checkout@v2
+      with:
+        fetch-depth: 0
 
     - name: Set up Go
       uses: actions/setup-go@v2
@@ -22,7 +24,6 @@ jobs:
       run: |
         git tag "v$(git show -s --format=%cd --date=format:%Y%m%d.%H%M%S)"
         git push --tags
-        sudo mv snyk-linux* /usr/local/bin/
 
     - name: Release binaries with GoReleaser
       uses: goreleaser/goreleaser-action@v2

README.md

@@ -0,0 +1,22 @@
+# mullvad-best-server
+![Build](https://github.com/bastiandoetsch/mullvad-best-server/actions/workflows/go.yml/badge.svg)
+
+Determines the mullvat.net wireguard server with the lowest latency. 
+
+## Installation
+
+Download binary from releases for your platform and unpack.
+
+## Usage
+### Default usage
+Execute `mullvad-best-server`. It outputs the code, e.g. `de05`. You can then connect to it with e.g. wireguard using the normal shell scripts.
+
+### Extended Output
+If you want the full server information, execute `mullvad-best-server -o json`. It returns the full json output  of the server information.
+
+### Specify Country
+The -c flag allows to give a country code. Else de/at/ch will be used.
+
+## Background
+The program uses `https://api.mullvad.net/www/relays/wireguard/` to get the current server list, pings the ones with the right country
+and outputs the server with the lowest ping.

main.go

@@ -2,36 +2,58 @@ package main
 
 import (
 	"encoding/json"
+	"flag"
+	"fmt"
 	"github.com/go-ping/ping"
+	"github.com/rs/zerolog"
 	"github.com/rs/zerolog/log"
 	"io"
 	"io/ioutil"
 	"net/http"
+	"strings"
 	"time"
 )
 
 var pings = make(map[string]time.Duration)
 
 func main() {
+	zerolog.SetGlobalLevel(zerolog.InfoLevel)
+	var outputFlag = flag.String("o", "short", "Output format. 'json' outputs server json")
+	var countryFlag = flag.String("c", "ch", "Server country code, e.g. ch for Switzerland")
+	flag.Parse()
 
 	servers := getServers()
-	bestIndex := selectBestServerIndex(servers)
+	bestIndex := selectBestServerIndex(servers, countryFlag)
-	log.Info().Interface("best", servers[bestIndex]).Msg("Best Latency Server found.")
+	log.Debug().Interface("server", servers[bestIndex]).Msg("Best latency server found.")
+	hostname := strings.Split(servers[bestIndex].Hostname, "-")[0]
+	if *outputFlag != "json" {
+		fmt.Println(hostname)
+	} else {
+		serverJson, err := json.Marshal(servers[bestIndex])
+		if err != nil {
+			log.Fatal().Err(err)
+		}
+		fmt.Println(string(serverJson))
+	}
 }
 
-func selectBestServerIndex(servers []server) int {
+func selectBestServerIndex(servers []server, country *string) int {
 	best := servers[0].Hostname
-	bestIndex := 0
+	bestIndex := -1
 	allowedCountries := map[string]string{}
+	if *country == "" {
 		allowedCountries["de"] = "1"
 		allowedCountries["ch"] = "1"
 		allowedCountries["at"] = "1"
+	} else {
+		allowedCountries[*country] = "1"
+	}
 	for i, server := range servers {
 		if server.Active && allowedCountries[server.CountryCode] != "" {
 			duration, err := serverLatency(server)
 			if err == nil {
 				pings[server.Hostname] = duration
-				if best == "" || pings[best] > pings[server.Hostname] {
+				if bestIndex == -1 || pings[best] > pings[server.Hostname] {
 					best = server.Hostname
 					bestIndex = i
 				}
@@ -69,7 +91,7 @@ func serverLatency(s server) (time.Duration, error) {
 	}
 	var duration time.Duration
 	pinger.OnRecv = func(pkt *ping.Packet) {
-		log.Info().Str("Server", s.Hostname).IPAddr("IP", pkt.IPAddr.IP).Dur("RTT", pkt.Rtt).Msg("Added server latency.")
+		log.Debug().Str("Server", s.Hostname).IPAddr("IP", pkt.IPAddr.IP).Dur("RTT", pkt.Rtt).Msg("Added server latency.")
 		duration = pkt.Rtt
 	}
 	err = pinger.Run()