Remove extraneous reaction remove/add, scw healthcheck kill

Author: denverquane

discord/eventHandler.go

@@ -265,14 +265,14 @@ func (bot *Bot) processTransition(phase game.Phase, dgsRequest GameStateRequest)
 	case game.MENU:
 		dgs.Edit(bot.SessionManager.GetPrimarySession(), bot.gameStateResponse(dgs, sett))
 		bot.applyToAll(dgs, false, false)
-		go dgs.RemoveAllReactions(bot.SessionManager.GetPrimarySession())
+		//go dgs.RemoveAllReactions(bot.SessionManager.GetPrimarySession())
 		break
 	case game.LOBBY:
 		delay := sett.Delays.GetDelay(oldPhase, phase)
 		bot.handleTrackedMembers(bot.SessionManager, sett, delay, NoPriority, dgsRequest)
 
 		dgs.Edit(bot.SessionManager.GetPrimarySession(), bot.gameStateResponse(dgs, sett))
-		go dgs.AddAllReactions(bot.SessionManager.GetPrimarySession(), bot.StatusEmojis[true])
+
 		break
 	case game.TASKS:
 		delay := sett.Delays.GetDelay(oldPhase, phase)

discord/healthCheck_test.go

@@ -0,0 +1,15 @@
+package discord
+
+import (
+	"os"
+	"testing"
+)
+
+func TestTerminateScalewayNode(t *testing.T) {
+	orgId := os.Getenv("SCW_ORGANIZATION_ID")
+	accessKey := os.Getenv("SCW_ACCESS_KEY")
+	secretKey := os.Getenv("SCW_SECRET_KEY")
+	nodeID := os.Getenv("SCW_NODE_ID")
+
+	TerminateScalewayNode(orgId, accessKey, secretKey, nodeID)
+}

discord/healthcheck.go

@@ -2,8 +2,12 @@ package discord
 
 import (
 	"github.com/gorilla/mux"
+	"github.com/scaleway/scaleway-sdk-go/api/k8s/v1"
+	"github.com/scaleway/scaleway-sdk-go/scw"
 	"log"
 	"net/http"
+	"os"
+	"strings"
 )
 
 func StartHealthCheckServer(port string) {
@@ -15,7 +19,7 @@ func StartHealthCheckServer(port string) {
 	})
 
 	r.HandleFunc("/ready", func(w http.ResponseWriter, r *http.Request) {
-		resp, err := http.Get("https://discordapp.com/api/v7/gateway")
+		resp, err := http.Get("https://discordapp.com/api/v8/gateway")
 		if err != nil {
 			log.Println(err)
 			w.WriteHeader(http.StatusInternalServerError)
@@ -28,7 +32,74 @@ func StartHealthCheckServer(port string) {
 		} else {
 			w.Write([]byte("unready"))
 		}
+
+		if resp.StatusCode == http.StatusTooManyRequests {
+			orgId := os.Getenv("SCW_ORGANIZATION_ID")
+			accessKey := os.Getenv("SCW_ACCESS_KEY")
+			secretKey := os.Getenv("SCW_SECRET_KEY")
+			nodeID := os.Getenv("SCW_NODE_ID")
+			if orgId == "" || accessKey == "" || secretKey == "" || nodeID == "" {
+				log.Println("One of the Scaleway credentials was null, not replacing any nodes!")
+				return
+			}
+
+			TerminateScalewayNode(orgId, accessKey, secretKey, nodeID)
+		}
 	})
 
 	http.ListenAndServe(":"+port, r)
 }
+
+func TerminateScalewayNode(org, access, secret, nodeID string) {
+	client, err := scw.NewClient(
+		// Get your credentials at https://console.scaleway.com/project/credentials
+		scw.WithDefaultOrganizationID(org),
+		scw.WithAuth(access, secret),
+	)
+	if err != nil {
+		panic(err)
+	}
+
+	kApi := k8s.NewAPI(client)
+
+	nodes, err := kApi.ListNodes(&k8s.ListNodesRequest{
+		Region:    scw.RegionFrPar,
+		ClusterID: os.Getenv("SCW_CLUSTER_ID"),
+		PoolID:    nil,
+		OrderBy:   "",
+		Page:      nil,
+		PageSize:  nil,
+		Name:      nil,
+		Status:    "",
+	})
+	if err != nil {
+		log.Println(err)
+		return
+	}
+	refUid := formatUUID(nodeID)
+	log.Println(refUid)
+	for _, v := range nodes.Nodes {
+		log.Println(v.ID)
+		if strings.HasPrefix(v.ID, refUid) {
+			nodeID = v.ID
+			break
+		}
+	}
+
+	_, err = kApi.ReplaceNode(&k8s.ReplaceNodeRequest{
+		Region: scw.RegionFrPar,
+		NodeID: nodeID,
+	})
+	if err != nil {
+		log.Println("Error replacing " + refUid + ": " + err.Error())
+	}
+}
+
+func formatUUID(id string) string {
+	ids := strings.Split(id, "-")
+	id = ids[len(ids)-1]
+	if len(id) < 20 {
+		return ""
+	}
+	return id[0:8] + "-" + id[8:12] + "-" + id[12:16] + "-" + id[16:20] + "-" + id[20:]
+}

discord/message_handlers.go

@@ -415,11 +415,5 @@ func (bot *Bot) handleGameStartMessage(s *discordgo.Session, m *discordgo.Messag
 	bot.RedisInterface.SetDiscordGameState(dgs, lock)
 
 	log.Println("Added self game state message")
-
-	if dgs.AmongUsData.GetPhase() != game.MENU {
-		for _, e := range bot.StatusEmojis[true] {
-			go dgs.AddReaction(s, e.FormatForReaction())
-		}
-		dgs.AddReaction(s, "❌")
-	}
+	go dgs.AddAllReactions(bot.SessionManager.GetPrimarySession(), bot.StatusEmojis[true])
 }

go.mod

@@ -15,6 +15,7 @@ require (
 	github.com/jackc/pgx/v4 v4.9.2
 	github.com/joho/godotenv v1.3.0
 	github.com/nicksnyder/go-i18n/v2 v2.1.1
+	github.com/scaleway/scaleway-sdk-go v1.0.0-beta.7
 	golang.org/x/crypto v0.0.0-20201112155050-0c6587e931a9 // indirect
 	golang.org/x/sys v0.0.0-20201113135734-0a15ea8d9b02 // indirect
 	golang.org/x/text v0.3.4

go.sum

@@ -31,6 +31,7 @@ github.com/denverquane/discordgo v0.24.1 h1:6XS7kRGgnZnbYbEkGNOLC6zDWtW2s3XJeVXM
 github.com/denverquane/discordgo v0.24.1/go.mod h1:VLV+ZuXWxlP2Tw88Yx7qgVApBy0yEMnvJQ1Us1UE4zE=
 github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78=
 github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc=
+github.com/dnaeon/go-vcr v1.0.1/go.mod h1:aBB1+wY4s93YsC3HHjMBMrwTj2R9FHDzUr9KyGc8n1E=
 github.com/envoyproxy/go-control-plane v0.9.1-0.20191026205805-5f8ba28d4473/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4=
 github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c=
 github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
@@ -178,6 +179,8 @@ github.com/rs/xid v1.2.1/go.mod h1:+uKXf+4Djp6Md1KODXJxgGQPKngRmWyn10oCKFzNHOQ=
 github.com/rs/zerolog v1.13.0/go.mod h1:YbFCdg8HfsridGWAh22vktObvhZbQsZXe4/zB0OKkWU=
 github.com/rs/zerolog v1.15.0/go.mod h1:xYTKnLHcpfU2225ny5qZjxnj9NvkumZYjJHlAThCjNc=
 github.com/satori/go.uuid v1.2.0/go.mod h1:dA0hQrYB0VpLJoorglMZABFdXlWrHn1NEOzdhQKdks0=
+github.com/scaleway/scaleway-sdk-go v1.0.0-beta.7 h1:Do8ksLD4Nr3pA0x0hnLOLftZgkiTDvwPDShRTUxtXpE=
+github.com/scaleway/scaleway-sdk-go v1.0.0-beta.7/go.mod h1:CJJ5VAbozOl0yEw7nHB9+7BXTJbIn6h7W+f6Gau5IP8=
 github.com/shopspring/decimal v0.0.0-20180709203117-cd690d0c9e24/go.mod h1:M+9NzErvs504Cn4c5DxATwIqPbtswREoFCre64PpcG4=
 github.com/shopspring/decimal v0.0.0-20200227202807-02e2044944cc h1:jUIKcSPO9MoMJBbEoyE/RJoE8vz7Mb8AjvifMMwSyvY=
 github.com/shopspring/decimal v0.0.0-20200227202807-02e2044944cc/go.mod h1:DKyhrW/HYNuLGql+MJL6WCR6knT2jwCFRcu2hWCYk4o=