adds leave command and bot to delete all commands for a bot
This commit is contained in:
parent
44c67ffd42
commit
fc2a748cd4
2 changed files with 74 additions and 75 deletions
60
cmd/delcombot/main.go
Normal file
60
cmd/delcombot/main.go
Normal file
|
@ -0,0 +1,60 @@
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"github.com/disgoorg/disgo"
|
||||||
|
"github.com/disgoorg/disgo/bot"
|
||||||
|
"github.com/disgoorg/disgo/cache"
|
||||||
|
"github.com/disgoorg/disgo/discord"
|
||||||
|
"github.com/disgoorg/disgo/gateway"
|
||||||
|
"github.com/disgoorg/log"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
token = os.Getenv("disgo_token")
|
||||||
|
noCommands []discord.ApplicationCommandCreate
|
||||||
|
)
|
||||||
|
|
||||||
|
// this bot should in theory delete all registered commands for a certain bot
|
||||||
|
func main() {
|
||||||
|
log.SetLevel(log.LevelDebug)
|
||||||
|
log.Info("starting delcombot...")
|
||||||
|
log.Info("disgo version: ", disgo.Version)
|
||||||
|
|
||||||
|
// permissions:
|
||||||
|
// intent:
|
||||||
|
client, err := disgo.New(token,
|
||||||
|
bot.WithGatewayConfigOpts(
|
||||||
|
gateway.WithIntents(gateway.IntentsNone),
|
||||||
|
),
|
||||||
|
bot.WithCacheConfigOpts(
|
||||||
|
cache.WithCaches(
|
||||||
|
cache.FlagsNone,
|
||||||
|
),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("error while building disgo instance: ", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
defer client.Close(context.TODO())
|
||||||
|
|
||||||
|
client.Caches().GuildsForEach(func(guild discord.Guild) {
|
||||||
|
if _, err = client.Rest().SetGuildCommands(client.ApplicationID(), guild.ID, noCommands); err != nil {
|
||||||
|
log.Info(fmt.Sprintf("error deleting guild commands from %v: ", guild.Name), err)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
if _, err = client.Rest().SetGlobalCommands(client.ApplicationID(), noCommands); err != nil {
|
||||||
|
log.Info("error deleting global commands", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err = client.OpenGateway(context.TODO()); err != nil {
|
||||||
|
log.Fatal("error while connecting to gateway: ", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Infof("delcombot removed all guild and global commands")
|
||||||
|
}
|
|
@ -48,7 +48,7 @@ func main() {
|
||||||
),
|
),
|
||||||
),
|
),
|
||||||
bot.WithEventListenerFunc(joinListener),
|
bot.WithEventListenerFunc(joinListener),
|
||||||
bot.WithEventListenerFunc(reactionListener),
|
bot.WithEventListenerFunc(leaveListener),
|
||||||
)
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Fatal("error while building disgo instance: ", err)
|
log.Fatal("error while building disgo instance: ", err)
|
||||||
|
@ -130,88 +130,27 @@ func joinListener(event *events.GuildMemberJoin) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func reactionListener(event *events.ComponentInteractionCreate) {
|
func leaveListener(event *events.GuildMemberLeave) {
|
||||||
|
user := event.Member.User
|
||||||
|
|
||||||
if event.ChannelID() != channelWelcomeID {
|
if event.GuildID.String() != registerGuildID.String() {
|
||||||
// reaction is not in welcome chat
|
event.Client().Logger().Error("leave from an other guild")
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
guild, err := event.Client().Rest().GetGuild(registerGuildID, false)
|
||||||
approverID := event.User().ID
|
|
||||||
if approverID.String() == event.Client().ID().String() {
|
|
||||||
// reaction user is bot itself
|
|
||||||
return
|
|
||||||
}
|
|
||||||
if event.ButtonInteractionData().CustomID() != "approve" {
|
|
||||||
// not approve button
|
|
||||||
return
|
|
||||||
}
|
|
||||||
approver, err := event.Client().Rest().GetMember(registerGuildID, approverID)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
event.Client().Logger().Error("could not find user: ", err)
|
event.Client().Logger().Error("could not find guild: ", err)
|
||||||
return
|
|
||||||
}
|
|
||||||
hasPermission := func() bool {
|
|
||||||
for _, roleID := range approver.RoleIDs {
|
|
||||||
if roleID.String() == roleAppleCoreID.String() {
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return false
|
|
||||||
}()
|
|
||||||
|
|
||||||
if hasPermission == false {
|
|
||||||
err = event.CreateMessage(discord.NewMessageCreateBuilder().
|
|
||||||
SetEphemeral(true).
|
|
||||||
SetContent("you don't have permissions to do this").
|
|
||||||
Build())
|
|
||||||
if err != nil {
|
|
||||||
event.Client().Logger().Error("could not respond to reaction: ", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
messageID := event.Message.ID
|
_, err = event.Client().Rest().CreateMessage(channelWelcomeID,
|
||||||
message, err := event.Client().Rest().GetMessage(channelWelcomeID, messageID)
|
discord.NewMessageCreateBuilder().
|
||||||
|
SetContent(
|
||||||
|
fmt.Sprintf("%v (`%v`) left %v", user.Mention(), user.Username, guild.Name)).
|
||||||
|
Build(),
|
||||||
|
)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
event.Client().Logger().Error("could not find message: ", err)
|
event.Client().Logger().Error("on sending leave message: ", err)
|
||||||
return
|
|
||||||
}
|
|
||||||
if len(message.Mentions) != 1 {
|
|
||||||
// there should be only one mention in the message
|
|
||||||
return
|
|
||||||
}
|
|
||||||
user := message.Mentions[0]
|
|
||||||
|
|
||||||
// add role to user (bot role needs to be above roleAppleID)
|
|
||||||
err = event.Client().Rest().AddMemberRole(registerGuildID, user.ID, roleAppleID)
|
|
||||||
if err != nil {
|
|
||||||
event.Client().Logger().Error("not able to add role: ", err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
// update embed (and time)
|
|
||||||
if len(message.Embeds) != 1 {
|
|
||||||
// there should be only one embed in the message
|
|
||||||
return
|
|
||||||
}
|
|
||||||
embed := message.Embeds[0]
|
|
||||||
updatedMessage := discord.NewEmbedBuilder().
|
|
||||||
SetColor(16068661).
|
|
||||||
SetTimestamp(time.Now()).
|
|
||||||
SetThumbnail(embed.Thumbnail.URL).
|
|
||||||
AddFields(embed.Fields...).
|
|
||||||
AddField("Approval", fmt.Sprintf("%v approved by %v (`%v`)", user.Mention(), approver.Mention(), approver.User.Username), false).
|
|
||||||
Build()
|
|
||||||
|
|
||||||
err = event.UpdateMessage(discord.NewMessageUpdateBuilder().
|
|
||||||
SetContent(message.Content).
|
|
||||||
SetEmbeds(updatedMessage).
|
|
||||||
ClearContainerComponents().
|
|
||||||
Build())
|
|
||||||
if err != nil {
|
|
||||||
event.Client().Logger().Error("could not update message: ", err)
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue