Files

191 lines
7.6 KiB
Go
Raw Permalink Normal View History

package store
import (
"context"
"database/sql"
"time"
)
// Групповые личные беседы (AGENT.md 7.8) живут в тех же таблицах, что и 1:1:
// channels.type = 'dm' + dm_participants. Отличие только в числе участников и
// владельце (channels.dm_owner_id): он добавляет/удаляет участников и
// переименовывает беседу. Беседа с двумя участниками снова считается обычной
// личной, с одним — удаляется (см. RemoveDMParticipant).
// CreateGroupDMChannel создаёт групповую беседу: владелец плюс приглашённые.
func (s *Store) CreateGroupDMChannel(ctx context.Context, ownerID uint64, name string, participantIDs []uint64) (*Channel, error) {
channelID := s.NextID()
members := make([]uint64, 0, len(participantIDs)+1)
members = append(members, ownerID)
for _, participantID := range participantIDs {
if participantID != ownerID {
members = append(members, participantID)
}
}
if err := s.InTx(ctx, func(tx *sql.Tx) error {
if _, err := tx.ExecContext(ctx, `
INSERT INTO channels (id, guild_id, type, name, position, dm_owner_id, created_at)
VALUES (?, NULL, 'dm', ?, 0, ?, ?)`,
int64(channelID), name, int64(ownerID), s.Now()); err != nil {
return err
}
for _, userID := range members {
if _, err := tx.ExecContext(ctx, `
INSERT INTO dm_participants (channel_id, user_id, joined_at) VALUES (?, ?, ?)`,
int64(channelID), int64(userID), s.Now()); err != nil {
return err
}
}
return nil
}); err != nil {
return nil, mapError(err)
}
return s.GetChannel(ctx, channelID)
}
// AddDMParticipant добавляет участника в беседу (повторный вызов — не ошибка).
func (s *Store) AddDMParticipant(ctx context.Context, channelID, userID uint64) error {
_, err := s.writer.ExecContext(ctx, `
INSERT INTO dm_participants (channel_id, user_id, joined_at) VALUES (?, ?, ?)
ON CONFLICT (channel_id, user_id) DO NOTHING`,
int64(channelID), int64(userID), s.Now())
return err
}
// RemoveDMParticipant убирает участника. Когда участников остаётся двое,
// беседа снова обычная личная (имя и владелец сбрасываются), а когда один —
// канал удаляется вместе с сообщениями (FK ON DELETE CASCADE) и файлами
// вложений: переписка с самим собой смысла не имеет.
func (s *Store) RemoveDMParticipant(ctx context.Context, channelID, userID uint64) error {
return s.InTx(ctx, func(tx *sql.Tx) error {
result, err := tx.ExecContext(ctx,
`DELETE FROM dm_participants WHERE channel_id = ? AND user_id = ?`,
int64(channelID), int64(userID))
if err != nil {
return err
}
if affected, err := result.RowsAffected(); err == nil && affected == 0 {
return ErrNotFound
}
var remaining int
if err := tx.QueryRowContext(ctx,
`SELECT COUNT(*) FROM dm_participants WHERE channel_id = ?`, int64(channelID)).Scan(&remaining); err != nil {
return err
}
switch {
case remaining >= 3:
// Владелец ушёл — передаём беседу самому «старому» участнику.
var ownerID sql.NullInt64
if err := tx.QueryRowContext(ctx,
`SELECT dm_owner_id FROM channels WHERE id = ?`, int64(channelID)).Scan(&ownerID); err != nil {
return err
}
if ownerID.Valid && uint64(ownerID.Int64) == userID {
var nextOwner uint64
if err := tx.QueryRowContext(ctx, `
SELECT user_id FROM dm_participants WHERE channel_id = ? ORDER BY joined_at, user_id LIMIT 1`,
int64(channelID)).Scan(&nextOwner); err != nil {
return err
}
if _, err := tx.ExecContext(ctx,
`UPDATE channels SET dm_owner_id = ? WHERE id = ?`, int64(nextOwner), int64(channelID)); err != nil {
return err
}
}
case remaining == 2:
// Беседа снова обычная личная: имя, владелец и иконка группы
// сбрасываются (AGENT.md 7.8, D-070). Файл иконки после этого
// становится сиротой и удаляется уборкой (AGENT.md 7.7).
if _, err := tx.ExecContext(ctx,
`UPDATE channels SET name = '', dm_owner_id = NULL, icon_file_id = NULL WHERE id = ?`,
int64(channelID)); err != nil {
return err
}
default:
if _, err := tx.ExecContext(ctx, `DELETE FROM channels WHERE id = ?`, int64(channelID)); err != nil {
return err
}
}
return nil
})
}
// RenameDMChannel задаёт имя групповой беседы.
func (s *Store) RenameDMChannel(ctx context.Context, channelID uint64, name string) error {
_, err := s.writer.ExecContext(ctx,
`UPDATE channels SET name = ? WHERE id = ? AND type = 'dm'`, name, int64(channelID))
return err
}
// DMParticipantCount считает участников беседы.
func (s *Store) DMParticipantCount(ctx context.Context, channelID uint64) (int, error) {
var count int
err := s.reader.QueryRowContext(ctx,
`SELECT COUNT(*) FROM dm_participants WHERE channel_id = ?`, int64(channelID)).Scan(&count)
return count, err
}
// DMParticipant — участник беседы для интерфейса (список участников группы).
type DMParticipant struct {
UserID uint64
Username string
DisplayName string
AvatarFileID *uint64
Status string
LastSeenAt *time.Time
IsOwner bool
IsCurrentUser bool
}
// DMParticipantProfiles возвращает участников беседы с профилями. Выражение
// «владелец» обёрнуто в COALESCE: у 1:1 dm_owner_id пуст, а сравнение с NULL
// даёт NULL, который нельзя прочитать в int.
func (s *Store) DMParticipantProfiles(ctx context.Context, channelID, currentUserID uint64) ([]DMParticipant, error) {
rows, err := s.reader.QueryContext(ctx, `
SELECT u.id, u.username, u.display_name, u.avatar_file_id, u.status, u.last_seen_at,
COALESCE(c.dm_owner_id = u.id, 0)
FROM dm_participants p
JOIN users u ON u.id = p.user_id
JOIN channels c ON c.id = p.channel_id
WHERE p.channel_id = ? AND u.deleted_at IS NULL
ORDER BY p.joined_at, u.id`, int64(channelID))
if err != nil {
return nil, err
}
defer rows.Close()
participants := make([]DMParticipant, 0, 4)
for rows.Next() {
var (
participant DMParticipant
avatarID sql.NullInt64
lastSeenAt sql.NullString
isOwner int
)
if err := rows.Scan(&participant.UserID, &participant.Username, &participant.DisplayName,
&avatarID, &participant.Status, &lastSeenAt, &isOwner); err != nil {
return nil, err
}
participant.AvatarFileID = optionalID(avatarID)
if lastSeenAt.Valid {
value := parseTimestamp(lastSeenAt.String)
participant.LastSeenAt = &value
}
participant.IsOwner = isOwner == 1
participant.IsCurrentUser = participant.UserID == currentUserID
participants = append(participants, participant)
}
return participants, rows.Err()
}
// GroupDMOwnerID возвращает владельца групповой беседы (nil — обычная 1:1).
func (s *Store) GroupDMOwnerID(ctx context.Context, channelID uint64) (*uint64, error) {
var ownerID sql.NullInt64
err := s.reader.QueryRowContext(ctx,
`SELECT dm_owner_id FROM channels WHERE id = ? AND type = 'dm'`, int64(channelID)).Scan(&ownerID)
if err != nil {
return nil, mapError(err)
}
return optionalID(ownerID), nil
}