introduce an alive server cycler

Author: Elfshot

.gitignore

@@ -1 +1,2 @@
-.vscode
+.vscode
+*temp*

aliveServers.go

@@ -0,0 +1,129 @@
+package ttRequests
+
+import (
+	"errors"
+	"time"
+
+	gocron "github.com/go-co-op/gocron"
+)
+
+type AliveServer struct {
+	Primary_server string
+	Servers        [2]serverType
+}
+
+type serverType struct {
+	Name      string
+	Id        uint8
+	CfxId     string
+	DirectUrl string
+	CfxUrl    string
+	AliveUrl  string
+	Alive     bool
+}
+
+var servers [2]serverType
+var aliveToggle bool = false
+var AliveServers AliveServer
+
+func checkInit() {
+	if aliveToggle != true {
+		Init()
+	}
+}
+
+func cronAlive() {
+	cron := gocron.NewScheduler(time.UTC)
+
+	cron.Every("45s").Do(checkServers)
+	cron.StartBlocking()
+}
+
+func Init() {
+	aliveToggle = true
+	servers = [2]serverType{
+		{
+			Name:      "NY-1",
+			Id:        1,
+			CfxId:     "2epova",
+			DirectUrl: "http://server.tycoon.community:30120",
+			CfxUrl:    "https://tycoon-2epova.users.cfx.re",
+			AliveUrl:  "",
+			Alive:     false,
+		},
+		{
+			Name:      "NY-2",
+			Id:        2,
+			CfxId:     "njyvop",
+			DirectUrl: "http://server.tycoon.community:30121",
+			CfxUrl:    "https://tycoon-njyvop.users.cfx.re",
+			AliveUrl:  "",
+			Alive:     false,
+		},
+	}
+	// AliveServers.Primary_server = servers[0].AliveUrl
+	AliveServers.Primary_server = ""
+	AliveServers.Servers = servers
+	checkServers()
+	// cron.Every("5s").Do(checkServers)
+	go cronAlive()
+}
+
+func checkServers() {
+	aliveDown := 0
+	for i := range &AliveServers.Servers {
+		server := &AliveServers.Servers[i]
+		aliveCheck(server)
+		if server.Alive {
+			AliveServers.Primary_server = server.AliveUrl
+		} else {
+			aliveDown++
+		}
+	}
+	if aliveDown == len(&AliveServers.Servers) {
+		AliveServers.Primary_server = ""
+	}
+	// fmt.Printf("AliveServers: %+v\n", AliveServers)
+}
+
+func aliveCheck(server *serverType) {
+	first, err := GetNoParse_(server.DirectUrl + "/status/alive")
+	if err != nil || (first.StatusCode != 200 && first.StatusCode != 204) {
+		second, err := GetNoParse_(server.CfxUrl + "/status/alive")
+
+		if err != nil || (second.StatusCode != 200 && second.StatusCode != 204) {
+			server.AliveUrl = server.DirectUrl
+			server.Alive = false
+		} else {
+			server.AliveUrl = server.CfxUrl
+			server.Alive = true
+		}
+	} else {
+		server.AliveUrl = server.DirectUrl
+		server.Alive = true
+	}
+}
+
+func PrimaryServerUrl() (string, error) {
+	if AliveServers.Primary_server != "" {
+		return AliveServers.Primary_server, nil
+	} else {
+		return "", errors.New("No online servers")
+	}
+}
+
+func ServerSpec(id uint8) (*serverType, error) {
+	if id < 1 || id > 2 {
+		return nil, errors.New("Server ID is out of range")
+	}
+	return &AliveServers.Servers[id-1], nil
+}
+
+func ServerSpecAlive(id uint8) (*serverType, error) {
+	server := &AliveServers.Servers[id-1]
+	if server.Alive {
+		return server, nil
+	} else {
+		return nil, errors.New("Requested server is not alive")
+	}
+}

go.mod

@@ -1,3 +1,10 @@
 module github.com/Elfshot/tt-api-wrapper
 
 go 1.18
+
+require github.com/go-co-op/gocron v1.15.0
+
+require (
+	github.com/robfig/cron/v3 v3.0.1 // indirect
+	golang.org/x/sync v0.0.0-20210220032951-036812b2e83c // indirect
+)

go.sum

@@ -0,0 +1,10 @@
+github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
+github.com/go-co-op/gocron v1.15.0 h1:XmiPazahD9aq0/QdK5toCVHfgTXfrZ/s83RpAgzr6SM=
+github.com/go-co-op/gocron v1.15.0/go.mod h1:On9zUZTv7FBeuj9D/cdYyAWcPUiLqqAx7nsPHd0EmKM=
+github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
+github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs=
+github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro=
+github.com/stretchr/testify v1.7.5 h1:s5PTfem8p8EbKQOctVV53k6jCJt3UX4IEJzwh+C324Q=
+golang.org/x/sync v0.0.0-20210220032951-036812b2e83c h1:5KslGYwFpkhGh+Q16bwMP3cOontH8FOep7tGV86Y7SQ=
+golang.org/x/sync v0.0.0-20210220032951-036812b2e83c/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
+gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=

httpRequest.go

@@ -1,4 +1,4 @@
-package tt_requests
+package ttRequests
 
 import (
 	"encoding/json"
@@ -11,7 +11,7 @@ import (
 
 var client = &http.Client{}
 
-func _Get(url string, strucVar any) (*any, error) {
+func Get_(url string, strucVar any) (*any, error) {
 	res, resErr := request(url)
 	if resErr != nil {
 		return nil, resErr
@@ -25,13 +25,20 @@ func _Get(url string, strucVar any) (*any, error) {
 	return &strucVar, nil
 }
 
-func _GetNoParse(url string) (*string, error) {
-	res, resErr := request(url)
-	if resErr != nil {
-		return nil, resErr
+func GetNoParse_(url string) (*http.Response, error) {
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return nil, err
 	}
+	req.Header.Set("X-Tycoon-Key", os.Getenv("TYCOON_KEY_PRIVATE"))
 
-	return &res, nil
+	resp, err := client.Do(req)
+	if err != nil { // Pass on any HTTP error
+		fmt.Println(err)
+		return nil, errors.New("HTTP Error: " + err.Error())
+	}
+
+	return resp, nil
 }
 
 func parse(body string, parsed any) (*any, error) {
@@ -47,7 +54,11 @@ func parse(body string, parsed any) (*any, error) {
 func request(url string) (string, error) {
 	// Make the request
 	req, err := http.NewRequest("GET", url, nil)
-	req.Header.Set("X-Tycoon-Key", os.Getenv("TYCOON_KEY"))
+	if err != nil {
+		return "", err
+	}
+
+	req.Header.Set("X-Tycoon-Key", os.Getenv("TYCOON_KEY_PRIVATE"))
 
 	resp, err := client.Do(req)
 	if err != nil { // Pass on any HTTP error

models/dataadv.go

@@ -1,4 +1,4 @@
-package tt_requests_models
+package ttRequests_models
 
 type UserData struct {
 	User_id   uint32

models/fxplayers.go

@@ -1,4 +1,4 @@
-package tt_requests_models
+package ttRequests_models
 
 type FxPlayers []struct {
 	Endpoint    string   `json:"endpoint"`

models/players.go

@@ -1,4 +1,4 @@
-package tt_requests_models
+package ttRequests_models
 
 // Dunno how I'd implement the mixed json array so...
 //

models/sotd.go

@@ -1,4 +1,4 @@
-package tt_requests_models
+package ttRequests_models
 
 type Sotd struct {
 	Aptitude string `json:"aptitude"`

ttRequests.go

@@ -1,14 +1,29 @@
-package tt_requests
+package ttRequests
 
 import (
 	"fmt"
 
 	models "github.com/Elfshot/tt-api-wrapper/models"
 )
 
+func primaryGet(append string, struc any) (*any, error) {
+	server, servErr := PrimaryServerUrl()
+	if servErr != nil {
+		return nil, servErr
+	}
+
+	_, err := Get_(server+append, struc)
+	if err != nil {
+		return nil, err
+	}
+
+	return &struc, nil
+}
+
 func Get_DataAdv(playerId uint32) (data *models.UserData, Error error) {
 	var res models.UserData
-	_, err := _Get(fmt.Sprintf("https://tycoon-2epova.users.cfx.re/status/dataadv/%d", playerId), &res)
+	_, err := primaryGet(fmt.Sprintf("/status/dataadv/%d", playerId), &res)
+
 	if err != nil {
 		return nil, err
 	}
@@ -17,7 +32,8 @@ func Get_DataAdv(playerId uint32) (data *models.UserData, Error error) {
 
 func Get_Players() (data *models.Players, Error error) {
 	var res models.Players
-	_, err := _Get("https://tycoon-2epova.users.cfx.re/status/players.json", &res)
+	_, err := primaryGet("/status/players.json", &res)
+
 	if err != nil {
 		return nil, err
 	}
@@ -26,7 +42,8 @@ func Get_Players() (data *models.Players, Error error) {
 
 func Get_FxPlayers() (data *models.FxPlayers, Error error) {
 	var res models.FxPlayers
-	_, err := _Get("https://tycoon-2epova.users.cfx.re/players.json", &res)
+	_, err := primaryGet("/players.json", &res)
+
 	if err != nil {
 		return nil, err
 	}
@@ -35,7 +52,8 @@ func Get_FxPlayers() (data *models.FxPlayers, Error error) {
 
 func Get_Sotd() (data *models.Sotd, Error error) {
 	var res models.Sotd
-	_, err := _Get("https://tycoon-2epova.users.cfx.re/status/sotd.json", &res)
+	_, err := primaryGet("/status/sotd.json", &res)
+
 	if err != nil {
 		return nil, err
 	}