package integration

import (
	"github.com/kaspanet/kaspad/domain/consensus/model/externalapi"
	"testing"

	"github.com/kaspanet/kaspad/app/appmessage"
	"github.com/kaspanet/kaspad/domain/consensus/utils/consensushashing"
)

func TestVirtualSelectedParentChain(t *testing.T) {
	// Setup a couple of kaspad instances
	kaspad1, kaspad2, _, teardown := standardSetup(t)
	defer teardown()

	// Register to virtual selected parent chain changes
	onVirtualSelectedParentChainChangedChan := make(chan *appmessage.VirtualSelectedParentChainChangedNotificationMessage)
	err := kaspad1.rpcClient.RegisterForVirtualSelectedParentChainChangedNotifications(
		func(notification *appmessage.VirtualSelectedParentChainChangedNotificationMessage) {
			onVirtualSelectedParentChainChangedChan <- notification
		})
	if err != nil {
		t.Fatalf("Failed to register for virtual selected parent chain change notifications: %s", err)
	}

	// In kaspad1, mine a chain over the genesis and make sure
	// each chain changed notifications contains only one entry
	// in `added` and nothing in `removed`
	chain1TipHash := consensushashing.BlockHash(kaspad1.config.NetParams().GenesisBlock)
	chain1TipHashString := chain1TipHash.String()
	const blockAmountToMine = 10
	for i := 0; i < blockAmountToMine; i++ {
		minedBlock := mineNextBlock(t, kaspad1)
		notification := <-onVirtualSelectedParentChainChangedChan
		if len(notification.RemovedChainBlockHashes) > 0 {
			t.Fatalf("RemovedChainBlockHashes is unexpectedly not empty")
		}
		if len(notification.AddedChainBlocks) != 1 {
			t.Fatalf("Unexpected length of AddedChainBlocks. Want: %d, got: %d",
				1, len(notification.AddedChainBlocks))
		}

		minedBlockHash := consensushashing.BlockHash(minedBlock)
		minedBlockHashString := minedBlockHash.String()
		if minedBlockHashString != notification.AddedChainBlocks[0].Hash {
			t.Fatalf("Unexpected block hash in AddedChainBlocks. Want: %s, got: %s",
				minedBlockHashString, notification.AddedChainBlocks[0].Hash)
		}
		chain1TipHashString = minedBlockHashString
	}

	// In kaspad2, mine a different chain of `blockAmountToMine` + 1
	// blocks over the genesis
	var chain2Tip *externalapi.DomainBlock
	for i := 0; i < blockAmountToMine+1; i++ {
		chain2Tip = mineNextBlock(t, kaspad2)
	}

	// Connect the two kaspads. This should trigger sync
	// between the two nodes
	connect(t, kaspad1, kaspad2)

	chain2TipHash := consensushashing.BlockHash(chain2Tip)
	chain2TipHashString := chain2TipHash.String()

	// For the first `blockAmountToMine - 1` blocks we don't expect
	// the chain to change at all
	for i := 0; i < blockAmountToMine-1; i++ {
		notification := <-onVirtualSelectedParentChainChangedChan
		if len(notification.RemovedChainBlockHashes) > 0 {
			t.Fatalf("RemovedChainBlockHashes is unexpectedly not empty")
		}
		if len(notification.AddedChainBlocks) > 0 {
			t.Fatalf("AddedChainBlocks is unexpectedly not empty")
		}
	}

	// Either the next block could cause a reorg or the one
	// after it
	potentialReorgNotification1 := <-onVirtualSelectedParentChainChangedChan
	potentialReorgNotification2 := <-onVirtualSelectedParentChainChangedChan
	var reorgNotification *appmessage.VirtualSelectedParentChainChangedNotificationMessage
	var nonReorgNotification *appmessage.VirtualSelectedParentChainChangedNotificationMessage
	if len(potentialReorgNotification1.RemovedChainBlockHashes) > 0 {
		reorgNotification = potentialReorgNotification1
		nonReorgNotification = potentialReorgNotification2
	} else {
		reorgNotification = potentialReorgNotification2
		nonReorgNotification = potentialReorgNotification1
	}

	// Make sure that the non-reorg notification has nothing
	// in `removed`
	if len(nonReorgNotification.RemovedChainBlockHashes) > 0 {
		t.Fatalf("nonReorgNotification.RemovedChainBlockHashes is unexpectedly not empty")
	}

	// Make sure that the reorg notification contains exactly
	// `blockAmountToMine` blocks in its `removed`
	if len(reorgNotification.RemovedChainBlockHashes) != blockAmountToMine {
		t.Fatalf("Unexpected length of reorgNotification.RemovedChainBlockHashes. Want: %d, got: %d",
			blockAmountToMine, len(reorgNotification.RemovedChainBlockHashes))
	}

	// Get the virtual selected parent chain from the tip of
	// the first chain
	virtualSelectedParentChainFromChain1Tip, err := kaspad1.rpcClient.GetVirtualSelectedParentChainFromBlock(chain1TipHashString)
	if err != nil {
		t.Fatalf("GetVirtualSelectedParentChainFromBlock failed: %s", err)
	}

	// Make sure that `blockAmountToMine` blocks were removed
	// and `blockAmountToMine + 1` blocks were added
	if len(virtualSelectedParentChainFromChain1Tip.RemovedChainBlockHashes) != blockAmountToMine {
		t.Fatalf("Unexpected length of virtualSelectedParentChainFromChain1Tip.RemovedChainBlockHashes. Want: %d, got: %d",
			blockAmountToMine, len(virtualSelectedParentChainFromChain1Tip.RemovedChainBlockHashes))
	}
	if len(virtualSelectedParentChainFromChain1Tip.AddedChainBlocks) != blockAmountToMine+1 {
		t.Fatalf("Unexpected length of virtualSelectedParentChainFromChain1Tip.AddedChainBlocks. Want: %d, got: %d",
			blockAmountToMine+1, len(virtualSelectedParentChainFromChain1Tip.AddedChainBlocks))
	}

	// Make sure that the last block in `added` is the tip
	// of chain2
	lastAddedChainBlock := virtualSelectedParentChainFromChain1Tip.AddedChainBlocks[len(virtualSelectedParentChainFromChain1Tip.AddedChainBlocks)-1]
	if lastAddedChainBlock.Hash != chain2TipHashString {
		t.Fatalf("Unexpected last added chain block. Want: %s, got: %s",
			chain2TipHashString, lastAddedChainBlock.Hash)
	}
}