//go:build system_test

package systemtests

import (
	"fmt"
	"os"
	"path/filepath"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
	"github.com/tidwall/gjson"

	systest "cosmossdk.io/systemtests"
)

func TestValidatorDoubleSign(t *testing.T) {
	// Scenario:
	//   given: a running chain
	//   when: a second instance with the same val key signs a block
	//   then: the validator is removed from the active set and jailed forever
	systest.Sut.ResetChain(t)
	cli := systest.NewCLIWrapper(t, systest.Sut, systest.Verbose)
	systest.Sut.StartChain(t)

	// Check the validator is in the active set
	rsp := cli.CustomQuery("q", "staking", "validators")
	t.Log(rsp)
	validatorPubKey := systest.LoadValidatorPubKeyForNode(t, systest.Sut, 0)
	rpc, pkBz := systest.Sut.RPCClient(t), validatorPubKey.Bytes()

	nodePowerBefore := systest.QueryCometValidatorPower(rpc, pkBz)
	require.NotEmpty(t, nodePowerBefore)
	t.Logf("nodePowerBefore: %v", nodePowerBefore)

	newNode := systest.Sut.AddFullnode(t, func(nodeNumber int, nodePath string) {
		valKeyFile := filepath.Join(systest.WorkDir, nodePath, "config", "priv_validator_key.json")
		_ = os.Remove(valKeyFile)
		_ = systest.MustCopyFile(filepath.Join(systest.Sut.NodeDir(0), "config", "priv_validator_key.json"), valKeyFile)
	})
	systest.Sut.AwaitNodeUp(t, fmt.Sprintf("http://%s:%d", newNode.IP, newNode.RPCPort))

	// let's wait some blocks to have evidence and update persisted
	var nodePowerAfter int64 = -1
	for i := 0; i < 100; i++ {
		systest.Sut.AwaitNextBlock(t, 6*time.Second)
		if nodePowerAfter = systest.QueryCometValidatorPower(rpc, pkBz); nodePowerAfter == 0 {
			break
		}
		t.Logf("wait %d", systest.Sut.CurrentHeight())
	}
	// then comet status updated
	require.Empty(t, nodePowerAfter)

	// and sdk status updated
	byzantineOperatorAddr := cli.GetKeyAddrPrefix("node0", "val")
	rsp = cli.CustomQuery("q", "staking", "validator", byzantineOperatorAddr)
	assert.True(t, gjson.Get(rsp, "validator.jailed").Bool(), rsp)

	// let's run for some blocks to confirm all good
	systest.Sut.AwaitNBlocks(t, 5)
}