package unpackstring

import (
	"errors"
	"strconv"
	"strings"
	"unicode"
)

// ErrInvalidString - Определение пользовательской ошибки для недопустимой строки.
var ErrInvalidString = errors.New("invalid string")

// state - Тип-перечисление для представления состояния автомата.
type state int

const (
	start state = iota
	escape
	number
)

// Unpack - Функция для распаковки строки в заданном формате.
func Unpack(str string) (string, error) {
	var currentState state = start
	var result strings.Builder
	numOfRepeat := 1

	var runeArray = []rune(str)

	//re, _ := regexp.Compile(`^\d|\d{2,}`)
	//
	//if re.MatchString(str) == true {
	//	return "", ErrInvalidString
	//}

	if len(runeArray) > 0 && unicode.IsDigit(runeArray[0]) {
		// Если первый символ является цифрой, возвращаем ошибку "недопустимой строки".
		return "first rune is digit", ErrInvalidString
	}

	for i, char := range runeArray {

		switch currentState {
		case start:
			if unicode.IsDigit(char) {
				currentState = number
				numOfRepeat, _ = strconv.Atoi(string(char))
				n := strconv.Itoa(numOfRepeat)
				repStr, err := repeatRune(runeArray[i-len(n)], numOfRepeat)
				if err != nil {
					// Если функция repeatRune возвращает ошибку, удаляем последний символ из результата.
					res := result.String()
					result.Reset()
					result.WriteString(res[0 : len(res)-1])
				}
				result.WriteString(repStr)
			} else if char == '\\' {
				currentState = escape
			} else {
				result.WriteRune(char)
				currentState = start
			}
		case number:
			if char == '\\' {
				currentState = escape
			} else if unicode.IsDigit(char) {
				var n = strconv.Itoa(numOfRepeat) + string(char)
				numOfRepeat, _ = strconv.Atoi(n)
				repStr, err := repeatRune(runeArray[i-len(n)], numOfRepeat)
				if err != nil {
					// Если функция repeatRune возвращает ошибку, удаляем последний символ из результата.
					res := result.String()
					result.Reset()
					result.WriteString(res[0 : len(res)-1])
				}
				result.WriteString(repStr)

			} else {
				result.WriteRune(char)
				currentState = start
			}

		case escape:
			// Если предыдущий символ - обратный слэш, добавляем текущий в результат.
			result.WriteRune(char)
			currentState = start
		}

	}

	return result.String(), nil

}

// repeatRune - Функция для повторения символа заданное количество раз.
func repeatRune(char rune, numOfRepeat int) (string, error) {
	if numOfRepeat > 0 {
		// Повторяем символ заданное количество раз и возвращаем строку.
		return strings.Repeat(string(char), numOfRepeat-1), nil
	}
	if numOfRepeat == 0 {
		// Если количество повторений равно нулю, возвращаем ошибку "нулевое повторение".
		return "", errors.New("zero repeat")
	}
	// Возвращаем пустую строку для отрицательных значений numOfRepeat.
	return "", errors.New("negative repeat")
}