models.go

package mqueue

import (
	"database/sql"
	"time"

	"github.com/jonas747/discordgo/v2"
	"github.com/jonas747/yagpdb/common"
)

// QueuedElement represents a queued message
type QueuedElement struct {
	// The channel to send the message in
	ChannelID int64 `json:"Channel"`
	GuildID   int64 `json:"Guild"`

	ID int64

	// Where this feed originated from, responsible for handling discord specific errors
	Source string
	// Could be stuff like reddit feed element id, youtube feed element id and so on
	SourceItemID string `json:"SourceID"`

	// The actual message as a simple string
	MessageStr string `json:",omitempty"`

	// The actual message as an embed
	MessageEmbed *discordgo.MessageEmbed `json:",omitempty"`

	UseWebhook      bool
	WebhookUsername string

	AllowedMentions discordgo.AllowedMentions `json:"allowed_mentions"`

	// When the queue grows, the feeds with the highest priority gets sent first
	Priority int

	CreatedAt time.Time
}

type webhook struct {
	ID    int64
	Token string

	GuildID   int64
	ChannelID int64

	Plugin string
}

func findCreateWebhook(guildID int64, channelID int64, plugin string, avatar string) (*webhook, error) {
	const query = `
SELECT id, guild_id, channel_id, token, plugin FROM mqueue_webhooks
WHERE guild_id=$1 AND channel_id=$2 AND plugin=$3;
`

	row := common.PQ.QueryRow(query, guildID, channelID, plugin)

	var hook webhook
	err := row.Scan(&hook.ID, &hook.GuildID, &hook.ChannelID, &hook.Token, &hook.Plugin)
	if err != nil {
		if err == sql.ErrNoRows {
			return createWebhook(guildID, channelID, plugin, avatar)
		}

		return nil, err
	}

	return &hook, nil
}

func createWebhook(guildID int64, channelID int64, plugin string, avatar string) (*webhook, error) {
	discordHook, err := common.BotSession.WebhookCreate(channelID, plugin, avatar)
	if err != nil {
		return nil, err
	}

	const query = `
INSERT INTO mqueue_webhooks (id, guild_id, channel_id, token, plugin)
VALUES ($1, $2, $3, $4, $5);
`

	_, err = common.PQ.Exec(query, discordHook.ID, guildID, channelID, discordHook.Token, plugin)
	if err != nil {
		return nil, err
	}

	return &webhook{
		ID:        discordHook.ID,
		Token:     discordHook.Token,
		GuildID:   guildID,
		ChannelID: channelID,
		Plugin:    plugin,
	}, nil
}