package multiaddr

import (
	"fmt"
)

// Split returns the sub-address portions of a multiaddr.
func Split(m Multiaddr) []Component {
	return m
}

// Join returns a combination of addresses.
// Note: This copies all the components from the input Multiaddrs. Depending on
// your use case, you may prefer to use `append(leftMA, rightMA...)` instead.
func Join(msInterfaces ...Multiaddrer) Multiaddr {
	ms := make([]Multiaddr, len(msInterfaces))
	for i, m := range msInterfaces {
		if m == nil {
			continue
		}
		ms[i] = m.Multiaddr()
	}
	size := 0
	for _, m := range ms {
		size += len(m)
	}
	if size == 0 {
		return nil
	}

	out := make([]Component, 0, size)
	for _, m := range ms {
		out = append(out, m...)
	}
	return out
}

// Cast re-casts a byte slice as a multiaddr.
func Cast(b []byte) (Multiaddr, error) {
	m, err := NewMultiaddrBytes(b)
	if err != nil {
		return nil, fmt.Errorf("multiaddr failed to parse: %s", err)
	}
	return m, nil
}

// StringCast like Cast, but parses a string.
func StringCast(s string) (Multiaddr, error) {
	m, err := NewMultiaddr(s)
	if err != nil {
		return nil, fmt.Errorf("multiaddr failed to parse: %s", err)
	}
	return m, nil
}

// SplitFirst returns the first component and the rest of the multiaddr.
func SplitFirst(m Multiaddr) (*Component, Multiaddr) {
	if len(m) == 0 {
		return nil, nil
	}
	if len(m) == 1 {
		return &m[0], nil
	}
	// defensive copy. Users can avoid by doing the split themselves.
	copyC := m[0]
	return &copyC, m[1:].copy()
}

// SplitLast returns the rest of the multiaddr and the last component.
func SplitLast(m Multiaddr) (Multiaddr, *Component) {
	if len(m) == 0 {
		return nil, nil
	}
	if len(m) == 1 {
		// We want to explicitly return a nil slice if the prefix is now empty.
		return nil, &m[0]
	}
	// defensive copy. Users can avoid by doing the split themselves.
	copyC := m[len(m)-1]
	return m[:len(m)-1].copy(), &copyC
}

// SplitFunc splits the multiaddr when the callback first returns true. The
// component on which the callback first returns will be included in the
// *second* multiaddr.
func SplitFunc(m Multiaddr, cb func(Component) bool) (Multiaddr, Multiaddr) {
	if len(m) == 0 {
		return nil, nil
	}

	idx := len(m)
	for i, c := range m {
		if cb(c) {
			idx = i
			break
		}
	}
	pre, post := m[:idx], m[idx:]
	if len(pre) == 0 {
		pre = nil
	}
	if len(post) == 0 {
		post = nil
	}
	// defensive copy. Users can avoid by doing the split themselves.
	return pre.copy(), post.copy()
}

// ForEach walks over the multiaddr, component by component.
//
// This function iterates over components.
// Return true to continue iteration, false to stop.
//
// Prefer a standard for range loop instead
// e.g. `for _, c := range m { ... }`
func ForEach(m Multiaddr, cb func(c Component, err error) bool) {
	if m == nil {
		return
	}
	for _, c := range m {
		if !cb(c, nil) {
			return
		}
	}
}