-
Notifications
You must be signed in to change notification settings - Fork 206
/
beautify.go
66 lines (60 loc) · 1.64 KB
/
beautify.go
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
// Copyright 2016 Documize Inc. <legal@documize.com>. All rights reserved.
//
// This software (Documize Community Edition) is licensed under
// GNU AGPL v3 http://www.gnu.org/licenses/agpl-3.0.en.html
//
// You can operate outside the AGPL restrictions by purchasing
// Documize Enterprise Edition and obtaining a commercial license
// by contacting <sales@documize.com>.
//
// https://documize.com
package stringutil
import (
"path/filepath"
"strings"
"unicode"
)
// BeautifyFilename takes a filename and attempts to turn it into a readable form,
// as TitleCase natural language, suitable for the top level of a Document.
func BeautifyFilename(fn string) string {
_, file := filepath.Split(fn)
splits := strings.Split(file, ".")
r := []rune(strings.Join(splits[:len(splits)-1], "."))
// make any non-letter/digit characters space
for i := range r {
if !(unicode.IsLetter(r[i]) || unicode.IsDigit(r[i]) || r[i] == '.') {
r[i] = ' '
}
}
// insert spaces in front of any Upper/Lowwer 2-letter combinations
addSpaces:
for i := range r {
if i > 1 { // do not insert a space at the start of the file name
if unicode.IsLower(r[i]) && unicode.IsUpper(r[i-1]) && r[i-2] != ' ' {
n := make([]rune, len(r)+1)
for j := 0; j < i-1; j++ {
n[j] = r[j]
}
n[i-1] = ' '
for j := i - 1; j < len(r); j++ {
n[j+1] = r[j]
}
r = n
goto addSpaces
}
}
}
// make the first letter of each word upper case
for i := range r {
switch i {
case 0:
r[i] = unicode.ToUpper(r[i])
case 1: // the zero element should never be space
default:
if r[i-1] == ' ' {
r[i] = unicode.ToUpper(r[i])
}
}
}
return string(r)
}