package web import ( "crypto/rand" "encoding/json" "fmt" "io/ioutil" "net/http" "strconv" "strings" log "github.com/sirupsen/logrus" "github.com/umahmood/haversine" "github.com/librespeed/speedtest/config" "github.com/librespeed/speedtest/results" ) var ( serverCoord haversine.Coord ) 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) { if conf.ServerLat != 0 || conf.ServerLng != 0 { log.Infof("Configured server coordinates: %.6f, %.6f", conf.ServerLat, conf.ServerLng) serverCoord.Lat = conf.ServerLat serverCoord.Lon = conf.ServerLng return } var ret results.IPInfoResponse resp, err := http.DefaultClient.Get(getIPInfoURL("")) if err != nil { log.Errorf("Error getting repsonse from ipinfo.io: %s", err) return } raw, err := ioutil.ReadAll(resp.Body) if err != nil { log.Errorf("Error reading response from ipinfo.io: %s", err) return } defer resp.Body.Close() if err := json.Unmarshal(raw, &ret); err != nil { log.Errorf("Error parsing response from ipinfo.io: %s", err) return } if ret.Location != "" { serverCoord, err = parseLocationString(ret.Location) if err != nil { log.Errorf("Cannot get server coordinates: %s", err) return } } log.Infof("Fetched server coordinates: %.6f, %.6f", serverCoord.Lat, serverCoord.Lon) } func parseLocationString(location string) (haversine.Coord, error) { var coord haversine.Coord parts := strings.Split(location, ",") if len(parts) != 2 { err := fmt.Errorf("unknown location format: %s", location) log.Error(err) return coord, err } lat, err := strconv.ParseFloat(parts[0], 64) if err != nil { log.Errorf("Error parsing latitude: %s", parts[0]) return coord, err } lng, err := strconv.ParseFloat(parts[1], 64) if err != nil { log.Errorf("Error parsing longitude: %s", parts[0]) return coord, err } coord.Lat = lat coord.Lon = lng return coord, nil } func calculateDistance(clientLocation string, unit string) string { clientCoord, err := parseLocationString(clientLocation) if err != nil { log.Errorf("Error parsing client coordinates: %s", err) return "" } dist, km := haversine.Distance(clientCoord, serverCoord) unitString := " mi" switch unit { case "km": dist = km unitString = " km" case "NM": dist = km * 0.539957 unitString = " NM" } return fmt.Sprintf("%.2f%s", dist, unitString) }