720 lines
24 KiB
Go
720 lines
24 KiB
Go
package userprojection
|
||
|
||
import (
|
||
"context"
|
||
|
||
"golang.org/x/sync/errgroup"
|
||
|
||
"telesrv/internal/domain"
|
||
"telesrv/internal/store"
|
||
)
|
||
|
||
// ProfilePhotoProvider returns current profile photos for a batch of owners.
|
||
type ProfilePhotoProvider interface {
|
||
CurrentProfilePhotos(ctx context.Context, ownerType domain.PeerType, ownerIDs []int64) (map[int64]domain.ProfilePhotoRef, error)
|
||
}
|
||
|
||
// ProfilePhotoKindProvider returns current profile/fallback photos for a batch of owners.
|
||
type ProfilePhotoKindProvider interface {
|
||
CurrentProfilePhotosKind(ctx context.Context, ownerType domain.PeerType, ownerIDs []int64, kind domain.ProfilePhotoKind) (map[int64]domain.ProfilePhotoRef, error)
|
||
}
|
||
|
||
// PrivacyEvaluator answers viewer-specific visibility for one user privacy key.
|
||
type PrivacyEvaluator interface {
|
||
CanSee(ctx context.Context, ownerUserID, viewerUserID int64, key domain.PrivacyKey) (bool, error)
|
||
}
|
||
|
||
// AccountFreezeProvider returns durable account freeze facts for a bounded
|
||
// batch. The projector only exposes them to viewers other than the frozen user.
|
||
type AccountFreezeProvider interface {
|
||
AccountFreezes(ctx context.Context, userIDs []int64) (map[int64]domain.AccountFreeze, error)
|
||
}
|
||
|
||
// BatchPrivacyEvaluator 批量评估多 owner 对单 viewer 的可见性，消除 projectBatch / fan-out
|
||
// 投影里 per-user 3×CanSee 的 N+1。可选：实现了它的 evaluator（privacy.Service）会被
|
||
// projectBatch 优先用批量预取，否则回退逐 CanSee。结果必须与逐 CanSee 字节等价。
|
||
type BatchPrivacyEvaluator interface {
|
||
CanSeeBatch(ctx context.Context, ownerUserIDs []int64, viewerUserID int64, keys []domain.PrivacyKey) (map[int64]map[domain.PrivacyKey]bool, error)
|
||
}
|
||
|
||
// MatrixPrivacyEvaluator 批量评估 owners×viewers×keys 的可见性矩阵（一次 ListPrivacyRules +
|
||
// 每 owner 一次 GetMany + 内存 Evaluate），供 ForViewers 把 fan-out 跨 viewer 投影的 privacy
|
||
// 查询从 O(viewer) 降到 O(owner)。可选：privacy.Service 实现了它，否则 ForViewers 回退逐 CanSee。
|
||
// 结果必须与逐 CanSee 字节等价。
|
||
type MatrixPrivacyEvaluator interface {
|
||
CanSeeMatrix(ctx context.Context, ownerUserIDs, viewerUserIDs []int64, keys []domain.PrivacyKey) (map[int64]map[int64]map[domain.PrivacyKey]bool, error)
|
||
}
|
||
|
||
// privacyProjectionKeys 是 projectBatch 投影会用到的 privacy key（phone/status/photo）。
|
||
var privacyProjectionKeys = []domain.PrivacyKey{
|
||
domain.PrivacyKeyPhoneNumber,
|
||
domain.PrivacyKeyStatusTimestamp,
|
||
domain.PrivacyKeyProfilePhoto,
|
||
}
|
||
|
||
// Projector builds the current viewer's user view for RPC response payloads.
|
||
// It intentionally stays in app/domain types; tg.* conversion remains in rpc.
|
||
type Projector struct {
|
||
contacts store.ContactStore
|
||
photos ProfilePhotoProvider
|
||
privacy PrivacyEvaluator
|
||
freezes AccountFreezeProvider
|
||
}
|
||
|
||
// Option configures a Projector.
|
||
type Option func(*Projector)
|
||
|
||
// WithContactStore enables viewer-specific contact name/phone projection.
|
||
func WithContactStore(c store.ContactStore) Option {
|
||
return func(p *Projector) { p.contacts = c }
|
||
}
|
||
|
||
// WithPhotoProvider enables current profile photo enrichment.
|
||
func WithPhotoProvider(photos ProfilePhotoProvider) Option {
|
||
return func(p *Projector) { p.photos = photos }
|
||
}
|
||
|
||
// WithPrivacyEvaluator enables profile/photo/status privacy projection.
|
||
func WithPrivacyEvaluator(privacy PrivacyEvaluator) Option {
|
||
return func(p *Projector) { p.privacy = privacy }
|
||
}
|
||
|
||
// WithAccountFreezeProvider enables viewer-scoped frozen-account visibility.
|
||
func WithAccountFreezeProvider(provider AccountFreezeProvider) Option {
|
||
return func(p *Projector) { p.freezes = provider }
|
||
}
|
||
|
||
// New creates a user projector.
|
||
func New(opts ...Option) *Projector {
|
||
p := &Projector{}
|
||
for _, opt := range opts {
|
||
opt(p)
|
||
}
|
||
return p
|
||
}
|
||
|
||
// ForViewer applies both current profile photos and owner-specific contact view.
|
||
func (p *Projector) ForViewer(ctx context.Context, viewerUserID int64, users []domain.User) ([]domain.User, error) {
|
||
users = sanitizeDeletedUsers(users)
|
||
if p == nil {
|
||
return users, nil
|
||
}
|
||
return projectBatch(ctx, p.contacts, p.photos, p.privacy, p.freezes, viewerUserID, users)
|
||
}
|
||
|
||
// One applies ForViewer to a single user.
|
||
func (p *Projector) One(ctx context.Context, viewerUserID int64, user domain.User) (domain.User, error) {
|
||
if p == nil {
|
||
return user, nil
|
||
}
|
||
projected, err := p.ForViewer(ctx, viewerUserID, []domain.User{user})
|
||
if err != nil || len(projected) == 0 {
|
||
return domain.User{}, err
|
||
}
|
||
return projected[0], nil
|
||
}
|
||
|
||
// ForViewers 跨多个 viewer 批量投影同一组 owner 用户（fan-out 模板化）。它把 per-viewer 各跑
|
||
// 一遍 ForViewer(=projectBatch) 的成本（O(viewer)×(photos+contacts+privacy) 查询）压成：
|
||
// - 一次 profile/fallback 头像批量（跨 viewer 复用）
|
||
// - O(owner) 次 GetReverseContacts（改名/电话覆盖，按 owner 反查 viewer）
|
||
// - O(owner) 次 GetMany + 一次 ListPrivacyRules（CanSeeMatrix 内做）
|
||
//
|
||
// 返回 map[viewerID][]domain.User，每个切片与对应 viewer 的 ForViewer(viewer, users) **字节等价，
|
||
// 唯一例外是 personal photo overlay**：v1 简化为 fan-out 模板不做 per-viewer personal photo
|
||
// （无 O(owner) 反查接口），客户端下次 getChannelDifference/getHistory 会走 projectBatch 完整投影自愈。
|
||
// 调用方传入的 users 不被修改（内部复制）。
|
||
func (p *Projector) ForViewers(ctx context.Context, viewerUserIDs []int64, users []domain.User) (map[int64][]domain.User, error) {
|
||
users = sanitizeDeletedUsers(users)
|
||
out := make(map[int64][]domain.User, len(viewerUserIDs))
|
||
if p == nil || len(users) == 0 {
|
||
for _, v := range viewerUserIDs {
|
||
out[v] = cloneUsers(users)
|
||
}
|
||
return out, nil
|
||
}
|
||
viewers := dedupNonZeroInt64(viewerUserIDs)
|
||
if len(viewers) == 0 {
|
||
for _, v := range viewerUserIDs {
|
||
out[v] = cloneUsers(users)
|
||
}
|
||
return out, nil
|
||
}
|
||
ids := uniqueUserIDs(users)
|
||
|
||
// 三组预取互不依赖（共享头像、反向联系人覆盖、privacy 矩阵），并发执行收敛成一波。
|
||
var (
|
||
profileRefs map[int64]domain.ProfilePhotoRef
|
||
fallbackRefs map[int64]domain.ProfilePhotoRef
|
||
contactsByViewer map[int64]map[int64]domain.Contact
|
||
matrix map[int64]map[int64]map[domain.PrivacyKey]bool
|
||
freezes map[int64]domain.AccountFreeze
|
||
)
|
||
g, gctx := errgroup.WithContext(ctx)
|
||
// 1) 共享头像：profile/fallback 一次批量，跨全部 viewer 复用；personal photo v1 跳过（见 doc）。
|
||
g.Go(func() error {
|
||
var err error
|
||
profileRefs, fallbackRefs, err = p.batchProfileFallbackPhotos(gctx, ids)
|
||
return err
|
||
})
|
||
// 2) 改名/电话覆盖：O(owner) 次 GetReverseContacts(owner, viewers) 重组为 [viewer][owner]Contact，
|
||
// 与 projectBatch 的 GetMany(viewer, owners) 命中同一条联系人记录（方向对称）。
|
||
g.Go(func() error {
|
||
var err error
|
||
contactsByViewer, err = p.reverseContactsByViewer(gctx, ids, viewers)
|
||
return err
|
||
})
|
||
// 3) privacy 可见性矩阵：O(owner) 查询；nil（无 MatrixPrivacyEvaluator）时 applyPrivacy 回退逐 CanSee。
|
||
if me, ok := p.privacy.(MatrixPrivacyEvaluator); ok && p.privacy != nil {
|
||
g.Go(func() error {
|
||
var err error
|
||
matrix, err = me.CanSeeMatrix(gctx, ids, viewers, privacyProjectionKeys)
|
||
return err
|
||
})
|
||
}
|
||
if p.freezes != nil && len(ids) > 0 {
|
||
g.Go(func() error {
|
||
var err error
|
||
freezes, err = p.freezes.AccountFreezes(gctx, ids)
|
||
return err
|
||
})
|
||
}
|
||
if err := g.Wait(); err != nil {
|
||
return nil, err
|
||
}
|
||
// 4) 逐 viewer 组装，复用与 projectBatch 完全相同的 apply* 链（personalRefs 传 nil）。
|
||
for _, viewer := range viewers {
|
||
projected := make([]domain.User, len(users))
|
||
copy(projected, users)
|
||
cache := make(map[int64]domain.User, len(projected))
|
||
for i := range projected {
|
||
u := projected[i]
|
||
if u.ID == 0 {
|
||
continue
|
||
}
|
||
if u.Deleted {
|
||
projected[i] = u.DeletedTombstone()
|
||
continue
|
||
}
|
||
if pj, ok := cache[u.ID]; ok {
|
||
projected[i] = pj
|
||
continue
|
||
}
|
||
pj := applyBasePhotos(u, profileRefs, fallbackRefs, nil, viewer)
|
||
if viewer != 0 && u.ID != viewer && u.ID != domain.OfficialSystemUserID && !u.Bot {
|
||
contact, found := contactsByViewer[viewer][u.ID]
|
||
pj = applyContactProjection(pj, contact, found)
|
||
var vis map[domain.PrivacyKey]bool
|
||
if matrix != nil {
|
||
vis = matrix[u.ID][viewer]
|
||
}
|
||
var perr error
|
||
pj, perr = applyPrivacy(ctx, p.privacy, viewer, pj, found, vis, profileRefs, fallbackRefs, nil)
|
||
if perr != nil {
|
||
return nil, perr
|
||
}
|
||
}
|
||
pj = applyAccountFreezeProjection(pj, viewer, freezes[u.ID])
|
||
cache[u.ID] = pj
|
||
projected[i] = pj
|
||
}
|
||
out[viewer] = projected
|
||
}
|
||
return out, nil
|
||
}
|
||
|
||
// batchProfileFallbackPhotos 取 owner 的 profile/fallback 头像（与 projectBatch 同逻辑），personal
|
||
// 头像不取（ForViewers v1 跳过）。photos 为 nil 时返回空 map（applyBasePhotos 视为无头像查询）。
|
||
func (p *Projector) batchProfileFallbackPhotos(ctx context.Context, ids []int64) (profileRefs, fallbackRefs map[int64]domain.ProfilePhotoRef, err error) {
|
||
profileRefs = map[int64]domain.ProfilePhotoRef{}
|
||
fallbackRefs = map[int64]domain.ProfilePhotoRef{}
|
||
if p.photos == nil || len(ids) == 0 {
|
||
return profileRefs, fallbackRefs, nil
|
||
}
|
||
if kindPhotos, ok := p.photos.(ProfilePhotoKindProvider); ok {
|
||
refs, err := kindPhotos.CurrentProfilePhotosKind(ctx, domain.PeerTypeUser, ids, domain.ProfilePhotoKindProfile)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
profileRefs = refs
|
||
refs, err = kindPhotos.CurrentProfilePhotosKind(ctx, domain.PeerTypeUser, ids, domain.ProfilePhotoKindFallback)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
fallbackRefs = refs
|
||
return profileRefs, fallbackRefs, nil
|
||
}
|
||
refs, err := p.photos.CurrentProfilePhotos(ctx, domain.PeerTypeUser, ids)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
return refs, fallbackRefs, nil
|
||
}
|
||
|
||
// reverseContactsByViewer 以 O(owner) 次 GetReverseContacts(owner, viewers) 取「每个 viewer 对各
|
||
// owner 的联系人记录」并重组为 map[viewer]map[owner]Contact。该记录与 projectBatch 的
|
||
// GetMany(viewer, owners)[owner] 是同一条（contacts 表上 (user_id=viewer, contact_user_id=owner)
|
||
// 的同一行，两端 store 均如此），用于 applyContactProjection 的改名/电话覆盖与 isContact 判定。
|
||
func (p *Projector) reverseContactsByViewer(ctx context.Context, ownerIDs, viewers []int64) (map[int64]map[int64]domain.Contact, error) {
|
||
out := make(map[int64]map[int64]domain.Contact, len(viewers))
|
||
if p.contacts == nil || len(ownerIDs) == 0 || len(viewers) == 0 {
|
||
return out, nil
|
||
}
|
||
for _, owner := range ownerIDs {
|
||
byViewer, err := p.contacts.GetReverseContacts(ctx, owner, viewers)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
for viewer, contact := range byViewer {
|
||
if out[viewer] == nil {
|
||
out[viewer] = make(map[int64]domain.Contact, len(ownerIDs))
|
||
}
|
||
out[viewer][owner] = contact
|
||
}
|
||
}
|
||
return out, nil
|
||
}
|
||
|
||
func cloneUsers(users []domain.User) []domain.User {
|
||
if len(users) == 0 {
|
||
return nil
|
||
}
|
||
out := make([]domain.User, len(users))
|
||
copy(out, users)
|
||
for i := range out {
|
||
out[i].ContactNoteEntities = append([]domain.MessageEntity(nil), out[i].ContactNoteEntities...)
|
||
out[i].RestrictionReasons = append([]domain.UserRestrictionReason(nil), out[i].RestrictionReasons...)
|
||
}
|
||
return out
|
||
}
|
||
|
||
func dedupNonZeroInt64(ids []int64) []int64 {
|
||
seen := make(map[int64]struct{}, len(ids))
|
||
out := make([]int64, 0, len(ids))
|
||
for _, id := range ids {
|
||
if id == 0 {
|
||
continue
|
||
}
|
||
if _, ok := seen[id]; ok {
|
||
continue
|
||
}
|
||
seen[id] = struct{}{}
|
||
out = append(out, id)
|
||
}
|
||
return out
|
||
}
|
||
|
||
// WithProfilePhotos enriches users with their current avatar from profile photo storage.
|
||
// The lookup is best-effort: a storage error keeps the original user list.
|
||
func WithProfilePhotos(ctx context.Context, photos ProfilePhotoProvider, users []domain.User) []domain.User {
|
||
users = sanitizeDeletedUsers(users)
|
||
if photos == nil || len(users) == 0 {
|
||
return users
|
||
}
|
||
ids := make([]int64, 0, len(users))
|
||
seen := make(map[int64]struct{}, len(users))
|
||
for _, u := range users {
|
||
if u.ID == 0 || u.Deleted {
|
||
continue
|
||
}
|
||
if _, ok := seen[u.ID]; ok {
|
||
continue
|
||
}
|
||
seen[u.ID] = struct{}{}
|
||
ids = append(ids, u.ID)
|
||
}
|
||
if len(ids) == 0 {
|
||
return users
|
||
}
|
||
refs, err := photos.CurrentProfilePhotos(ctx, domain.PeerTypeUser, ids)
|
||
if err != nil || len(refs) == 0 {
|
||
return users
|
||
}
|
||
out := make([]domain.User, len(users))
|
||
copy(out, users)
|
||
for i := range out {
|
||
if ref, ok := refs[out[i].ID]; ok {
|
||
applyPhotoRef(&out[i], ref)
|
||
}
|
||
}
|
||
return out
|
||
}
|
||
|
||
// ForViewer applies the owner-specific user view that Telegram clients expect.
|
||
// In particular, phone is visible for self and contacts; non-contacts should not
|
||
// receive a phone field because TDesktop will prefer it over the public name.
|
||
func ForViewer(ctx context.Context, contacts store.ContactStore, viewerUserID int64, users []domain.User) ([]domain.User, error) {
|
||
users = sanitizeDeletedUsers(users)
|
||
if contacts == nil || viewerUserID == 0 || len(users) == 0 {
|
||
return users, nil
|
||
}
|
||
out := make([]domain.User, len(users))
|
||
copy(out, users)
|
||
cache := make(map[int64]domain.User, len(users))
|
||
for i := range out {
|
||
u := out[i]
|
||
if u.ID == 0 || u.Deleted || u.ID == viewerUserID || u.ID == domain.OfficialSystemUserID || u.Bot {
|
||
continue
|
||
}
|
||
if projected, ok := cache[u.ID]; ok {
|
||
out[i] = projected
|
||
continue
|
||
}
|
||
projected, err := projectOne(ctx, contacts, viewerUserID, u)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
cache[u.ID] = projected
|
||
out[i] = projected
|
||
}
|
||
return out, nil
|
||
}
|
||
|
||
// One applies ForViewer to a single user.
|
||
func One(ctx context.Context, contacts store.ContactStore, viewerUserID int64, user domain.User) (domain.User, error) {
|
||
projected, err := ForViewer(ctx, contacts, viewerUserID, []domain.User{user})
|
||
if err != nil || len(projected) == 0 {
|
||
return domain.User{}, err
|
||
}
|
||
return projected[0], nil
|
||
}
|
||
|
||
func projectBatch(ctx context.Context, contacts store.ContactStore, photos ProfilePhotoProvider, privacy PrivacyEvaluator, freezesProvider AccountFreezeProvider, viewerUserID int64, users []domain.User) ([]domain.User, error) {
|
||
if len(users) == 0 {
|
||
return users, nil
|
||
}
|
||
out := make([]domain.User, len(users))
|
||
copy(out, users)
|
||
out = sanitizeDeletedUsers(out)
|
||
ids := uniqueUserIDs(out)
|
||
var (
|
||
profileRefs = map[int64]domain.ProfilePhotoRef{}
|
||
fallbackRefs = map[int64]domain.ProfilePhotoRef{}
|
||
personalRefs = map[int64]domain.ProfilePhotoRef{}
|
||
contactsByID map[int64]domain.Contact
|
||
visibility map[int64]map[domain.PrivacyKey]bool
|
||
freezes map[int64]domain.AccountFreeze
|
||
)
|
||
// 这些预取查询互不依赖（头像 profile/fallback、联系人 GetMany/PersonalPhotos、privacy 可见性），
|
||
// 并发执行把 ~6 次串行 round-trip 收敛成一波；每个 goroutine 只写自己那一个变量，组装循环在
|
||
// Wait 之后串行进行（纯内存、无查询），无数据竞争。
|
||
g, gctx := errgroup.WithContext(ctx)
|
||
if photos != nil && len(ids) > 0 {
|
||
if kindPhotos, ok := photos.(ProfilePhotoKindProvider); ok {
|
||
g.Go(func() error {
|
||
refs, err := kindPhotos.CurrentProfilePhotosKind(gctx, domain.PeerTypeUser, ids, domain.ProfilePhotoKindProfile)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
profileRefs = refs
|
||
return nil
|
||
})
|
||
g.Go(func() error {
|
||
refs, err := kindPhotos.CurrentProfilePhotosKind(gctx, domain.PeerTypeUser, ids, domain.ProfilePhotoKindFallback)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
fallbackRefs = refs
|
||
return nil
|
||
})
|
||
} else {
|
||
g.Go(func() error {
|
||
refs, err := photos.CurrentProfilePhotos(gctx, domain.PeerTypeUser, ids)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
profileRefs = refs
|
||
return nil
|
||
})
|
||
}
|
||
}
|
||
if contacts != nil && viewerUserID != 0 && len(ids) > 0 {
|
||
g.Go(func() error {
|
||
m, err := contacts.GetMany(gctx, viewerUserID, ids)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
contactsByID = m
|
||
return nil
|
||
})
|
||
g.Go(func() error {
|
||
refs, err := contacts.PersonalPhotos(gctx, viewerUserID, ids)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
personalRefs = refs
|
||
return nil
|
||
})
|
||
}
|
||
// 批量预取 privacy 可见性（若 evaluator 支持）：把 per-user 3×CanSee×2行 的 N+1 降到
|
||
// 一次 ListPrivacyRules + 一次 GetReverseContacts + 内存 Evaluate；nil 时 applyPrivacy 回退逐 CanSee。
|
||
g.Go(func() error {
|
||
v, err := prefetchPrivacyVisibility(gctx, privacy, viewerUserID, out)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
visibility = v
|
||
return nil
|
||
})
|
||
if freezesProvider != nil && len(ids) > 0 {
|
||
g.Go(func() error {
|
||
m, err := freezesProvider.AccountFreezes(gctx, ids)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
freezes = m
|
||
return nil
|
||
})
|
||
}
|
||
if err := g.Wait(); err != nil {
|
||
return nil, err
|
||
}
|
||
cache := make(map[int64]domain.User, len(out))
|
||
for i := range out {
|
||
u := out[i]
|
||
if u.ID == 0 {
|
||
continue
|
||
}
|
||
if u.Deleted {
|
||
out[i] = u.DeletedTombstone()
|
||
continue
|
||
}
|
||
if projected, ok := cache[u.ID]; ok {
|
||
out[i] = projected
|
||
continue
|
||
}
|
||
projected := applyBasePhotos(u, profileRefs, fallbackRefs, personalRefs, viewerUserID)
|
||
// bot 与系统账号豁免联系人/privacy 投影：官方 bot 无 phone/last seen，
|
||
// 不参与联系人改名与隐私裁剪。
|
||
if viewerUserID != 0 && u.ID != viewerUserID && u.ID != domain.OfficialSystemUserID && !u.Bot {
|
||
contact, found := contactsByID[u.ID]
|
||
projected = applyContactProjection(projected, contact, found)
|
||
var err error
|
||
projected, err = applyPrivacy(ctx, privacy, viewerUserID, projected, found, visibility[u.ID], profileRefs, fallbackRefs, personalRefs)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
}
|
||
projected = applyAccountFreezeProjection(projected, viewerUserID, freezes[u.ID])
|
||
cache[u.ID] = projected
|
||
out[i] = projected
|
||
}
|
||
return out, nil
|
||
}
|
||
|
||
func applyAccountFreezeProjection(user domain.User, viewerUserID int64, freeze domain.AccountFreeze) domain.User {
|
||
// Base users and self users must never retain a viewer-scoped restriction.
|
||
user.RestrictionReasons = nil
|
||
if user.Deleted || viewerUserID == 0 || user.ID == 0 || user.ID == viewerUserID || !freeze.Frozen {
|
||
return user
|
||
}
|
||
user.RestrictionReasons = domain.AccountFrozenRestrictionReasons()
|
||
return user
|
||
}
|
||
|
||
func prefetchPrivacyVisibility(ctx context.Context, privacy PrivacyEvaluator, viewerUserID int64, users []domain.User) (map[int64]map[domain.PrivacyKey]bool, error) {
|
||
if privacy == nil || viewerUserID == 0 {
|
||
return nil, nil
|
||
}
|
||
batch, ok := privacy.(BatchPrivacyEvaluator)
|
||
if !ok {
|
||
return nil, nil
|
||
}
|
||
ids := make([]int64, 0, len(users))
|
||
seen := make(map[int64]struct{}, len(users))
|
||
for _, u := range users {
|
||
if u.ID == 0 || u.Deleted || u.ID == viewerUserID || u.ID == domain.OfficialSystemUserID || u.Bot {
|
||
continue
|
||
}
|
||
if _, ok := seen[u.ID]; ok {
|
||
continue
|
||
}
|
||
seen[u.ID] = struct{}{}
|
||
ids = append(ids, u.ID)
|
||
}
|
||
if len(ids) == 0 {
|
||
return nil, nil
|
||
}
|
||
return batch.CanSeeBatch(ctx, ids, viewerUserID, privacyProjectionKeys)
|
||
}
|
||
|
||
func projectOne(ctx context.Context, contacts store.ContactStore, viewerUserID int64, user domain.User) (domain.User, error) {
|
||
if user.Deleted {
|
||
return user.DeletedTombstone(), nil
|
||
}
|
||
contact, found, err := contacts.Get(ctx, viewerUserID, user.ID)
|
||
if err != nil {
|
||
return domain.User{}, err
|
||
}
|
||
return applyContactProjection(user, contact, found), nil
|
||
}
|
||
|
||
func uniqueUserIDs(users []domain.User) []int64 {
|
||
seen := make(map[int64]struct{}, len(users))
|
||
ids := make([]int64, 0, len(users))
|
||
for _, user := range users {
|
||
if user.ID == 0 || user.Deleted {
|
||
continue
|
||
}
|
||
if _, ok := seen[user.ID]; ok {
|
||
continue
|
||
}
|
||
seen[user.ID] = struct{}{}
|
||
ids = append(ids, user.ID)
|
||
}
|
||
return ids
|
||
}
|
||
|
||
func applyBasePhotos(user domain.User, profileRefs, fallbackRefs, personalRefs map[int64]domain.ProfilePhotoRef, viewerUserID int64) domain.User {
|
||
if user.Deleted {
|
||
return user.DeletedTombstone()
|
||
}
|
||
if !hasPhotoLookups(profileRefs, fallbackRefs, personalRefs) {
|
||
return user
|
||
}
|
||
clearPhoto(&user)
|
||
if viewerUserID != 0 && user.ID != viewerUserID {
|
||
if ref, ok := personalRefs[user.ID]; ok && ref.PhotoID != 0 {
|
||
ref.Personal = true
|
||
applyPhotoRef(&user, ref)
|
||
return user
|
||
}
|
||
}
|
||
if ref, ok := profileRefs[user.ID]; ok && ref.PhotoID != 0 {
|
||
applyPhotoRef(&user, ref)
|
||
return user
|
||
}
|
||
if ref, ok := fallbackRefs[user.ID]; ok && ref.PhotoID != 0 {
|
||
applyPhotoRef(&user, ref)
|
||
}
|
||
return user
|
||
}
|
||
|
||
func applyContactProjection(user domain.User, contact domain.Contact, found bool) domain.User {
|
||
if user.Deleted {
|
||
return user.DeletedTombstone()
|
||
}
|
||
if !found {
|
||
user.Phone = ""
|
||
user.Contact = false
|
||
user.Mutual = false
|
||
user.CloseFriend = false
|
||
user.ContactNote = ""
|
||
user.ContactNoteEntities = nil
|
||
return user
|
||
}
|
||
user.Contact = true
|
||
user.Mutual = contact.Mutual || contact.User.Mutual
|
||
user.CloseFriend = contact.CloseFriend || contact.User.CloseFriend
|
||
user.ContactNote = contact.Note
|
||
user.ContactNoteEntities = append([]domain.MessageEntity(nil), contact.NoteEntities...)
|
||
if contact.User.Phone != "" {
|
||
user.Phone = contact.User.Phone
|
||
} else {
|
||
user.Phone = contact.Phone
|
||
}
|
||
if contact.User.FirstName != "" || contact.User.LastName != "" {
|
||
user.FirstName = contact.User.FirstName
|
||
user.LastName = contact.User.LastName
|
||
} else if contact.FirstName != "" || contact.LastName != "" {
|
||
user.FirstName = contact.FirstName
|
||
user.LastName = contact.LastName
|
||
}
|
||
return user
|
||
}
|
||
|
||
func applyPrivacy(ctx context.Context, privacy PrivacyEvaluator, viewerUserID int64, user domain.User, isContact bool, vis map[domain.PrivacyKey]bool, profileRefs, fallbackRefs, personalRefs map[int64]domain.ProfilePhotoRef) (domain.User, error) {
|
||
if user.Deleted {
|
||
return user.DeletedTombstone(), nil
|
||
}
|
||
if privacy == nil {
|
||
return user, nil
|
||
}
|
||
// vis 为批量预取结果（projectBatch 一次 ListPrivacyRules+GetReverseContacts 算得）；
|
||
// 为 nil 时回退逐 CanSee，二者结果等价。
|
||
canSee := func(key domain.PrivacyKey) (bool, error) {
|
||
if vis != nil {
|
||
return vis[key], nil
|
||
}
|
||
return privacy.CanSee(ctx, user.ID, viewerUserID, key)
|
||
}
|
||
phoneAllowed, err := canSee(domain.PrivacyKeyPhoneNumber)
|
||
if err != nil {
|
||
return domain.User{}, err
|
||
}
|
||
if !phoneAllowed && !isContact {
|
||
user.Phone = ""
|
||
}
|
||
statusAllowed, err := canSee(domain.PrivacyKeyStatusTimestamp)
|
||
if err != nil {
|
||
return domain.User{}, err
|
||
}
|
||
if !statusAllowed {
|
||
user.LastSeenAt = 0
|
||
if user.Status.Kind == domain.UserStatusOnline || user.Status.Kind == domain.UserStatusOffline {
|
||
user.Status = domain.UserStatus{Kind: domain.UserStatusRecently}
|
||
}
|
||
}
|
||
if ref, ok := personalRefs[user.ID]; ok && ref.PhotoID != 0 {
|
||
ref.Personal = true
|
||
applyPhotoRef(&user, ref)
|
||
return user, nil
|
||
}
|
||
if !hasPhotoLookups(profileRefs, fallbackRefs, personalRefs) && user.PhotoID == 0 {
|
||
return user, nil
|
||
}
|
||
profileAllowed, err := canSee(domain.PrivacyKeyProfilePhoto)
|
||
if err != nil {
|
||
return domain.User{}, err
|
||
}
|
||
if profileAllowed {
|
||
if ref, ok := profileRefs[user.ID]; ok && ref.PhotoID != 0 {
|
||
applyPhotoRef(&user, ref)
|
||
return user, nil
|
||
}
|
||
}
|
||
if ref, ok := fallbackRefs[user.ID]; ok && ref.PhotoID != 0 {
|
||
applyPhotoRef(&user, ref)
|
||
return user, nil
|
||
}
|
||
clearPhoto(&user)
|
||
return user, nil
|
||
}
|
||
|
||
func hasPhotoLookups(profileRefs, fallbackRefs, personalRefs map[int64]domain.ProfilePhotoRef) bool {
|
||
return len(profileRefs) != 0 || len(fallbackRefs) != 0 || len(personalRefs) != 0
|
||
}
|
||
|
||
func applyPhotoRef(user *domain.User, ref domain.ProfilePhotoRef) {
|
||
user.PhotoID = ref.PhotoID
|
||
user.PhotoDCID = ref.DCID
|
||
user.PhotoStripped = append([]byte(nil), ref.Stripped...)
|
||
user.PhotoPersonal = ref.Personal
|
||
user.PhotoHasVideo = ref.HasVideo
|
||
}
|
||
|
||
func clearPhoto(user *domain.User) {
|
||
user.PhotoID = 0
|
||
user.PhotoDCID = 0
|
||
user.PhotoStripped = nil
|
||
user.PhotoPersonal = false
|
||
user.PhotoHasVideo = false
|
||
}
|
||
|
||
func sanitizeDeletedUsers(users []domain.User) []domain.User {
|
||
var out []domain.User
|
||
for i, user := range users {
|
||
if !user.Deleted {
|
||
continue
|
||
}
|
||
if out == nil {
|
||
out = make([]domain.User, len(users))
|
||
copy(out, users)
|
||
}
|
||
out[i] = user.DeletedTombstone()
|
||
}
|
||
if out != nil {
|
||
return out
|
||
}
|
||
return users
|
||
}
|