package tests

import (
	"crypto/sha256"
	"strings"
	"testing"

	"github.com/mr-tron/base58"
	"github.com/nspcc-dev/neo-go/pkg/core"
	"github.com/nspcc-dev/neo-go/pkg/util"
	"github.com/nspcc-dev/neo-go/pkg/vm/stackitem"
	"github.com/nspcc-dev/neo-go/pkg/wallet"
	"github.com/nspcc-dev/neofs-contract/container"
	"github.com/nspcc-dev/neofs-contract/nns"
	"github.com/stretchr/testify/require"
)

const containerPath = "../container"

const (
	containerFee      = 0_0100_0000
	containerAliasFee = 0_0050_0000
)

func deployContainerContract(t *testing.T, bc *core.Blockchain, addrNetmap, addrBalance, addrNNS util.Uint160) util.Uint160 {
	args := make([]interface{}, 6)
	args[0] = int64(0)
	args[1] = addrNetmap
	args[2] = addrBalance
	args[3] = util.Uint160{} // not needed for now
	args[4] = addrNNS
	args[5] = "neofs"
	return DeployContract(t, bc, containerPath, args)
}

func prepareContainerContract(t *testing.T, bc *core.Blockchain) (util.Uint160, util.Uint160) {
	addrNNS := DeployContract(t, bc, nnsPath, nil)

	ctrNetmap, err := ContractInfo(CommitteeAcc.Contract.ScriptHash(), netmapPath)
	require.NoError(t, err)

	ctrBalance, err := ContractInfo(CommitteeAcc.Contract.ScriptHash(), balancePath)
	require.NoError(t, err)

	ctrContainer, err := ContractInfo(CommitteeAcc.Contract.ScriptHash(), containerPath)
	require.NoError(t, err)

	deployNetmapContract(t, bc, ctrBalance.Hash, ctrContainer.Hash,
		container.RegistrationFeeKey, int64(containerFee),
		container.AliasFeeKey, int64(containerAliasFee))
	balHash := deployBalanceContract(t, bc, ctrNetmap.Hash, ctrContainer.Hash)
	return deployContainerContract(t, bc, ctrNetmap.Hash, ctrBalance.Hash, addrNNS), balHash
}

func setContainerOwner(c []byte, acc *wallet.Account) {
	owner, _ := base58.Decode(acc.Address)
	copy(c[6:], owner)
}

type testContainer struct {
	id                     [32]byte
	value, sig, pub, token []byte
}

func dummyContainer(owner *wallet.Account) testContainer {
	value := randomBytes(100)
	value[1] = 0 // zero offset
	setContainerOwner(value, owner)

	return testContainer{
		id:    sha256.Sum256(value),
		value: value,
		sig:   randomBytes(64),
		pub:   randomBytes(33),
		token: randomBytes(42),
	}
}

func TestContainerPut(t *testing.T) {
	bc := NewChain(t)
	h, balanceHash := prepareContainerContract(t, bc)

	acc := NewAccount(t, bc)
	c := dummyContainer(acc)

	putArgs := []interface{}{c.value, c.sig, c.pub, c.token}
	tx := PrepareInvoke(t, bc, CommitteeAcc, h, "put", putArgs...)
	AddBlock(t, bc, tx)
	CheckFault(t, bc, tx.Hash(), "insufficient balance to create container")

	balanceMint(t, bc, acc, balanceHash, containerFee*1, []byte{})

	tx = PrepareInvoke(t, bc, acc, h, "put", putArgs...)
	AddBlock(t, bc, tx)
	CheckFault(t, bc, tx.Hash(), "alphabet witness check failed")

	tx = PrepareInvoke(t, bc, CommitteeAcc, h, "put", putArgs...)
	AddBlockCheckHalt(t, bc, tx)

	t.Run("with nice names", func(t *testing.T) {
		nnsHash := contracts[nnsPath].Hash

		balanceMint(t, bc, acc, balanceHash, containerFee*1, []byte{})

		putArgs := []interface{}{c.value, c.sig, c.pub, c.token, "mycnt", ""}
		t.Run("no fee for alias", func(t *testing.T) {
			tx = PrepareInvoke(t, bc, acc, h, "putNamed", putArgs...)
			AddBlock(t, bc, tx)
			CheckFault(t, bc, tx.Hash(), "insufficient balance to create container")
		})

		balanceMint(t, bc, acc, balanceHash, containerAliasFee*1, []byte{})

		tx = PrepareInvoke(t, bc, CommitteeAcc, h, "putNamed", putArgs...)
		AddBlockCheckHalt(t, bc, tx)

		tx = PrepareInvoke(t, bc, acc, nnsHash, "resolve", "mycnt.neofs", int64(nns.TXT))
		CheckTestInvoke(t, bc, tx, stackitem.NewArray([]stackitem.Item{
			stackitem.NewByteArray([]byte(base58.Encode(c.id[:]))),
		}))

		t.Run("name is already taken", func(t *testing.T) {
			tx = PrepareInvoke(t, bc, CommitteeAcc, h, "putNamed", putArgs...)
			AddBlock(t, bc, tx)
			CheckFault(t, bc, tx.Hash(), "name is already taken")
		})

		tx = PrepareInvoke(t, bc, CommitteeAcc, h, "delete", c.id[:], c.sig, c.token)
		AddBlockCheckHalt(t, bc, tx)

		tx = PrepareInvoke(t, bc, CommitteeAcc, nnsHash, "resolve", "mycnt.neofs", int64(nns.TXT))
		CheckTestInvoke(t, bc, tx, stackitem.Null{})

		t.Run("register in advance", func(t *testing.T) {
			c.value[len(c.value)-1] = 10
			c.id = sha256.Sum256(c.value)

			t.Run("bad domain owner", func(t *testing.T) {
				tx = PrepareInvoke(t, bc, acc, nnsHash, "register",
					"baddomain.neofs", acc.Contract.ScriptHash(),
					"whateveriwant@world.com", int64(0), int64(0), int64(0), int64(0))
				AddBlockCheckHalt(t, bc, tx)

				tx = PrepareInvoke(t, bc, acc, h, "putNamed",
					c.value, c.sig, c.pub, c.token, "baddomain", "neofs")
				AddBlock(t, bc, tx)
				CheckFault(t, bc, tx.Hash(), "committee or container contract must own registered domain")
			})

			tx = PrepareInvoke(t, bc, CommitteeAcc, nnsHash, "register",
				"second.neofs", CommitteeAcc.Contract.ScriptHash(),
				"whateveriwant@world.com", int64(0), int64(0), int64(0), int64(0))
			AddBlockCheckHalt(t, bc, tx)

			balanceMint(t, bc, acc, balanceHash, (containerFee+containerAliasFee)*1, []byte{})

			putArgs := []interface{}{c.value, c.sig, c.pub, c.token, "second", "neofs"}
			tx = PrepareInvoke(t, bc, []*wallet.Account{CommitteeAcc, acc}, h, "putNamed", putArgs...)
			AddBlockCheckHalt(t, bc, tx)

			tx = PrepareInvoke(t, bc, CommitteeAcc, nnsHash, "resolve", "second.neofs", int64(nns.TXT))
			CheckTestInvoke(t, bc, tx, stackitem.NewArray([]stackitem.Item{
				stackitem.NewByteArray([]byte(base58.Encode(c.id[:]))),
			}))
		})
	})
}

func TestContainerDelete(t *testing.T) {
	bc := NewChain(t)
	h, balanceHash := prepareContainerContract(t, bc)

	acc := NewAccount(t, bc)
	c := dummyContainer(acc)

	balanceMint(t, bc, acc, balanceHash, containerFee*1, []byte{})
	tx := PrepareInvoke(t, bc, CommitteeAcc, h, "put",
		c.value, c.sig, c.pub, c.token)
	AddBlockCheckHalt(t, bc, tx)

	tx = PrepareInvoke(t, bc, acc, h, "delete", c.id[:], c.sig, c.token)
	AddBlock(t, bc, tx)
	CheckFault(t, bc, tx.Hash(), "delete: alphabet witness check failed")

	tx = PrepareInvoke(t, bc, CommitteeAcc, h, "delete", c.id[:], c.sig, c.token)
	AddBlockCheckHalt(t, bc, tx)

	tx = PrepareInvoke(t, bc, acc, h, "get", c.id[:])
	_, err := TestInvoke(bc, tx)
	require.Error(t, err)
	require.True(t, strings.Contains(err.Error(), container.NotFoundError))
}

func TestContainerOwner(t *testing.T) {
	bc := NewChain(t)
	h, balanceHash := prepareContainerContract(t, bc)

	acc := NewAccount(t, bc)
	balanceMint(t, bc, acc, balanceHash, containerFee*1, []byte{})

	c := dummyContainer(acc)
	tx := PrepareInvoke(t, bc, CommitteeAcc, h, "put", c.value, c.sig, c.pub, c.token)
	AddBlockCheckHalt(t, bc, tx)

	t.Run("missing container", func(t *testing.T) {
		id := c.id
		id[0] ^= 0xFF
		tx = PrepareInvoke(t, bc, CommitteeAcc, h, "owner", id[:])
		_, err := TestInvoke(bc, tx)
		require.Error(t, err)
		require.True(t, strings.Contains(err.Error(), container.NotFoundError))
	})

	tx = PrepareInvoke(t, bc, CommitteeAcc, h, "owner", c.id[:])
	owner, _ := base58.Decode(acc.Address)
	CheckTestInvoke(t, bc, tx, stackitem.NewBuffer(owner))
}

func TestContainerGet(t *testing.T) {
	bc := NewChain(t)
	h, balanceHash := prepareContainerContract(t, bc)

	acc := NewAccount(t, bc)
	balanceMint(t, bc, acc, balanceHash, containerFee*1, []byte{})

	c := dummyContainer(acc)
	tx := PrepareInvoke(t, bc, CommitteeAcc, h, "put", c.value, c.sig, c.pub, c.token)
	AddBlockCheckHalt(t, bc, tx)

	t.Run("missing container", func(t *testing.T) {
		id := c.id
		id[0] ^= 0xFF
		tx = PrepareInvoke(t, bc, CommitteeAcc, h, "get", id[:])
		_, err := TestInvoke(bc, tx)
		require.Error(t, err)
		require.True(t, strings.Contains(err.Error(), container.NotFoundError))
	})

	tx = PrepareInvoke(t, bc, CommitteeAcc, h, "get", c.id[:])
	CheckTestInvoke(t, bc, tx, stackitem.NewStruct([]stackitem.Item{
		stackitem.NewByteArray(c.value),
		stackitem.NewByteArray(c.sig),
		stackitem.NewByteArray(c.pub),
		stackitem.NewByteArray(c.token),
	}))
}

type eacl struct {
	value []byte
	sig   []byte
	pub   []byte
	token []byte
}

func dummyEACL(containerID [32]byte) eacl {
	e := make([]byte, 50)
	copy(e[6:], containerID[:])
	return eacl{
		value: e,
		sig:   randomBytes(64),
		pub:   randomBytes(33),
		token: randomBytes(42),
	}
}

func TestContainerSetEACL(t *testing.T) {
	bc := NewChain(t)
	h, balanceHash := prepareContainerContract(t, bc)

	acc := NewAccount(t, bc)
	balanceMint(t, bc, acc, balanceHash, containerFee*1, []byte{})

	c := dummyContainer(acc)
	tx := PrepareInvoke(t, bc, CommitteeAcc, h, "put", c.value, c.sig, c.pub, c.token)
	AddBlockCheckHalt(t, bc, tx)

	t.Run("missing container", func(t *testing.T) {
		id := c.id
		id[0] ^= 0xFF
		e := dummyEACL(id)
		tx = PrepareInvoke(t, bc, CommitteeAcc, h, "setEACL", e.value, e.sig, e.pub, e.token)
		_, err := TestInvoke(bc, tx)
		require.Error(t, err)
		require.True(t, strings.Contains(err.Error(), container.NotFoundError))
	})

	e := dummyEACL(c.id)
	tx = PrepareInvoke(t, bc, acc, h, "setEACL", e.value, e.sig, e.pub, e.token)
	AddBlock(t, bc, tx)
	CheckFault(t, bc, tx.Hash(), "setEACL: alphabet witness check failed")

	tx = PrepareInvoke(t, bc, CommitteeAcc, h, "setEACL", e.value, e.sig, e.pub, e.token)
	AddBlockCheckHalt(t, bc, tx)

	tx = PrepareInvoke(t, bc, CommitteeAcc, h, "eACL", c.id[:])
	CheckTestInvoke(t, bc, tx, stackitem.NewStruct([]stackitem.Item{
		stackitem.NewByteArray(e.value),
		stackitem.NewByteArray(e.sig),
		stackitem.NewByteArray(e.pub),
		stackitem.NewByteArray(e.token),
	}))
}