gmessages/database/message.go

211 lines
6.7 KiB
Go
Raw Normal View History

// mautrix-gmessages - A Matrix-Google Messages puppeting bridge.
// Copyright (C) 2023 Tulir Asokan
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
package database
import (
"context"
"database/sql"
"errors"
2023-07-11 22:57:07 +00:00
"fmt"
"strings"
"time"
2023-08-08 13:13:44 +00:00
"go.mau.fi/util/dbutil"
"maunium.net/go/mautrix/id"
2023-07-17 13:51:31 +00:00
"go.mau.fi/mautrix-gmessages/libgm/gmproto"
)
type MessageQuery struct {
db *Database
}
func (mq *MessageQuery) New() *Message {
return &Message{
db: mq.db,
}
}
func (mq *MessageQuery) getDB() *Database {
return mq.db
}
const (
getMessageByIDQuery = `
SELECT conv_id, conv_receiver, id, mxid, mx_room, sender, timestamp, status FROM message
WHERE conv_receiver=$1 AND id=$2
`
2023-07-11 22:57:07 +00:00
getLastMessageInChatQuery = `
SELECT conv_id, conv_receiver, id, mxid, mx_room, sender, timestamp, status FROM message
2023-07-11 22:57:07 +00:00
WHERE conv_id=$1 AND conv_receiver=$2
ORDER BY timestamp DESC LIMIT 1
`
2023-08-10 08:30:17 +00:00
getLastMessageInChatWithMXIDQuery = `
SELECT conv_id, conv_receiver, id, mxid, mx_room, sender, timestamp, status FROM message
2023-08-10 08:30:17 +00:00
WHERE conv_id=$1 AND conv_receiver=$2 AND mxid NOT LIKE '$fake::%'
ORDER BY timestamp DESC LIMIT 1
`
getMessageByMXIDQuery = `
SELECT conv_id, conv_receiver, id, mxid, mx_room, sender, timestamp, status FROM message
WHERE mxid=$1
`
2023-09-04 22:46:56 +00:00
deleteAllMessagesInChatQuery = `
2023-08-21 16:42:57 +00:00
DELETE FROM message WHERE conv_id=$1 AND conv_receiver=$2
`
2023-09-04 22:46:56 +00:00
insertMessageQuery = `
INSERT INTO message (conv_id, conv_receiver, id, mxid, mx_room, sender, timestamp, status)
VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
`
massInsertMessageQueryPrefix = `
INSERT INTO message (conv_id, conv_receiver, id, mxid, mx_room, sender, timestamp, status)
VALUES
`
updateMessageQuery = `
UPDATE message
SET conv_id=$1, mxid=$4, mx_room=$5, sender=$6, timestamp=$7, status=$8
WHERE conv_receiver=$2 AND id=$3
`
updateMessageStatusQuery = "UPDATE message SET status=$1, timestamp=$2 WHERE conv_receiver=$3 AND id=$4"
deleteMessageQuery = "DELETE FROM message WHERE conv_id=$1 AND conv_receiver=$2 AND id=$3"
)
func (mq *MessageQuery) GetByID(ctx context.Context, receiver int, messageID string) (*Message, error) {
return get[*Message](mq, ctx, getMessageByIDQuery, receiver, messageID)
}
func (mq *MessageQuery) GetByMXID(ctx context.Context, mxid id.EventID) (*Message, error) {
return get[*Message](mq, ctx, getMessageByMXIDQuery, mxid)
}
2023-07-11 22:57:07 +00:00
func (mq *MessageQuery) GetLastInChat(ctx context.Context, chat Key) (*Message, error) {
return get[*Message](mq, ctx, getLastMessageInChatQuery, chat.ID, chat.Receiver)
}
2023-08-10 08:30:17 +00:00
func (mq *MessageQuery) GetLastInChatWithMXID(ctx context.Context, chat Key) (*Message, error) {
return get[*Message](mq, ctx, getLastMessageInChatWithMXIDQuery, chat.ID, chat.Receiver)
}
2023-08-21 16:42:57 +00:00
func (mq *MessageQuery) DeleteAllInChat(ctx context.Context, chat Key) error {
2023-09-04 22:46:56 +00:00
_, err := mq.db.Conn(ctx).ExecContext(ctx, deleteAllMessagesInChatQuery, chat.ID, chat.Receiver)
2023-08-21 16:42:57 +00:00
return err
}
type MediaPart struct {
EventID id.EventID `json:"mxid,omitempty"`
PendingMedia bool `json:"pending_media,omitempty"`
}
2023-07-11 22:57:07 +00:00
type MessageStatus struct {
Type gmproto.MessageStatusType `json:"type,omitempty"`
MediaStatus string `json:"media_status,omitempty"`
MediaParts map[string]MediaPart `json:"media_parts,omitempty"`
2023-08-09 16:49:36 +00:00
PartCount int `json:"part_count,omitempty"`
MSSSent bool `json:"mss_sent,omitempty"`
MSSFailSent bool `json:"mss_fail_sent,omitempty"`
MSSDeliverySent bool `json:"mss_delivery_sent,omitempty"`
ReadReceiptSent bool `json:"read_receipt_sent,omitempty"`
}
func (ms *MessageStatus) HasPendingMediaParts() bool {
for _, part := range ms.MediaParts {
if part.PendingMedia {
return true
}
}
return false
2023-07-11 22:57:07 +00:00
}
type Message struct {
db *Database
Chat Key
ID string
MXID id.EventID
RoomID id.RoomID
Sender string
Timestamp time.Time
2023-07-11 22:57:07 +00:00
Status MessageStatus
}
func (msg *Message) Scan(row dbutil.Scannable) (*Message, error) {
var ts int64
err := row.Scan(&msg.Chat.ID, &msg.Chat.Receiver, &msg.ID, &msg.MXID, &msg.RoomID, &msg.Sender, &ts, dbutil.JSON{Data: &msg.Status})
if errors.Is(err, sql.ErrNoRows) {
return nil, nil
} else if err != nil {
return nil, err
}
if ts != 0 {
2023-07-02 20:47:31 +00:00
msg.Timestamp = time.UnixMicro(ts)
}
return msg, nil
}
func (msg *Message) sqlVariables() []any {
return []any{msg.Chat.ID, msg.Chat.Receiver, msg.ID, msg.MXID, msg.RoomID, msg.Sender, msg.Timestamp.UnixMicro(), dbutil.JSON{Data: &msg.Status}}
}
func (msg *Message) Insert(ctx context.Context) error {
2023-09-04 22:46:56 +00:00
_, err := msg.db.Conn(ctx).ExecContext(ctx, insertMessageQuery, msg.sqlVariables()...)
return err
}
2023-07-11 22:57:07 +00:00
func (mq *MessageQuery) MassInsert(ctx context.Context, messages []*Message) error {
valueStringFormat := "($1, $2, $%d, $%d, $3, $%d, $%d, $%d)"
2023-07-11 22:57:07 +00:00
if mq.db.Dialect == dbutil.SQLite {
valueStringFormat = strings.ReplaceAll(valueStringFormat, "$", "?")
}
placeholders := make([]string, len(messages))
params := make([]any, 3+len(messages)*5)
2023-07-11 22:57:07 +00:00
params[0] = messages[0].Chat.ID
params[1] = messages[0].Chat.Receiver
params[2] = messages[0].RoomID
2023-07-11 22:57:07 +00:00
for i, msg := range messages {
baseIndex := 3 + i*5
2023-07-11 22:57:07 +00:00
params[baseIndex] = msg.ID
params[baseIndex+1] = msg.MXID
params[baseIndex+2] = msg.Sender
params[baseIndex+3] = msg.Timestamp.UnixMicro()
params[baseIndex+4] = dbutil.JSON{Data: &msg.Status}
placeholders[i] = fmt.Sprintf(valueStringFormat, baseIndex+1, baseIndex+2, baseIndex+3, baseIndex+4, baseIndex+5)
}
2023-09-04 22:46:56 +00:00
query := massInsertMessageQueryPrefix + strings.Join(placeholders, ",")
2023-07-11 22:57:07 +00:00
_, err := mq.db.Conn(ctx).ExecContext(ctx, query, params...)
return err
}
func (msg *Message) Update(ctx context.Context) error {
2023-09-04 22:46:56 +00:00
_, err := msg.db.Conn(ctx).ExecContext(ctx, updateMessageQuery, msg.sqlVariables()...)
return err
}
2023-07-11 22:57:07 +00:00
func (msg *Message) UpdateStatus(ctx context.Context) error {
2023-09-04 22:46:56 +00:00
_, err := msg.db.Conn(ctx).ExecContext(ctx, updateMessageStatusQuery, dbutil.JSON{Data: &msg.Status}, msg.Timestamp.UnixMicro(), msg.Chat.Receiver, msg.ID)
2023-07-11 22:57:07 +00:00
return err
}
func (msg *Message) Delete(ctx context.Context) error {
2023-09-04 22:46:56 +00:00
_, err := msg.db.Conn(ctx).ExecContext(ctx, deleteMessageQuery, msg.Chat.ID, msg.Chat.Receiver, msg.ID)
return err
}
2023-08-10 08:30:17 +00:00
func (msg *Message) IsFakeMXID() bool {
return strings.HasPrefix(msg.MXID.String(), "$fake:")
}