95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package utils
|
|
|
|
import "github.com/bwmarrin/discordgo"
|
|
|
|
type MessageCreate struct {
|
|
*discordgo.MessageCreate
|
|
Session *discordgo.Session
|
|
}
|
|
|
|
type MessageSender struct {
|
|
Embeds []*discordgo.MessageEmbed
|
|
Content string
|
|
Components []discordgo.MessageComponent
|
|
Ephemeral bool
|
|
Reply bool
|
|
AllowedMentions *discordgo.MessageAllowedMentions
|
|
m any
|
|
}
|
|
|
|
func NewMessageSender(m any) *MessageSender {
|
|
return &MessageSender{m: m}
|
|
}
|
|
|
|
func (s *MessageSender) AddEmbeds(embeds ...*discordgo.MessageEmbed) *MessageSender {
|
|
s.Embeds = append(s.Embeds, embeds...)
|
|
return s
|
|
}
|
|
|
|
func (s *MessageSender) AddComponents(components ...discordgo.MessageComponent) *MessageSender {
|
|
s.Components = append(s.Components, components...)
|
|
return s
|
|
}
|
|
|
|
func (s *MessageSender) SetContent(content string) *MessageSender {
|
|
s.Content = content
|
|
return s
|
|
}
|
|
|
|
func (s *MessageSender) SetEphemeral(ephemeral bool) *MessageSender {
|
|
s.Ephemeral = ephemeral
|
|
return s
|
|
}
|
|
|
|
func (s *MessageSender) SetReply(reply bool) *MessageSender {
|
|
s.Reply = reply
|
|
return s
|
|
}
|
|
|
|
func (s *MessageSender) SetAllowedMentions(allowedMentions discordgo.MessageAllowedMentions) *MessageSender {
|
|
s.AllowedMentions = &allowedMentions
|
|
return s
|
|
}
|
|
|
|
func (s *MessageSender) Send() {
|
|
switch m := s.m.(type) {
|
|
case *MessageCreate:
|
|
var reference *discordgo.MessageReference = nil
|
|
|
|
if s.Reply {
|
|
reference = m.Reference()
|
|
}
|
|
|
|
m.Session.ChannelMessageSendComplex(m.ChannelID, &discordgo.MessageSend{
|
|
Content: s.Content,
|
|
Embeds: s.Embeds,
|
|
Components: s.Components,
|
|
AllowedMentions: s.AllowedMentions,
|
|
Reference: reference,
|
|
})
|
|
return
|
|
case *InteractionCreate:
|
|
var flags discordgo.MessageFlags
|
|
|
|
if s.Ephemeral {
|
|
flags = discordgo.MessageFlagsEphemeral
|
|
}
|
|
|
|
if m.Replied || m.Deferred {
|
|
m.EditReply(&discordgo.WebhookEdit{
|
|
Content: &s.Content,
|
|
Embeds: &s.Embeds,
|
|
Components: &s.Components,
|
|
})
|
|
return
|
|
}
|
|
|
|
m.Reply(&discordgo.InteractionResponseData{
|
|
Content: s.Content,
|
|
Embeds: s.Embeds,
|
|
Components: s.Components,
|
|
Flags: flags,
|
|
})
|
|
return
|
|
}
|
|
}
|