photoprism/pkg/sanitize/id.go
2021-12-14 18:34:52 +01:00

42 lines
692 B
Go

package sanitize
import (
"strconv"
"strings"
)
// IdString removes invalid character from an id string.
func IdString(s string) string {
if s == "" || len(s) > 256 {
return ""
}
s = strings.ToLower(s)
// Remove all invalid characters.
s = strings.Map(func(r rune) rune {
if (r < '0' || r > '9') && (r < 'a' || r > 'z') && r != '-' && r != '_' && r != ':' {
return -1
}
return r
}, s)
return s
}
// IdUint converts the string converted to an unsigned integer and 0 if the string is invalid.
func IdUint(s string) uint {
if s == "" || len(s) > 64 {
return 0
}
result, err := strconv.ParseUint(s, 10, 32)
if err != nil {
return 0
}
return uint(result)
}