package query_test

import (
	"context"
	"encoding/base64"
	"fmt"
	"math/big"
	"regexp"
	"strconv"
	"strings"
	"testing"
	"time"

	"github.com/tutus-one/tutus-chain/internal/random"
	"github.com/tutus-one/tutus-chain/internal/testcli"
	"github.com/tutus-one/tutus-chain/pkg/config"
	"github.com/tutus-one/tutus-chain/pkg/core/native/nativehashes"
	"github.com/tutus-one/tutus-chain/pkg/core/state"
	"github.com/tutus-one/tutus-chain/pkg/core/transaction"
	"github.com/tutus-one/tutus-chain/pkg/crypto/keys"
	"github.com/tutus-one/tutus-chain/pkg/encoding/address"
	"github.com/tutus-one/tutus-chain/pkg/encoding/fixedn"
	"github.com/tutus-one/tutus-chain/pkg/rpcclient"
	"github.com/tutus-one/tutus-chain/pkg/rpcclient/actor"
	"github.com/tutus-one/tutus-chain/pkg/rpcclient/gas"
	"github.com/tutus-one/tutus-chain/pkg/rpcclient/notary"
	"github.com/tutus-one/tutus-chain/pkg/smartcontract/trigger"
	"github.com/tutus-one/tutus-chain/pkg/util"
	"github.com/tutus-one/tutus-chain/pkg/vm"
	"github.com/tutus-one/tutus-chain/pkg/vm/stackitem"
	"github.com/tutus-one/tutus-chain/pkg/vm/vmstate"
	"github.com/tutus-one/tutus-chain/pkg/wallet"
	"github.com/stretchr/testify/require"
)

func TestQueryTx(t *testing.T) {
	e := testcli.NewExecutorSuspended(t)

	w, err := wallet.NewWalletFromFile("../testdata/testwallet.json")
	require.NoError(t, err)

	transferArgs := []string{
		"neo-go", "wallet", "nep17", "transfer",
		"--rpc-endpoint", "http://" + e.RPC.Addresses()[0],
		"--wallet", testcli.ValidatorWallet,
		"--to", w.Accounts[0].Address,
		"--token", "NEO",
		"--from", testcli.ValidatorAddr,
		"--force",
	}

	e.In.WriteString("one\r")
	e.Run(t, append(transferArgs, "--amount", "1")...)
	line := e.GetNextLine(t)
	txHash, err := util.Uint256DecodeStringLE(line)
	require.NoError(t, err)

	tx, ok := e.Chain.GetMemPool().TryGetValue(txHash)
	require.True(t, ok)

	args := []string{"neo-go", "query", "tx", "--rpc-endpoint", "http://" + e.RPC.Addresses()[0]}
	e.Run(t, append(args, txHash.StringLE())...)
	e.CheckNextLine(t, `Hash:\s+`+txHash.StringLE())
	e.CheckNextLine(t, `OnChain:\s+false`)
	e.CheckNextLine(t, `ValidUntil:\s+`+strconv.FormatUint(uint64(tx.ValidUntilBlock), 10))
	e.CheckEOF(t)

	go e.Chain.Run()
	require.Eventually(t, func() bool { _, aerErr := e.Chain.GetAppExecResults(txHash, trigger.Application); return aerErr == nil }, time.Second*2, time.Millisecond*50)

	e.Run(t, append(args, txHash.StringLE())...)
	e.CheckNextLine(t, `Hash:\s+`+txHash.StringLE())
	e.CheckNextLine(t, `OnChain:\s+true`)

	_, height, err := e.Chain.GetTransaction(txHash)
	require.NoError(t, err)
	e.CheckNextLine(t, `BlockHash:\s+`+e.Chain.GetHeaderHash(height).StringLE())
	e.CheckNextLine(t, `Success:\s+true`)
	e.CheckEOF(t)

	t.Run("verbose", func(t *testing.T) {
		e.Run(t, append(args, "--verbose", txHash.StringLE())...)
		compareQueryTxVerbose(t, e, tx)

		t.Run("FAULT", func(t *testing.T) {
			e.In.WriteString("one\r")
			e.Run(t, "neo-go", "contract", "invokefunction",
				"--rpc-endpoint", "http://"+e.RPC.Addresses()[0],
				"--wallet", testcli.ValidatorWallet,
				"--address", testcli.ValidatorAddr,
				"--force",
				random.Uint160().StringLE(),
				"randomMethod")

			e.CheckNextLine(t, `Warning:`)
			e.CheckNextLine(t, "Sending transaction")
			line := strings.TrimPrefix(e.GetNextLine(t), "Sent invocation transaction ")
			txHash, err := util.Uint256DecodeStringLE(line)
			require.NoError(t, err)

			require.Eventually(t, func() bool { _, aerErr := e.Chain.GetAppExecResults(txHash, trigger.Application); return aerErr == nil }, time.Second*2, time.Millisecond*50)

			tx, _, err := e.Chain.GetTransaction(txHash)
			require.NoError(t, err)
			e.Run(t, append(args, "--verbose", txHash.StringLE())...)
			compareQueryTxVerbose(t, e, tx)
		})
	})

	t.Run("invalid", func(t *testing.T) {
		t.Run("missing tx argument", func(t *testing.T) {
			e.RunWithError(t, args...)
		})
		t.Run("excessive arguments", func(t *testing.T) {
			e.RunWithError(t, append(args, txHash.StringLE(), txHash.StringLE())...)
		})
		t.Run("invalid hash", func(t *testing.T) {
			e.RunWithError(t, append(args, "notahash")...)
		})
		t.Run("good hash, missing tx", func(t *testing.T) {
			e.RunWithError(t, append(args, random.Uint256().StringLE())...)
		})
	})
}

func compareQueryTxVerbose(t *testing.T, e *testcli.Executor, tx *transaction.Transaction, fallbacks ...int) {
	if len(fallbacks) > 0 {
		e.CheckNextLine(t, `Current height:\s+(\d|\.)+`)
	}
	e.CheckNextLine(t, `Hash:\s+`+tx.Hash().StringLE())
	e.CheckNextLine(t, `OnChain:\s+`+strconv.FormatBool(len(fallbacks) == 0))
	var (
		res []state.AppExecResult
		err error
	)
	if len(fallbacks) == 0 {
		_, height, err := e.Chain.GetTransaction(tx.Hash())
		require.NoError(t, err)
		e.CheckNextLine(t, `BlockHash:\s+`+e.Chain.GetHeaderHash(height).StringLE())
		res, _ = e.Chain.GetAppExecResults(tx.Hash(), trigger.Application)
		e.CheckNextLine(t, fmt.Sprintf(`Success:\s+%t`, res[0].VMState == vmstate.Halt))
	} else {
		e.CheckNextLine(t, `ValidUntil:\s+`+strconv.FormatUint(uint64(tx.ValidUntilBlock), 10))
		e.CheckNextLine(t, `Fallbacks:\s+`+strconv.Itoa(len(fallbacks)))
	}

	for _, s := range tx.Signers {
		e.CheckNextLine(t, fmt.Sprintf(`Signer:\s+%s\s*\(%s\)`, address.Uint160ToString(s.Account), s.Scopes.String()))
	}
	e.CheckNextLine(t, `SystemFee:\s+`+fixedn.Fixed8(tx.SystemFee).String()+" GAS$")
	e.CheckNextLine(t, `NetworkFee:\s+`+fixedn.Fixed8(tx.NetworkFee).String()+" GAS$")
	e.CheckNextLine(t, `Script:\s+`+regexp.QuoteMeta(base64.StdEncoding.EncodeToString(tx.Script)))
	if len(fallbacks) == 0 {
		c := vm.NewContext(tx.Script)
		n := 0
		for ; c.NextIP() < c.LenInstr(); _, _, err = c.Next() {
			require.NoError(t, err)
			n++
		}
		e.CheckScriptDump(t, n)

		if res[0].VMState != vmstate.Halt {
			e.CheckNextLine(t, `Exception:\s+`+regexp.QuoteMeta(res[0].FaultException))
		}
	}

	e.CheckEOF(t)
}

func TestQueryHeight(t *testing.T) {
	e := testcli.NewExecutor(t, true)

	args := []string{"neo-go", "query", "height", "--rpc-endpoint", "http://" + e.RPC.Addresses()[0]}
	e.Run(t, args...)
	e.CheckNextLine(t, `^Latest block: [0-9]+$`)
	e.CheckNextLine(t, `^Validated state: [0-9]+$`)
	e.CheckEOF(t)
	t.Run("excessive arguments", func(t *testing.T) {
		e.RunWithError(t, append(args, "something")...)
	})
}

func TestQueryNotaryPool(t *testing.T) {
	e := testcli.NewExecutorWithConfig(t, true, true, func(cfg *config.Config) {
		cfg.ProtocolConfiguration.Hardforks = map[string]uint32{
			config.HFFaun.String(): 0,
		}
	})
	endpoint := "http://" + e.RPC.Addresses()[0]

	w, err := wallet.NewWalletFromFile("../testdata/testwallet.json")
	require.NoError(t, err)
	acc := w.Accounts[0]
	require.NoError(t, acc.Decrypt(testcli.TestWalletPass, w.Scrypt))

	// Transfer some GAS to the account.
	transferArgs := []string{
		"neo-go", "wallet", "nep17", "transfer",
		"--rpc-endpoint", "http://" + e.RPC.Addresses()[0],
		"--wallet", testcli.ValidatorWallet,
		"--to", acc.Address,
		"--token", "GAS",
		"--amount", "10000",
		"--from", testcli.ValidatorAddr,
		"--force",
	}
	e.In.WriteString("one\r")
	e.Run(t, transferArgs...)
	line := e.GetNextLine(t)
	txHash, err := util.Uint256DecodeStringLE(line)
	require.NoError(t, err)
	require.Eventually(t, func() bool { _, aerErr := e.Chain.GetAppExecResults(txHash, trigger.Application); return aerErr == nil }, time.Second*2, time.Millisecond*50)

	// Make a notary deposit.
	c, err := rpcclient.New(context.Background(), endpoint, rpcclient.Options{})
	require.NoError(t, err)
	require.NoError(t, c.Init())
	signer := actor.SignerAccount{
		Signer:  transaction.Signer{Account: acc.ScriptHash(), Scopes: transaction.Global},
		Account: acc,
	}
	act, err := actor.New(c, []actor.SignerAccount{signer})
	require.NoError(t, err)
	g := gas.New(act)
	h, _, err := g.Transfer(acc.ScriptHash(), nativehashes.Notary, big.NewInt(20000000), []any{stackitem.Null{}, e.Chain.BlockHeight() + 1000})
	require.NoError(t, err)
	require.Eventually(t, func() bool { _, aerErr := e.Chain.GetAppExecResults(h, trigger.Application); return aerErr == nil }, time.Second*2, time.Millisecond*50)

	// Submit some notary request.
	pk, err := keys.NewPrivateKey()
	require.NoError(t, err)
	fakeAcc := notary.FakeSimpleAccount(pk.PublicKey())
	ntr, err := notary.NewActor(c, []actor.SignerAccount{
		signer,
		{
			Signer:  transaction.Signer{Account: fakeAcc.ScriptHash(), Scopes: transaction.Global},
			Account: fakeAcc,
		},
	}, acc)
	require.NoError(t, err)
	tx, err := ntr.MakeCall(nativehashes.GasToken, "transfer", acc.ScriptHash(), acc.ScriptHash(), big.NewInt(1), nil)
	mainH, _, vub, err := ntr.Notarize(tx, err)
	require.NoError(t, err)

	// Check the notary pool.
	args := []string{"neo-go", "query", "notarypool", "--rpc-endpoint", endpoint}
	t.Run("silent", func(t *testing.T) {
		e.Run(t, args...)
		e.CheckNextLine(t, `Current height:\s+(\d|\.)+`)
		e.CheckNextLine(t, `Hash:\s+`+mainH.StringLE())
		e.CheckNextLine(t, `OnChain:\s+false`)
		e.CheckNextLine(t, `ValidUntil:\s+`+strconv.FormatUint(uint64(vub), 10))
		e.CheckNextLine(t, `Fallbacks:\s+1`)
		e.CheckEOF(t)
	})

	t.Run("verbose", func(t *testing.T) {
		e.Run(t, append(args, "--verbose")...)
		compareQueryTxVerbose(t, e, tx, 1)
	})

	t.Run("invalid", func(t *testing.T) {
		t.Run("excessive arguments", func(t *testing.T) {
			e.RunWithError(t, append(args, "bla")...)
		})
	})
}