package governance

import (
	"encoding/binary"
	"encoding/hex"
	"sort"
	"strings"

	"git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client"
	frostfscontract "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client/frostfs"
	nmClient "git.frostfs.info/TrueCloudLab/frostfs-node/pkg/morph/client/netmap"
	"github.com/nspcc-dev/neo-go/pkg/crypto/keys"
	"github.com/nspcc-dev/neo-go/pkg/util"
	"go.uber.org/zap"
)

const (
	alphabetUpdateIDPrefix = "AlphabetUpdate"
)

func (gp *Processor) processAlphabetSync(txHash util.Uint256) {
	if !gp.alphabetState.IsAlphabet() {
		gp.log.Info("non alphabet mode, ignore alphabet sync")
		return
	}

	mainnetAlphabet, err := gp.mainnetClient.NeoFSAlphabetList()
	if err != nil {
		gp.log.Error("can't fetch alphabet list from main net",
			zap.String("error", err.Error()))
		return
	}

	sidechainAlphabet, err := gp.morphClient.Committee()
	if err != nil {
		gp.log.Error("can't fetch alphabet list from side chain",
			zap.String("error", err.Error()))
		return
	}

	newAlphabet, err := newAlphabetList(sidechainAlphabet, mainnetAlphabet)
	if err != nil {
		gp.log.Error("can't merge alphabet lists from main net and side chain",
			zap.String("error", err.Error()))
		return
	}

	if newAlphabet == nil {
		gp.log.Info("no governance update, alphabet list has not been changed")
		return
	}

	gp.log.Info("alphabet list has been changed, starting update",
		zap.String("side_chain_alphabet", prettyKeys(sidechainAlphabet)),
		zap.String("new_alphabet", prettyKeys(newAlphabet)),
	)

	votePrm := VoteValidatorPrm{
		Validators: newAlphabet,
		Hash:       &txHash,
	}

	// 1. Vote to sidechain committee via alphabet contracts.
	err = gp.voter.VoteForSidechainValidator(votePrm)
	if err != nil {
		gp.log.Error("can't vote for side chain committee",
			zap.String("error", err.Error()))
	}

	// 2. Update NeoFSAlphabet role in the sidechain.
	gp.updateNeoFSAlphabetRoleInSidechain(sidechainAlphabet, newAlphabet, txHash)

	// 3. Update notary role in the sidechain.
	gp.updateNotaryRoleInSidechain(newAlphabet, txHash)

	// 4. Update FrostFS contract in the mainnet.
	gp.updateFrostFSContractInMainnet(newAlphabet)

	gp.log.Info("finished alphabet list update")
}

func prettyKeys(keys keys.PublicKeys) string {
	const delimiter = ","

	sb := strings.Builder{}
	for _, key := range keys {
		sb.WriteString(hex.EncodeToString(key.Bytes()))
		sb.WriteString(delimiter)
	}

	return strings.TrimRight(sb.String(), delimiter)
}

func (gp *Processor) updateNeoFSAlphabetRoleInSidechain(sidechainAlphabet, newAlphabet keys.PublicKeys, txHash util.Uint256) {
	innerRing, err := gp.irFetcher.InnerRingKeys()
	if err != nil {
		gp.log.Error("can't fetch inner ring list from side chain",
			zap.String("error", err.Error()))
		return
	}

	newInnerRing, err := updateInnerRing(innerRing, sidechainAlphabet, newAlphabet)
	if err != nil {
		gp.log.Error("can't create new inner ring list with new alphabet keys",
			zap.String("error", err.Error()))
		return
	}

	sort.Sort(newInnerRing)

	gp.log.Info("update of the inner ring list",
		zap.String("before", prettyKeys(innerRing)),
		zap.String("after", prettyKeys(newInnerRing)),
	)

	if gp.notaryDisabled {
		updPrm := nmClient.UpdateIRPrm{}

		updPrm.SetKeys(newInnerRing)
		updPrm.SetHash(txHash)

		err = gp.netmapClient.UpdateInnerRing(updPrm)
	} else {
		updPrm := client.UpdateAlphabetListPrm{}

		updPrm.SetList(newInnerRing)
		updPrm.SetHash(txHash)

		err = gp.morphClient.UpdateNeoFSAlphabetList(updPrm)
	}

	if err != nil {
		gp.log.Error("can't update inner ring list with new alphabet keys",
			zap.String("error", err.Error()))
	}
}

func (gp *Processor) updateNotaryRoleInSidechain(newAlphabet keys.PublicKeys, txHash util.Uint256) {
	if gp.notaryDisabled {
		return
	}

	updPrm := client.UpdateNotaryListPrm{}

	updPrm.SetList(newAlphabet)
	updPrm.SetHash(txHash)

	err := gp.morphClient.UpdateNotaryList(updPrm)
	if err != nil {
		gp.log.Error("can't update list of notary nodes in side chain",
			zap.String("error", err.Error()))
	}
}

func (gp *Processor) updateFrostFSContractInMainnet(newAlphabet keys.PublicKeys) {
	epoch := gp.epochState.EpochCounter()

	buf := make([]byte, 8)
	binary.LittleEndian.PutUint64(buf, epoch)

	id := append([]byte(alphabetUpdateIDPrefix), buf...)

	prm := frostfscontract.AlphabetUpdatePrm{}

	prm.SetID(id)
	prm.SetPubs(newAlphabet)

	err := gp.frostfsClient.AlphabetUpdate(prm)
	if err != nil {
		gp.log.Error("can't update list of alphabet nodes in frostfs contract",
			zap.String("error", err.Error()))
	}
}