package textfmt

import "strings"

func getWords(text string) []string {
	var words []string
	raw := strings.Split(text, " ")
	for _, r := range raw {
		if r == "" {
			continue
		}

		words = append(words, r)
	}

	return words
}

func lineLength(words []string) int {
	if len(words) == 0 {
		return 0
	}

	var l int
	for _, w := range words {
		r := []rune(w)
		l += len(r)
	}

	return l + len(words) - 1
}

func singleLine(text string) string {
	var l []string
	p := strings.Split(text, "\n")
	for _, part := range p {
		part = strings.TrimSpace(part)
		if part == "" {
			continue
		}

		l = append(l, part)
	}

	return strings.Join(l, " ")
}

func wrap(text string, width, firstIndent, restIndent int) string {
	var (
		lines       []string
		currentLine []string
		currentLen int
	)

	words := getWords(text)
	for _, w := range words {
		maxw := width - restIndent
		if len(lines) == 0 {
			maxw = width - firstIndent
		}

		currentLine = append(currentLine, w)
		if lineLength(currentLine) > maxw {
			currentLine = currentLine[:len(currentLine)-1]
			lines = append(lines, strings.Join(currentLine, " "))
			currentLine = []string{w}
		}
	}

	if len(currentLine) > 0 {
		lines = append(lines, strings.Join(currentLine, " "))
	}

	return strings.Join(lines, "\n")
}