92 lines
2.1 KiB
Go
92 lines
2.1 KiB
Go
package commands
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"math/rand"
|
|
"time"
|
|
|
|
tempest "github.com/Amatsagu/Tempest"
|
|
)
|
|
|
|
type MessageOfTheDay struct {
|
|
Date time.Time
|
|
Message tempest.Message
|
|
}
|
|
|
|
type PinnedData struct {
|
|
Messages []tempest.Message
|
|
Count int
|
|
MessageOfTheDay *MessageOfTheDay
|
|
}
|
|
|
|
// not ideal but good enough for this
|
|
var data map[string]*PinnedData = make(map[string]*PinnedData)
|
|
|
|
func InitPinned(c tempest.Client, channelIDs []tempest.Snowflake) error {
|
|
for _, cid := range channelIDs {
|
|
messages, err := getPinnedMessages(c.Rest, cid.String())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
data[cid.String()] = &PinnedData{
|
|
Messages: messages,
|
|
Count: len(messages),
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
var Pinned tempest.Command = tempest.Command{
|
|
Name: "pinned",
|
|
Description: "will show a random pinned message",
|
|
Options: []tempest.Option{},
|
|
SlashCommandHandler: func(itx tempest.CommandInteraction) {
|
|
channelData, exist := data[itx.ChannelId.String()]
|
|
if !exist {
|
|
itx.SendLinearReply("no data for this channel", false)
|
|
}
|
|
|
|
|
|
if channelData.Count == 0 {
|
|
itx.SendLinearReply("no pinned messages", false)
|
|
return
|
|
}
|
|
|
|
today := time.Now()
|
|
todayDate := time.Now().Format("2006/01/02")
|
|
currentMessageDate := channelData.MessageOfTheDay.Date.Format("2006/01/02")
|
|
|
|
if channelData.MessageOfTheDay == nil || todayDate != currentMessageDate {
|
|
newMessage := channelData.Messages[rand.Intn(channelData.Count)]
|
|
|
|
channelData.MessageOfTheDay = &MessageOfTheDay{
|
|
Date: today,
|
|
Message: newMessage,
|
|
}
|
|
}
|
|
|
|
if _, err := itx.Client.SendMessage(itx.ChannelId, channelData.MessageOfTheDay.Message); err != nil {
|
|
itx.SendLinearReply(err.Error(), false)
|
|
}
|
|
},
|
|
}
|
|
|
|
func getPinnedMessages(rest tempest.Rest, channelID string) ([]tempest.Message, error) {
|
|
route := fmt.Sprintf("/channels/%s/pins", channelID)
|
|
bytes, err := rest.Request("GET", route, nil)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
messages := []tempest.Message{}
|
|
if err = json.Unmarshal(bytes, &messages); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return messages, nil
|
|
}
|
|
|