162 lines
3.5 KiB
Go
162 lines
3.5 KiB
Go
package web
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"math"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/librespeed/speedtest/config"
|
|
"github.com/librespeed/speedtest/results"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
var (
|
|
serverLat, serverLng float64
|
|
)
|
|
|
|
func getRandomData(length int) []byte {
|
|
data := make([]byte, length)
|
|
if _, err := rand.Read(data); err != nil {
|
|
log.Fatalf("Failed to generate random data: %s", err)
|
|
}
|
|
return data
|
|
}
|
|
|
|
func getIPInfoURL(address string) string {
|
|
apiKey := config.LoadedConfig().IPInfoAPIKey
|
|
|
|
ipInfoURL := `https://ipinfo.io/%s/json`
|
|
if address != "" {
|
|
ipInfoURL = fmt.Sprintf(ipInfoURL, address)
|
|
} else {
|
|
ipInfoURL = "https://ipinfo.io/json"
|
|
}
|
|
|
|
if apiKey != "" {
|
|
ipInfoURL += "?token=" + apiKey
|
|
}
|
|
|
|
return ipInfoURL
|
|
}
|
|
|
|
func getIPInfo(addr string) results.IPInfoResponse {
|
|
var ret results.IPInfoResponse
|
|
resp, err := http.DefaultClient.Get(getIPInfoURL(addr))
|
|
if err != nil {
|
|
log.Errorf("Error getting response from ipinfo.io: %s", err)
|
|
return ret
|
|
}
|
|
|
|
raw, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
log.Errorf("Error reading response from ipinfo.io: %s", err)
|
|
return ret
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if err := json.Unmarshal(raw, &ret); err != nil {
|
|
log.Errorf("Error parsing response from ipinfo.io: %s", err)
|
|
}
|
|
|
|
return ret
|
|
}
|
|
|
|
func SetServerLocation(conf *config.Config) (float64, float64) {
|
|
if conf.ServerLat > 0 && conf.ServerLng > 0 {
|
|
log.Infof("Configured server coordinates: %.6f, %.6f", conf.ServerLat, conf.ServerLng)
|
|
return conf.ServerLat, conf.ServerLng
|
|
}
|
|
|
|
var ret results.IPInfoResponse
|
|
resp, err := http.DefaultClient.Get(getIPInfoURL(""))
|
|
if err != nil {
|
|
log.Errorf("Error getting repsonse from ipinfo.io: %s", err)
|
|
return 0, 0
|
|
}
|
|
raw, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
log.Errorf("Error reading response from ipinfo.io: %s", err)
|
|
return 0, 0
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if err := json.Unmarshal(raw, &ret); err != nil {
|
|
log.Errorf("Error parsing response from ipinfo.io: %s", err)
|
|
return 0, 0
|
|
}
|
|
|
|
var lat, lng float64
|
|
if ret.Location != "" {
|
|
lat, lng = parseLocationString(ret.Location)
|
|
}
|
|
|
|
log.Infof("Fetched server coordinates: %.6f, %.6f", lat, lng)
|
|
|
|
return lat, lng
|
|
}
|
|
|
|
func parseLocationString(location string) (float64, float64) {
|
|
parts := strings.Split(location, ",")
|
|
if len(parts) != 2 {
|
|
log.Errorf("Unknown location format: %s", location)
|
|
return 0, 0
|
|
}
|
|
|
|
lat, err := strconv.ParseFloat(parts[0], 64)
|
|
if err != nil {
|
|
log.Errorf("Error parsing latitude: %s", parts[0])
|
|
return 0, 0
|
|
}
|
|
|
|
lng, err := strconv.ParseFloat(parts[1], 64)
|
|
if err != nil {
|
|
log.Errorf("Error parsing longitude: %s", parts[0])
|
|
return 0, 0
|
|
}
|
|
|
|
return lat, lng
|
|
}
|
|
|
|
func calculateDistance(clientLocation string, unit string) string {
|
|
clientLat, clientLng := parseLocationString(clientLocation)
|
|
|
|
radlat1 := float64(math.Pi * serverLat / 180)
|
|
radlat2 := float64(math.Pi * clientLat / 180)
|
|
|
|
theta := float64(serverLng - clientLng)
|
|
radtheta := float64(math.Pi * theta / 180)
|
|
|
|
dist := math.Sin(radlat1)*math.Sin(radlat2) + math.Cos(radlat1)*math.Cos(radlat2)*math.Cos(radtheta)
|
|
|
|
if dist > 1 {
|
|
dist = 1
|
|
}
|
|
|
|
dist = math.Acos(dist)
|
|
dist = dist * 180 / math.Pi
|
|
dist = dist * 60 * 1.1515
|
|
|
|
unitString := " mi"
|
|
switch unit {
|
|
case "km":
|
|
dist = dist * 1.609344
|
|
unitString = " km"
|
|
case "NM":
|
|
dist = dist * 0.8684
|
|
unitString = " NM"
|
|
}
|
|
|
|
return fmt.Sprintf("%d%s", round(dist), unitString)
|
|
}
|
|
|
|
func round(v float64) int {
|
|
r := int(math.Round(v))
|
|
return 10 * ((r + 9) / 10)
|
|
}
|