Replaced regexp in ExtractValuesFromTemplate function and updated usage

Author: ReverseTM

doc/en/usage.md

@@ -247,9 +247,21 @@ all its features and available functions in this [documentation](https://pkg.go.
 
 Accessing Data:
 
-In a template, data is accessed using `.`(the object or value passed to the template)
+In a template, data is accessed using `.` (the object or value passed to the template)
 and the field name, for example: `{{ .var }}`.
 
+> **Important**: only the following characters are allowed in variable names:
+>
+> - letters from any alphabet (Unicode category L*).
+> - numbers (Unicode category Nd), but not as the first character.
+> - the underscore character `_`.
+>
+> Any other characters — spaces, periods, hyphens, quotation marks, punctuation marks,
+  etc. — cannot be used in the name itself.
+>
+> If you need to access a variable whose name is considered invalid,
+  refer to it using the `index` function, specifying the variable name in quotation marks, for example, `{{ index . "field-with-dash" }}`.
+
 Function calls:
 
 - direct call: `{{ upper .name }}`.

doc/ru/usage.md

@@ -253,9 +253,21 @@ open_ai:
 
 Доступ к данным:
 
-Обращение к данным в шаблоне выполняется с помощью `.`(объект или значение, переданное шаблону)
+Обращение к данным в шаблоне выполняется с помощью `.` (объект или значение, переданное шаблону)
 и имени переменной, например, `{{ .var }}`.
 
+> **Важно**: в именах переменных допустимы только следующие символы:
+>
+> - буквы любого алфавита (Unicode-категории L*).
+> - цифры (Unicode-категории Nd), но не первым символом.
+> - знак подчёркивания `_`.
+>
+> Любые другие символы — пробелы, точки, дефисы, кавычки, символы пунктуации
+  и т.д. — в самом имени использовать нельзя.
+>
+> Если вам необходимо получить доступ к переменной, имя которой считается недопустимым,
+  обращайтесь к ней через функцию `index`, указав имя переменной в кавычках, например `{{ index . "field-with-dash" }}`.
+
 Вызовы функций:
 
 - прямой вызов: `{{ upper .name }}`.

internal/generator/common/utils.go

@@ -361,12 +361,21 @@ func CtxClosed(ctx context.Context) bool {
 }
 
 func ExtractValuesFromTemplate(template string) []string {
-	re := regexp.MustCompile(`{{.*?\.([^\s|}]+).*?}}`)
-	matches := re.FindAllStringSubmatch(template, -1)
+	// regexp for templates like {{ .name }}
+	reField := regexp.MustCompile(`{{\s*(?:\w+\s+)?\.([^\s"|}]+).*?}}`)
+	matchesField := reField.FindAllStringSubmatch(template, -1)
 
-	values := make([]string, 0, len(matches))
+	// regexp for templates like {{ index . "name-with-specific-symbols" }}
+	reMapKey := regexp.MustCompile(`{{\s*index\s+\.\s+"([^"]+)".*?}}`)
+	matchesMapKeys := reMapKey.FindAllStringSubmatch(template, -1)
 
-	for _, match := range matches {
+	values := make([]string, 0, len(matchesField)+len(matchesMapKeys))
+
+	for _, match := range matchesField {
+		values = append(values, match[1])
+	}
+
+	for _, match := range matchesMapKeys {
 		values = append(values, match[1])
 	}
 

internal/generator/common/utils_test.go

@@ -718,8 +718,23 @@ func TestExtractValuesFromTemplate(t *testing.T) {
 		},
 		{
 			name:     "Template with functions",
-			template: "{{ upper .foo | lower }}@{{ .boo }}",
-			expected: []string{"foo", "boo"},
+			template: "{{ upper .foo | lower }}",
+			expected: []string{"foo"},
+		},
+		{
+			name:     "Template with index function",
+			template: `{{ index . foo }}.{{ index . "boo" }}`,
+			expected: []string{"boo"},
+		},
+		{
+			name:     "Mixed template",
+			template: `{{ index . "foo" }}.{{ lower .boo }}|{{.coo}}`,
+			expected: []string{"boo", "coo", "foo"},
+		},
+		{
+			name:     "Field name in quotation marks",
+			template: `{{ ."foo" }}`,
+			expected: []string{},
 		},
 		{
 			name:     "Invalid template",

internal/generator/usecase/general/generator/value/string.go

@@ -486,6 +486,9 @@ func (g *StringGenerator) Value(number float64, rowValues map[string]any) (any,
 //nolint:cyclop
 func (g *StringGenerator) ValuesCount() float64 {
 	if g.Template != "" {
+		// Using `distinct` or `ordered` parameters with templates
+		// is not possible, we cannot guarantee that these parameters
+		// will be met, so we just need to return something other than 0.
 		return 1.0
 	}