fix: single letter words resolves #39

tympanix · Jun 18, 2018 · 84bde13 · 84bde13
1 parent 6e5aab7
commit 84bde13
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/parse/parse.go b/parse/parse.go
@@ -53,7 +53,7 @@ func Filename(filename string) string {
 	return strings.TrimSuffix(f, filepath.Ext(f))
 }
 
-var abbreviationRegexp = regexp.MustCompile(`[A-Z]\s[A-Z](\s[A-Z])*`)
+var abbreviationRegexp = regexp.MustCompile(`[A-Z]\s[A-Z](\s[A-Z])*\s`)
 var illegalcharsRegexp = regexp.MustCompile(`[^\p{L}0-9\s&'_\(\)-]`)
 var spaceReplaceRegexp = regexp.MustCompile(`[\.\s_]+`)
 
@@ -63,7 +63,7 @@ func CleanName(name string) string {
 	name = illegalcharsRegexp.ReplaceAllString(name, "")
 
 	name = abbreviationRegexp.ReplaceAllStringFunc(name, func(match string) string {
-		return strings.Replace(match, " ", "", -1)
+		return strings.Replace(match, " ", "", -1) + " "
 	})
 
 	return strings.TrimSpace(name)

diff --git a/parse/parse_test.go b/parse/parse_test.go
@@ -18,6 +18,8 @@ func TestCleanName(t *testing.T) {
 	assert.Equal(t, "this is a (test)", CleanName("this?_=is#.,a_(test)"))
 	assert.Equal(t, "abc ABC abc", CleanName("abc.A.B.C.abc"))
 	assert.Equal(t, "abc ABC abc", CleanName("abc A B C abc"))
+	assert.Equal(t, "A Good Day To Die Hard", CleanName("A.Good.Day.To.Die.Hard"))
+	assert.Equal(t, "This Is A Test", CleanName("This.Is.A.Test"))
 }
 
 func TestFileName(t *testing.T) {