photoprism/pkg/rnd/validation.go

107 lines
1.9 KiB
Go

package rnd
import (
"strconv"
"strings"
"time"
)
// IsUnique checks if the string is a valid unique ID.
func IsUnique(s string, prefix byte) bool {
if IsUUID(s) {
// Standard UUID.
return true
}
return IsUID(s, prefix)
}
// IsUID checks if the string is a valid entity UID.
func IsUID(s string, prefix byte) bool {
// Check length.
if len(s) != 16 {
return false
}
// Check prefix.
if prefix != 0 && s[0] != prefix {
return false
}
// Check character range.
if !IsAlnum(s) {
return false
}
// Check timestamp.
if ts := s[1:7]; ts == "000000" {
return true
} else if t, err := strconv.ParseInt(ts, 36, 64); err != nil {
return false
} else if t < 1483228800 || t > time.Now().UTC().Unix() {
return false
}
// Valid.
return true
}
// InvalidUID checks if the UID is empty or invalid.
func InvalidUID(s string, prefix byte) bool {
return !IsUID(s, prefix)
}
// IsUUID tests if the string looks like a standard UUID.
func IsUUID(s string) bool {
return len(s) == 36 && IsHex(s)
}
// SanitizeUUID normalizes UUIDs found in XMP or Exif metadata.
func SanitizeUUID(s string) string {
if s == "" {
return ""
}
s = strings.Replace(strings.TrimSpace(s), "\"", "", -1)
if start := strings.LastIndex(s, ":"); start != -1 {
s = s[start+1:]
}
if !IsUUID(s) {
return ""
}
return strings.ToLower(s)
}
// IsAlnum returns true if the string only contains alphanumeric ascii chars without whitespace.
func IsAlnum(s string) bool {
if s == "" {
return false
}
for _, r := range s {
if (r < '0' || r > '9') && (r < 'A' || r > 'Z') && (r < 'a' || r > 'z') {
return false
}
}
return true
}
// IsHex returns true if the string only contains hex numbers, dashes and letters without whitespace.
func IsHex(s string) bool {
if s == "" {
return false
}
for _, r := range s {
if (r < '0' || r > '9') && (r < 'a' || r > 'f') && (r < 'A' || r > 'F') && r != '-' {
return false
}
}
return true
}