FlameInTheDark / dtbot

bot/messages.go

package bot

// BotMessages contains map with key = guild ID, value = array of messages IDs
type BotMessages struct {

type name will be used as bot.BotMessages by other packages, and that stutters; consider calling this Messages

	Messages map[string][]string
}

// NewMessagesMap creates map of bot's messages
func NewMessagesMap() *BotMessages {
	return &BotMessages{Messages: make(map[string][]string)}
}

// Add adds bot message to index
func (m *BotMessages) Add(ctx *Context, messageID string) {
	m.Messages[ctx.Message.ChannelID] = append(m.Messages[ctx.Message.ChannelID], messageID)
	if len(m.Messages[ctx.Message.ChannelID]) > ctx.Conf.General.MessagePool {
		m.Messages[ctx.Message.ChannelID] = m.Messages[ctx.Message.ChannelID][1:]
	}
}

// Clear deletes bot messages
func (m *BotMessages) Clear(ctx *Context, from int) {
	channelID := ctx.Message.ChannelID
	if from >= len(m.Messages[channelID]) {
		return
	}
	if len(m.Messages[channelID][:(len(m.Messages[channelID])-1)-from]) > 0 {
		err := ctx.Discord.ChannelMessagesBulkDelete(ctx.Message.ChannelID, m.Messages[channelID][:(len(m.Messages[channelID])-1)-from])
		if err != nil {
			ctx.Log("Message", ctx.Guild.ID, err.Error())
		}
		m.Messages[channelID] = m.Messages[channelID][(len(m.Messages[channelID])-1)-from:]
	}
}