examples: Add autocomplete example

2024-11-27 09:12:53 +00:00 · 2021-11-14 09:34:55 -07:00 · 2021-11-14 09:34:55 -07:00 · 9289303255
parent 2ec0e6b011
commit 9289303255
1 changed files with 131 additions and 0 deletions
--- a/0-examples/autocomplete/main.go
+++ b/0-examples/autocomplete/main.go
@ -0,0 +1,131 @@
+package main
+
+import (
+	"context"
+	"log"
+	"os"
+	"strings"
+
+	"github.com/diamondburned/arikawa/v3/api"
+	"github.com/diamondburned/arikawa/v3/discord"
+	"github.com/diamondburned/arikawa/v3/gateway"
+	"github.com/diamondburned/arikawa/v3/state"
+	"github.com/diamondburned/arikawa/v3/utils/json/option"
+)
+
+// To run, do `GUILD_ID="GUILD ID" BOT_TOKEN="TOKEN HERE" go run .`
+
+func main() {
+	guildID := discord.GuildID(mustSnowflakeEnv("GUILD_ID"))
+
+	token := os.Getenv("BOT_TOKEN")
+	if token == "" {
+		log.Fatalln("No $BOT_TOKEN given.")
+	}
+
+	s, err := state.New("Bot " + token)
+	if err != nil {
+		log.Fatalln("Session failed:", err)
+		return
+	}
+
+	app, err := s.CurrentApplication()
+	if err != nil {
+		log.Fatalln("Failed to get application ID:", err)
+	}
+
+	s.AddHandler(func(e *gateway.InteractionCreateEvent) {
+		var resp api.InteractionResponse
+		switch d := e.Data.(type) {
+		case *discord.CommandInteraction:
+			content := option.NewNullableString("Pong: " + d.Options[0].String() + "!")
+			resp = api.InteractionResponse{
+				Type: api.MessageInteractionWithSource,
+				Data: &api.InteractionResponseData{
+					Content: content,
+				},
+			}
+		case *discord.AutocompleteInteraction:
+			allChoices := []api.AutocompleteChoice{
+				{Name: "Choice A", Value: "Choice A"},
+				{Name: "Choice B", Value: "Choice B"},
+				{Name: "Choice C", Value: "Choice C"},
+				{Name: "Abc Def", Value: "Abcdef"},
+				{Name: "Ghi Jkl", Value: "Ghijkl"},
+				{Name: "Mno Pqr", Value: "Mnopqr"},
+				{Name: "Stu Vwx", Value: "Stuvwx"},
+			}
+			query := strings.ToLower(d.Options[0].Value)
+			var choices []api.AutocompleteChoice
+			for _, choice := range allChoices {
+				if strings.HasPrefix(strings.ToLower(choice.Name), query) ||
+					strings.HasPrefix(strings.ToLower(choice.Value), query) {
+					choices = append(choices, choice)
+				}
+			}
+			resp = api.InteractionResponse{
+				Type: api.AutocompleteResult,
+				Data: &api.InteractionResponseData{
+					Choices: &choices,
+				},
+			}
+		default:
+			return
+		}
+
+		if err := s.RespondInteraction(e.ID, e.Token, resp); err != nil {
+			log.Println("failed to send interaction callback:", err)
+		}
+	})
+
+	s.AddIntents(gateway.IntentGuilds)
+	s.AddIntents(gateway.IntentGuildMessages)
+
+	if err := s.Open(context.Background()); err != nil {
+		log.Fatalln("failed to open:", err)
+	}
+	defer s.Close()
+
+	log.Println("Gateway connected. Getting all guild commands.")
+
+	commands, err := s.GuildCommands(app.ID, guildID)
+	if err != nil {
+		log.Fatalln("failed to get guild commands:", err)
+	}
+
+	for _, command := range commands {
+		log.Println("Existing command", command.Name, "found.")
+	}
+
+	newCommands := []api.CreateCommandData{
+		{
+			Name:        "ping",
+			Description: "Basic ping command.",
+			Options: []discord.CommandOption{
+				&discord.StringOption{
+					OptionName:   "text",
+					Description:  "Text to echo back",
+					Autocomplete: true,
+				},
+			},
+		},
+	}
+
+	for _, command := range newCommands {
+		_, err := s.CreateGuildCommand(app.ID, guildID, command)
+		if err != nil {
+			log.Fatalln("failed to create guild command:", err)
+		}
+	}
+
+	// Block forever.
+	select {}
+}
+
+func mustSnowflakeEnv(env string) discord.Snowflake {
+	s, err := discord.ParseSnowflake(os.Getenv(env))
+	if err != nil {
+		log.Fatalf("Invalid snowflake for $%s: %v", env, err)
+	}
+	return s
+}