Joren/HwidBot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

Steal slash commands for testing purposes

Browse Source
This commit is contained in:
Joren 2024-07-06 17:37:07 +02:00
parent f24c836cbc
commit 6de41e0180
Signed by: Joren
GPG Key ID: 280E33DFBC0F1B55
1 changed files with 164 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

165
main.go
View File

@ -7,6 +7,7 @@ import (
"os"
"os/signal"
"syscall"
"strings"
"github.com/BurntSushi/toml"
"github.com/bwmarrin/discordgo"
@ -26,6 +27,7 @@ type Config struct {
type Discord struct {
Token string `toml:"client"`
AppID string `toml:"appid"`
GuildID string `toml:"guildid"`
}
@ -52,7 +54,6 @@ func connectDb(config Config) (*sql.DB, error) {
config.Database.Password,
config.Database.Name,
)
log.Println(connectionString)
db, err := sql.Open("postgres", connectionString)
if err != nil {
@ -90,6 +91,149 @@ func init() {
}
}
func searchLink(message, format, sep string) string {
return fmt.Sprintf(format, strings.Join(
strings.Split(
message,
" ",
),
sep,
))
}
var (
commands = []discordgo.ApplicationCommand{
{
Name: "rickroll-em",
Type: discordgo.UserApplicationCommand,
},
{
Name: "google-it",
Type: discordgo.MessageApplicationCommand,
},
{
Name: "stackoverflow-it",
Type: discordgo.MessageApplicationCommand,
},
{
Name: "godoc-it",
Type: discordgo.MessageApplicationCommand,
},
{
Name: "discordjs-it",
Type: discordgo.MessageApplicationCommand,
},
{
Name: "discordpy-it",
Type: discordgo.MessageApplicationCommand,
},
}
commandsHandlers = map[string]func(s *discordgo.Session, i *discordgo.InteractionCreate){
"rickroll-em": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: "Operation rickroll has begun",
Flags: discordgo.MessageFlagsEphemeral,
},
})
if err != nil {
panic(err)
}
ch, err := s.UserChannelCreate(
i.ApplicationCommandData().TargetID,
)
if err != nil {
_, err = s.FollowupMessageCreate(i.Interaction, true, &discordgo.WebhookParams{
Content: fmt.Sprintf("Mission failed. Cannot send a message to this user: %q", err.Error()),
Flags: discordgo.MessageFlagsEphemeral,
})
if err != nil {
panic(err)
}
}
_, err = s.ChannelMessageSend(
ch.ID,
fmt.Sprintf("%s sent you this: https://youtu.be/dQw4w9WgXcQ", i.Member.Mention()),
)
if err != nil {
panic(err)
}
},
"google-it": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: searchLink(
i.ApplicationCommandData().Resolved.Messages[i.ApplicationCommandData().TargetID].Content,
"https://google.com/search?q=%s", "+"),
Flags: discordgo.MessageFlagsEphemeral,
},
})
if err != nil {
panic(err)
}
},
"stackoverflow-it": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: searchLink(
i.ApplicationCommandData().Resolved.Messages[i.ApplicationCommandData().TargetID].Content,
"https://stackoverflow.com/search?q=%s", "+"),
Flags: discordgo.MessageFlagsEphemeral,
},
})
if err != nil {
panic(err)
}
},
"godoc-it": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: searchLink(
i.ApplicationCommandData().Resolved.Messages[i.ApplicationCommandData().TargetID].Content,
"https://pkg.go.dev/search?q=%s", "+"),
Flags: discordgo.MessageFlagsEphemeral,
},
})
if err != nil {
panic(err)
}
},
"discordjs-it": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: searchLink(
i.ApplicationCommandData().Resolved.Messages[i.ApplicationCommandData().TargetID].Content,
"https://discord.js.org/#/docs/main/stable/search?query=%s", "+"),
Flags: discordgo.MessageFlagsEphemeral,
},
})
if err != nil {
panic(err)
}
},
"discordpy-it": func(s *discordgo.Session, i *discordgo.InteractionCreate) {
err := s.InteractionRespond(i.Interaction, &discordgo.InteractionResponse{
Type: discordgo.InteractionResponseChannelMessageWithSource,
Data: &discordgo.InteractionResponseData{
Content: searchLink(
i.ApplicationCommandData().Resolved.Messages[i.ApplicationCommandData().TargetID].Content,
"https://discordpy.readthedocs.io/en/stable/search.html?q=%s", "+"),
Flags: discordgo.MessageFlagsEphemeral,
},
})
if err != nil {
panic(err)
}
},
}
)
func main() {
if client == nil {
log.Println("Bot client is not initialized")
@ -100,6 +244,25 @@ func main() {
log.Println("Bot is online")
})
client.AddHandler(func(client *discordgo.Session, i *discordgo.InteractionCreate) {
if h, ok := commandsHandlers[i.ApplicationCommandData().Name]; ok {
h(client, i)
}
})
cmdIDs := make(map[string]string, len(commands))
for _, cmd := range commands {
rcmd, err := client.ApplicationCommandCreate(config.Discord.AppID, config.Discord.GuildID, &cmd)
if err != nil {
log.Fatalf("Cannot create slash command %q: %v", cmd.Name, err)
}
cmdIDs[rcmd.ID] = rcmd.Name
}
err := client.Open()
if err != nil {
log.Println("Error opening connection:", err)