package dex

import (
	"crypto/ecdsa"
	"fmt"
	"math/big"
	"testing"
	"time"

	coreCommon "github.com/dexon-foundation/dexon-consensus/common"
	coreTypes "github.com/dexon-foundation/dexon-consensus/core/types"

	"github.com/dexon-foundation/dexon/common"
	"github.com/dexon-foundation/dexon/consensus/dexcon"
	"github.com/dexon-foundation/dexon/core"
	"github.com/dexon-foundation/dexon/core/types"
	"github.com/dexon-foundation/dexon/core/vm"
	"github.com/dexon-foundation/dexon/crypto"
	"github.com/dexon-foundation/dexon/ethdb"
	"github.com/dexon-foundation/dexon/params"
	"github.com/dexon-foundation/dexon/rlp"
)

func TestPreparePayload(t *testing.T) {
	key, err := crypto.GenerateKey()
	if err != nil {
		t.Errorf("hex to ecdsa error: %v", err)
	}

	dex, err := newTestDexonWithGenesis(key)
	if err != nil {
		t.Errorf("new test dexon error: %v", err)
	}

	signer := types.NewEIP155Signer(dex.chainConfig.ChainID)

	var expectTx types.Transactions
	for i := 0; i < 5; i++ {
		tx, err := addTx(dex, i, signer, key)
		if err != nil {
			t.Errorf("add tx error: %v", err)
		}
		expectTx = append(expectTx, tx)
	}

	// This transaction will not be included.
	_, err = addTx(dex, 100, signer, key)
	if err != nil {
		t.Errorf("add tx error: %v", err)
	}

	chainID := new(big.Int).Mod(crypto.PubkeyToAddress(key.PublicKey).Big(),
		big.NewInt(int64(dex.chainConfig.Dexcon.NumChains)))
	var chainNum uint32
	for chainNum = 0; chainNum < dex.chainConfig.Dexcon.NumChains; chainNum++ {

		payload, err := dex.app.PreparePayload(coreTypes.Position{ChainID: chainNum})
		if err != nil {
			t.Errorf("prepare payload error: %v", err)
		}

		var transactions types.Transactions
		err = rlp.DecodeBytes(payload, &transactions)
		if err != nil {
			t.Errorf("rlp decode error: %v", err)
		}

		// Only one chain id allow prepare transactions.
		if chainID.Uint64() == uint64(chainNum) && len(transactions) != 5 {
			t.Errorf("incorrect transaction num expect %v but %v", 5, len(transactions))
		} else if chainID.Uint64() != uint64(chainNum) && len(transactions) != 0 {
			t.Errorf("expect empty slice but %v", len(transactions))
		}

		for i, tx := range transactions {
			if expectTx[i].Gas() != tx.Gas() {
				t.Errorf("unexpected gas expect %v but %v", expectTx[i].Gas(), tx.Gas())
			}

			if expectTx[i].Hash() != tx.Hash() {
				t.Errorf("unexpected hash expect %v but %v", expectTx[i].Hash(), tx.Hash())
			}

			if expectTx[i].Nonce() != tx.Nonce() {
				t.Errorf("unexpected nonce expect %v but %v", expectTx[i].Nonce(), tx.Nonce())
			}

			if expectTx[i].GasPrice().Uint64() != tx.GasPrice().Uint64() {
				t.Errorf("unexpected gas price expect %v but %v",
					expectTx[i].GasPrice().Uint64(), tx.GasPrice().Uint64())
			}
		}
	}
}

func TestPrepareWitness(t *testing.T) {
	key, err := crypto.GenerateKey()
	if err != nil {
		t.Errorf("hex to ecdsa error: %v", err)
	}

	dex, err := newTestDexonWithGenesis(key)
	if err != nil {
		t.Errorf("new test dexon error: %v", err)
	}

	currentBlock := dex.blockchain.CurrentBlock()

	witness, err := dex.app.PrepareWitness(0)
	if err != nil {
		t.Errorf("prepare witness error: %v", err)
	}

	if witness.Height != currentBlock.NumberU64() {
		t.Errorf("unexpeted witness height %v", witness.Height)
	}

	var witnessData types.WitnessData
	err = rlp.DecodeBytes(witness.Data, &witnessData)
	if err != nil {
		t.Errorf("rlp decode error: %v", err)
	}

	if witnessData.Root != currentBlock.Root() {
		t.Errorf("expect root %v but %v", currentBlock.Root(), witnessData.Root)
	}

	if witnessData.ReceiptHash != currentBlock.ReceiptHash() {
		t.Errorf("expect receipt hash %v but %v", currentBlock.ReceiptHash(), witnessData.ReceiptHash)
	}

	if _, err := dex.app.PrepareWitness(999); err == nil {
		t.Errorf("it must be get error from prepare")
	} else {
		t.Logf("Nice error: %v", err)
	}
}

func TestVerifyBlock(t *testing.T) {
	key, err := crypto.GenerateKey()
	if err != nil {
		t.Errorf("hex to ecdsa error: %v", err)
	}

	dex, err := newTestDexonWithGenesis(key)
	if err != nil {
		t.Errorf("new test dexon error: %v", err)
	}

	// Prepare first confirmed block.
	prepareConfirmedBlocks(dex, []*ecdsa.PrivateKey{key}, 0)

	chainID := new(big.Int).Mod(crypto.PubkeyToAddress(key.PublicKey).Big(),
		big.NewInt(int64(dex.chainConfig.Dexcon.NumChains)))

	// Prepare normal block.
	block := coreTypes.NewBlock()
	block.Hash = coreCommon.NewRandomHash()
	block.Position.ChainID = uint32(chainID.Uint64())
	block.Position.Height = 1
	block.Payload, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 100)
	if err != nil {
		t.Errorf("prepare data error: %v", err)
	}

	// Expect ok.
	status := dex.app.VerifyBlock(block)
	if status != coreTypes.VerifyOK {
		t.Errorf("verify fail: %v", status)
	}

	// Prepare invalid nonce tx.
	block = coreTypes.NewBlock()
	block.Hash = coreCommon.NewRandomHash()
	block.Position.ChainID = uint32(chainID.Uint64())
	block.Position.Height = 1
	block.Payload, block.Witness, err = prepareDataWithoutTxPool(dex, key, 1, 100)
	if err != nil {
		t.Errorf("prepare data error: %v", err)
	}

	// Expect invalid block.
	status = dex.app.VerifyBlock(block)
	if status != coreTypes.VerifyInvalidBlock {
		t.Errorf("verify fail: %v", status)
	}

	// Prepare invalid block height.
	block = coreTypes.NewBlock()
	block.Hash = coreCommon.NewRandomHash()
	block.Position.ChainID = uint32(chainID.Uint64())
	block.Position.Height = 2
	block.Payload, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 100)
	if err != nil {
		t.Errorf("prepare data error: %v", err)
	}

	// Expect retry later.
	status = dex.app.VerifyBlock(block)
	if status != coreTypes.VerifyRetryLater {
		t.Errorf("verify fail expect retry later but get %v", status)
	}

	// Prepare reach block limit.
	block = coreTypes.NewBlock()
	block.Hash = coreCommon.NewRandomHash()
	block.Position.ChainID = uint32(chainID.Uint64())
	block.Position.Height = 1
	block.Payload, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 10000)
	if err != nil {
		t.Errorf("prepare data error: %v", err)
	}

	// Expect invalid block.
	status = dex.app.VerifyBlock(block)
	if status != coreTypes.VerifyInvalidBlock {
		t.Errorf("verify fail expect invalid block but get %v", status)
	}

	// Prepare insufficient funds.
	block = coreTypes.NewBlock()
	block.Hash = coreCommon.NewRandomHash()
	block.Position.ChainID = uint32(chainID.Uint64())
	block.Position.Height = 1
	_, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 0)
	if err != nil {
		t.Errorf("prepare data error: %v", err)
	}

	signer := types.NewEIP155Signer(dex.chainConfig.ChainID)
	tx := types.NewTransaction(
		0,
		common.BytesToAddress([]byte{9}),
		big.NewInt(50000000000000001),
		params.TxGas,
		big.NewInt(1),
		nil)
	tx, err = types.SignTx(tx, signer, key)
	if err != nil {
		return
	}

	block.Payload, err = rlp.EncodeToBytes(types.Transactions{tx})
	if err != nil {
		return
	}

	// expect invalid block
	status = dex.app.VerifyBlock(block)
	if status != coreTypes.VerifyInvalidBlock {
		t.Errorf("verify fail expect invalid block but get %v", status)
	}

	// Prepare invalid intrinsic gas.
	block = coreTypes.NewBlock()
	block.Hash = coreCommon.NewRandomHash()
	block.Position.ChainID = uint32(chainID.Uint64())
	block.Position.Height = 1
	_, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 0)
	if err != nil {
		t.Errorf("prepare data error: %v", err)
	}

	signer = types.NewEIP155Signer(dex.chainConfig.ChainID)
	tx = types.NewTransaction(
		0,
		common.BytesToAddress([]byte{9}),
		big.NewInt(1),
		params.TxGas,
		big.NewInt(1),
		make([]byte, 1))
	tx, err = types.SignTx(tx, signer, key)
	if err != nil {
		return
	}

	block.Payload, err = rlp.EncodeToBytes(types.Transactions{tx})
	if err != nil {
		return
	}

	// Expect invalid block.
	status = dex.app.VerifyBlock(block)
	if status != coreTypes.VerifyInvalidBlock {
		t.Errorf("verify fail expect invalid block but get %v", status)
	}

	// Prepare invalid transactions with nonce.
	block = coreTypes.NewBlock()
	block.Hash = coreCommon.NewRandomHash()
	block.Position.ChainID = uint32(chainID.Uint64())
	block.Position.Height = 1
	_, block.Witness, err = prepareDataWithoutTxPool(dex, key, 0, 0)
	if err != nil {
		t.Errorf("prepare data error: %v", err)
	}

	signer = types.NewEIP155Signer(dex.chainConfig.ChainID)
	tx1 := types.NewTransaction(
		0,
		common.BytesToAddress([]byte{9}),
		big.NewInt(1),
		params.TxGas,
		big.NewInt(1),
		make([]byte, 1))
	tx1, err = types.SignTx(tx, signer, key)
	if err != nil {
		return
	}

	// Invalid nonce.
	tx2 := types.NewTransaction(
		2,
		common.BytesToAddress([]byte{9}),
		big.NewInt(1),
		params.TxGas,
		big.NewInt(1),
		make([]byte, 1))
	tx2, err = types.SignTx(tx, signer, key)
	if err != nil {
		return
	}

	block.Payload, err = rlp.EncodeToBytes(types.Transactions{tx1, tx2})
	if err != nil {
		return
	}

	// Expect invalid block.
	status = dex.app.VerifyBlock(block)
	if status != coreTypes.VerifyInvalidBlock {
		t.Errorf("verify fail expect invalid block but get %v", status)
	}
}

func TestBlockConfirmed(t *testing.T) {
	key, err := crypto.GenerateKey()
	if err != nil {
		t.Errorf("hex to ecdsa error: %v", err)
	}

	dex, err := newTestDexonWithGenesis(key)
	if err != nil {
		t.Errorf("new test dexon error: %v", err)
	}

	chainID := new(big.Int).Mod(crypto.PubkeyToAddress(key.PublicKey).Big(),
		big.NewInt(int64(dex.chainConfig.Dexcon.NumChains)))

	var (
		expectCost    big.Int
		expectNonce   uint64
		expectCounter uint64
	)
	for i := 0; i < 10; i++ {
		var startNonce int
		if expectNonce != 0 {
			startNonce = int(expectNonce) + 1
		}
		payload, witness, cost, nonce, err := prepareData(dex, key, startNonce, 50)
		if err != nil {
			t.Errorf("prepare data error: %v", err)
		}
		expectCost.Add(&expectCost, &cost)
		expectNonce = nonce

		block := coreTypes.NewBlock()
		block.Hash = coreCommon.NewRandomHash()
		block.Witness = witness
		block.Payload = payload
		block.Position.ChainID = uint32(chainID.Uint64())

		dex.app.BlockConfirmed(*block)
		expectCounter++
	}

	info := dex.app.blockchain.GetAddressInfo(uint32(chainID.Uint64()),
		crypto.PubkeyToAddress(key.PublicKey))

	if info.Counter != expectCounter {
		t.Errorf("expect address counter is %v but %v", expectCounter, info.Counter)
	}

	if info.Cost.Cmp(&expectCost) != 0 {
		t.Errorf("expect address cost is %v but %v", expectCost.Uint64(), info.Cost.Uint64())
	}

	if info.Nonce != expectNonce {
		t.Errorf("expect address nonce is %v but %v", expectNonce, info.Nonce)
	}
}

func TestBlockDelivered(t *testing.T) {
	key, err := crypto.GenerateKey()
	if err != nil {
		t.Errorf("hex to ecdsa error: %v", err)
	}

	dex, err := newTestDexonWithGenesis(key)
	if err != nil {
		t.Errorf("new test dexon error: %v", err)
	}

	address := crypto.PubkeyToAddress(key.PublicKey)
	firstBlocksInfo, err := prepareConfirmedBlocks(dex, []*ecdsa.PrivateKey{key}, 50)
	if err != nil {
		t.Errorf("preapare confirmed block error: %v", err)
	}

	dex.app.BlockDelivered(firstBlocksInfo[0].Block.Hash, firstBlocksInfo[0].Block.Position,
		coreTypes.FinalizationResult{
			Timestamp: time.Now(),
			Height:    1,
		})

	_, pendingState := dex.blockchain.GetPending()
	currentBlock := dex.blockchain.CurrentBlock()
	if currentBlock.NumberU64() != 0 {
		t.Errorf("unexpected current block number %v", currentBlock.NumberU64())
	}

	pendingNonce := pendingState.GetNonce(address)
	if pendingNonce != firstBlocksInfo[0].Nonce+1 {
		t.Errorf("unexpected pending state nonce %v", pendingNonce)
	}

	balance := pendingState.GetBalance(address)
	if new(big.Int).Add(balance, &firstBlocksInfo[0].Cost).Cmp(big.NewInt(50000000000000000)) != 0 {
		t.Errorf("unexpected pending state balance %v", balance)
	}

	// prepare second block to witness first block
	secondBlocksInfo, err := prepareConfirmedBlocks(dex, []*ecdsa.PrivateKey{key}, 25)
	if err != nil {
		t.Errorf("preapare confirmed block error: %v", err)
	}

	dex.app.BlockDelivered(secondBlocksInfo[0].Block.Hash, secondBlocksInfo[0].Block.Position,
		coreTypes.FinalizationResult{
			Timestamp: time.Now(),
			Height:    2,
		})

	// second block witness first block, so current block number should be 1
	currentBlock = dex.blockchain.CurrentBlock()
	if currentBlock.NumberU64() != 1 {
		t.Errorf("unexpected current block number %v", currentBlock.NumberU64())
	}

	currentState, err := dex.blockchain.State()
	if err != nil {
		t.Errorf("current state error: %v", err)
	}

	currentNonce := currentState.GetNonce(address)
	if currentNonce != firstBlocksInfo[0].Nonce+1 {
		t.Errorf("unexpected current state nonce %v", currentNonce)
	}

	balance = currentState.GetBalance(address)
	if new(big.Int).Add(balance, &firstBlocksInfo[0].Cost).Cmp(big.NewInt(50000000000000000)) != 0 {
		t.Errorf("unexpected current state balance %v", balance)
	}
}

func BenchmarkBlockDeliveredFlow(b *testing.B) {
	key, err := crypto.GenerateKey()
	if err != nil {
		b.Errorf("hex to ecdsa error: %v", err)
		return
	}

	dex, err := newTestDexonWithGenesis(key)
	if err != nil {
		b.Errorf("new test dexon error: %v", err)
	}

	b.ResetTimer()
	for i := 1; i <= b.N; i++ {
		blocksInfo, err := prepareConfirmedBlocks(dex, []*ecdsa.PrivateKey{key}, 100)
		if err != nil {
			b.Errorf("preapare confirmed block error: %v", err)
			return
		}

		dex.app.BlockDelivered(blocksInfo[0].Block.Hash, blocksInfo[0].Block.Position,
			coreTypes.FinalizationResult{
				Timestamp: time.Now(),
				Height:    uint64(i),
			})
	}
}

func newTestDexonWithGenesis(allocKey *ecdsa.PrivateKey) (*Dexon, error) {
	db := ethdb.NewMemDatabase()

	testBankAddress := crypto.PubkeyToAddress(allocKey.PublicKey)
	genesis := core.DefaultTestnetGenesisBlock()
	genesis.Alloc = core.GenesisAlloc{
		testBankAddress: {
			Balance: big.NewInt(100000000000000000),
			Staked:  big.NewInt(50000000000000000),
		},
	}
	chainConfig, _, err := core.SetupGenesisBlock(db, genesis)
	if err != nil {
		return nil, err
	}

	key, err := crypto.GenerateKey()
	if err != nil {
		return nil, err
	}

	config := Config{PrivateKey: key}
	vmConfig := vm.Config{IsBlockProposer: true}

	engine := dexcon.New()

	dex := &Dexon{
		chainDb:     db,
		chainConfig: chainConfig,
		networkID:   config.NetworkId,
		engine:      engine,
	}

	dex.blockchain, err = core.NewBlockChain(db, nil, chainConfig, engine, vmConfig, nil)
	if err != nil {
		return nil, err
	}

	txPoolConfig := core.DefaultTxPoolConfig
	dex.txPool = core.NewTxPool(txPoolConfig, chainConfig, dex.blockchain, true)

	dex.APIBackend = &DexAPIBackend{dex, nil}
	dex.governance = NewDexconGovernance(dex.APIBackend, dex.chainConfig, config.PrivateKey)
	engine.SetConfigFetcher(dex.governance)
	dex.app = NewDexconApp(dex.txPool, dex.blockchain, dex.governance, db, &config)

	return dex, nil
}

// Add tx into tx pool.
func addTx(dex *Dexon, nonce int, signer types.Signer, key *ecdsa.PrivateKey) (
	*types.Transaction, error) {
	tx := types.NewTransaction(
		uint64(nonce),
		common.BytesToAddress([]byte{9}),
		big.NewInt(int64(nonce*1)),
		params.TxGas,
		big.NewInt(1),
		nil)
	tx, err := types.SignTx(tx, signer, key)
	if err != nil {
		return nil, err
	}

	if err := dex.txPool.AddRemote(tx); err != nil {
		return nil, err
	}

	return tx, nil
}

// Prepare data with given transaction number and start nonce.
func prepareData(dex *Dexon, key *ecdsa.PrivateKey, startNonce, txNum int) (
	payload []byte, witness coreTypes.Witness, cost big.Int, nonce uint64, err error) {
	signer := types.NewEIP155Signer(dex.chainConfig.ChainID)
	chainID := new(big.Int).Mod(crypto.PubkeyToAddress(key.PublicKey).Big(),
		big.NewInt(int64(dex.chainConfig.Dexcon.NumChains)))

	for n := startNonce; n < startNonce+txNum; n++ {
		var tx *types.Transaction
		tx, err = addTx(dex, n, signer, key)
		if err != nil {
			return
		}

		cost.Add(&cost, tx.Cost())
		nonce = uint64(n)
	}

	payload, err = dex.app.PreparePayload(coreTypes.Position{ChainID: uint32(chainID.Uint64())})
	if err != nil {
		return
	}

	witness, err = dex.app.PrepareWitness(0)
	if err != nil {
		return
	}

	return
}

func prepareDataWithoutTxPool(dex *Dexon, key *ecdsa.PrivateKey, startNonce, txNum int) (
	payload []byte, witness coreTypes.Witness, err error) {
	signer := types.NewEIP155Signer(dex.chainConfig.ChainID)

	var transactions types.Transactions
	for n := startNonce; n < startNonce+txNum; n++ {
		tx := types.NewTransaction(
			uint64(n),
			common.BytesToAddress([]byte{9}),
			big.NewInt(int64(n*1)),
			params.TxGas,
			big.NewInt(1),
			nil)
		tx, err = types.SignTx(tx, signer, key)
		if err != nil {
			return
		}
		transactions = append(transactions, tx)
	}

	payload, err = rlp.EncodeToBytes(&transactions)
	if err != nil {
		return
	}

	witness, err = dex.app.PrepareWitness(0)
	if err != nil {
		return
	}

	return
}

func prepareConfirmedBlocks(dex *Dexon, keys []*ecdsa.PrivateKey, txNum int) (blocksInfo []struct {
	Block *coreTypes.Block
	Cost  big.Int
	Nonce uint64
}, err error) {
	for _, key := range keys {
		address := crypto.PubkeyToAddress(key.PublicKey)
		chainID := new(big.Int).Mod(address.Big(),
			big.NewInt(int64(dex.chainConfig.Dexcon.NumChains)))

		// Prepare one block for pending.
		var (
			payload []byte
			witness coreTypes.Witness
			cost    big.Int
			nonce   uint64
		)
		startNonce := dex.txPool.State().GetNonce(address)
		payload, witness, cost, nonce, err = prepareData(dex, key, int(startNonce), txNum)
		if err != nil {
			err = fmt.Errorf("prepare data error: %v", err)
			return
		}

		block := coreTypes.NewBlock()
		block.Hash = coreCommon.NewRandomHash()
		block.Witness = witness
		block.Payload = payload
		block.Position.ChainID = uint32(chainID.Uint64())

		status := dex.app.VerifyBlock(block)
		if status != coreTypes.VerifyOK {
			err = fmt.Errorf("verify fail: %v", status)
			return
		}

		dex.app.BlockConfirmed(*block)

		blocksInfo = append(blocksInfo, struct {
			Block *coreTypes.Block
			Cost  big.Int
			Nonce uint64
		}{Block: block, Cost: cost, Nonce: nonce})
	}

	return
}