123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- 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
- }
|