package cfgutil

import (
	"net"
)

// NormalizeAddress returns the normalized form of the address, adding a default port if necessary. An error is returned
// if the address, even without a port, is not valid.
func NormalizeAddress(addr string, defaultPort string) (hostport string, e error) {
	// If the first SplitHostPort errors because of a missing port and not for an invalid host, add the port. If the
	// second SplitHostPort fails, then a port is not missing and the original error should be returned.
	host, port, origErr := net.SplitHostPort(addr)
	if origErr == nil {
		return net.JoinHostPort(host, port), nil
	}
	addr = net.JoinHostPort(addr, defaultPort)
	_, _, e = net.SplitHostPort(addr)
	if e != nil {
		E.Ln(e)
		return "", origErr
	}
	return addr, nil
}

// NormalizeAddresses returns a new slice with all the passed peer addresses normalized with the given default port, and
// all duplicates removed.
func NormalizeAddresses(addrs []string, defaultPort string) ([]string, error) {
	var (
		normalized = make([]string, 0, len(addrs))
		seenSet    = make(map[string]struct{})
	)
	for _, addr := range addrs {
		normalizedAddr, e := NormalizeAddress(addr, defaultPort)
		if e != nil {
			E.Ln(e)
			return nil, e
		}
		_, seen := seenSet[normalizedAddr]
		if !seen {
			normalized = append(normalized, normalizedAddr)
			seenSet[normalizedAddr] = struct{}{}
		}
	}
	return normalized, nil
}