package innerring

import (
	"strings"
	"testing"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client"
	"github.com/nspcc-dev/neo-go/pkg/util"
	"github.com/spf13/viper"
	"github.com/stretchr/testify/require"
)

func TestParseContractsSuccess(t *testing.T) {
	t.Parallel()
	file := strings.NewReader(`
contracts:
 frostfs: ee3dee6d05dc79c24a5b8f6985e10d68b7cacc62
 processing: 597f5894867113a41e192801709c02497f611de8
 audit: 219e37aed2180b87e7fe945dbf97d67125e8d73f
 balance: d2aa48d14b17b11bc4c68205027884a96706dd16
 container: ed4a7a66fe3f9bfe50f214b49be8f215a3c886b6
 frostfsid: 9f5866decbc751a099e74c7c7bc89f609201755a
 netmap: 83c600c81d47a1b1b7cf58eb49ae7ee7240dc742
 proxy: abc8794bb40a21f2db5f21ae62741eb46c8cad1c
 alphabet:
  amount: 2
  az: c1d211fceeb4b1dc76b8e4054d11fdf887e418ea
  buky: e2ba789320899658b100f331bdebb74474757920
`)

	v := viper.New()
	v.SetConfigType("yaml")
	err := v.ReadConfig(file)
	require.NoError(t, err, "read config file failed")

	t.Run("all enabled", func(t *testing.T) {
		t.Parallel()
		c, err := parseContracts(v, nil, false, false, false)
		require.NoError(t, err, "failed to parse contracts")

		frostfsExp, _ := util.Uint160DecodeStringLE("ee3dee6d05dc79c24a5b8f6985e10d68b7cacc62")
		require.Equal(t, frostfsExp, c.frostfs, "invalid frostfs")

		processingExp, _ := util.Uint160DecodeStringLE("597f5894867113a41e192801709c02497f611de8")
		require.Equal(t, processingExp, c.processing, "invalid processing")

		balanceExp, _ := util.Uint160DecodeStringLE("d2aa48d14b17b11bc4c68205027884a96706dd16")
		require.Equal(t, balanceExp, c.balance, "invalid balance")

		containerExp, _ := util.Uint160DecodeStringLE("ed4a7a66fe3f9bfe50f214b49be8f215a3c886b6")
		require.Equal(t, containerExp, c.container, "invalid container")

		frostfsIDExp, _ := util.Uint160DecodeStringLE("9f5866decbc751a099e74c7c7bc89f609201755a")
		require.Equal(t, frostfsIDExp, c.frostfsID, "invalid frostfsID")

		netmapIDExp, _ := util.Uint160DecodeStringLE("83c600c81d47a1b1b7cf58eb49ae7ee7240dc742")
		require.Equal(t, netmapIDExp, c.netmap, "invalid netmap")

		proxyExp, _ := util.Uint160DecodeStringLE("abc8794bb40a21f2db5f21ae62741eb46c8cad1c")
		require.Equal(t, proxyExp, c.proxy, "invalid proxy")

		require.Equal(t, 2, len(c.alphabet), "invalid alphabet contracts length")

		azExp, _ := util.Uint160DecodeStringLE("c1d211fceeb4b1dc76b8e4054d11fdf887e418ea")
		require.Equal(t, azExp, c.alphabet[az], "invalid az")

		bukyExp, _ := util.Uint160DecodeStringLE("e2ba789320899658b100f331bdebb74474757920")
		require.Equal(t, bukyExp, c.alphabet[buky], "invalid buky")
	})

	t.Run("all disabled", func(t *testing.T) {
		t.Parallel()
		c, err := parseContracts(v, nil, true, true, true)
		require.NoError(t, err, "failed to parse contracts")

		require.Equal(t, util.Uint160{}, c.frostfs, "invalid frostfs")

		require.Equal(t, util.Uint160{}, c.processing, "invalid processing")

		balanceExp, _ := util.Uint160DecodeStringLE("d2aa48d14b17b11bc4c68205027884a96706dd16")
		require.Equal(t, balanceExp, c.balance, "invalid balance")

		containerExp, _ := util.Uint160DecodeStringLE("ed4a7a66fe3f9bfe50f214b49be8f215a3c886b6")
		require.Equal(t, containerExp, c.container, "invalid container")

		frostfsIDExp, _ := util.Uint160DecodeStringLE("9f5866decbc751a099e74c7c7bc89f609201755a")
		require.Equal(t, frostfsIDExp, c.frostfsID, "invalid frostfsID")

		netmapIDExp, _ := util.Uint160DecodeStringLE("83c600c81d47a1b1b7cf58eb49ae7ee7240dc742")
		require.Equal(t, netmapIDExp, c.netmap, "invalid netmap")

		require.Equal(t, util.Uint160{}, c.proxy, "invalid proxy")

		require.Equal(t, 2, len(c.alphabet), "invalid alphabet contracts length")

		azExp, _ := util.Uint160DecodeStringLE("c1d211fceeb4b1dc76b8e4054d11fdf887e418ea")
		require.Equal(t, azExp, c.alphabet[az], "invalid az")

		bukyExp, _ := util.Uint160DecodeStringLE("e2ba789320899658b100f331bdebb74474757920")
		require.Equal(t, bukyExp, c.alphabet[buky], "invalid buky")
	})

	t.Run("main notary & side notary disabled", func(t *testing.T) {
		t.Parallel()
		c, err := parseContracts(v, nil, false, true, true)
		require.NoError(t, err, "failed to parse contracts")

		frostfsExp, _ := util.Uint160DecodeStringLE("ee3dee6d05dc79c24a5b8f6985e10d68b7cacc62")
		require.Equal(t, frostfsExp, c.frostfs, "invalid frostfs")

		require.Equal(t, util.Uint160{}, c.processing, "invalid processing")

		balanceExp, _ := util.Uint160DecodeStringLE("d2aa48d14b17b11bc4c68205027884a96706dd16")
		require.Equal(t, balanceExp, c.balance, "invalid balance")

		containerExp, _ := util.Uint160DecodeStringLE("ed4a7a66fe3f9bfe50f214b49be8f215a3c886b6")
		require.Equal(t, containerExp, c.container, "invalid container")

		frostfsIDExp, _ := util.Uint160DecodeStringLE("9f5866decbc751a099e74c7c7bc89f609201755a")
		require.Equal(t, frostfsIDExp, c.frostfsID, "invalid frostfsID")

		netmapIDExp, _ := util.Uint160DecodeStringLE("83c600c81d47a1b1b7cf58eb49ae7ee7240dc742")
		require.Equal(t, netmapIDExp, c.netmap, "invalid netmap")

		require.Equal(t, util.Uint160{}, c.proxy, "invalid proxy")

		require.Equal(t, 2, len(c.alphabet), "invalid alphabet contracts length")

		azExp, _ := util.Uint160DecodeStringLE("c1d211fceeb4b1dc76b8e4054d11fdf887e418ea")
		require.Equal(t, azExp, c.alphabet[az], "invalid az")

		bukyExp, _ := util.Uint160DecodeStringLE("e2ba789320899658b100f331bdebb74474757920")
		require.Equal(t, bukyExp, c.alphabet[buky], "invalid buky")
	})
}

func TestParseContractsInvalid(t *testing.T) {
	t.Parallel()
	t.Run("invalid audit contract", func(t *testing.T) {
		t.Parallel()
		file := strings.NewReader(`
contracts:
 frostfs: invalid_data
 processing: 597f5894867113a41e192801709c02497f611de8
 audit: 219e37aed2180b87e7fe945dbf97d67125e8d73f
 balance: d2aa48d14b17b11bc4c68205027884a96706dd16
 container: ed4a7a66fe3f9bfe50f214b49be8f215a3c886b6
 frostfsid: 9f5866decbc751a099e74c7c7bc89f609201755a
 netmap: 83c600c81d47a1b1b7cf58eb49ae7ee7240dc742
 proxy: abc8794bb40a21f2db5f21ae62741eb46c8cad1c
 alphabet:
  amount: 2
  az: c1d211fceeb4b1dc76b8e4054d11fdf887e418ea
  buky: e2ba789320899658b100f331bdebb74474757920
`)

		v := viper.New()
		v.SetConfigType("yaml")
		err := v.ReadConfig(file)
		require.NoError(t, err, "read config file failed")

		_, err = parseContracts(v, nil, false, false, false)
		require.Error(t, err, "unexpected success")
	})

	t.Run("invalid alphabet count", func(t *testing.T) {
		t.Parallel()
		file := strings.NewReader(`
contracts:
 frostfs: ee3dee6d05dc79c24a5b8f6985e10d68b7cacc62
 processing: 597f5894867113a41e192801709c02497f611de8
 audit: 219e37aed2180b87e7fe945dbf97d67125e8d73f
 balance: d2aa48d14b17b11bc4c68205027884a96706dd16
 container: ed4a7a66fe3f9bfe50f214b49be8f215a3c886b6
 frostfsid: 9f5866decbc751a099e74c7c7bc89f609201755a
 netmap: 83c600c81d47a1b1b7cf58eb49ae7ee7240dc742
 proxy: abc8794bb40a21f2db5f21ae62741eb46c8cad1c
 alphabet:
  amount: 3
  az: c1d211fceeb4b1dc76b8e4054d11fdf887e418ea
  buky: e2ba789320899658b100f331bdebb74474757920
`)

		v := viper.New()
		v.SetConfigType("yaml")
		err := v.ReadConfig(file)
		require.NoError(t, err, "read config file failed")

		azExp, _ := util.Uint160DecodeStringLE("c1d211fceeb4b1dc76b8e4054d11fdf887e418ea")
		bukyExp, _ := util.Uint160DecodeStringLE("e2ba789320899658b100f331bdebb74474757920")

		morph := &testParserMorph{
			values: map[string]util.Uint160{
				"az":   azExp,
				"buky": bukyExp,
			},
		}

		_, err = parseContracts(v, morph, false, false, false)
		require.ErrorContains(t, err, "could not read all contracts: required 3, read 2", "unexpected success")
	})
}

type testParserMorph struct {
	values map[string]util.Uint160
}

func (m *testParserMorph) NNSContractAddress(name string) (sh util.Uint160, err error) {
	if value, found := m.values[name]; found {
		return value, nil
	}
	return util.Uint160{}, client.ErrNNSRecordNotFound
}