TerraOceanPlugin/GoBot/discord/modules/embed.go

package modules

import (
	"github.com/bwmarrin/discordgo"

	"TerraOceanBot/discord/config"
)

type embed struct {
	*discordgo.MessageEmbed
}

func newEmbed() *embed {
	return &embed{&discordgo.MessageEmbed{}}
}

func initNewEmbed(title, description string, colour int) *embed {
	return newEmbed().SetTitle(title).SetDescription(description).SetColour(colour)
}

func initNewErrorEmbed(description string) *embed {
	e := newEmbed()
	e.Color = 0xE00000
	e.Title = "发生错误！An error has occurred!"
	e.Description = description
	return e
}

func (e *embed) SetTitle(title string) *embed {
	e.Title = title
	return e
}

func (e *embed) SetColour(colour int) *embed {
	e.Color = colour
	return e
}

func (e *embed) SetDescription(description string) *embed {
	e.Description = description
	return e
}

func (e *embed) AddField(name, value string) *embed {
	e.Fields = append(e.Fields, &discordgo.MessageEmbedField{
		Name:  name,
		Value: value,
	})
	return e
}

func (e *embed) AddInlineField(name, value string) *embed {
	e.Fields = append(e.Fields, &discordgo.MessageEmbedField{
		Name:   name,
		Value:  value,
		Inline: true,
	})
	return e
}

func (e *embed) UpdateVoteStatus(isPositive bool) *embed {
	if isPositive {
		return e.SetColour(config.VotePassColour).SetTitle(config.VotePassTitle)
	}
	return e.SetColour(config.VoteRejectColour).SetTitle(config.VoteRejectTitle)
}

func (e *embed) Build() *discordgo.MessageEmbed {
	return e.MessageEmbed
}

func (e *embed) Send(s *discordgo.Session, channelID string) (*discordgo.Message, error) {
	return s.ChannelMessageSendEmbed(channelID, e.Build())
}

func (e *embed) SendPM(s *discordgo.Session, userID string) (*discordgo.Message, error) {
	channel, err := s.UserChannelCreate(userID)
	if err != nil {
		auditError(s, "", err)
		return nil, err
	}
	return e.Send(s, channel.ID)
}

func (e *embed) Edit(s *discordgo.Session, channelID, messageID string) {
	s.ChannelMessageEditEmbed(channelID, messageID, e.Build())
}