So far so good!

main.go

@@ -0,0 +1,174 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"log"
+	"math/rand"
+	"os"
+	"os/signal"
+	"time"
+
+	"github.com/diamondburned/arikawa/v3/api"
+	"github.com/diamondburned/arikawa/v3/api/cmdroute"
+	"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"
+	"github.com/joho/godotenv"
+	openai "github.com/sashabaranov/go-openai"
+)
+
+var commands = []api.CreateCommandData{
+	{
+		Name:        "ping",
+		Description: "ping pong!",
+	},
+	{
+		Name:        "echo",
+		Description: "echo back the argument",
+		Options: []discord.CommandOption{
+			&discord.StringOption{
+				OptionName:  "argument",
+				Description: "what's echoed back",
+				Required:    true,
+			},
+		},
+	},
+	{
+		Name:        "thonk",
+		Description: "biiiig thonk",
+	},
+	{
+		Name:        "ask",
+		Description: "Ask Himbot!",
+		Options: []discord.CommandOption{
+			&discord.StringOption{
+				OptionName:  "argument",
+				Description: "the prompt",
+				Required:    true,
+			},
+		},
+	},
+}
+
+func main() {
+	godotenv.Load(".env")
+
+	token := os.Getenv("DISCORD_TOKEN")
+	if token == "" {
+		log.Fatalln("No $DISCORD_TOKEN given.")
+	}
+
+	h := newHandler(state.New("Bot " + token))
+	h.s.AddInteractionHandler(h)
+	h.s.AddIntents(gateway.IntentGuilds)
+	h.s.AddHandler(func(*gateway.ReadyEvent) {
+		me, _ := h.s.Me()
+		log.Println("connected to the gateway as", me.Tag())
+	})
+
+	if err := cmdroute.OverwriteCommands(h.s, commands); err != nil {
+		log.Fatalln("cannot update commands:", err)
+	}
+
+	ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt)
+	defer cancel()
+
+	if err := h.s.Connect(ctx); err != nil {
+		log.Fatalln("cannot connect:", err)
+	}
+}
+
+type handler struct {
+	*cmdroute.Router
+	s *state.State
+}
+
+func newHandler(s *state.State) *handler {
+	h := &handler{s: s}
+
+	h.Router = cmdroute.NewRouter()
+	// Automatically defer handles if they're slow.
+	h.Use(cmdroute.Deferrable(s, cmdroute.DeferOpts{}))
+	h.AddFunc("ping", h.cmdPing)
+	h.AddFunc("echo", h.cmdEcho)
+	h.AddFunc("thonk", h.cmdThonk)
+	h.AddFunc("ask", h.cmdAsk)
+
+	return h
+}
+
+func (h *handler) cmdPing(ctx context.Context, cmd cmdroute.CommandData) *api.InteractionResponseData {
+	return &api.InteractionResponseData{
+		Content: option.NewNullableString("Pong!"),
+	}
+}
+
+func (h *handler) cmdEcho(ctx context.Context, data cmdroute.CommandData) *api.InteractionResponseData {
+	var options struct {
+		Arg string `discord:"argument"`
+	}
+
+	if err := data.Options.Unmarshal(&options); err != nil {
+		return errorResponse(err)
+	}
+
+	return &api.InteractionResponseData{
+		Content:         option.NewNullableString(options.Arg),
+		AllowedMentions: &api.AllowedMentions{}, // don't mention anyone
+	}
+}
+
+func (h *handler) cmdAsk(ctx context.Context, data cmdroute.CommandData) *api.InteractionResponseData {
+	var options struct {
+		Arg string `discord:"argument"`
+	}
+
+	if err := data.Options.Unmarshal(&options); err != nil {
+		return errorResponse(err)
+	}
+
+	client := openai.NewClient(os.Getenv("OPENAI_API_KEY"))
+
+	resp, err := client.CreateChatCompletion(
+		context.Background(),
+		openai.ChatCompletionRequest{
+			Model: openai.GPT4TurboPreview,
+			Messages: []openai.ChatCompletionMessage{
+				{
+					Role:    openai.ChatMessageRoleUser,
+					Content: options.Arg,
+				},
+			},
+		},
+	)
+
+	if err != nil {
+		fmt.Printf("ChatCompletion error: %v\n", err)
+		return &api.InteractionResponseData{
+			Content:         option.NewNullableString("ChatCompletion Error!"),
+			AllowedMentions: &api.AllowedMentions{}, // don't mention anyone
+		}
+	}
+
+	return &api.InteractionResponseData{
+		Content:         option.NewNullableString(resp.Choices[0].Message.Content),
+		AllowedMentions: &api.AllowedMentions{}, // don't mention anyone
+	}
+}
+
+func (h *handler) cmdThonk(ctx context.Context, data cmdroute.CommandData) *api.InteractionResponseData {
+	time.Sleep(time.Duration(3+rand.Intn(5)) * time.Second)
+	return &api.InteractionResponseData{
+		Content: option.NewNullableString("https://tenor.com/view/thonk-thinking-sun-thonk-sun-thinking-sun-gif-14999983"),
+	}
+}
+
+func errorResponse(err error) *api.InteractionResponseData {
+	return &api.InteractionResponseData{
+		Content:         option.NewNullableString("**Error:** " + err.Error()),
+		Flags:           discord.EphemeralMessage,
+		AllowedMentions: &api.AllowedMentions{ /* none */ },
+	}
+}