package main import ( "github.com/bwmarrin/discordgo" "github.com/sirupsen/logrus" "golang.org/x/exp/slices" ) // Populate Users map for the first launch // This variable will be keep as sort of cache for all // operation func populateUserRole(discordClient *discordgo.Session) error { Users = make(map[string]*UserRole) APIResponse, err := getUsers42CTF() if err != nil { return err } for id, rank := range APIResponse { tmp, err := discordClient.GuildMember(guild, id) if err == nil { if slices.Contains(tmp.Roles, rolesID["top1"]) { Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top1"]} } else if slices.Contains(tmp.Roles, rolesID["top10"]) { Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top10"]} } else if slices.Contains(tmp.Roles, rolesID["top50"]) { Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top50"]} } else { Users[id] = &UserRole{Rank: rank, CurrentRole: ""} } } else { logrus.Error("ID not found", id) } } return nil } func updateUserRole(discordClient *discordgo.Session, dryrun bool) error { APIResponse, err := getUsers42CTF() if err != nil { return err } for id, rank := range APIResponse { tmp, err := discordClient.GuildMember(guild, id) if err == nil { if slices.Contains(tmp.Roles, rolesID["top1"]) { Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top1"]} } else if slices.Contains(tmp.Roles, rolesID["top10"]) { Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top10"]} } else if slices.Contains(tmp.Roles, rolesID["top50"]) { Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top50"]} } else { Users[id] = &UserRole{Rank: rank, CurrentRole: ""} } } else { logrus.Error("ID not found", id) } if Users[id].Rank != rank { logrus.Info("New rank (", rank, ") for user ", id) if Users[id].CurrentRole != "" { logrus.Info("Remove role for ", id) removeRole(discordClient, guild, id, Users[id].CurrentRole, dryrun) Users[id] = &UserRole{Rank: rank, CurrentRole: ""} } if rank == 1 { logrus.Info("Add role `top1` for ", id) addRole(discordClient, guild, id, rolesID["top1"], dryrun) Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top1"]} } else if rank <= 10 { logrus.Info("Add role `top10` for ", id) addRole(discordClient, guild, id, rolesID["top10"], dryrun) Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top10"]} } else if rank <= 50 { logrus.Info("Add role `top50` for ", id) addRole(discordClient, guild, id, rolesID["top50"], dryrun) Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top50"]} } } } return nil } func forceUpdate(discordClient *discordgo.Session, dryrun bool) (error) { APIResponse, err := getUsers42CTF() if err != nil { return err } for id, rank := range APIResponse { tmp, err := discordClient.GuildMember(guild, id) if err == nil { if slices.Contains(tmp.Roles, rolesID["top1"]) { removeRole(discordClient, guild, id, rolesID["top1"], dryrun) } if slices.Contains(tmp.Roles, rolesID["top10"]) { removeRole(discordClient, guild, id, rolesID["top10"], dryrun) } if slices.Contains(tmp.Roles, rolesID["top50"]) { removeRole(discordClient, guild, id, rolesID["top50"], dryrun) } Users[id] = &UserRole{Rank: rank, CurrentRole: ""} } else { logrus.Error("ID not found", id) } if rank == 1 { logrus.Info("Add role `top1` for ", id) addRole(discordClient, guild, id, rolesID["top1"], dryrun) Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top1"]} } else if rank <= 10 { logrus.Info("Add role `top10` for ", id) addRole(discordClient, guild, id, rolesID["top10"], dryrun) Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top10"]} } else if rank <= 50 { logrus.Info("Add role `top50` for ", id) addRole(discordClient, guild, id, rolesID["top50"], dryrun) Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top50"]} } } return nil }