76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
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_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) 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) GetByMatrixID(key PortalKey, matrixEventID id.EventID) *Reaction {
|
|
query := reactionSelect + " WHERE channel_id=$1 AND receiver=$2" +
|
|
" AND matrix_event_id=$3"
|
|
|
|
return rq.get(query, key.ChannelID, key.Receiver, matrixEventID)
|
|
}
|
|
|
|
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)
|
|
}
|