summaryrefslogtreecommitdiff
path: root/internal/comic/text_validation.go
blob: 31f1bbac983799dce4a071d158236aaff98accd8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
package comic

import (
	"fmt"
	"strings"
	"unicode"
)

var promptLeakMarkers = []string{
	"words to include",
	"required words",
	"character guide",
	"comic title",
	"panel script",
	"mandatory language rule",
	"mandatory panel layout",
	"mandatory speech bubbles",
	"strict consistency rules",
	"story excerpt",
	"story teaser",
	"story ending hint",
	"---character guide---",
	"---comic title---",
	"---panel script---",
}

func validateTextScript(label, text, script string) error {
	text = strings.TrimSpace(text)
	if text == "" {
		return fmt.Errorf("%s is empty", label)
	}

	switch {
	case strings.EqualFold(script, "Cyrillic"):
		if containsLatinLetters(text) {
			return fmt.Errorf("%s contains Latin letters despite %s script", label, script)
		}
	case strings.EqualFold(script, "Latin"):
		if containsCyrillicLetters(text) {
			return fmt.Errorf("%s contains Cyrillic letters despite %s script", label, script)
		}
	}
	return nil
}

func containsLatinLetters(text string) bool {
	for _, r := range text {
		if unicode.Is(unicode.Latin, r) {
			return true
		}
	}
	return false
}

func containsCyrillicLetters(text string) bool {
	for _, r := range text {
		if unicode.Is(unicode.Cyrillic, r) {
			return true
		}
	}
	return false
}

func validateNoPromptLeakage(label, text string) error {
	lower := strings.ToLower(text)
	for _, marker := range promptLeakMarkers {
		if strings.Contains(lower, marker) {
			return fmt.Errorf("%s contains prompt leakage marker %q", label, marker)
		}
	}
	return nil
}