// Copyright 2023 The go-ethereum Authors
// This file is part of the go-ethereum library.
//
// The go-ethereum library is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// The go-ethereum library is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.

package ethapi

import (
	"context"
	"encoding/json"
	"hash"
	"math/big"
	"testing"

	"github.com/XinFinOrg/XDPoSChain/common"
	"github.com/XinFinOrg/XDPoSChain/core"
	"github.com/XinFinOrg/XDPoSChain/core/rawdb"
	"github.com/XinFinOrg/XDPoSChain/core/state"
	"github.com/XinFinOrg/XDPoSChain/core/types"
	"github.com/XinFinOrg/XDPoSChain/crypto/keccak"
	"github.com/XinFinOrg/XDPoSChain/params"
	"github.com/XinFinOrg/XDPoSChain/rpc"
	"github.com/stretchr/testify/require"
)

// testHasher is the helper tool for transaction/receipt list hashing.
// The original hasher is trie, in order to get rid of import cycle,
// use the testing hasher instead.
type testHasher struct {
	hasher hash.Hash
}

func newHasher() *testHasher {
	return &testHasher{hasher: keccak.NewLegacyKeccak256()}
}

func (h *testHasher) Reset() {
	h.hasher.Reset()
}

func (h *testHasher) Update(key, val []byte) error {
	h.hasher.Write(key)
	h.hasher.Write(val)
	return nil
}

func (h *testHasher) Hash() common.Hash {
	return common.BytesToHash(h.hasher.Sum(nil))
}

func TestRPCMarshalBlock(t *testing.T) {
	var (
		txs []*types.Transaction
		to  = common.BytesToAddress([]byte{0x11})
	)
	for i := uint64(1); i <= 4; i++ {
		var tx *types.Transaction
		if i%2 == 0 {
			tx = types.NewTx(&types.LegacyTx{
				Nonce:    i,
				GasPrice: big.NewInt(11111),
				Gas:      1111,
				To:       &to,
				Value:    big.NewInt(111),
				Data:     []byte{0x11, 0x11, 0x11},
			})
		} else {
			tx = types.NewTx(&types.AccessListTx{
				ChainID:  big.NewInt(1337),
				Nonce:    i,
				GasPrice: big.NewInt(11111),
				Gas:      1111,
				To:       &to,
				Value:    big.NewInt(111),
				Data:     []byte{0x11, 0x11, 0x11},
			})
		}
		txs = append(txs, tx)
	}
	block := types.NewBlock(&types.Header{Number: big.NewInt(100)}, &types.Body{Transactions: txs}, nil, newHasher())

	var testSuite = []struct {
		inclTx bool
		fullTx bool
		want   string
	}{
		// without txs
		{
			inclTx: false,
			fullTx: false,
			want: `{
				"difficulty":"0x0",
				"extraData":"0x",
				"gasLimit":"0x0",
				"gasUsed":"0x0",
				"hash":"0x2cb4e4b5b5be5a2520377e87e8d7d2cf83fc0783fa6518d67b9606d3c5317b50",
				"logsBloom":"0x00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000",
				"miner":"0x0000000000000000000000000000000000000000",
				"mixHash":"0x0000000000000000000000000000000000000000000000000000000000000000",
				"nonce":"0x0000000000000000",
				"number":"0x64",
				"parentHash":"0x0000000000000000000000000000000000000000000000000000000000000000",
				"penalties":"0x",
				"receiptsRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421",
				"sha3Uncles":"0x1dcc4de8dec75d7aab85b567b6ccd41ad312451b948a7413f0a142fd40d49347",
				"size":"0x299",
				"stateRoot":"0x0000000000000000000000000000000000000000000000000000000000000000",
				"timestamp":"0x0",
				"transactionsRoot":"0x661a9febcfa8f1890af549b874faf9fa274aede26ef489d9db0b25daa569450e",
				"uncles":[],
				"validator":"0x",
				"validators":"0x"
			}`,
		},
		// only tx hashes
		{
			inclTx: true,
			fullTx: false,
			want: `{
				"difficulty":"0x0",
				"extraData":"0x",
				"gasLimit":"0x0",
				"gasUsed":"0x0",
				"hash":"0x2cb4e4b5b5be5a2520377e87e8d7d2cf83fc0783fa6518d67b9606d3c5317b50",
				"logsBloom":"0x00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000",
				"miner":"0x0000000000000000000000000000000000000000",
				"mixHash":"0x0000000000000000000000000000000000000000000000000000000000000000",
				"nonce":"0x0000000000000000",
				"number":"0x64",
				"parentHash":"0x0000000000000000000000000000000000000000000000000000000000000000",
				"penalties":"0x",
				"receiptsRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421",
				"sha3Uncles":"0x1dcc4de8dec75d7aab85b567b6ccd41ad312451b948a7413f0a142fd40d49347",
				"size":"0x299",
				"stateRoot":"0x0000000000000000000000000000000000000000000000000000000000000000",
				"timestamp":"0x0",
				"transactions": [
					"0x7d39df979e34172322c64983a9ad48302c2b889e55bda35324afecf043a77605",
					"0x9bba4c34e57c875ff57ac8d172805a26ae912006985395dc1bdf8f44140a7bf4",
					"0x98909ea1ff040da6be56bc4231d484de1414b3c1dac372d69293a4beb9032cb5",
					"0x12e1f81207b40c3bdcc13c0ee18f5f86af6d31754d57a0ea1b0d4cfef21abef1"
				],
				"transactionsRoot":"0x661a9febcfa8f1890af549b874faf9fa274aede26ef489d9db0b25daa569450e",
				"uncles":[],
				"validator":"0x",
				"validators":"0x"
			}`,
		},

		// full tx details
		{
			inclTx: true,
			fullTx: true,
			want: `{
				"difficulty":"0x0",
				"extraData":"0x",
				"gasLimit":"0x0",
				"gasUsed":"0x0",
				"hash":"0x2cb4e4b5b5be5a2520377e87e8d7d2cf83fc0783fa6518d67b9606d3c5317b50",
				"logsBloom":"0x00000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000000",
				"miner":"0x0000000000000000000000000000000000000000",
				"mixHash":"0x0000000000000000000000000000000000000000000000000000000000000000",
				"nonce":"0x0000000000000000",
				"number":"0x64",
				"parentHash":"0x0000000000000000000000000000000000000000000000000000000000000000",
				"penalties":"0x",
				"receiptsRoot":"0x56e81f171bcc55a6ff8345e692c0f86e5b48e01b996cadc001622fb5e363b421",
				"sha3Uncles":"0x1dcc4de8dec75d7aab85b567b6ccd41ad312451b948a7413f0a142fd40d49347",
				"size":"0x299",
				"stateRoot":"0x0000000000000000000000000000000000000000000000000000000000000000",
				"timestamp":"0x0",
				"transactions": [
					{
						"blockHash":"0x2cb4e4b5b5be5a2520377e87e8d7d2cf83fc0783fa6518d67b9606d3c5317b50",
						"blockNumber":"0x64",
						"from":"0x0000000000000000000000000000000000000000",
						"gas":"0x457",
						"gasPrice":"0x2b67",
						"hash":"0x7d39df979e34172322c64983a9ad48302c2b889e55bda35324afecf043a77605",
						"input":"0x111111",
						"nonce":"0x1",
						"to":"0x0000000000000000000000000000000000000011",
						"transactionIndex":"0x0",
						"value":"0x6f",
						"type":"0x1",
						"accessList":[],
						"chainId":"0x539",
						"v":"0x0",
						"r":"0x0",
						"s":"0x0",
						"yParity":"0x0"
					},
					{
						"blockHash":"0x2cb4e4b5b5be5a2520377e87e8d7d2cf83fc0783fa6518d67b9606d3c5317b50",
						"blockNumber":"0x64",
						"from":"0x0000000000000000000000000000000000000000",
						"gas":"0x457",
						"gasPrice":"0x2b67",
						"hash":"0x9bba4c34e57c875ff57ac8d172805a26ae912006985395dc1bdf8f44140a7bf4",
						"input":"0x111111",
						"nonce":"0x2",
						"to":"0x0000000000000000000000000000000000000011",
						"transactionIndex":"0x1",
						"value":"0x6f",
						"type":"0x0",
						"chainId":"0x7fffffffffffffee",
						"v":"0x0",
						"r":"0x0",
						"s":"0x0"
					},
					{
						"blockHash":"0x2cb4e4b5b5be5a2520377e87e8d7d2cf83fc0783fa6518d67b9606d3c5317b50",
						"blockNumber":"0x64",
						"from":"0x0000000000000000000000000000000000000000",
						"gas":"0x457",
						"gasPrice":"0x2b67",
						"hash":"0x98909ea1ff040da6be56bc4231d484de1414b3c1dac372d69293a4beb9032cb5",
						"input":"0x111111",
						"nonce":"0x3",
						"to":"0x0000000000000000000000000000000000000011",
						"transactionIndex":"0x2",
						"value":"0x6f",
						"type":"0x1",
						"accessList":[],
						"chainId":"0x539",
						"v":"0x0",
						"r":"0x0",
						"s":"0x0",
						"yParity":"0x0"
					},
					{
						"blockHash":"0x2cb4e4b5b5be5a2520377e87e8d7d2cf83fc0783fa6518d67b9606d3c5317b50",
						"blockNumber":"0x64",
						"from":"0x0000000000000000000000000000000000000000",
						"gas":"0x457",
						"gasPrice":"0x2b67",
						"hash":"0x12e1f81207b40c3bdcc13c0ee18f5f86af6d31754d57a0ea1b0d4cfef21abef1",
						"input":"0x111111",
						"nonce":"0x4",
						"to":"0x0000000000000000000000000000000000000011",
						"transactionIndex":"0x3",
						"value":"0x6f",
						"type":"0x0",
						"chainId":"0x7fffffffffffffee",
						"v":"0x0",
						"r":"0x0",
						"s":"0x0"
					}
				],
				"transactionsRoot":"0x661a9febcfa8f1890af549b874faf9fa274aede26ef489d9db0b25daa569450e",
				"uncles":[],
				"validator":"0x",
				"validators":"0x"
			}`,
		},
	}

	for i, tc := range testSuite {
		resp := RPCMarshalBlock(block, tc.inclTx, tc.fullTx, params.MainnetChainConfig)
		out, err := json.Marshal(resp)
		if err != nil {
			t.Errorf("test %d: json marshal error: %v", i, err)
			continue
		}
		require.JSONEqf(t, tc.want, string(out), "test %d", i)
	}
}

type storageBackendMock struct {
	*backendMock
	stateDB *state.StateDB
	header  *types.Header
}

func (b *storageBackendMock) StateAndHeaderByNumberOrHash(ctx context.Context, blockNrOrHash rpc.BlockNumberOrHash) (*state.StateDB, *types.Header, error) {
	return b.stateDB, b.header, nil
}

func TestGetStorageValues(t *testing.T) {
	t.Parallel()

	var (
		addr1 = common.HexToAddress("0x1111")
		addr2 = common.HexToAddress("0x2222")
		slot0 = common.Hash{}
		slot1 = common.BigToHash(big.NewInt(1))
		slot2 = common.BigToHash(big.NewInt(2))
		val0  = common.BigToHash(big.NewInt(42))
		val1  = common.BigToHash(big.NewInt(100))
		val2  = common.BigToHash(big.NewInt(200))

		genesis = &core.Genesis{
			Config: params.MergedTestChainConfig,
			Alloc: types.GenesisAlloc{
				addr1: {
					Balance: big.NewInt(params.Ether),
					Storage: map[common.Hash]common.Hash{
						slot0: val0,
						slot1: val1,
					},
				},
				addr2: {
					Balance: big.NewInt(params.Ether),
					Storage: map[common.Hash]common.Hash{
						slot2: val2,
					},
				},
			},
		}
	)
	db := rawdb.NewMemoryDatabase()
	block := genesis.MustCommit(db)
	stateDB, err := state.New(block.Root(), state.NewDatabase(db))
	if err != nil {
		t.Fatalf("failed to create state db: %v", err)
	}
	backend := &storageBackendMock{
		backendMock: newBackendMock(),
		stateDB:     stateDB,
		header:      block.Header(),
	}
	api := NewBlockChainAPI(backend, nil)
	latest := rpc.BlockNumberOrHashWithNumber(rpc.LatestBlockNumber)

	// Happy path: multiple addresses, multiple slots.
	result, err := api.GetStorageValues(context.Background(), map[common.Address][]common.Hash{
		addr1: {slot0, slot1},
		addr2: {slot2},
	}, latest)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	if len(result) != 2 {
		t.Fatalf("expected 2 addresses in result, got %d", len(result))
	}
	if got := common.BytesToHash(result[addr1][0]); got != val0 {
		t.Errorf("addr1 slot0: want %x, got %x", val0, got)
	}
	if got := common.BytesToHash(result[addr1][1]); got != val1 {
		t.Errorf("addr1 slot1: want %x, got %x", val1, got)
	}
	if got := common.BytesToHash(result[addr2][0]); got != val2 {
		t.Errorf("addr2 slot2: want %x, got %x", val2, got)
	}

	// Missing slot returns zero.
	result, err = api.GetStorageValues(context.Background(), map[common.Address][]common.Hash{
		addr1: {common.HexToHash("0xff")},
	}, latest)
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}
	if got := common.BytesToHash(result[addr1][0]); got != (common.Hash{}) {
		t.Errorf("missing slot: want zero, got %x", got)
	}

	// Empty request returns error.
	_, err = api.GetStorageValues(context.Background(), map[common.Address][]common.Hash{}, latest)
	if err == nil {
		t.Fatal("expected error for empty request")
	}

	// Exceeding slot limit returns error.
	tooMany := make([]common.Hash, maxGetStorageSlots+1)
	for i := range tooMany {
		tooMany[i] = common.BigToHash(big.NewInt(int64(i)))
	}
	_, err = api.GetStorageValues(context.Background(), map[common.Address][]common.Hash{
		addr1: tooMany,
	}, latest)
	if err == nil {
		t.Fatal("expected error for exceeding slot limit")
	}
}