mirror of
https://github.com/usememos/memos.git
synced 2024-09-21 23:05:59 +08:00
58 lines
1.4 KiB
Go
58 lines
1.4 KiB
Go
package v2
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/usememos/memos/internal/util"
|
|
)
|
|
|
|
const (
|
|
UserNamePrefix = "users/"
|
|
InboxNamePrefix = "inboxes/"
|
|
)
|
|
|
|
// GetNameParentTokens returns the tokens from a resource name.
|
|
func GetNameParentTokens(name string, tokenPrefixes ...string) ([]string, error) {
|
|
parts := strings.Split(name, "/")
|
|
if len(parts) != 2*len(tokenPrefixes) {
|
|
return nil, errors.Errorf("invalid request %q", name)
|
|
}
|
|
|
|
var tokens []string
|
|
for i, tokenPrefix := range tokenPrefixes {
|
|
if fmt.Sprintf("%s/", parts[2*i]) != tokenPrefix {
|
|
return nil, errors.Errorf("invalid prefix %q in request %q", tokenPrefix, name)
|
|
}
|
|
if parts[2*i+1] == "" {
|
|
return nil, errors.Errorf("invalid request %q with empty prefix %q", name, tokenPrefix)
|
|
}
|
|
tokens = append(tokens, parts[2*i+1])
|
|
}
|
|
return tokens, nil
|
|
}
|
|
|
|
// ExtractUsernameFromName returns the username from a resource name.
|
|
func ExtractUsernameFromName(name string) (string, error) {
|
|
tokens, err := GetNameParentTokens(name, UserNamePrefix)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return tokens[0], nil
|
|
}
|
|
|
|
// ExtractInboxIDFromName returns the inbox ID from a resource name.
|
|
func ExtractInboxIDFromName(name string) (int32, error) {
|
|
tokens, err := GetNameParentTokens(name, InboxNamePrefix)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
id, err := util.ConvertStringToInt32(tokens[0])
|
|
if err != nil {
|
|
return 0, errors.Errorf("invalid inbox ID %q", tokens[0])
|
|
}
|
|
return id, nil
|
|
}
|