42CTF/bot
42CTF
/
bot
3
1
Fork 2
Code Issues Pull Requests Projects Releases Wiki Activity

fix(bot): Check if user exist before perform action #6

Merged
Starthur merged 2 commits from Starthur/bot:main into main 2023-02-07 09:25:07 +01:00
Conversation 0 Commits 2 Files Changed 4 +53 -37
4 changed files with 53 additions and 37 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit d62b7beaf0 - Show all commits

8
discord.go
View File

@ -9,7 +9,9 @@ func removeRole(discordClient *discordgo.Session, guildID string, userID string,
if ! dryrun {
err := discordClient.GuildMemberRoleRemove(guild, userID, roleID)
if err != nil {
logrus.Error(err.Error())
logrus.WithFields(logrus.Fields{
"id": userID,
}).Error(err.Error())
}
}
}
@ -18,7 +20,9 @@ func addRole(discordClient *discordgo.Session, guildID string, userID string, ro
if ! dryrun {
err := discordClient.GuildMemberRoleAdd(guild, userID, roleID)
if err != nil {
logrus.Error(err.Error())
logrus.WithFields(logrus.Fields{
"id": userID,
}).Error(err.Error())
}
}
}

2
docker-compose.yaml
View File

@ -5,4 +5,4 @@ services:
build: .
container_name: bot_discord
env_file: ./env_file
entrypoint: bot --guild=943459216901955604 --top1=798638767359524875 --top10=801787467064672286 --top50=803729539145924649
entrypoint: bot --guild=606162827274616845 --top1=798638767359524875 --top10=801787467064672286 --top50=803729539145924649

2
main.go
View File

@ -16,7 +16,7 @@ type UserRole struct {
}
// Global var (I know, and i don't care)
var api string = "https://preprod.42ctf.org/api/bot/discord?token=" + os.Getenv("AUTH_TOKEN")
var api string = os.Getenv("API_URL")
var guild string
var rolesID map[string]string
var Users map[string]*UserRole

22
rank.go
View File

@ -29,7 +29,10 @@ func populateUserRole(discordClient *discordgo.Session) error {
Users[id] = &UserRole{Rank: rank, CurrentRole: ""}
}
} else {
logrus.Error("ID not found", id)
logrus.WithFields(logrus.Fields{
"id": id,
"rank": rank,
}).Error(err.Error())
}
}
return nil
@ -53,13 +56,18 @@ func updateUserRole(discordClient *discordgo.Session, dryrun bool) error {
Users[id] = &UserRole{Rank: rank, CurrentRole: ""}
}
} else {
logrus.Error("ID not found", id)
logrus.WithFields(logrus.Fields{
"id": id,
"rank": rank,
}).Error(err.Error())
}
if Users[id].Rank != rank {
user, exist := Users[id]
if exist {
if user.Rank != rank {
logrus.Info("New rank (", rank, ") for user ", id)
if Users[id].CurrentRole != "" {
if user.CurrentRole != "" {
logrus.Info("Remove role for ", id)
removeRole(discordClient, guild, id, Users[id].CurrentRole, dryrun)
removeRole(discordClient, guild, id, user.CurrentRole, dryrun)
Users[id] = &UserRole{Rank: rank, CurrentRole: ""}
}
if rank == 1 {
@ -77,6 +85,7 @@ func updateUserRole(discordClient *discordgo.Session, dryrun bool) error {
}
}
}
}
return nil
}
@ -101,6 +110,8 @@ func forceUpdate(discordClient *discordgo.Session, dryrun bool) (error) {
} else {
logrus.Error("ID not found", id)
}
_, exist := Users[id]
if exist {
if rank == 1 {
logrus.Info("Add role `top1` for ", id)
addRole(discordClient, guild, id, rolesID["top1"], dryrun)
@ -115,5 +126,6 @@ func forceUpdate(discordClient *discordgo.Session, dryrun bool) (error) {
Users[id] = &UserRole{Rank: rank, CurrentRole: rolesID["top50"]}
}
}
}
return nil
}