Autisticats/mautrix-discord
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Finish up reaction removals from both sides

Browse Source 
Also implement redactions on the matrix side
This commit is contained in:
Gary Kramlich
2022-02-08 03:51:29 -06:00
parent 9d13905a77
commit 8380c4b1b0
6 changed files with 135 additions and 78 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
database/message.go
View File

@@ -54,6 +54,19 @@ func (m *Message) Insert() {
}
}
func (m *Message) Delete() {
query := "DELETE FROM message" +
" WHERE channel_id=$1 AND receiver=$2 AND discord_message_id=$3 AND" +
" matrix_message_id=$4"
_, err := m.db.Exec(query, m.Channel.ChannelID, m.Channel.Receiver,
m.DiscordID, m.MatrixID)
if err != nil {
m.log.Warnfln("Failed to delete %s@%s: %v", m.Channel, m.DiscordID, err)
}
}
func (m *Message) UpdateMatrixID(mxid id.EventID) {
query := "UPDATE message SET matrix_message_id=$1 WHERE channel_id=$2" +
"AND receiver=$3 AND discord_message_id=$4"

4
database/migrations/01-initial.sql
View File

@@ -59,11 +59,9 @@ CREATE TABLE reaction (
matrix_name TEXT,
matrix_url TEXT,
discord_name TEXT,
discord_id TEXT,
CHECK ((discord_name IS NULL AND discord_id IS NOT NULL) OR (discord_name IS NOT NULL AND discord_id IS NULL)),
UNIQUE (discord_name, discord_id, author_id, discord_message_id, channel_id, receiver),
UNIQUE (discord_id, author_id, discord_message_id, channel_id, receiver),
FOREIGN KEY(channel_id, receiver) REFERENCES portal(channel_id, receiver) ON DELETE CASCADE
);

31
database/reaction.go
View File

@@ -23,19 +23,18 @@ type Reaction struct {
MatrixName string
MatrixURL string // Used for custom emoji
DiscordName string // Used for unicode emoji
DiscordID string // Used for custom emoji
DiscordID string // The id or unicode of the emoji for discord
}
func (r *Reaction) Scan(row Scannable) *Reaction {
var discordName, discordID sql.NullString
var discordID sql.NullString
err := row.Scan(
&r.Channel.ChannelID, &r.Channel.Receiver,
&r.DiscordMessageID, &r.MatrixEventID,
&r.AuthorID,
&r.MatrixName, &r.MatrixURL,
&discordName, &discordID)
&discordID)
if err != nil {
if !errors.Is(err, sql.ErrNoRows) {
@@ -45,7 +44,6 @@ func (r *Reaction) Scan(row Scannable) *Reaction {
return nil
}
r.DiscordName = discordName.String
r.DiscordID = discordID.String
return r
@@ -54,14 +52,10 @@ func (r *Reaction) Scan(row Scannable) *Reaction {
func (r *Reaction) Insert() {
query := "INSERT INTO reaction" +
" (channel_id, receiver, discord_message_id, matrix_event_id," +
" author_id, matrix_name, matrix_url, discord_name, discord_id)" +
" VALUES($1, $2, $3, $4, $5, $6, $7, $8, $9);"
" author_id, matrix_name, matrix_url, discord_id)" +
" VALUES($1, $2, $3, $4, $5, $6, $7, $8);"
var discordName, discordID sql.NullString
if r.DiscordName != "" {
discordName = sql.NullString{r.DiscordName, true}
}
var discordID sql.NullString
if r.DiscordID != "" {
discordID = sql.NullString{r.DiscordID, true}
@@ -73,7 +67,7 @@ func (r *Reaction) Insert() {
r.DiscordMessageID, r.MatrixEventID,
r.AuthorID,
r.MatrixName, r.MatrixURL,
discordName, discordID,
discordID,
)
if err != nil {
@@ -90,14 +84,9 @@ func (r *Reaction) Update() {
func (r *Reaction) Delete() {
query := "DELETE FROM reaction WHERE" +
" channel_id=$1 AND receiver=$2 AND discord_message_id=$3 AND" +
" author_id=$4 AND discord_name=$5 AND discord_id=$6"
var discordName, discordID sql.NullString
if r.DiscordName != "" {
discordName = sql.NullString{r.DiscordName, true}
}
" author_id=$4 AND discord_id=$5"
var discordID sql.NullString
if r.DiscordID != "" {
discordID = sql.NullString{r.DiscordID, true}
}
@@ -106,7 +95,7 @@ func (r *Reaction) Delete() {
query,
r.Channel.ChannelID, r.Channel.Receiver,
r.DiscordMessageID, r.AuthorID,
discordName, discordID,
discordID,
)
if err != nil {

15
database/reactionquery.go
View File

@@ -13,7 +13,7 @@ type ReactionQuery struct {
const (
reactionSelect = "SELECT channel_id, receiver, discord_message_id," +
" matrix_event_id, author_id, matrix_name, matrix_url, " +
" discord_name, discord_id FROM reaction"
" discord_id FROM reaction"
)
func (rq *ReactionQuery) New() *Reaction {
@@ -51,13 +51,6 @@ func (rq *ReactionQuery) getAll(query string, args ...interface{}) []*Reaction {
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"
@@ -65,11 +58,11 @@ func (rq *ReactionQuery) GetByDiscordID(key PortalKey, discordMessageID, discord
return rq.get(query, key.ChannelID, key.Receiver, discordMessageID, discordID)
}
func (rq *ReactionQuery) GetByMatrixName(key PortalKey, matrixEventID id.EventID, matrixName string) *Reaction {
func (rq *ReactionQuery) GetByMatrixID(key PortalKey, matrixEventID id.EventID) *Reaction {
query := reactionSelect + " WHERE channel_id=$1 AND receiver=$2" +
" AND matrix_event_id=$3 AND matrix_name=$4"
" AND matrix_event_id=$3"
return rq.get(query, key.ChannelID, key.Receiver, matrixEventID, matrixName)
return rq.get(query, key.ChannelID, key.Receiver, matrixEventID)
}
func (rq *ReactionQuery) get(query string, args ...interface{}) *Reaction {