Re-organisation. Added file utility to calculate upload/download slot count by connection speed (we should probably add a config value in the file server to override this, but it's good for people who are lazy)

files/files.go

@@ -0,0 +1,48 @@
+package files
+
+import (
+	"github.com/showwin/speedtest-go/speedtest"
+)
+
+type ConnectionStatistics struct {
+	downloadBytes uint64 // Server download speed in bytes
+	uploadBytes   uint64 // Upload speed in bytes
+	downloadSlots uint16 // How fany files can be downloaded by the server at once before queueing kicks in
+	uploadSlots   uint16 // How many uploads we can do at once
+
+	// Slots are calculated with a minimum acceptable speed of 128KiB/s per slot
+	// Using uint16 for this MIGHT get a bit munted if someone tries to use something faster than a 25gbit connection
+	// or something but I really, really doubt anyone is going to do that. Problem for later me if it turns into an issue
+}
+
+func CalculateConnectionStatistics() ConnectionStatistics {
+	var speedtestClient = speedtest.New()
+	serverList, _ := speedtestClient.FetchServers()
+	// Picks server with the lowest ping
+	targets, _ := serverList.FindServer([]int{})
+	for _, s := range targets {
+		s.PingTest(nil)
+		s.DownloadTest()
+		s.UploadTest()
+		connStats := ConnectionStatistics{
+			downloadBytes: uint64(s.DLSpeed), // Casting since speedtest-go dev implemented this as float64
+			uploadBytes:   uint64(s.ULSpeed),
+			downloadSlots: uint16(uint64(s.DLSpeed) / 131072), // 128KiB per slot
+			uploadSlots:   uint16(uint64(s.ULSpeed) / 131072), // Casting to avoid float
+		}
+
+		if connStats.downloadSlots > 1000 {
+			connStats.downloadSlots = 1000 // Max slot count should be 1000 to avoid DDoS-like behaviour
+		}
+		if connStats.uploadSlots > 1000 {
+			connStats.uploadSlots = 1000
+		}
+		s.Context.Reset()
+		return connStats
+
+	}
+	// If all else fails... fuckin' yolo it
+	connStats := ConnectionStatistics{0, 0, 1000, 1000}
+	return connStats
+
+}

files/files_test.go

@@ -0,0 +1,13 @@
+package files
+
+import (
+	"fmt"
+	"strconv"
+	"testing"
+)
+
+func TestCalculateConnectionStatistics(t *testing.T) {
+	connStats := CalculateConnectionStatistics()
+	fmt.Printf(" Download: %s bytes/s\n Upload: %s bytes/s\n Estimated Client Slots: %s\n Estimated Internal Slots: %s\n", strconv.FormatUint(connStats.downloadBytes, 10), strconv.FormatUint(connStats.uploadBytes, 10), strconv.Itoa(int(connStats.uploadSlots)), strconv.Itoa(int(connStats.downloadSlots)))
+
+}