package rpc

import (
	"context"
	"testing"

	"github.com/iamxvbaba/td/clock"
	"github.com/iamxvbaba/td/tg"
	"go.uber.org/zap"

	usernamesapp "telesrv/internal/app/usernames"
	"telesrv/internal/domain"
)

var _ usernamesapp.PeerUsernameNotifier = (*Router)(nil)

func TestNotifyPeerUsernamesChangedUserPushesPreloadedVector(t *testing.T) {
	const (
		targetID = int64(2002)
		viewerID = int64(1001)
	)
	users := &verifiedNotifyUsers{
		user:     domain.User{ID: targetID, FirstName: "Owner", Username: "owner_slot"},
		found:    true,
		audience: []int64{targetID, viewerID},
	}
	registry := newFakeUsernameRegistry()
	registry.byPeer[domain.Peer{Type: domain.PeerTypeUser, ID: targetID}] = []domain.Username{
		{Username: "owner_slot", Editable: true, Active: true, SortOrder: 1},
		{Username: "nft4", Active: true, SortOrder: 0, CollectibleID: 7},
	}
	sessions := &captureSessions{onlineUserIDs: []int64{targetID, viewerID}}
	r := New(Config{}, Deps{Users: users, Usernames: registry, Sessions: sessions}, zap.NewNop(), clock.System)

	if err := r.NotifyPeerUsernamesChanged(context.Background(), domain.Peer{
		Type: domain.PeerTypeUser, ID: targetID,
	}); err != nil {
		t.Fatalf("notify user usernames: %v", err)
	}
	if registry.peerCalls != 1 || registry.batchCalls != 0 {
		t.Fatalf("registry reads = peer %d / batch %d, want one peer-wide read", registry.peerCalls, registry.batchCalls)
	}
	updates := sessions.lastUserPush().(*tg.Updates)
	user := updates.Users[0].(*tg.User)
	assertVectorOnlyUsernames(t, "pushed user", user, []string{"nft4", "owner_slot"})
}

func TestNotifyPeerUsernamesChangedChannelPushesPreloadedVector(t *testing.T) {
	const (
		channelID = int64(4004)
		ownerID   = int64(3003)
		memberID  = int64(3004)
	)
	channels := &verifiedNotifyChannels{channel: domain.Channel{
		ID: channelID, AccessHash: 44, CreatorUserID: ownerID,
		Title: "Collectible", Username: "channel_slot", Broadcast: true,
	}}
	registry := newFakeUsernameRegistry()
	registry.byPeer[domain.Peer{Type: domain.PeerTypeChannel, ID: channelID}] = []domain.Username{
		{Username: "channel_slot", Editable: true, Active: true, SortOrder: 1},
		{Username: "collectible", Active: true, SortOrder: 0, CollectibleID: 9},
	}
	sessions := &captureSessions{
		onlineUserIDs:  []int64{ownerID, memberID},
		channelMembers: map[int64][]int64{channelID: {ownerID, memberID}},
	}
	r := New(Config{}, Deps{
		Channels: channels, Usernames: registry, Sessions: sessions,
	}, zap.NewNop(), clock.System)

	if err := r.NotifyPeerUsernamesChanged(context.Background(), domain.Peer{
		Type: domain.PeerTypeChannel, ID: channelID,
	}); err != nil {
		t.Fatalf("notify channel usernames: %v", err)
	}
	if registry.peerCalls != 1 || registry.batchCalls != 0 {
		t.Fatalf("registry reads = peer %d / batch %d, want one peer-wide read", registry.peerCalls, registry.batchCalls)
	}
	updates := sessions.lastUserPush().(*tg.Updates)
	channel := updates.Chats[0].(*tg.Channel)
	assertVectorOnlyUsernames(t, "pushed channel", channel, []string{"collectible", "channel_slot"})
}