mirror of
https://github.com/diamondburned/arikawa.git
synced 2024-11-04 22:14:30 +00:00
135 lines
3 KiB
Go
135 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/diamondburned/arikawa/bot"
|
|
"github.com/diamondburned/arikawa/bot/extras/arguments"
|
|
"github.com/diamondburned/arikawa/discord"
|
|
"github.com/diamondburned/arikawa/gateway"
|
|
)
|
|
|
|
type Bot struct {
|
|
// Context must not be embedded.
|
|
Ctx *bot.Context
|
|
}
|
|
|
|
func (bot *Bot) Help(m *gateway.MessageCreateEvent) error {
|
|
_, err := bot.Ctx.SendMessage(m.ChannelID, bot.Ctx.Help(), nil)
|
|
return err
|
|
}
|
|
|
|
func (bot *Bot) Add(m *gateway.MessageCreateEvent, a, b int) error {
|
|
content := fmt.Sprintf("%d + %d = %d", a, b, a+b)
|
|
|
|
_, err := bot.Ctx.SendMessage(m.ChannelID, content, nil)
|
|
return err
|
|
}
|
|
|
|
func (bot *Bot) Ping(m *gateway.MessageCreateEvent) error {
|
|
_, err := bot.Ctx.SendMessage(m.ChannelID, "Pong!", nil)
|
|
return err
|
|
}
|
|
|
|
func (bot *Bot) Say(m *gateway.MessageCreateEvent, f *arguments.Flag) error {
|
|
args := f.String()
|
|
if args == "" {
|
|
// Empty message, ignore
|
|
return nil
|
|
}
|
|
|
|
_, err := bot.Ctx.SendMessage(m.ChannelID, args, nil)
|
|
return err
|
|
}
|
|
|
|
// Repeat tells the bot to wait for the user's response, then repeat what they
|
|
// said.
|
|
func (bot *Bot) Repeat(m *gateway.MessageCreateEvent) error {
|
|
_, err := bot.Ctx.SendMessage(m.ChannelID,
|
|
"What do you want me to say?", nil)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
|
|
defer cancel()
|
|
|
|
v := bot.Ctx.WaitFor(ctx, func(v interface{}) bool {
|
|
// Incoming event is a message create event:
|
|
mg, ok := v.(*gateway.MessageCreateEvent)
|
|
if !ok {
|
|
return false
|
|
}
|
|
|
|
// Message is from the same author:
|
|
return mg.Author.ID == m.Author.ID
|
|
})
|
|
|
|
if v == nil {
|
|
return errors.New("Timed out waiting for response.")
|
|
}
|
|
|
|
ev := v.(*gateway.MessageCreateEvent)
|
|
|
|
_, err = bot.Ctx.SendMessage(m.ChannelID, ev.Content, nil)
|
|
return err
|
|
}
|
|
|
|
func (bot *Bot) Embed(
|
|
m *gateway.MessageCreateEvent, f *arguments.Flag) error {
|
|
|
|
fs := arguments.NewFlagSet()
|
|
|
|
var (
|
|
title = fs.String("title", "", "Title")
|
|
author = fs.String("author", "", "Author")
|
|
footer = fs.String("footer", "", "Footer")
|
|
color = fs.String("color", "#FFFFFF", "Color in hex format #hhhhhh")
|
|
)
|
|
|
|
if err := f.With(fs.FlagSet); err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(fs.Args()) < 1 {
|
|
return fmt.Errorf("Usage: embed [flags] content...\n" + fs.Usage())
|
|
}
|
|
|
|
// Check if the color string is valid.
|
|
if !strings.HasPrefix(*color, "#") || len(*color) != 7 {
|
|
return errors.New("Invalid color, format must be #hhhhhh")
|
|
}
|
|
|
|
// Parse the color into decimal numbers.
|
|
colorHex, err := strconv.ParseInt((*color)[1:], 16, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// Make a new embed
|
|
embed := discord.Embed{
|
|
Title: *title,
|
|
Description: strings.Join(fs.Args(), " "),
|
|
Color: discord.Color(colorHex),
|
|
}
|
|
|
|
if *author != "" {
|
|
embed.Author = &discord.EmbedAuthor{
|
|
Name: *author,
|
|
}
|
|
}
|
|
if *footer != "" {
|
|
embed.Footer = &discord.EmbedFooter{
|
|
Text: *footer,
|
|
}
|
|
}
|
|
|
|
_, err = bot.Ctx.SendMessage(m.ChannelID, "", &embed)
|
|
return err
|
|
}
|