feat: sync chatlist sharing support
This commit is contained in:
A
parent
ec6e8fd13d
commit
2a9c12263f
62 changed files with 3408 additions and 212 deletions
834
internal/app/chatlists/service.go
Normal file
834
internal/app/chatlists/service.go
Normal file
|
|
@ -0,0 +1,834 @@
|
|||
package chatlists
|
||||
|
||||
import (
|
||||
"context"
|
||||
"crypto/rand"
|
||||
"encoding/base64"
|
||||
"errors"
|
||||
"fmt"
|
||||
"unicode/utf8"
|
||||
|
||||
"telesrv/internal/domain"
|
||||
"telesrv/internal/links"
|
||||
"telesrv/internal/store"
|
||||
)
|
||||
|
||||
type Service struct {
|
||||
chatlists store.ChatlistStore
|
||||
dialogs store.DialogStore
|
||||
channels ChannelService
|
||||
premium PremiumChecker
|
||||
newSlug func() (string, error)
|
||||
}
|
||||
|
||||
type Option func(*Service)
|
||||
|
||||
// ChannelService is the domain-only channel dependency used for shared-folder
|
||||
// peer membership side effects.
|
||||
type ChannelService interface {
|
||||
GetChannel(ctx context.Context, userID, channelID int64) (domain.ChannelView, error)
|
||||
InviteToChannel(ctx context.Context, userID, channelID int64, userIDs []int64, date int) (domain.CreateChannelResult, error)
|
||||
JoinChannel(ctx context.Context, userID, channelID int64, date int) (domain.CreateChannelResult, error)
|
||||
LeaveChannel(ctx context.Context, userID, channelID int64, date int) (domain.CreateChannelResult, error)
|
||||
}
|
||||
|
||||
type PremiumChecker func(ctx context.Context, userID int64) bool
|
||||
|
||||
func WithChannels(channels ChannelService) Option {
|
||||
return func(s *Service) {
|
||||
s.channels = channels
|
||||
}
|
||||
}
|
||||
|
||||
func WithPremiumChecker(fn PremiumChecker) Option {
|
||||
return func(s *Service) {
|
||||
s.premium = fn
|
||||
}
|
||||
}
|
||||
|
||||
func WithSlugGenerator(fn func() (string, error)) Option {
|
||||
return func(s *Service) {
|
||||
if fn != nil {
|
||||
s.newSlug = fn
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func NewService(chatlists store.ChatlistStore, dialogs store.DialogStore, opts ...Option) *Service {
|
||||
s := &Service{
|
||||
chatlists: chatlists,
|
||||
dialogs: dialogs,
|
||||
newSlug: randomChatlistSlug,
|
||||
}
|
||||
for _, opt := range opts {
|
||||
opt(s)
|
||||
}
|
||||
return s
|
||||
}
|
||||
|
||||
func (s *Service) ExportInvite(ctx context.Context, userID int64, filterID int, title string, peers []domain.DialogFolderPeer, date int) (domain.DialogFolder, domain.ChatlistInvite, error) {
|
||||
if err := validateChatlistUserFilter(userID, filterID); err != nil {
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
if utf8.RuneCountInString(title) > domain.MaxDialogFolderTitleRunes {
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
folder, err := s.ownerFolder(ctx, userID, filterID)
|
||||
if err != nil {
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
if !folderShareable(folder) {
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, domain.ErrChatlistNotShareable
|
||||
}
|
||||
selected, err := selectChatlistPeers(folder, peers, true)
|
||||
if err != nil {
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
if err := s.validateShareableChannelPeers(ctx, userID, selected); err != nil {
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
count, err := s.chatlists.CountActiveInvites(ctx, userID, filterID)
|
||||
if err != nil {
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
if count >= s.invitesLimit(ctx, userID) {
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, domain.ErrChatlistInvitesTooMuch
|
||||
}
|
||||
folder = exportedChatlistFolder(folder, true)
|
||||
var lastErr error
|
||||
for i := 0; i < 8; i++ {
|
||||
slug, err := s.newSlug()
|
||||
if err != nil {
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
invite := domain.ChatlistInvite{
|
||||
OwnerUserID: userID,
|
||||
FilterID: filterID,
|
||||
Slug: slug,
|
||||
Title: title,
|
||||
Peers: selected,
|
||||
Date: date,
|
||||
}
|
||||
saved, err := s.chatlists.SaveInvite(ctx, invite)
|
||||
if err == nil {
|
||||
if err := s.dialogs.UpsertFolder(ctx, userID, folder); err != nil {
|
||||
rollbackErr := s.deleteInviteIfSaved(ctx, userID, filterID, saved.Slug)
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, errors.Join(fmt.Errorf("save exported chatlist folder: %w", err), rollbackErr)
|
||||
}
|
||||
return folder, saved, nil
|
||||
}
|
||||
if !errors.Is(err, domain.ErrChatlistSlugOccupied) {
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
lastErr = err
|
||||
}
|
||||
if lastErr == nil {
|
||||
lastErr = domain.ErrChatlistSlugOccupied
|
||||
}
|
||||
return domain.DialogFolder{}, domain.ChatlistInvite{}, lastErr
|
||||
}
|
||||
|
||||
func (s *Service) ListInvites(ctx context.Context, userID int64, filterID int) ([]domain.ChatlistInvite, error) {
|
||||
if err := validateChatlistUserFilter(userID, filterID); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if _, err := s.ownerFolder(ctx, userID, filterID); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return s.chatlists.ListInvites(ctx, userID, filterID)
|
||||
}
|
||||
|
||||
func (s *Service) EditInvite(ctx context.Context, userID int64, filterID int, slug string, title *string, peers *[]domain.DialogFolderPeer, revoke bool) (domain.ChatlistInvite, error) {
|
||||
if err := validateChatlistUserFilter(userID, filterID); err != nil {
|
||||
return domain.ChatlistInvite{}, err
|
||||
}
|
||||
slug = CleanSlug(slug)
|
||||
if !links.ValidChatlistSlug(slug) {
|
||||
return domain.ChatlistInvite{}, domain.ErrChatlistInviteInvalid
|
||||
}
|
||||
folder, err := s.ownerFolder(ctx, userID, filterID)
|
||||
if err != nil {
|
||||
return domain.ChatlistInvite{}, err
|
||||
}
|
||||
existing, found, err := s.chatlists.GetInvite(ctx, userID, filterID, slug)
|
||||
if err != nil {
|
||||
return domain.ChatlistInvite{}, err
|
||||
}
|
||||
if !found {
|
||||
return domain.ChatlistInvite{}, domain.ErrChatlistInviteExpired
|
||||
}
|
||||
if title != nil {
|
||||
if utf8.RuneCountInString(*title) > domain.MaxDialogFolderTitleRunes {
|
||||
return domain.ChatlistInvite{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
existing.Title = *title
|
||||
}
|
||||
if peers != nil {
|
||||
selected, err := selectChatlistPeers(folder, *peers, true)
|
||||
if err != nil {
|
||||
return domain.ChatlistInvite{}, err
|
||||
}
|
||||
if err := s.validateShareableChannelPeers(ctx, userID, selected); err != nil {
|
||||
return domain.ChatlistInvite{}, err
|
||||
}
|
||||
existing.Peers = selected
|
||||
}
|
||||
if revoke {
|
||||
existing.Revoked = true
|
||||
}
|
||||
return s.chatlists.SaveInvite(ctx, existing)
|
||||
}
|
||||
|
||||
func (s *Service) DeleteInvite(ctx context.Context, userID int64, filterID int, slug string) (domain.DialogFolder, bool, error) {
|
||||
if err := validateChatlistUserFilter(userID, filterID); err != nil {
|
||||
return domain.DialogFolder{}, false, err
|
||||
}
|
||||
slug = CleanSlug(slug)
|
||||
if !links.ValidChatlistSlug(slug) {
|
||||
return domain.DialogFolder{}, false, domain.ErrChatlistInviteInvalid
|
||||
}
|
||||
folder, err := s.ownerFolder(ctx, userID, filterID)
|
||||
if err != nil {
|
||||
return domain.DialogFolder{}, false, err
|
||||
}
|
||||
deleted, err := s.chatlists.DeleteInvite(ctx, userID, filterID, slug)
|
||||
if err != nil || !deleted {
|
||||
return domain.DialogFolder{}, false, err
|
||||
}
|
||||
count, err := s.chatlists.CountInvites(ctx, userID, filterID)
|
||||
if err != nil {
|
||||
return domain.DialogFolder{}, false, err
|
||||
}
|
||||
if count > 0 || !folder.HasMyInvites {
|
||||
return domain.DialogFolder{}, false, nil
|
||||
}
|
||||
folder = exportedChatlistFolder(folder, false)
|
||||
if err := s.dialogs.UpsertFolder(ctx, userID, folder); err != nil {
|
||||
return domain.DialogFolder{}, false, fmt.Errorf("clear exported chatlist folder flag: %w", err)
|
||||
}
|
||||
return folder, true, nil
|
||||
}
|
||||
|
||||
func (s *Service) CheckInvite(ctx context.Context, userID int64, slug string) (domain.ChatlistInvitePreview, error) {
|
||||
if userID == 0 {
|
||||
return domain.ChatlistInvitePreview{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
slug = CleanSlug(slug)
|
||||
if !links.ValidChatlistSlug(slug) {
|
||||
return domain.ChatlistInvitePreview{}, domain.ErrChatlistInviteInvalid
|
||||
}
|
||||
invite, folder, err := s.inviteWithFolder(ctx, slug)
|
||||
if err != nil {
|
||||
return domain.ChatlistInvitePreview{}, err
|
||||
}
|
||||
out := domain.ChatlistInvitePreview{Invite: invite, OwnerFolder: folder}
|
||||
if userID == invite.OwnerUserID {
|
||||
ownerFolder := exportedChatlistFolder(folder, true)
|
||||
out.LocalFolder = &ownerFolder
|
||||
out.Already = peersIntersection(invite.Peers, folderPeers(ownerFolder))
|
||||
return out, nil
|
||||
}
|
||||
membership, found, err := s.chatlists.GetMembershipBySlug(ctx, userID, slug)
|
||||
if err != nil {
|
||||
return domain.ChatlistInvitePreview{}, err
|
||||
}
|
||||
if !found {
|
||||
out.Missing = cloneFolderPeers(invite.Peers)
|
||||
return out, nil
|
||||
}
|
||||
local, found, err := s.dialogs.GetFolder(ctx, userID, membership.LocalFilterID)
|
||||
if err != nil {
|
||||
return domain.ChatlistInvitePreview{}, err
|
||||
}
|
||||
if !found {
|
||||
return domain.ChatlistInvitePreview{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
out.Membership = &membership
|
||||
out.LocalFolder = &local
|
||||
out.Already = peersIntersection(invite.Peers, folderPeers(local))
|
||||
out.Missing = peersDifference(invite.Peers, folderPeers(local))
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func (s *Service) JoinInvite(ctx context.Context, userID int64, slug string, peers []domain.DialogFolderPeer, date int) (domain.ChatlistJoinResult, error) {
|
||||
if userID == 0 {
|
||||
return domain.ChatlistJoinResult{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
slug = CleanSlug(slug)
|
||||
if !links.ValidChatlistSlug(slug) {
|
||||
return domain.ChatlistJoinResult{}, domain.ErrChatlistInviteInvalid
|
||||
}
|
||||
invite, ownerFolder, err := s.inviteWithFolder(ctx, slug)
|
||||
if err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
}
|
||||
if userID == invite.OwnerUserID {
|
||||
return domain.ChatlistJoinResult{Folder: exportedChatlistFolder(ownerFolder, true), Date: date}, nil
|
||||
}
|
||||
if existing, found, err := s.chatlists.GetMembershipBySlug(ctx, userID, slug); err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
} else if found {
|
||||
folder, found, err := s.dialogs.GetFolder(ctx, userID, existing.LocalFilterID)
|
||||
if err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
}
|
||||
if !found {
|
||||
return domain.ChatlistJoinResult{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
return domain.ChatlistJoinResult{Folder: folder, Membership: existing, Date: date}, nil
|
||||
}
|
||||
selected, err := selectInvitePeers(invite, peers, true)
|
||||
if err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
}
|
||||
count, err := s.chatlists.CountMemberships(ctx, userID)
|
||||
if err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
}
|
||||
if count >= s.joinedLimit(ctx, userID) {
|
||||
return domain.ChatlistJoinResult{}, domain.ErrChatlistsTooMuch
|
||||
}
|
||||
channelResults, err := s.joinChannelPeers(ctx, invite.OwnerUserID, userID, selected, date)
|
||||
if err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
}
|
||||
filterID, err := s.nextLocalFilterID(ctx, userID)
|
||||
if err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
}
|
||||
folder := importedChatlistFolder(ownerFolder, filterID, selected)
|
||||
membership := domain.ChatlistMembership{
|
||||
UserID: userID,
|
||||
LocalFilterID: filterID,
|
||||
OwnerUserID: invite.OwnerUserID,
|
||||
OwnerFilterID: invite.FilterID,
|
||||
Slug: invite.Slug,
|
||||
Date: date,
|
||||
}
|
||||
if err := s.dialogs.UpsertFolder(ctx, userID, folder); err != nil {
|
||||
if rollbackErr := s.leaveJoinedChannelResults(ctx, userID, channelResults, date); rollbackErr != nil {
|
||||
return domain.ChatlistJoinResult{}, errors.Join(fmt.Errorf("save joined chatlist folder: %w", err), rollbackErr)
|
||||
}
|
||||
return domain.ChatlistJoinResult{}, fmt.Errorf("save joined chatlist folder: %w", err)
|
||||
}
|
||||
if err := s.chatlists.SaveMembership(ctx, membership); err != nil {
|
||||
rollbackErr := errors.Join(
|
||||
s.dialogs.DeleteFolder(ctx, userID, filterID),
|
||||
s.leaveJoinedChannelResults(ctx, userID, channelResults, date),
|
||||
)
|
||||
return domain.ChatlistJoinResult{}, errors.Join(err, rollbackErr)
|
||||
}
|
||||
return domain.ChatlistJoinResult{Folder: folder, Membership: membership, Date: date, ChannelResults: channelResults}, nil
|
||||
}
|
||||
|
||||
func (s *Service) GetUpdates(ctx context.Context, userID int64, localFilterID int) (domain.ChatlistUpdates, error) {
|
||||
membership, folder, invite, err := s.memberFolderInvite(ctx, userID, localFilterID)
|
||||
if err != nil {
|
||||
if errors.Is(err, domain.ErrChatlistInviteExpired) {
|
||||
return domain.ChatlistUpdates{Membership: membership}, nil
|
||||
}
|
||||
if errors.Is(err, domain.ErrChatlistInvalid) {
|
||||
if owner, ownerErr := s.ownerFolder(ctx, userID, localFilterID); ownerErr == nil && owner.IsChatlist && owner.HasMyInvites {
|
||||
return domain.ChatlistUpdates{}, nil
|
||||
}
|
||||
}
|
||||
return domain.ChatlistUpdates{}, err
|
||||
}
|
||||
if membership.HiddenUpdates {
|
||||
return domain.ChatlistUpdates{Membership: membership}, nil
|
||||
}
|
||||
return domain.ChatlistUpdates{
|
||||
Membership: membership,
|
||||
Missing: peersDifference(invite.Peers, folderPeers(folder)),
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (s *Service) JoinUpdates(ctx context.Context, userID int64, localFilterID int, peers []domain.DialogFolderPeer, date int) (domain.ChatlistJoinResult, error) {
|
||||
membership, folder, invite, err := s.memberFolderInvite(ctx, userID, localFilterID)
|
||||
if err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
}
|
||||
selected, err := selectInvitePeers(invite, peers, true)
|
||||
if err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
}
|
||||
if _, err := s.chatlists.SetMembershipHidden(ctx, userID, membership.LocalFilterID, false); err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
}
|
||||
channelResults, err := s.joinChannelPeers(ctx, membership.OwnerUserID, userID, selected, date)
|
||||
if err != nil {
|
||||
return domain.ChatlistJoinResult{}, err
|
||||
}
|
||||
folder.IncludePeers = mergeFolderPeers(folder.IncludePeers, selected)
|
||||
folder.IsChatlist = true
|
||||
if err := s.dialogs.UpsertFolder(ctx, userID, folder); err != nil {
|
||||
if rollbackErr := s.leaveJoinedChannelResults(ctx, userID, channelResults, date); rollbackErr != nil {
|
||||
return domain.ChatlistJoinResult{}, errors.Join(fmt.Errorf("save chatlist updates: %w", err), rollbackErr)
|
||||
}
|
||||
return domain.ChatlistJoinResult{}, fmt.Errorf("save chatlist updates: %w", err)
|
||||
}
|
||||
return domain.ChatlistJoinResult{Folder: folder, Membership: membership, ChannelResults: channelResults}, nil
|
||||
}
|
||||
|
||||
func (s *Service) HideUpdates(ctx context.Context, userID int64, localFilterID int) error {
|
||||
if err := validateChatlistUserFilter(userID, localFilterID); err != nil {
|
||||
return err
|
||||
}
|
||||
if _, found, err := s.chatlists.GetMembershipByLocalFilter(ctx, userID, localFilterID); err != nil {
|
||||
return err
|
||||
} else if !found {
|
||||
return domain.ErrChatlistInvalid
|
||||
}
|
||||
_, err := s.chatlists.SetMembershipHidden(ctx, userID, localFilterID, true)
|
||||
return err
|
||||
}
|
||||
|
||||
func (s *Service) Leave(ctx context.Context, userID int64, localFilterID int, peers []domain.DialogFolderPeer, date int) (domain.ChatlistLeaveResult, error) {
|
||||
if err := validateChatlistUserFilter(userID, localFilterID); err != nil {
|
||||
return domain.ChatlistLeaveResult{}, err
|
||||
}
|
||||
if _, found, err := s.chatlists.GetMembershipByLocalFilter(ctx, userID, localFilterID); err != nil {
|
||||
return domain.ChatlistLeaveResult{}, err
|
||||
} else if !found {
|
||||
return domain.ChatlistLeaveResult{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
folder, found, err := s.dialogs.GetFolder(ctx, userID, localFilterID)
|
||||
if err != nil {
|
||||
return domain.ChatlistLeaveResult{}, err
|
||||
}
|
||||
if !found {
|
||||
return domain.ChatlistLeaveResult{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
selected, err := selectPeersFromAllowed(folderPeerMap(channelFolderPeers(folderPeers(folder))), peers, false)
|
||||
if err != nil {
|
||||
return domain.ChatlistLeaveResult{}, err
|
||||
}
|
||||
channelResults, err := s.leaveChannelPeers(ctx, userID, selected, date)
|
||||
if err != nil {
|
||||
return domain.ChatlistLeaveResult{}, err
|
||||
}
|
||||
if err := s.dialogs.DeleteFolder(ctx, userID, localFilterID); err != nil {
|
||||
return domain.ChatlistLeaveResult{}, err
|
||||
}
|
||||
if _, err := s.chatlists.DeleteMembershipByLocalFilter(ctx, userID, localFilterID); err != nil {
|
||||
restoreErr := s.dialogs.UpsertFolder(ctx, userID, folder)
|
||||
return domain.ChatlistLeaveResult{}, errors.Join(err, restoreErr)
|
||||
}
|
||||
return domain.ChatlistLeaveResult{FilterID: localFilterID, ChannelResults: channelResults, RequestedLeaves: selected}, nil
|
||||
}
|
||||
|
||||
func (s *Service) LeaveSuggestions(ctx context.Context, userID int64, localFilterID int) ([]domain.DialogFolderPeer, error) {
|
||||
if err := validateChatlistUserFilter(userID, localFilterID); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if _, found, err := s.chatlists.GetMembershipByLocalFilter(ctx, userID, localFilterID); err != nil {
|
||||
return nil, err
|
||||
} else if !found {
|
||||
return nil, domain.ErrChatlistInvalid
|
||||
}
|
||||
folder, found, err := s.dialogs.GetFolder(ctx, userID, localFilterID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if !found {
|
||||
return nil, domain.ErrChatlistInvalid
|
||||
}
|
||||
return channelFolderPeers(folderPeers(folder)), nil
|
||||
}
|
||||
|
||||
func (s *Service) ownerFolder(ctx context.Context, userID int64, filterID int) (domain.DialogFolder, error) {
|
||||
if s == nil || s.chatlists == nil || s.dialogs == nil {
|
||||
return domain.DialogFolder{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
folder, found, err := s.dialogs.GetFolder(ctx, userID, filterID)
|
||||
if err != nil {
|
||||
return domain.DialogFolder{}, err
|
||||
}
|
||||
if !found {
|
||||
return domain.DialogFolder{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
folder.ID = filterID
|
||||
return folder, nil
|
||||
}
|
||||
|
||||
func (s *Service) inviteWithFolder(ctx context.Context, slug string) (domain.ChatlistInvite, domain.DialogFolder, error) {
|
||||
invite, found, err := s.chatlists.GetInviteBySlug(ctx, slug)
|
||||
if err != nil {
|
||||
return domain.ChatlistInvite{}, domain.DialogFolder{}, err
|
||||
}
|
||||
if !found {
|
||||
return domain.ChatlistInvite{}, domain.DialogFolder{}, domain.ErrChatlistInviteExpired
|
||||
}
|
||||
folder, found, err := s.dialogs.GetFolder(ctx, invite.OwnerUserID, invite.FilterID)
|
||||
if err != nil {
|
||||
return domain.ChatlistInvite{}, domain.DialogFolder{}, err
|
||||
}
|
||||
if !found {
|
||||
return domain.ChatlistInvite{}, domain.DialogFolder{}, domain.ErrChatlistInviteExpired
|
||||
}
|
||||
return invite, exportedChatlistFolder(folder, true), nil
|
||||
}
|
||||
|
||||
func (s *Service) memberFolderInvite(ctx context.Context, userID int64, localFilterID int) (domain.ChatlistMembership, domain.DialogFolder, domain.ChatlistInvite, error) {
|
||||
if err := validateChatlistUserFilter(userID, localFilterID); err != nil {
|
||||
return domain.ChatlistMembership{}, domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
membership, found, err := s.chatlists.GetMembershipByLocalFilter(ctx, userID, localFilterID)
|
||||
if err != nil {
|
||||
return domain.ChatlistMembership{}, domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
if !found {
|
||||
return domain.ChatlistMembership{}, domain.DialogFolder{}, domain.ChatlistInvite{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
folder, found, err := s.dialogs.GetFolder(ctx, userID, localFilterID)
|
||||
if err != nil {
|
||||
return membership, domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
if !found {
|
||||
return membership, domain.DialogFolder{}, domain.ChatlistInvite{}, domain.ErrChatlistInvalid
|
||||
}
|
||||
invite, found, err := s.chatlists.GetInviteBySlug(ctx, membership.Slug)
|
||||
if err != nil {
|
||||
return membership, domain.DialogFolder{}, domain.ChatlistInvite{}, err
|
||||
}
|
||||
if !found {
|
||||
return membership, folder, domain.ChatlistInvite{}, domain.ErrChatlistInviteExpired
|
||||
}
|
||||
return membership, folder, invite, nil
|
||||
}
|
||||
|
||||
func (s *Service) nextLocalFilterID(ctx context.Context, userID int64) (int, error) {
|
||||
list, err := s.dialogs.ListFolders(ctx, userID)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
used := make(map[int]struct{}, len(list.Folders))
|
||||
for _, folder := range list.Folders {
|
||||
used[folder.ID] = struct{}{}
|
||||
}
|
||||
for id := domain.DialogCustomFolderMinID; id < domain.DialogCustomFolderMinID+domain.MaxDialogFolders; id++ {
|
||||
if _, ok := used[id]; !ok {
|
||||
return id, nil
|
||||
}
|
||||
}
|
||||
return 0, domain.ErrChatlistsTooMuch
|
||||
}
|
||||
|
||||
func validateChatlistUserFilter(userID int64, filterID int) error {
|
||||
if userID == 0 || filterID < domain.DialogCustomFolderMinID {
|
||||
return domain.ErrChatlistInvalid
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func folderShareable(folder domain.DialogFolder) bool {
|
||||
if len(folder.ExcludePeers) > 0 || folder.Contacts || folder.NonContacts || folder.Groups ||
|
||||
folder.Broadcasts || folder.Bots || folder.ExcludeMuted || folder.ExcludeRead || folder.ExcludeArchived {
|
||||
return false
|
||||
}
|
||||
return len(folder.IncludePeers)+len(folder.PinnedPeers) > 0
|
||||
}
|
||||
|
||||
func (s *Service) validateShareableChannelPeers(ctx context.Context, userID int64, peers []domain.DialogFolderPeer) error {
|
||||
if len(peers) == 0 {
|
||||
return domain.ErrChatlistPeersEmpty
|
||||
}
|
||||
for _, item := range peers {
|
||||
if item.Peer.Type != domain.PeerTypeChannel || item.Peer.ID == 0 {
|
||||
return domain.ErrChatlistNotShareable
|
||||
}
|
||||
if s.channels == nil {
|
||||
continue
|
||||
}
|
||||
view, err := s.channels.GetChannel(ctx, userID, item.Peer.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !channelShareableForChatlist(view) {
|
||||
return domain.ErrChatlistNotShareable
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func channelShareableForChatlist(view domain.ChannelView) bool {
|
||||
if view.Channel.Deleted || view.Forbidden {
|
||||
return false
|
||||
}
|
||||
return channelCanInviteForChatlist(view) || channelPublicJoinableForChatlist(view)
|
||||
}
|
||||
|
||||
func channelCanInviteForChatlist(view domain.ChannelView) bool {
|
||||
switch view.Self.Role {
|
||||
case domain.ChannelRoleCreator:
|
||||
return true
|
||||
case domain.ChannelRoleAdmin:
|
||||
if view.Self.AdminRights.InviteUsers {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func channelPublicJoinableForChatlist(view domain.ChannelView) bool {
|
||||
if view.Channel.Deleted || view.Forbidden {
|
||||
return false
|
||||
}
|
||||
return view.Channel.Username != "" && !view.Channel.JoinRequest
|
||||
}
|
||||
|
||||
func (s *Service) joinChannelPeers(ctx context.Context, ownerUserID, userID int64, peers []domain.DialogFolderPeer, date int) ([]domain.CreateChannelResult, error) {
|
||||
if s.channels == nil {
|
||||
return nil, nil
|
||||
}
|
||||
type joinPlan struct {
|
||||
peer domain.DialogFolderPeer
|
||||
useInvite bool
|
||||
}
|
||||
channelPeers := channelFolderPeers(peers)
|
||||
plans := make([]joinPlan, 0, len(channelPeers))
|
||||
for _, item := range channelPeers {
|
||||
view, err := s.channels.GetChannel(ctx, ownerUserID, item.Peer.ID)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if !channelShareableForChatlist(view) {
|
||||
return nil, domain.ErrChatlistNotShareable
|
||||
}
|
||||
plans = append(plans, joinPlan{peer: item, useInvite: channelCanInviteForChatlist(view)})
|
||||
}
|
||||
results := make([]domain.CreateChannelResult, 0, len(plans))
|
||||
for _, plan := range plans {
|
||||
var res domain.CreateChannelResult
|
||||
var err error
|
||||
if plan.useInvite {
|
||||
res, err = s.channels.InviteToChannel(ctx, ownerUserID, plan.peer.Peer.ID, []int64{userID}, date)
|
||||
} else {
|
||||
res, err = s.channels.JoinChannel(ctx, userID, plan.peer.Peer.ID, date)
|
||||
}
|
||||
if err != nil {
|
||||
if errors.Is(err, domain.ErrUserAlreadyParticipant) {
|
||||
continue
|
||||
}
|
||||
return results, err
|
||||
}
|
||||
results = append(results, res)
|
||||
}
|
||||
return results, nil
|
||||
}
|
||||
|
||||
func (s *Service) leaveChannelPeers(ctx context.Context, userID int64, peers []domain.DialogFolderPeer, date int) ([]domain.CreateChannelResult, error) {
|
||||
if s.channels == nil {
|
||||
return nil, nil
|
||||
}
|
||||
results := make([]domain.CreateChannelResult, 0)
|
||||
for _, item := range channelFolderPeers(peers) {
|
||||
res, err := s.channels.LeaveChannel(ctx, userID, item.Peer.ID, date)
|
||||
if err != nil {
|
||||
if errors.Is(err, domain.ErrUserNotParticipant) || errors.Is(err, domain.ErrChannelPrivate) {
|
||||
continue
|
||||
}
|
||||
return results, err
|
||||
}
|
||||
results = append(results, res)
|
||||
}
|
||||
return results, nil
|
||||
}
|
||||
|
||||
func (s *Service) leaveJoinedChannelResults(ctx context.Context, userID int64, results []domain.CreateChannelResult, date int) error {
|
||||
if s.channels == nil || len(results) == 0 {
|
||||
return nil
|
||||
}
|
||||
var joinedErr error
|
||||
for _, res := range results {
|
||||
if res.Channel.ID == 0 {
|
||||
continue
|
||||
}
|
||||
if _, err := s.channels.LeaveChannel(ctx, userID, res.Channel.ID, date); err != nil &&
|
||||
!errors.Is(err, domain.ErrUserNotParticipant) &&
|
||||
!errors.Is(err, domain.ErrChannelPrivate) {
|
||||
joinedErr = errors.Join(joinedErr, err)
|
||||
}
|
||||
}
|
||||
return joinedErr
|
||||
}
|
||||
|
||||
func (s *Service) deleteInviteIfSaved(ctx context.Context, ownerUserID int64, filterID int, slug string) error {
|
||||
if slug == "" {
|
||||
return nil
|
||||
}
|
||||
if _, err := s.chatlists.DeleteInvite(ctx, ownerUserID, filterID, slug); err != nil {
|
||||
return fmt.Errorf("rollback chatlist invite: %w", err)
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Service) invitesLimit(ctx context.Context, userID int64) int {
|
||||
if s != nil && s.premium != nil && s.premium(ctx, userID) {
|
||||
return domain.MaxChatlistInvitesPremium
|
||||
}
|
||||
return domain.MaxChatlistInvitesDefault
|
||||
}
|
||||
|
||||
func (s *Service) joinedLimit(ctx context.Context, userID int64) int {
|
||||
if s != nil && s.premium != nil && s.premium(ctx, userID) {
|
||||
return domain.MaxChatlistsJoinedPremium
|
||||
}
|
||||
return domain.MaxChatlistsJoinedDefault
|
||||
}
|
||||
|
||||
func exportedChatlistFolder(folder domain.DialogFolder, hasMyInvites bool) domain.DialogFolder {
|
||||
folder.Contacts = false
|
||||
folder.NonContacts = false
|
||||
folder.Groups = false
|
||||
folder.Broadcasts = false
|
||||
folder.Bots = false
|
||||
folder.ExcludeMuted = false
|
||||
folder.ExcludeRead = false
|
||||
folder.ExcludeArchived = false
|
||||
folder.ExcludePeers = nil
|
||||
folder.IsChatlist = true
|
||||
folder.HasMyInvites = hasMyInvites
|
||||
return cloneDialogFolder(folder)
|
||||
}
|
||||
|
||||
func importedChatlistFolder(source domain.DialogFolder, filterID int, peers []domain.DialogFolderPeer) domain.DialogFolder {
|
||||
source = exportedChatlistFolder(source, false)
|
||||
source.ID = filterID
|
||||
source.PinnedPeers = nil
|
||||
source.IncludePeers = cloneFolderPeers(peers)
|
||||
source.HasMyInvites = false
|
||||
return source
|
||||
}
|
||||
|
||||
func selectChatlistPeers(folder domain.DialogFolder, requested []domain.DialogFolderPeer, requireNonEmpty bool) ([]domain.DialogFolderPeer, error) {
|
||||
allowed := folderPeerMap(folderPeers(folder))
|
||||
return selectPeersFromAllowed(allowed, requested, requireNonEmpty)
|
||||
}
|
||||
|
||||
func selectInvitePeers(invite domain.ChatlistInvite, requested []domain.DialogFolderPeer, requireNonEmpty bool) ([]domain.DialogFolderPeer, error) {
|
||||
if len(requested) == 0 {
|
||||
requested = invite.Peers
|
||||
}
|
||||
return selectPeersFromAllowed(folderPeerMap(invite.Peers), requested, requireNonEmpty)
|
||||
}
|
||||
|
||||
func selectPeersFromAllowed(allowed map[domain.Peer]domain.DialogFolderPeer, requested []domain.DialogFolderPeer, requireNonEmpty bool) ([]domain.DialogFolderPeer, error) {
|
||||
if len(requested) > domain.MaxChatlistInvitePeers {
|
||||
return nil, domain.ErrChatlistPeersTooMuch
|
||||
}
|
||||
out := make([]domain.DialogFolderPeer, 0, len(requested))
|
||||
seen := make(map[domain.Peer]struct{}, len(requested))
|
||||
for _, item := range requested {
|
||||
if item.Peer.Type == "" || item.Peer.ID == 0 {
|
||||
return nil, domain.ErrChatlistInvalid
|
||||
}
|
||||
allowedPeer, ok := allowed[item.Peer]
|
||||
if !ok {
|
||||
return nil, domain.ErrChatlistInvalid
|
||||
}
|
||||
if _, ok := seen[item.Peer]; ok {
|
||||
continue
|
||||
}
|
||||
seen[item.Peer] = struct{}{}
|
||||
if item.AccessHash == 0 {
|
||||
item.AccessHash = allowedPeer.AccessHash
|
||||
}
|
||||
out = append(out, item)
|
||||
}
|
||||
if requireNonEmpty && len(out) == 0 {
|
||||
return nil, domain.ErrChatlistPeersEmpty
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func channelFolderPeers(peers []domain.DialogFolderPeer) []domain.DialogFolderPeer {
|
||||
out := make([]domain.DialogFolderPeer, 0, len(peers))
|
||||
for _, item := range peers {
|
||||
if item.Peer.Type == domain.PeerTypeChannel && item.Peer.ID != 0 {
|
||||
out = append(out, item)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func folderPeers(folder domain.DialogFolder) []domain.DialogFolderPeer {
|
||||
return mergeFolderPeers(folder.PinnedPeers, folder.IncludePeers)
|
||||
}
|
||||
|
||||
func folderPeerMap(peers []domain.DialogFolderPeer) map[domain.Peer]domain.DialogFolderPeer {
|
||||
out := make(map[domain.Peer]domain.DialogFolderPeer, len(peers))
|
||||
for _, item := range peers {
|
||||
if item.Peer.Type == "" || item.Peer.ID == 0 {
|
||||
continue
|
||||
}
|
||||
if _, ok := out[item.Peer]; !ok {
|
||||
out[item.Peer] = item
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func mergeFolderPeers(a, b []domain.DialogFolderPeer) []domain.DialogFolderPeer {
|
||||
out := make([]domain.DialogFolderPeer, 0, len(a)+len(b))
|
||||
seen := make(map[domain.Peer]struct{}, len(a)+len(b))
|
||||
for _, list := range [][]domain.DialogFolderPeer{a, b} {
|
||||
for _, item := range list {
|
||||
if item.Peer.Type == "" || item.Peer.ID == 0 {
|
||||
continue
|
||||
}
|
||||
if _, ok := seen[item.Peer]; ok {
|
||||
continue
|
||||
}
|
||||
seen[item.Peer] = struct{}{}
|
||||
out = append(out, item)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func peersDifference(all, existing []domain.DialogFolderPeer) []domain.DialogFolderPeer {
|
||||
existingMap := folderPeerMap(existing)
|
||||
out := make([]domain.DialogFolderPeer, 0)
|
||||
for _, item := range all {
|
||||
if _, ok := existingMap[item.Peer]; !ok {
|
||||
out = append(out, item)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func peersIntersection(all, existing []domain.DialogFolderPeer) []domain.DialogFolderPeer {
|
||||
existingMap := folderPeerMap(existing)
|
||||
out := make([]domain.DialogFolderPeer, 0)
|
||||
for _, item := range all {
|
||||
if _, ok := existingMap[item.Peer]; ok {
|
||||
out = append(out, item)
|
||||
}
|
||||
}
|
||||
return out
|
||||
}
|
||||
|
||||
func cloneDialogFolder(folder domain.DialogFolder) domain.DialogFolder {
|
||||
folder.TitleEntities = append([]domain.MessageEntity(nil), folder.TitleEntities...)
|
||||
folder.PinnedPeers = cloneFolderPeers(folder.PinnedPeers)
|
||||
folder.IncludePeers = cloneFolderPeers(folder.IncludePeers)
|
||||
folder.ExcludePeers = cloneFolderPeers(folder.ExcludePeers)
|
||||
return folder
|
||||
}
|
||||
|
||||
func cloneFolderPeers(peers []domain.DialogFolderPeer) []domain.DialogFolderPeer {
|
||||
return append([]domain.DialogFolderPeer(nil), peers...)
|
||||
}
|
||||
|
||||
func CleanSlug(raw string) string {
|
||||
return links.CleanChatlistSlug(raw)
|
||||
}
|
||||
|
||||
func randomChatlistSlug() (string, error) {
|
||||
var b [12]byte
|
||||
if _, err := rand.Read(b[:]); err != nil {
|
||||
return "", fmt.Errorf("chatlist slug rand: %w", err)
|
||||
}
|
||||
return base64.RawURLEncoding.EncodeToString(b[:]), nil
|
||||
}
|
||||
488
internal/app/chatlists/service_test.go
Normal file
488
internal/app/chatlists/service_test.go
Normal file
|
|
@ -0,0 +1,488 @@
|
|||
package chatlists
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"telesrv/internal/domain"
|
||||
"telesrv/internal/links"
|
||||
"telesrv/internal/store/memory"
|
||||
)
|
||||
|
||||
func TestSharedFolderInviteJoinUpdatesAndLeave(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
dialogs := memory.NewDialogStore()
|
||||
chatlists := memory.NewChatlistStore()
|
||||
svc := NewService(chatlists, dialogs, WithSlugGenerator(func() (string, error) { return "slug-one", nil }))
|
||||
|
||||
ownerID := int64(1001)
|
||||
viewerID := int64(2002)
|
||||
peerA := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 3001}, AccessHash: 31}
|
||||
peerB := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 3002}, AccessHash: 32}
|
||||
peerC := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 3003}, AccessHash: 33}
|
||||
|
||||
if err := dialogs.UpsertFolder(ctx, ownerID, domain.DialogFolder{
|
||||
ID: 2,
|
||||
Title: "Team",
|
||||
IncludePeers: []domain.DialogFolderPeer{peerA},
|
||||
}); err != nil {
|
||||
t.Fatalf("seed owner folder: %v", err)
|
||||
}
|
||||
|
||||
folder, invite, err := svc.ExportInvite(ctx, ownerID, 2, "Main link", []domain.DialogFolderPeer{peerA}, 100)
|
||||
if err != nil {
|
||||
t.Fatalf("ExportInvite: %v", err)
|
||||
}
|
||||
if invite.Slug != "slug-one" || !folder.IsChatlist || !folder.HasMyInvites {
|
||||
t.Fatalf("export = folder %+v invite %+v, want chatlist with invite slug", folder, invite)
|
||||
}
|
||||
persisted, found, err := dialogs.GetFolder(ctx, ownerID, 2)
|
||||
if err != nil || !found || !persisted.IsChatlist || !persisted.HasMyInvites {
|
||||
t.Fatalf("persisted folder = %+v found %v err %v, want exported chatlist", persisted, found, err)
|
||||
}
|
||||
|
||||
preview, err := svc.CheckInvite(ctx, viewerID, "https://t.me/addlist/slug-one")
|
||||
if err != nil {
|
||||
t.Fatalf("CheckInvite before join: %v", err)
|
||||
}
|
||||
if len(preview.Missing) != 1 || preview.Missing[0].Peer != peerA.Peer || preview.LocalFolder != nil {
|
||||
t.Fatalf("preview before join = %+v, want missing peerA and no local folder", preview)
|
||||
}
|
||||
|
||||
joined, err := svc.JoinInvite(ctx, viewerID, "slug-one", nil, 101)
|
||||
if err != nil {
|
||||
t.Fatalf("JoinInvite: %v", err)
|
||||
}
|
||||
if joined.Folder.ID != 2 || !joined.Folder.IsChatlist || joined.Folder.HasMyInvites || len(joined.Folder.IncludePeers) != 1 {
|
||||
t.Fatalf("joined folder = %+v, want imported local chatlist with peerA", joined.Folder)
|
||||
}
|
||||
|
||||
preview, err = svc.CheckInvite(ctx, viewerID, "slug-one")
|
||||
if err != nil {
|
||||
t.Fatalf("CheckInvite after join: %v", err)
|
||||
}
|
||||
if preview.LocalFolder == nil || preview.Membership == nil || len(preview.Already) != 1 || len(preview.Missing) != 0 {
|
||||
t.Fatalf("preview after join = %+v, want already imported", preview)
|
||||
}
|
||||
|
||||
if err := dialogs.UpsertFolder(ctx, ownerID, domain.DialogFolder{
|
||||
ID: 2,
|
||||
Title: "Team",
|
||||
IsChatlist: true,
|
||||
HasMyInvites: true,
|
||||
IncludePeers: []domain.DialogFolderPeer{peerA, peerB},
|
||||
}); err != nil {
|
||||
t.Fatalf("extend owner folder: %v", err)
|
||||
}
|
||||
if _, err := svc.EditInvite(ctx, ownerID, 2, "slug-one", nil, &[]domain.DialogFolderPeer{peerA, peerB}, false); err != nil {
|
||||
t.Fatalf("EditInvite add peerB: %v", err)
|
||||
}
|
||||
updates, err := svc.GetUpdates(ctx, viewerID, joined.Folder.ID)
|
||||
if err != nil {
|
||||
t.Fatalf("GetUpdates: %v", err)
|
||||
}
|
||||
if len(updates.Missing) != 1 || updates.Missing[0].Peer != peerB.Peer {
|
||||
t.Fatalf("updates = %+v, want missing peerB", updates)
|
||||
}
|
||||
ownerUpdates, err := svc.GetUpdates(ctx, ownerID, folder.ID)
|
||||
if err != nil {
|
||||
t.Fatalf("GetUpdates owner exported folder: %v", err)
|
||||
}
|
||||
if len(ownerUpdates.Missing) != 0 {
|
||||
t.Fatalf("owner updates = %+v, want empty", ownerUpdates)
|
||||
}
|
||||
updated, err := svc.JoinUpdates(ctx, viewerID, joined.Folder.ID, []domain.DialogFolderPeer{peerB}, 102)
|
||||
if err != nil {
|
||||
t.Fatalf("JoinUpdates: %v", err)
|
||||
}
|
||||
if len(updated.Folder.IncludePeers) != 2 {
|
||||
t.Fatalf("updated folder peers = %+v, want 2 peers", updated.Folder.IncludePeers)
|
||||
}
|
||||
|
||||
if err := dialogs.UpsertFolder(ctx, ownerID, domain.DialogFolder{
|
||||
ID: 2,
|
||||
Title: "Team",
|
||||
IsChatlist: true,
|
||||
HasMyInvites: true,
|
||||
IncludePeers: []domain.DialogFolderPeer{peerA, peerB, peerC},
|
||||
}); err != nil {
|
||||
t.Fatalf("extend owner folder with peerC: %v", err)
|
||||
}
|
||||
if _, err := svc.EditInvite(ctx, ownerID, 2, "slug-one", nil, &[]domain.DialogFolderPeer{peerA, peerB, peerC}, false); err != nil {
|
||||
t.Fatalf("EditInvite add peerC: %v", err)
|
||||
}
|
||||
if err := svc.HideUpdates(ctx, viewerID, joined.Folder.ID); err != nil {
|
||||
t.Fatalf("HideUpdates: %v", err)
|
||||
}
|
||||
updates, err = svc.GetUpdates(ctx, viewerID, joined.Folder.ID)
|
||||
if err != nil {
|
||||
t.Fatalf("GetUpdates after hide: %v", err)
|
||||
}
|
||||
if len(updates.Missing) != 0 {
|
||||
t.Fatalf("updates after hide = %+v, want empty", updates)
|
||||
}
|
||||
|
||||
suggestions, err := svc.LeaveSuggestions(ctx, viewerID, joined.Folder.ID)
|
||||
if err != nil {
|
||||
t.Fatalf("LeaveSuggestions: %v", err)
|
||||
}
|
||||
if len(suggestions) != 2 {
|
||||
t.Fatalf("leave suggestions = %+v, want current two imported peers", suggestions)
|
||||
}
|
||||
if _, err := svc.Leave(ctx, viewerID, joined.Folder.ID, suggestions, 103); err != nil {
|
||||
t.Fatalf("Leave: %v", err)
|
||||
}
|
||||
if _, found, err := dialogs.GetFolder(ctx, viewerID, joined.Folder.ID); err != nil || found {
|
||||
t.Fatalf("local folder after leave found=%v err=%v, want deleted", found, err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestDeleteLastSharedFolderInviteClearsOwnerFlag(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
dialogs := memory.NewDialogStore()
|
||||
chatlists := memory.NewChatlistStore()
|
||||
slugs := []string{"slug-a", "slug-b"}
|
||||
svc := NewService(chatlists, dialogs, WithSlugGenerator(func() (string, error) {
|
||||
slug := slugs[0]
|
||||
slugs = slugs[1:]
|
||||
return slug, nil
|
||||
}))
|
||||
|
||||
ownerID := int64(1001)
|
||||
peer := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 3001}, AccessHash: 31}
|
||||
if err := dialogs.UpsertFolder(ctx, ownerID, domain.DialogFolder{
|
||||
ID: 2,
|
||||
Title: "Team",
|
||||
IncludePeers: []domain.DialogFolderPeer{peer},
|
||||
}); err != nil {
|
||||
t.Fatalf("seed owner folder: %v", err)
|
||||
}
|
||||
if _, _, err := svc.ExportInvite(ctx, ownerID, 2, "one", []domain.DialogFolderPeer{peer}, 100); err != nil {
|
||||
t.Fatalf("ExportInvite one: %v", err)
|
||||
}
|
||||
if _, _, err := svc.ExportInvite(ctx, ownerID, 2, "two", []domain.DialogFolderPeer{peer}, 101); err != nil {
|
||||
t.Fatalf("ExportInvite two: %v", err)
|
||||
}
|
||||
if folder, changed, err := svc.DeleteInvite(ctx, ownerID, 2, "slug-a"); err != nil || changed || folder.HasMyInvites {
|
||||
t.Fatalf("DeleteInvite first = folder %+v changed %v err %v, want no folder update", folder, changed, err)
|
||||
}
|
||||
persisted, found, err := dialogs.GetFolder(ctx, ownerID, 2)
|
||||
if err != nil || !found || !persisted.HasMyInvites {
|
||||
t.Fatalf("folder after first delete = %+v found %v err %v, want has_my_invites", persisted, found, err)
|
||||
}
|
||||
folder, changed, err := svc.DeleteInvite(ctx, ownerID, 2, "slug-b")
|
||||
if err != nil || !changed || folder.HasMyInvites {
|
||||
t.Fatalf("DeleteInvite last = folder %+v changed %v err %v, want cleared flag", folder, changed, err)
|
||||
}
|
||||
persisted, found, err = dialogs.GetFolder(ctx, ownerID, 2)
|
||||
if err != nil || !found || persisted.HasMyInvites {
|
||||
t.Fatalf("folder after last delete = %+v found %v err %v, want has_my_invites=false", persisted, found, err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSharedFolderSlugValidation(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
svc := NewService(memory.NewChatlistStore(), memory.NewDialogStore())
|
||||
if _, err := svc.CheckInvite(ctx, 2002, "bad!"); !errors.Is(err, domain.ErrChatlistInviteInvalid) {
|
||||
t.Fatalf("CheckInvite bad slug err = %v, want ErrChatlistInviteInvalid", err)
|
||||
}
|
||||
longSlug := strings.Repeat("a", links.MaxChatlistSlugBytes+1)
|
||||
if _, err := svc.JoinInvite(ctx, 2002, longSlug, nil, 100); !errors.Is(err, domain.ErrChatlistInviteInvalid) {
|
||||
t.Fatalf("JoinInvite long slug err = %v, want ErrChatlistInviteInvalid", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRevokedSharedFolderInviteRemainsListedButCannotBeImported(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
dialogs := memory.NewDialogStore()
|
||||
chatlists := memory.NewChatlistStore()
|
||||
svc := NewService(chatlists, dialogs, WithSlugGenerator(func() (string, error) { return "slug-revoke", nil }))
|
||||
|
||||
ownerID := int64(1001)
|
||||
viewerID := int64(2002)
|
||||
peer := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 3001}, AccessHash: 31}
|
||||
if err := dialogs.UpsertFolder(ctx, ownerID, domain.DialogFolder{
|
||||
ID: 2,
|
||||
Title: "Team",
|
||||
IncludePeers: []domain.DialogFolderPeer{peer},
|
||||
}); err != nil {
|
||||
t.Fatalf("seed owner folder: %v", err)
|
||||
}
|
||||
if _, _, err := svc.ExportInvite(ctx, ownerID, 2, "Main link", []domain.DialogFolderPeer{peer}, 100); err != nil {
|
||||
t.Fatalf("ExportInvite: %v", err)
|
||||
}
|
||||
revoked, err := svc.EditInvite(ctx, ownerID, 2, "slug-revoke", nil, nil, true)
|
||||
if err != nil {
|
||||
t.Fatalf("EditInvite revoke: %v", err)
|
||||
}
|
||||
if !revoked.Revoked {
|
||||
t.Fatalf("revoked invite = %+v, want Revoked=true", revoked)
|
||||
}
|
||||
invites, err := svc.ListInvites(ctx, ownerID, 2)
|
||||
if err != nil {
|
||||
t.Fatalf("ListInvites: %v", err)
|
||||
}
|
||||
if len(invites) != 1 || !invites[0].Revoked {
|
||||
t.Fatalf("listed invites = %+v, want revoked invite visible to owner", invites)
|
||||
}
|
||||
if _, err := svc.CheckInvite(ctx, viewerID, "slug-revoke"); !errors.Is(err, domain.ErrChatlistInviteExpired) {
|
||||
t.Fatalf("CheckInvite revoked err = %v, want ErrChatlistInviteExpired", err)
|
||||
}
|
||||
if _, err := svc.JoinInvite(ctx, viewerID, "slug-revoke", nil, 101); !errors.Is(err, domain.ErrChatlistInviteExpired) {
|
||||
t.Fatalf("JoinInvite revoked err = %v, want ErrChatlistInviteExpired", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSharedFolderJoinUpdatesAndLeaveUseChannelMemberships(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
dialogs := memory.NewDialogStore()
|
||||
channels := &fakeChatlistChannels{}
|
||||
svc := NewService(
|
||||
memory.NewChatlistStore(),
|
||||
dialogs,
|
||||
WithChannels(channels),
|
||||
WithSlugGenerator(func() (string, error) { return "slug-channel", nil }),
|
||||
)
|
||||
|
||||
ownerID := int64(1001)
|
||||
viewerID := int64(2002)
|
||||
peerA := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 3001}, AccessHash: 31}
|
||||
peerB := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 3002}, AccessHash: 32}
|
||||
if err := dialogs.UpsertFolder(ctx, ownerID, domain.DialogFolder{
|
||||
ID: 2,
|
||||
Title: "Team",
|
||||
IncludePeers: []domain.DialogFolderPeer{peerA, peerB},
|
||||
}); err != nil {
|
||||
t.Fatalf("seed owner folder: %v", err)
|
||||
}
|
||||
|
||||
if _, _, err := svc.ExportInvite(ctx, ownerID, 2, "Main link", []domain.DialogFolderPeer{peerA, peerB}, 100); err != nil {
|
||||
t.Fatalf("ExportInvite: %v", err)
|
||||
}
|
||||
if got := channels.getIDs; len(got) != 2 || got[0] != peerA.Peer.ID || got[1] != peerB.Peer.ID {
|
||||
t.Fatalf("shareability channel checks = %v, want peerA/peerB", got)
|
||||
}
|
||||
|
||||
joined, err := svc.JoinInvite(ctx, viewerID, "slug-channel", []domain.DialogFolderPeer{peerA}, 101)
|
||||
if err != nil {
|
||||
t.Fatalf("JoinInvite peerA: %v", err)
|
||||
}
|
||||
if got := channels.inviteIDs; len(got) != 1 || got[0] != peerA.Peer.ID {
|
||||
t.Fatalf("invite channel calls = %v, want peerA", got)
|
||||
}
|
||||
if len(joined.ChannelResults) != 1 || joined.ChannelResults[0].Channel.ID != peerA.Peer.ID {
|
||||
t.Fatalf("join channel results = %+v, want peerA result", joined.ChannelResults)
|
||||
}
|
||||
|
||||
updated, err := svc.JoinUpdates(ctx, viewerID, joined.Folder.ID, []domain.DialogFolderPeer{peerB}, 102)
|
||||
if err != nil {
|
||||
t.Fatalf("JoinUpdates peerB: %v", err)
|
||||
}
|
||||
if got := channels.inviteIDs; len(got) != 2 || got[1] != peerB.Peer.ID {
|
||||
t.Fatalf("invite channel calls after updates = %v, want peerB appended", got)
|
||||
}
|
||||
if len(updated.ChannelResults) != 1 || updated.ChannelResults[0].Channel.ID != peerB.Peer.ID {
|
||||
t.Fatalf("update channel results = %+v, want peerB result", updated.ChannelResults)
|
||||
}
|
||||
|
||||
leave, err := svc.Leave(ctx, viewerID, joined.Folder.ID, []domain.DialogFolderPeer{peerA}, 103)
|
||||
if err != nil {
|
||||
t.Fatalf("Leave peerA: %v", err)
|
||||
}
|
||||
if got := channels.leaveIDs; len(got) != 1 || got[0] != peerA.Peer.ID {
|
||||
t.Fatalf("leave channel calls = %v, want peerA", got)
|
||||
}
|
||||
if len(leave.ChannelResults) != 1 || leave.ChannelResults[0].Channel.ID != peerA.Peer.ID {
|
||||
t.Fatalf("leave channel results = %+v, want peerA result", leave.ChannelResults)
|
||||
}
|
||||
}
|
||||
|
||||
func TestSharedFolderPublicPeerFallsBackToSelfJoin(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
dialogs := memory.NewDialogStore()
|
||||
channels := &fakeChatlistChannels{publicOnly: true}
|
||||
svc := NewService(
|
||||
memory.NewChatlistStore(),
|
||||
dialogs,
|
||||
WithChannels(channels),
|
||||
WithSlugGenerator(func() (string, error) { return "slug-public", nil }),
|
||||
)
|
||||
|
||||
ownerID := int64(1001)
|
||||
viewerID := int64(2002)
|
||||
peer := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 3001}, AccessHash: 31}
|
||||
if err := dialogs.UpsertFolder(ctx, ownerID, domain.DialogFolder{
|
||||
ID: 2,
|
||||
Title: "Public",
|
||||
IncludePeers: []domain.DialogFolderPeer{peer},
|
||||
}); err != nil {
|
||||
t.Fatalf("seed owner folder: %v", err)
|
||||
}
|
||||
if _, _, err := svc.ExportInvite(ctx, ownerID, 2, "", []domain.DialogFolderPeer{peer}, 100); err != nil {
|
||||
t.Fatalf("ExportInvite: %v", err)
|
||||
}
|
||||
if _, err := svc.JoinInvite(ctx, viewerID, "slug-public", []domain.DialogFolderPeer{peer}, 101); err != nil {
|
||||
t.Fatalf("JoinInvite public peer: %v", err)
|
||||
}
|
||||
if len(channels.inviteIDs) != 0 {
|
||||
t.Fatalf("invite channel calls = %v, want none for public-only owner", channels.inviteIDs)
|
||||
}
|
||||
if got := channels.joinIDs; len(got) != 1 || got[0] != peer.Peer.ID {
|
||||
t.Fatalf("join channel calls = %v, want self-join peer", got)
|
||||
}
|
||||
}
|
||||
|
||||
func TestExportInviteRejectsRuleBasedFolder(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
dialogs := memory.NewDialogStore()
|
||||
svc := NewService(memory.NewChatlistStore(), dialogs, WithSlugGenerator(func() (string, error) { return "slug-two", nil }))
|
||||
ownerID := int64(1001)
|
||||
peer := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 3001}, AccessHash: 31}
|
||||
if err := dialogs.UpsertFolder(ctx, ownerID, domain.DialogFolder{
|
||||
ID: 2,
|
||||
Title: "Rule based",
|
||||
Groups: true,
|
||||
IncludePeers: []domain.DialogFolderPeer{peer},
|
||||
}); err != nil {
|
||||
t.Fatalf("seed folder: %v", err)
|
||||
}
|
||||
if _, _, err := svc.ExportInvite(ctx, ownerID, 2, "", []domain.DialogFolderPeer{peer}, 0); !errors.Is(err, domain.ErrChatlistNotShareable) {
|
||||
t.Fatalf("ExportInvite rule folder err = %v, want ErrChatlistNotShareable", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestExportInviteRejectsUserPeers(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
dialogs := memory.NewDialogStore()
|
||||
svc := NewService(memory.NewChatlistStore(), dialogs, WithSlugGenerator(func() (string, error) { return "slug-user-peer", nil }))
|
||||
ownerID := int64(1001)
|
||||
peer := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeUser, ID: 2002}, AccessHash: 22}
|
||||
if err := dialogs.UpsertFolder(ctx, ownerID, domain.DialogFolder{
|
||||
ID: 2,
|
||||
Title: "Users",
|
||||
IncludePeers: []domain.DialogFolderPeer{peer},
|
||||
}); err != nil {
|
||||
t.Fatalf("seed folder: %v", err)
|
||||
}
|
||||
if _, _, err := svc.ExportInvite(ctx, ownerID, 2, "", []domain.DialogFolderPeer{peer}, 0); !errors.Is(err, domain.ErrChatlistNotShareable) {
|
||||
t.Fatalf("ExportInvite user peer err = %v, want ErrChatlistNotShareable", err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestChatlistInviteLimitUsesPremiumTier(t *testing.T) {
|
||||
ctx := context.Background()
|
||||
dialogs := memory.NewDialogStore()
|
||||
chatlists := memory.NewChatlistStore()
|
||||
ownerID := int64(1001)
|
||||
peer := domain.DialogFolderPeer{Peer: domain.Peer{Type: domain.PeerTypeChannel, ID: 3001}, AccessHash: 31}
|
||||
if err := dialogs.UpsertFolder(ctx, ownerID, domain.DialogFolder{
|
||||
ID: 2,
|
||||
Title: "Team",
|
||||
IncludePeers: []domain.DialogFolderPeer{peer},
|
||||
}); err != nil {
|
||||
t.Fatalf("seed folder: %v", err)
|
||||
}
|
||||
|
||||
nextSlug := 0
|
||||
slugger := func() (string, error) {
|
||||
nextSlug++
|
||||
return "slug-limit-" + string(rune('a'+nextSlug)), nil
|
||||
}
|
||||
svc := NewService(chatlists, dialogs, WithSlugGenerator(slugger))
|
||||
for i := 0; i < domain.MaxChatlistInvitesDefault; i++ {
|
||||
if _, _, err := svc.ExportInvite(ctx, ownerID, 2, "", []domain.DialogFolderPeer{peer}, i+1); err != nil {
|
||||
t.Fatalf("ExportInvite default #%d: %v", i+1, err)
|
||||
}
|
||||
}
|
||||
if _, _, err := svc.ExportInvite(ctx, ownerID, 2, "", []domain.DialogFolderPeer{peer}, 10); !errors.Is(err, domain.ErrChatlistInvitesTooMuch) {
|
||||
t.Fatalf("ExportInvite over default limit err = %v, want ErrChatlistInvitesTooMuch", err)
|
||||
}
|
||||
if _, err := svc.EditInvite(ctx, ownerID, 2, "slug-limit-b", nil, nil, true); err != nil {
|
||||
t.Fatalf("EditInvite revoke first default link: %v", err)
|
||||
}
|
||||
if _, _, err := svc.ExportInvite(ctx, ownerID, 2, "", []domain.DialogFolderPeer{peer}, 11); err != nil {
|
||||
t.Fatalf("ExportInvite after revoked link freed active limit: %v", err)
|
||||
}
|
||||
|
||||
premiumSvc := NewService(
|
||||
chatlists,
|
||||
dialogs,
|
||||
WithSlugGenerator(slugger),
|
||||
WithPremiumChecker(func(context.Context, int64) bool { return true }),
|
||||
)
|
||||
if _, _, err := premiumSvc.ExportInvite(ctx, ownerID, 2, "", []domain.DialogFolderPeer{peer}, 11); err != nil {
|
||||
t.Fatalf("ExportInvite premium extra link: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
type fakeChatlistChannels struct {
|
||||
getIDs []int64
|
||||
inviteIDs []int64
|
||||
joinIDs []int64
|
||||
leaveIDs []int64
|
||||
publicOnly bool
|
||||
}
|
||||
|
||||
func (f *fakeChatlistChannels) GetChannel(_ context.Context, userID, channelID int64) (domain.ChannelView, error) {
|
||||
f.getIDs = append(f.getIDs, channelID)
|
||||
role := domain.ChannelRoleCreator
|
||||
if f.publicOnly {
|
||||
role = domain.ChannelRoleMember
|
||||
}
|
||||
return domain.ChannelView{
|
||||
Channel: domain.Channel{
|
||||
ID: channelID,
|
||||
AccessHash: channelID * 10,
|
||||
Title: "Team",
|
||||
Username: "team",
|
||||
Megagroup: true,
|
||||
},
|
||||
Self: domain.ChannelMember{
|
||||
ChannelID: channelID,
|
||||
UserID: userID,
|
||||
Role: role,
|
||||
Status: domain.ChannelMemberActive,
|
||||
},
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (f *fakeChatlistChannels) InviteToChannel(_ context.Context, userID, channelID int64, userIDs []int64, date int) (domain.CreateChannelResult, error) {
|
||||
f.inviteIDs = append(f.inviteIDs, channelID)
|
||||
if len(userIDs) == 0 {
|
||||
return domain.CreateChannelResult{}, domain.ErrUsersTooMuch
|
||||
}
|
||||
return fakeChatlistChannelResult(userIDs[0], channelID, domain.ChannelMemberActive, date), nil
|
||||
}
|
||||
|
||||
func (f *fakeChatlistChannels) JoinChannel(_ context.Context, userID, channelID int64, date int) (domain.CreateChannelResult, error) {
|
||||
f.joinIDs = append(f.joinIDs, channelID)
|
||||
return fakeChatlistChannelResult(userID, channelID, domain.ChannelMemberActive, date), nil
|
||||
}
|
||||
|
||||
func (f *fakeChatlistChannels) LeaveChannel(_ context.Context, userID, channelID int64, date int) (domain.CreateChannelResult, error) {
|
||||
f.leaveIDs = append(f.leaveIDs, channelID)
|
||||
return fakeChatlistChannelResult(userID, channelID, domain.ChannelMemberLeft, date), nil
|
||||
}
|
||||
|
||||
func fakeChatlistChannelResult(userID, channelID int64, status domain.ChannelMemberStatus, date int) domain.CreateChannelResult {
|
||||
return domain.CreateChannelResult{
|
||||
Channel: domain.Channel{
|
||||
ID: channelID,
|
||||
AccessHash: channelID * 10,
|
||||
Title: "Team",
|
||||
Username: "team",
|
||||
Megagroup: true,
|
||||
Date: date,
|
||||
},
|
||||
Members: []domain.ChannelMember{{
|
||||
ChannelID: channelID,
|
||||
UserID: userID,
|
||||
Role: domain.ChannelRoleMember,
|
||||
Status: status,
|
||||
JoinedAt: date,
|
||||
}},
|
||||
Recipients: []int64{userID},
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue