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 }