Hw02 unpack string

hw02_unpack_string/go.mod

@@ -1,4 +1,4 @@
-module github.com/fixme_my_friend/hw02_unpack_string
+module github.com/DimVlas/otus_hw/hw02_unpack_string
 
 go 1.19
 

hw02_unpack_string/unpack.go

@@ -2,11 +2,85 @@ package hw02unpackstring
 
 import (
 	"errors"
+	"fmt"
+	"strconv"
+	"strings"
 )
 
-var ErrInvalidString = errors.New("invalid string")
+var (
+	ErrInvalidString = errors.New("invalid string")
 
-func Unpack(_ string) (string, error) {
-	// Place your code here.
-	return "", nil
+	nums = []rune{'0', '1', '2', '3', '4', '5', '6', '7', '8', '9'}
+
+	numZero = '0'
+
+	bslash = '\\'
+)
+
+// True - если руна является цифрой.
+func IsDigit(r rune) bool {
+	for _, num := range nums {
+		if num == r {
+			return true
+		}
+	}
+	return false
+}
+
+func Unpack(text string) (string, error) {
+	if text == "" { // с пустой строкой ничего не делаем
+		return "", nil
+	}
+
+	runes := []rune(text)
+
+	if IsDigit(runes[0]) { // если первая руна цифра
+		return "", ErrInvalidString
+	}
+
+	var res strings.Builder
+
+	lenRunes := len(runes)
+	for i := 0; i < lenRunes; i++ {
+		if IsDigit(runes[i]) {
+			return "", ErrInvalidString
+		}
+
+		if runes[i] == bslash { // текущий символ слэш
+			if i == lenRunes-1 { // это последний символ
+				return "", ErrInvalidString
+			}
+
+			if !IsDigit(runes[i+1]) && runes[i+1] != bslash { // следующий символ не цифра и не слэш
+				return "", ErrInvalidString
+			}
+
+			i++ // нужно обработать следующий символ как обычный
+		}
+
+		if i == lenRunes-1 { // это последний символ
+			res.WriteRune(runes[i])
+			break
+		}
+		if !IsDigit(runes[i+1]) { // следующий символ не цифра
+			res.WriteRune(runes[i])
+			continue
+		}
+		if IsDigit(runes[i+1]) { // следующий символ цифра
+			if runes[i+1] == numZero { // следующий символ '0'
+				i++
+				continue
+			}
+			n, err := strconv.Atoi(string(runes[i+1]))
+			if err != nil {
+				return "", fmt.Errorf("error converting rune '%q' to number: %w", runes[i+1], err)
+			}
+
+			res.WriteString(strings.Repeat(string(runes[i]), n))
+			i++
+			continue
+		}
+	}
+
+	return res.String(), nil
 }

hw02_unpack_string/unpack_test.go

@@ -13,14 +13,17 @@ func TestUnpack(t *testing.T) {
 		expected string
 	}{
 		{input: "a4bc2d5e", expected: "aaaabccddddde"},
+		{input: "a4bc2d5e3", expected: "aaaabccdddddeee"},
+		{input: "a4bc2d5e0", expected: "aaaabccddddd"},
+		{input: "a+3b", expected: "a+++b"},
 		{input: "abccd", expected: "abccd"},
 		{input: "", expected: ""},
 		{input: "aaa0b", expected: "aab"},
 		// uncomment if task with asterisk completed
-		// {input: `qwe\4\5`, expected: `qwe45`},
-		// {input: `qwe\45`, expected: `qwe44444`},
-		// {input: `qwe\\5`, expected: `qwe\\\\\`},
-		// {input: `qwe\\\3`, expected: `qwe\3`},
+		{input: `qwe\4\5`, expected: `qwe45`},
+		{input: `qwe\45`, expected: `qwe44444`},
+		{input: `qwe\\5`, expected: `qwe\\\\\`},
+		{input: `qwe\\\3`, expected: `qwe\3`},
 	}
 
 	for _, tc := range tests {
@@ -34,7 +37,7 @@ func TestUnpack(t *testing.T) {
 }
 
 func TestUnpackInvalidString(t *testing.T) {
-	invalidStrings := []string{"3abc", "45", "aaa10b"}
+	invalidStrings := []string{"3abc", "45", "aaa10b", "aaa+b10", `aaaa\`, `aaa\\\b`}
 	for _, tc := range invalidStrings {
 		tc := tc
 		t.Run(tc, func(t *testing.T) {
@@ -43,3 +46,23 @@ func TestUnpackInvalidString(t *testing.T) {
 		})
 	}
 }
+
+func TestIsDigit(t *testing.T) {
+	tests := []struct {
+		input    rune
+		expected bool
+	}{
+		{input: 'a', expected: false},
+		{input: '5', expected: true},
+		{input: '!', expected: false},
+		{input: '\n', expected: false},
+	}
+
+	for _, tc := range tests {
+		tc := tc
+		t.Run(string(tc.input), func(t *testing.T) {
+			result := IsDigit(tc.input)
+			require.Equal(t, tc.expected, result)
+		})
+	}
+}