Add the db for reactions and removals from discord

database/reactionquery.go

@@ -0,0 +1,82 @@
+package database
+
+import (
+	log "maunium.net/go/maulogger/v2"
+	"maunium.net/go/mautrix/id"
+)
+
+type ReactionQuery struct {
+	db  *Database
+	log log.Logger
+}
+
+const (
+	reactionSelect = "SELECT channel_id, receiver, discord_message_id," +
+		" matrix_event_id, author_id, matrix_name, matrix_url, " +
+		" discord_name, discord_id FROM reaction"
+)
+
+func (rq *ReactionQuery) New() *Reaction {
+	return &Reaction{
+		db:  rq.db,
+		log: rq.log,
+	}
+}
+
+func (rq *ReactionQuery) GetAllByDiscordID(key PortalKey, discordMessageID string) []*Reaction {
+	query := reactionSelect + " WHERE channel_id=$1 AND receiver=$2 AND" +
+		" discord_message_id=$3"
+
+	return rq.getAll(query, key.ChannelID, key.Receiver, discordMessageID)
+}
+
+func (rq *ReactionQuery) GetAllByMatrixID(key PortalKey, matrixEventID id.EventID) []*Reaction {
+	query := reactionSelect + " WHERE channel_id=$1 AND receiver=$2 AND" +
+		" matrix_event_id=$3"
+
+	return rq.getAll(query, key.ChannelID, key.Receiver, matrixEventID)
+}
+
+func (rq *ReactionQuery) getAll(query string, args ...interface{}) []*Reaction {
+	rows, err := rq.db.Query(query)
+	if err != nil || rows == nil {
+		return nil
+	}
+
+	reactions := []*Reaction{}
+	for rows.Next() {
+		reactions = append(reactions, rq.New().Scan(rows))
+	}
+
+	return reactions
+}
+
+func (rq *ReactionQuery) GetByDiscordName(key PortalKey, discordMessageID, discordName string) *Reaction {
+	query := reactionSelect + " WHERE channel_id=$1 AND receiver=$2" +
+		" AND discord_message_id=$3 AND discord_name=$4"
+
+	return rq.get(query, key.ChannelID, key.Receiver, discordMessageID, discordName)
+}
+
+func (rq *ReactionQuery) GetByDiscordID(key PortalKey, discordMessageID, discordID string) *Reaction {
+	query := reactionSelect + " WHERE channel_id=$1 AND receiver=$2" +
+		" AND discord_message_id=$3 AND discord_id=$4"
+
+	return rq.get(query, key.ChannelID, key.Receiver, discordMessageID, discordID)
+}
+
+func (rq *ReactionQuery) GetByMatrixName(key PortalKey, matrixEventID id.EventID, matrixName string) *Reaction {
+	query := reactionSelect + " WHERE channel_id=$1 AND receiver=$2" +
+		" AND matrix_event_id=$3 AND matrix_name=$4"
+
+	return rq.get(query, key.ChannelID, key.Receiver, matrixEventID, matrixName)
+}
+
+func (rq *ReactionQuery) get(query string, args ...interface{}) *Reaction {
+	row := rq.db.QueryRow(query, args...)
+	if row == nil {
+		return nil
+	}
+
+	return rq.New().Scan(row)
+}