95 lines
1.9 KiB
Go
95 lines
1.9 KiB
Go
package httpapi
|
|
|
|
import (
|
|
"encoding/base64"
|
|
"encoding/binary"
|
|
"encoding/json"
|
|
"fmt"
|
|
"strconv"
|
|
"strings"
|
|
)
|
|
|
|
type userID int64
|
|
|
|
func (id *userID) UnmarshalJSON(raw []byte) error {
|
|
value, err := parseUserIDJSON(raw)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
*id = userID(value)
|
|
return nil
|
|
}
|
|
|
|
type optionalUserID struct {
|
|
value int64
|
|
set bool
|
|
}
|
|
|
|
func (id *optionalUserID) UnmarshalJSON(raw []byte) error {
|
|
if strings.TrimSpace(string(raw)) == "null" {
|
|
return nil
|
|
}
|
|
value, err := parseUserIDJSON(raw)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
id.set = true
|
|
id.value = value
|
|
return nil
|
|
}
|
|
|
|
func (id optionalUserID) ptr() *int64 {
|
|
if !id.set {
|
|
return nil
|
|
}
|
|
return &id.value
|
|
}
|
|
|
|
func parseUserIDJSON(raw []byte) (int64, error) {
|
|
var text string
|
|
if err := json.Unmarshal(raw, &text); err == nil {
|
|
return parseUserID(text)
|
|
}
|
|
return 0, fmt.Errorf("user id must be a string")
|
|
}
|
|
|
|
func parseUserID(value string) (int64, error) {
|
|
value = strings.TrimSpace(value)
|
|
if value == "" {
|
|
return 0, fmt.Errorf("empty user id")
|
|
}
|
|
if strings.HasPrefix(value, "usr") {
|
|
decoded, err := base64.URLEncoding.WithPadding(base64.NoPadding).DecodeString(strings.TrimPrefix(value, "usr"))
|
|
if err != nil || len(decoded) < 8 {
|
|
return 0, fmt.Errorf("invalid user id")
|
|
}
|
|
return int64(binary.LittleEndian.Uint64(decoded[:8])), nil
|
|
}
|
|
return strconv.ParseInt(value, 10, 64)
|
|
}
|
|
|
|
func formatUserID(id int64) string {
|
|
if id == 0 {
|
|
return ""
|
|
}
|
|
src := make([]byte, 8)
|
|
dst := make([]byte, base64.URLEncoding.WithPadding(base64.NoPadding).EncodedLen(8))
|
|
binary.LittleEndian.PutUint64(src, uint64(id))
|
|
base64.URLEncoding.WithPadding(base64.NoPadding).Encode(dst, src)
|
|
return "usr" + string(dst)
|
|
}
|
|
|
|
func parseUserIDList(values []userID) []int64 {
|
|
items := make([]int64, 0, len(values))
|
|
seen := make(map[int64]bool, len(values))
|
|
for _, value := range values {
|
|
id := int64(value)
|
|
if id == 0 || seen[id] {
|
|
continue
|
|
}
|
|
seen[id] = true
|
|
items = append(items, id)
|
|
}
|
|
return items
|
|
}
|