bot/main.go

66 lines
1.4 KiB
Go
Raw Permalink Normal View History

2023-02-06 21:58:22 +01:00
package main
import (
"fmt"
"os"
"time"
"github.com/alexflint/go-arg"
"github.com/bwmarrin/discordgo"
"github.com/sirupsen/logrus"
)
type UserRole struct {
Rank int
CurrentRole string
}
// Global var (I know, and i don't care)
var api string = os.Getenv("API_URL")
2023-02-06 21:58:22 +01:00
var guild string
var rolesID map[string]string
var Users map[string]*UserRole
var args struct {
Guild string `arg:"-g,required,env:GUILD_ID"`
Top1 string `arg:"env:TOP_1"`
Top10 string `arg:"env:TOP_10"`
Top50 string `arg:"env:TOP_50"`
Wait int `default:"2" help:"Minutes between run"`
Force bool
DryRun bool
}
func main() {
fmt.Println(len(os.Args), os.Args)
arg.MustParse(&args)
guild = args.Guild
rolesID = make(map[string]string)
rolesID["top1"] = args.Top1
rolesID["top10"] = args.Top10
rolesID["top50"] = args.Top50
discord, err := discordgo.New("Bot " + os.Getenv("DISCORD_TOKEN"))
if err != nil {
logrus.Fatal(err.Error())
}
if err := populateUserRole(discord); err != nil {
logrus.Fatal(err.Error())
}
if args.DryRun {
logrus.Warn("Running in Dry Run mode")
}
if args.Force {
logrus.Warn("Force update of all users")
forceUpdate(discord, args.DryRun)
}
for {
logrus.Info("Start update")
updateUserRole(discord, args.DryRun)
logrus.Info("End update, sleep for 2 minutes")
time.Sleep(time.Duration(args.Wait) * time.Minute)
}
}