package control_test

import (
	"bytes"
	"path/filepath"
	"strconv"
	"testing"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/blobovniczatree"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/local_object_storage/blobstor/fstree"
	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/services/control"
	"github.com/google/uuid"
)

func TestNetmap_StableMarshal(t *testing.T) {
	testStableMarshal(t, generateNetmap(), new(control.Netmap), func(m1, m2 protoMessage) bool {
		return equalNetmaps(m1.(*control.Netmap), m2.(*control.Netmap))
	})
}

func generateNetmap() *control.Netmap {
	nm := new(control.Netmap)
	nm.SetEpoch(13)

	const nodeCount = 2

	nodes := make([]*control.NodeInfo, 0, nodeCount)

	for i := 0; i < nodeCount; i++ {
		n := new(control.NodeInfo)
		n.SetPublicKey(testData(33))
		n.SetAddresses([]string{testString(), testString()})
		n.SetState(control.NetmapStatus_ONLINE)

		const attrCount = 2

		attrs := make([]*control.NodeInfo_Attribute, 0, attrCount)

		for j := 0; j < attrCount; j++ {
			a := new(control.NodeInfo_Attribute)
			a.SetKey(testString())
			a.SetValue(testString())

			const parentsCount = 2

			parents := make([]string, 0, parentsCount)

			for k := 0; k < parentsCount; k++ {
				parents = append(parents, testString())
			}

			a.SetParents(parents)

			attrs = append(attrs, a)
		}

		n.SetAttributes(attrs)

		nodes = append(nodes, n)
	}

	nm.SetNodes(nodes)

	return nm
}

func equalNetmaps(nm1, nm2 *control.Netmap) bool {
	if nm1.GetEpoch() != nm2.GetEpoch() {
		return false
	}

	n1, n2 := nm1.GetNodes(), nm2.GetNodes()

	if len(n1) != len(n2) {
		return false
	}

	for i := range n1 {
		if !equalNodeInfos(n1[i], n2[i]) {
			return false
		}
	}

	return true
}

func equalNodeInfos(n1, n2 *control.NodeInfo) bool {
	if !bytes.Equal(n1.GetPublicKey(), n2.GetPublicKey()) ||
		n1.GetState() != n2.GetState() {
		return false
	}

	na1, na2 := n1.GetAddresses(), n2.GetAddresses()

	if len(na1) != len(na2) {
		return false
	}

	for i := range na1 {
		if na1[i] != na2[i] {
			return false
		}
	}

	a1, a2 := n1.GetAttributes(), n2.GetAttributes()

	if len(a1) != len(a2) {
		return false
	}

	for i := range a1 {
		if a1[i].GetKey() != a2[i].GetKey() || a1[i].GetValue() != a2[i].GetValue() {
			return false
		}

		p1, p2 := a1[i].GetParents(), a2[i].GetParents()

		if len(p1) != len(p2) {
			return false
		}

		for j := range p1 {
			if p1[j] != p2[j] {
				return false
			}
		}
	}

	return true
}

func generateShardInfo(id int) *control.ShardInfo {
	si := new(control.ShardInfo)

	path := "/nice/dir/awesome/files/" + strconv.Itoa(id)

	uid, _ := uuid.NewRandom()
	bin, _ := uid.MarshalBinary()

	si.SetID(bin)
	si.SetMode(control.ShardMode_READ_WRITE)
	si.SetMetabasePath(filepath.Join(path, "meta"))
	si.Blobstor = []*control.BlobstorInfo{
		{Type: fstree.Type, Path: filepath.Join(path, "fstree")},
		{Type: blobovniczatree.Type, Path: filepath.Join(path, "blobtree")},
	}
	si.SetWriteCachePath(filepath.Join(path, "writecache"))
	si.SetPiloramaPath(filepath.Join(path, "pilorama"))

	return si
}