//go:build system_test

package systemtests

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

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

	systest "cosmossdk.io/systemtests"
)

func TestChainExportImport(t *testing.T) {
	// Scenario:
	//   given: a state dump from a running chain
	//   when: new chain is initialized with exported state
	//   then: the chain should start and produce blocks
	systest.Sut.ResetChain(t)
	cli := systest.NewCLIWrapper(t, systest.Sut, systest.Verbose)
	systest.Sut.StartChain(t)

	grantee := cli.GetKeyAddr("node1")
	rsp3 := cli.RunAndWait("tx", "authz", "grant", grantee, "send", "--spend-limit=1000stake", "--from=node0", "--fees=1stake")
	systest.RequireTxSuccess(t, rsp3)
	systest.Sut.StopChain()

	outFile := filepath.Join(t.TempDir(), "exported_genesis.json")
	cli.RunCommandWithArgs("genesis", "export", "--home="+systest.Sut.NodeDir(0), "--output-document="+outFile)
	exportedContent, err := os.ReadFile(outFile)
	require.NoError(t, err)
	exportedState := gjson.Get(string(exportedContent), "app_state").Raw

	systest.Sut.ModifyGenesisJSON(t, func(genesis []byte) []byte {
		state, err := sjson.SetRawBytes(genesis, "app_state", []byte(exportedState))
		require.NoError(t, err)
		return state
	})
	systest.Sut.StartChain(t)
	systest.Sut.AwaitNBlocks(t, 2)
	systest.Sut.StopChain()
}

func TestExportCmd_WithHeight(t *testing.T) {
	systest.Sut.ResetChain(t)
	cli := systest.NewCLIWrapper(t, systest.Sut, systest.Verbose)

	systest.Sut.StartChain(t)

	// Wait 10s for producing blocks
	systest.Sut.AwaitNBlocks(t, 10)

	systest.Sut.StopChain()

	testCases := []struct {
		name          string
		args          []string
		expZeroHeight bool
	}{
		{"should export correct height", []string{"genesis", "export", "--home", systest.Sut.NodeDir(0), disabledLog}, false},
		{"should export correct height with --height", []string{"genesis", "export", "--height=5", "--home", systest.Sut.NodeDir(0), disabledLog}, false},
		{"should export height 0 with --for-zero-height", []string{"genesis", "export", "--for-zero-height=true", "--home", systest.Sut.NodeDir(0), disabledLog}, true},
	}

	for _, tc := range testCases {
		res := cli.
			WithRunErrorsIgnored().
			WithRunSingleOutput(). // pebbledb prints logs to stderr, we cannot override the logger in store/v2 and cosmos-db. This isn't problematic in a real-world scenario, but it makes it hard to test the output
			RunCommandWithArgs(tc.args...)
		height := gjson.Get(res, "initial_height").Int()
		if tc.expZeroHeight {
			require.Equal(t, height, int64(0))
		} else {
			require.Greater(t, height, int64(0))
		}

		// Check consensus params of exported state
		maxGas := gjson.Get(res, "consensus.params.block.max_gas").Int()
		require.Equal(t, maxGas, int64(systest.MaxGas))
	}
}

func TestExportCmd_WithFileFlag(t *testing.T) {
	systest.Sut.ResetChain(t)
	cli := systest.NewCLIWrapper(t, systest.Sut, systest.Verbose)
	exportFile := "foobar.json"

	systest.Sut.StartChain(t)

	// Wait 10s for producing blocks
	systest.Sut.AwaitNBlocks(t, 10)

	systest.Sut.StopChain()

	testCases := []struct {
		name   string
		args   []string
		expErr bool
		errMsg string
	}{
		{"invalid home dir", []string{"genesis", "export", "--home=foo"}, true, "no such file or directory"},
		{"should export state to the specified file", []string{"genesis", "export", fmt.Sprintf("--output-document=%s", exportFile), "--home", systest.Sut.NodeDir(0)}, false, ""},
	}

	for _, tc := range testCases {
		if tc.expErr {
			assertOutput := func(_ assert.TestingT, gotErr error, gotOutputs ...interface{}) bool {
				require.Contains(t, gotOutputs[0], tc.errMsg)
				return false
			}
			cli.WithRunErrorMatcher(assertOutput).RunCommandWithArgs(tc.args...)
		} else {
			cli.RunCommandWithArgs(tc.args...)
			require.FileExists(t, exportFile)
			err := os.Remove(exportFile)
			require.NoError(t, err)
		}
	}
}