fix typo in README

.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

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2022 Bastian Doetsch
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,13 +1,32 @@
 # mullvad-best-server
-Determines the mullvat.net server with the lowest latency. 
+![Build](https://github.com/bastiandoetsch/mullvad-best-server/actions/workflows/go.yml/badge.svg)
+
+Determines the mullvad.net 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.
 
+### Command line parameters
+
+```angular2html
+Usage of dist/mullvad-best-server_darwin_amd64/mullvad-best-server:
+-c string
+  Server country code, e.g. ch for Switzerland (default "ch")
+-o string
+  Output format. 'json' outputs server json
+-t string
+  Server type, e.g. wireguard (default "wireguard")
+```
+
+If you want the full server information, execute `mullvad-best-server -o json`. It returns the full json output  of the server information.
+The `-c` flag allows to give a country code. Else `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
+The program uses `https://api.mullvad.net/www/relays/<SERVER_TYPE/` to get the current server list, pings the ones with the right country
 and outputs the server with the lowest ping.

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"encoding/json"
+	"flag"
 	"fmt"
 	"github.com/go-ping/ping"
 	"github.com/rs/zerolog"
@@ -17,26 +18,36 @@ var pings = make(map[string]time.Duration)
 
 func main() {
 	zerolog.SetGlobalLevel(zerolog.InfoLevel)
-	servers := getServers()
+	var outputFlag = flag.String("o", "", "Output format. 'json' outputs server json")
-	bestIndex := selectBestServerIndex(servers)
+	var countryFlag = flag.String("c", "ch", "Server country code, e.g. ch for Switzerland")
-	log.Debug().Interface("server", servers[bestIndex]).Msg("Best latency server found.")
+	var typeFlag = flag.String("t", "wireguard", "Server type, e.g. wireguard")
-	hostname := strings.Split(servers[bestIndex].Hostname, "-")[0]
+	flag.Parse()
-	fmt.Println(hostname)
+
+	servers := getServers(*typeFlag)
+	bestIndex := selectBestServerIndex(servers, *countryFlag)
+	best := servers[bestIndex]
+	log.Debug().Interface("server", best).Msg("Best latency server found.")
+	hostname := strings.TrimSuffix(best.Hostname, "-wireguard")
+	if *outputFlag != "json" {
+		fmt.Println(hostname)
+	} else {
+		serverJson, err := json.Marshal(best)
+		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{}
-	allowedCountries["de"] = "1"
-	allowedCountries["ch"] = "1"
-	allowedCountries["at"] = "1"
 	for i, server := range servers {
-		if server.Active && allowedCountries[server.CountryCode] != "" {
+		if server.Active && server.CountryCode == country {
 			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
 				}
@@ -46,8 +57,8 @@ func selectBestServerIndex(servers []server) int {
 	return bestIndex
 }
 
-func getServers() []server {
+func getServers(serverType string) []server {
-	resp, err := http.Get("https://api.mullvad.net/www/relays/wireguard/")
+	resp, err := http.Get("https://api.mullvad.net/www/relays/" + serverType + "/")
 	if err != nil {
 		log.Fatal().Err(err)
 	}