package attributes

import (
	"errors"
	"fmt"
	"strings"

	"github.com/nspcc-dev/neofs-sdk-go/netmap"
)

const (
	pairSeparator     = "/"
	keyValueSeparator = ":"
)

var (
	errEmptyChain      = errors.New("empty attribute chain")
	errNonUniqueBucket = errors.New("attributes must contain unique keys")
	errUnexpectedKey   = errors.New("attributes contain unexpected key")
)

// ParseV2Attributes parses strings like "K1:V1/K2:V2/K3:V3" into netmap
// attributes. Supports escaped symbols "\:", "\/" and "\\".
func ParseV2Attributes(attrs []string, excl []string) ([]netmap.NodeAttribute, error) {
	restricted := make(map[string]struct{}, len(excl))
	for i := range excl {
		restricted[excl[i]] = struct{}{}
	}

	cache := make(map[string]*netmap.NodeAttribute, len(attrs))
	result := make([]netmap.NodeAttribute, 0, len(attrs))

	for i := range attrs {
		line := strings.Trim(attrs[i], pairSeparator)
		line = replaceEscaping(line, false) // replaced escaped symbols with non-printable symbols
		chain := strings.Split(line, pairSeparator)
		if len(chain) == 0 {
			return nil, errEmptyChain
		}

		var parentKey string // backtrack parents in next pairs

		for j := range chain {
			pair := strings.Split(chain[j], keyValueSeparator)
			if len(pair) != 2 {
				return nil, fmt.Errorf("incorrect attribute pair %s", chain[j])
			}

			key := pair[0]
			value := pair[1]

			attribute, present := cache[key]
			if present && attribute.Value() != value {
				return nil, errNonUniqueBucket
			}

			if _, ok := restricted[key]; ok {
				return nil, errUnexpectedKey
			}

			if !present {
				result = append(result, netmap.NodeAttribute{})
				attribute = &result[len(result)-1]
				cache[key] = attribute

				// replace non-printable symbols with escaped symbols without escape character
				key = replaceEscaping(key, true)
				value = replaceEscaping(value, true)

				attribute.SetKey(key)
				attribute.SetValue(value)
			}

			if parentKey != "" {
				parentKeys := attribute.ParentKeys()
				if !hasString(parentKeys, parentKey) {
					attribute.SetParentKeys(append(parentKeys, parentKey)...)
				}
			}

			parentKey = key
		}
	}

	return result, nil
}

func replaceEscaping(target string, rollback bool) (s string) {
	const escChar = `\`

	var (
		oldPairSep = escChar + pairSeparator
		oldKVSep   = escChar + keyValueSeparator
		oldEsc     = escChar + escChar
		newPairSep = string(uint8(1))
		newKVSep   = string(uint8(2))
		newEsc     = string(uint8(3))
	)

	if rollback {
		oldPairSep, oldKVSep, oldEsc = newPairSep, newKVSep, newEsc
		newPairSep = pairSeparator
		newKVSep = keyValueSeparator
		newEsc = escChar
	}

	s = strings.ReplaceAll(target, oldEsc, newEsc)
	s = strings.ReplaceAll(s, oldPairSep, newPairSep)
	s = strings.ReplaceAll(s, oldKVSep, newKVSep)

	return
}

func hasString(lst []string, v string) bool {
	for i := range lst {
		if lst[i] == v {
			return true
		}
	}
	return false
}