Handle patterns in vocab

jdkato · Oct 16, 2023 · d74a03f · d74a03f
1 parent 0089863
commit d74a03f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/strcase/sentence.go b/strcase/sentence.go
@@ -74,8 +74,11 @@ func (sc *SentenceConverter) Convert(s string) string {
 
 func (sc *SentenceConverter) inVocab(s string) string {
 	for _, token := range sc.vocab {
+		matched, _ := regexp.MatchString(token, s)
 		if strings.ToLower(token) == strings.ToLower(s) {
 			return token
+		} else if matched {
+			return s
 		}
 	}
 	return ""

diff --git a/strcase/title.go b/strcase/title.go
@@ -102,8 +102,11 @@ func (tc *TitleConverter) Convert(s string) string {
 
 func (tc *TitleConverter) inVocab(s string) string {
 	for _, token := range tc.vocab {
+		matched, _ := regexp.MatchString(token, s)
 		if strings.ToLower(token) == strings.ToLower(s) {
 			return token
+		} else if matched {
+			return s
 		}
 	}
 	return ""