mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-11-19 05:12:48 +00:00
126 lines
3.1 KiB
Go
126 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"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"
|
|
)
|
|
|
|
// To run, do `BOT_TOKEN="TOKEN HERE" go run .`
|
|
|
|
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",
|
|
},
|
|
}
|
|
|
|
func main() {
|
|
token := os.Getenv("BOT_TOKEN")
|
|
if token == "" {
|
|
log.Fatalln("No $BOT_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 := overwriteCommands(h.s); 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)
|
|
}
|
|
}
|
|
|
|
func overwriteCommands(s *state.State) error {
|
|
return cmdroute.OverwriteCommands(s, commands)
|
|
}
|
|
|
|
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)
|
|
|
|
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) 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 */ },
|
|
}
|
|
}
|