chore: refresh gramsrv public release
This commit is contained in:
A
parent
75cebe8dbf
commit
70b6820474
1274 changed files with 378751 additions and 59919 deletions
194
internal/store/postgres/channel_message_helpers.go
Normal file
194
internal/store/postgres/channel_message_helpers.go
Normal file
|
|
@ -0,0 +1,194 @@
|
|||
package postgres
|
||||
|
||||
import (
|
||||
"context"
|
||||
"database/sql"
|
||||
"telesrv/internal/domain"
|
||||
"telesrv/internal/store/postgres/sqlcgen"
|
||||
)
|
||||
|
||||
func (s *ChannelStore) eventForChannelMessage(ctx context.Context, channelID int64, messageID int) (domain.ChannelUpdateEvent, error) {
|
||||
row := s.db.QueryRow(ctx, `
|
||||
SELECT channel_id, pts, pts_count, date, event_type, message_id, message_ids::text, sender_user_id, user_ids::text, payload::text
|
||||
FROM channel_update_events
|
||||
WHERE channel_id = $1 AND message_id = $2 AND event_type = $3
|
||||
ORDER BY pts ASC LIMIT 1`, channelID, messageID, string(domain.ChannelUpdateNewMessage))
|
||||
event, _, err := scanChannelEvent(row)
|
||||
return event, err
|
||||
}
|
||||
|
||||
func scanChannelMessage(row rowScanner) (domain.ChannelMessage, error) {
|
||||
var msg domain.ChannelMessage
|
||||
var fromType string
|
||||
var sendAsType sql.NullString
|
||||
var sendAsID sql.NullInt64
|
||||
var replyMsgID, replyTopID int
|
||||
var replyPeerType string
|
||||
var replyPeerID int64
|
||||
var discussionChannelID int64
|
||||
var discussionMessageID int
|
||||
var entities, reply, forward, action string
|
||||
var mediaJSON string
|
||||
var replyMarkupJSON string
|
||||
var savedPeerType string
|
||||
var savedPeerID int64
|
||||
if err := row.Scan(
|
||||
&msg.ChannelID, &msg.ID, &msg.RandomID, &msg.SenderUserID, &fromType, &msg.From.ID,
|
||||
&sendAsType, &sendAsID, &msg.Date, &msg.EditDate, &msg.Post, &msg.Silent, &msg.NoForwards,
|
||||
&msg.Body, &entities, &reply, &replyMsgID, &replyPeerType, &replyPeerID, &replyTopID,
|
||||
&forward, &discussionChannelID, &discussionMessageID, &action, &msg.Pts, &msg.Deleted, &mediaJSON,
|
||||
&replyMarkupJSON, &msg.TTLPeriod, &msg.ExpiresAt, &msg.ViewsCount, &msg.PostAuthor, &msg.Pinned, &msg.ViaBotID, &msg.GroupedID, &msg.FromBoostsApplied, &savedPeerType, &savedPeerID,
|
||||
); err != nil {
|
||||
return domain.ChannelMessage{}, err
|
||||
}
|
||||
msg.From.Type = domain.PeerType(fromType)
|
||||
msg.SavedPeer = domain.Peer{Type: domain.PeerType(savedPeerType), ID: savedPeerID}
|
||||
if sendAsType.Valid && sendAsID.Valid {
|
||||
msg.SendAs = &domain.Peer{Type: domain.PeerType(sendAsType.String), ID: sendAsID.Int64}
|
||||
}
|
||||
parsedEntities, err := decodeMessageEntities(entities)
|
||||
if err != nil {
|
||||
return domain.ChannelMessage{}, err
|
||||
}
|
||||
msg.Entities = parsedEntities
|
||||
msg.ReplyTo = channelMessageReplyFromColumns(decodeJSONPtr[domain.MessageReply](reply), replyMsgID, replyPeerType, replyPeerID, replyTopID)
|
||||
msg.Forward = decodeJSONPtr[domain.MessageForward](forward)
|
||||
if discussionChannelID != 0 && discussionMessageID != 0 {
|
||||
msg.Discussion = &domain.ChannelDiscussionRef{ChannelID: discussionChannelID, MessageID: discussionMessageID}
|
||||
}
|
||||
msg.Action = decodeJSONPtr[domain.ChannelMessageAction](action)
|
||||
msg.Media, err = decodeMessageMedia(mediaJSON)
|
||||
if err != nil {
|
||||
return domain.ChannelMessage{}, err
|
||||
}
|
||||
msg.ReplyMarkup, err = decodeReplyMarkup(replyMarkupJSON)
|
||||
if err != nil {
|
||||
return domain.ChannelMessage{}, err
|
||||
}
|
||||
return msg, nil
|
||||
}
|
||||
|
||||
func scanChannelMessageWithCount(row rowScanner) (domain.ChannelMessage, int, error) {
|
||||
var msg domain.ChannelMessage
|
||||
var fromType string
|
||||
var sendAsType sql.NullString
|
||||
var sendAsID sql.NullInt64
|
||||
var replyMsgID, replyTopID int
|
||||
var replyPeerType string
|
||||
var replyPeerID int64
|
||||
var discussionChannelID int64
|
||||
var discussionMessageID int
|
||||
var entities, reply, forward, action string
|
||||
var count int
|
||||
var mediaJSON string
|
||||
var replyMarkupJSON string
|
||||
var savedPeerType string
|
||||
var savedPeerID int64
|
||||
if err := row.Scan(
|
||||
&msg.ChannelID, &msg.ID, &msg.RandomID, &msg.SenderUserID, &fromType, &msg.From.ID,
|
||||
&sendAsType, &sendAsID, &msg.Date, &msg.EditDate, &msg.Post, &msg.Silent, &msg.NoForwards,
|
||||
&msg.Body, &entities, &reply, &replyMsgID, &replyPeerType, &replyPeerID, &replyTopID,
|
||||
&forward, &discussionChannelID, &discussionMessageID, &action, &msg.Pts, &msg.Deleted, &mediaJSON,
|
||||
&replyMarkupJSON, &msg.TTLPeriod, &msg.ExpiresAt, &msg.ViewsCount, &msg.PostAuthor, &msg.Pinned, &msg.ViaBotID, &msg.GroupedID, &msg.FromBoostsApplied, &savedPeerType, &savedPeerID, &count,
|
||||
); err != nil {
|
||||
return domain.ChannelMessage{}, 0, err
|
||||
}
|
||||
msg.From.Type = domain.PeerType(fromType)
|
||||
msg.SavedPeer = domain.Peer{Type: domain.PeerType(savedPeerType), ID: savedPeerID}
|
||||
if sendAsType.Valid && sendAsID.Valid {
|
||||
msg.SendAs = &domain.Peer{Type: domain.PeerType(sendAsType.String), ID: sendAsID.Int64}
|
||||
}
|
||||
parsedEntities, err := decodeMessageEntities(entities)
|
||||
if err != nil {
|
||||
return domain.ChannelMessage{}, 0, err
|
||||
}
|
||||
msg.Entities = parsedEntities
|
||||
msg.ReplyTo = channelMessageReplyFromColumns(decodeJSONPtr[domain.MessageReply](reply), replyMsgID, replyPeerType, replyPeerID, replyTopID)
|
||||
msg.Forward = decodeJSONPtr[domain.MessageForward](forward)
|
||||
if discussionChannelID != 0 && discussionMessageID != 0 {
|
||||
msg.Discussion = &domain.ChannelDiscussionRef{ChannelID: discussionChannelID, MessageID: discussionMessageID}
|
||||
}
|
||||
msg.Action = decodeJSONPtr[domain.ChannelMessageAction](action)
|
||||
msg.Media, err = decodeMessageMedia(mediaJSON)
|
||||
if err != nil {
|
||||
return domain.ChannelMessage{}, 0, err
|
||||
}
|
||||
msg.ReplyMarkup, err = decodeReplyMarkup(replyMarkupJSON)
|
||||
if err != nil {
|
||||
return domain.ChannelMessage{}, 0, err
|
||||
}
|
||||
return msg, count, nil
|
||||
}
|
||||
|
||||
func channelMessageReplyFromColumns(reply *domain.MessageReply, msgID int, peerType string, peerID int64, topID int) *domain.MessageReply {
|
||||
if reply != nil {
|
||||
if reply.MessageID == 0 {
|
||||
reply.MessageID = msgID
|
||||
}
|
||||
if reply.TopMessageID == 0 {
|
||||
reply.TopMessageID = topID
|
||||
}
|
||||
if reply.Peer.ID == 0 && peerType != "" && peerID != 0 {
|
||||
reply.Peer = domain.Peer{Type: domain.PeerType(peerType), ID: peerID}
|
||||
}
|
||||
if reply.MessageID <= 0 && reply.TopMessageID <= 0 {
|
||||
return nil
|
||||
}
|
||||
return reply
|
||||
}
|
||||
if msgID <= 0 && topID <= 0 {
|
||||
return nil
|
||||
}
|
||||
out := &domain.MessageReply{
|
||||
MessageID: msgID,
|
||||
TopMessageID: topID,
|
||||
}
|
||||
if peerType != "" && peerID != 0 {
|
||||
out.Peer = domain.Peer{Type: domain.PeerType(peerType), ID: peerID}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func collectChannelMessageRefs(msg domain.ChannelMessage, currentChannelID int64, userRefs, channelRefs map[int64]struct{}) {
|
||||
if msg.SenderUserID != 0 {
|
||||
userRefs[msg.SenderUserID] = struct{}{}
|
||||
}
|
||||
addPeerRef(msg.From, currentChannelID, userRefs, channelRefs)
|
||||
if msg.SendAs != nil {
|
||||
addPeerRef(*msg.SendAs, currentChannelID, userRefs, channelRefs)
|
||||
}
|
||||
if msg.Forward != nil {
|
||||
addPeerRef(msg.Forward.From, currentChannelID, userRefs, channelRefs)
|
||||
}
|
||||
if msg.ViaBotID != 0 {
|
||||
userRefs[msg.ViaBotID] = struct{}{}
|
||||
}
|
||||
if msg.ReplyTo != nil {
|
||||
addPeerRef(msg.ReplyTo.Peer, currentChannelID, userRefs, channelRefs)
|
||||
}
|
||||
if msg.Action != nil {
|
||||
for _, id := range msg.Action.UserIDs {
|
||||
if id != 0 {
|
||||
userRefs[id] = struct{}{}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
type pgChannelMessageIDAllocator struct {
|
||||
db sqlcgen.DBTX
|
||||
}
|
||||
|
||||
func (a pgChannelMessageIDAllocator) NextChannelMessageID(ctx context.Context, channelID int64) (int, error) {
|
||||
current, err := a.CurrentChannelMessageID(ctx, channelID)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return current + 1, nil
|
||||
}
|
||||
|
||||
func (a pgChannelMessageIDAllocator) CurrentChannelMessageID(ctx context.Context, channelID int64) (int, error) {
|
||||
var id int
|
||||
err := a.db.QueryRow(ctx, `SELECT COALESCE(MAX(id), 0) FROM channel_messages WHERE channel_id = $1`, channelID).Scan(&id)
|
||||
return id, err
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue