109 lines
2 KiB
Go
109 lines
2 KiB
Go
package namer
|
||
|
||
import (
|
||
"crypto/md5"
|
||
"encoding/hex"
|
||
"fmt"
|
||
"regexp"
|
||
"strconv"
|
||
"strings"
|
||
"sync"
|
||
)
|
||
|
||
var claimNameRegexp = regexp.MustCompile(`[=&#:$@%？?;、\\"/<>%{}|｜^~\x60[\]\s]`)
|
||
|
||
type Namer struct {
|
||
mu *sync.Mutex
|
||
names map[string]bool
|
||
}
|
||
|
||
func NewNamer() *Namer {
|
||
return &Namer{
|
||
mu: &sync.Mutex{},
|
||
names: make(map[string]bool),
|
||
}
|
||
}
|
||
|
||
func (n *Namer) SetNames(names map[string]bool) {
|
||
n.names = names
|
||
}
|
||
|
||
func (n *Namer) GetNextName(prefix string) string {
|
||
n.mu.Lock()
|
||
defer n.mu.Unlock()
|
||
|
||
attempt := 1
|
||
var name string
|
||
for {
|
||
name = getClaimNameFromTitle(prefix, attempt)
|
||
if _, exists := n.names[name]; !exists {
|
||
break
|
||
}
|
||
attempt++
|
||
}
|
||
|
||
//if for some reasons the title can't be converted in a valid claim name (too short or not latin) then we use a hash
|
||
attempt = 1
|
||
if len(name) < 2 {
|
||
sum := md5.Sum([]byte(prefix))
|
||
for {
|
||
name = fmt.Sprintf("%s-%d", hex.EncodeToString(sum[:])[:15], attempt)
|
||
if _, exists := n.names[name]; !exists {
|
||
break
|
||
}
|
||
attempt++
|
||
}
|
||
}
|
||
|
||
n.names[name] = true
|
||
|
||
return name
|
||
}
|
||
|
||
// TODO: clean this up some
|
||
func getClaimNameFromTitle(title string, attempt int) string {
|
||
suffix := ""
|
||
if attempt > 1 {
|
||
suffix = "-" + strconv.Itoa(attempt)
|
||
}
|
||
maxLen := 40 - len(suffix)
|
||
|
||
chunks := strings.Split(strings.ToLower(strings.Trim(claimNameRegexp.ReplaceAllString(title, "-"), "-")), "-")
|
||
|
||
name := chunks[0]
|
||
if len(name) > maxLen {
|
||
return truncateUnicode(name, maxLen) + suffix
|
||
}
|
||
|
||
for _, chunk := range chunks[1:] {
|
||
if chunk == "" {
|
||
continue
|
||
}
|
||
tmpName := name + "-" + chunk
|
||
if len(tmpName) > maxLen {
|
||
if len(name) < 20 {
|
||
name = truncateUnicode(tmpName, maxLen-len(name))
|
||
}
|
||
break
|
||
}
|
||
name = tmpName
|
||
}
|
||
|
||
return name + suffix
|
||
}
|
||
|
||
func truncateUnicode(name string, limit int) string {
|
||
reNameBlacklist := regexp.MustCompile(`(&|>|<|\/|:|\n|\r)*`)
|
||
name = reNameBlacklist.ReplaceAllString(name, "")
|
||
result := name
|
||
chars := 0
|
||
for i := range name {
|
||
if chars >= limit {
|
||
result = name[:i]
|
||
break
|
||
}
|
||
chars++
|
||
}
|
||
return result
|
||
}
|