chore: update user setting validator

This commit is contained in:
Steven
2022-08-20 21:51:28 +08:00
parent 3b1bb4a95d
commit c60bb12424
3 changed files with 62 additions and 4 deletions

View File

@ -1,5 +1,10 @@
package api
import (
"encoding/json"
"fmt"
)
type UserSettingKey string
const (
@ -20,6 +25,11 @@ func (key UserSettingKey) String() string {
return ""
}
var (
UserSettingLocaleValue = []string{"en", "zh"}
UserSettingMemoVisibilityValue = []Visibility{Privite, Protected, Public}
)
type UserSetting struct {
UserID int
Key UserSettingKey `json:"key"`
@ -33,6 +43,48 @@ type UserSettingUpsert struct {
Value string `json:"value"`
}
func (upsert UserSettingUpsert) Validate() error {
if upsert.Key == UserSettingLocaleKey {
var localeValue string
err := json.Unmarshal([]byte(upsert.Value), &localeValue)
if err != nil {
return fmt.Errorf("failed to unmarshal user setting locale value")
}
invalid := true
for _, value := range UserSettingLocaleValue {
if localeValue == value {
invalid = false
break
}
}
if invalid {
return fmt.Errorf("invalid user setting locale value")
}
} else if upsert.Key == UserSettingMemoVisibilityKey {
var memoVisibilityValue Visibility
err := json.Unmarshal([]byte(upsert.Value), &memoVisibilityValue)
if err != nil {
return fmt.Errorf("failed to unmarshal user setting memo visibility value")
}
invalid := true
for _, value := range UserSettingMemoVisibilityValue {
if memoVisibilityValue == value {
invalid = false
break
}
}
if invalid {
return fmt.Errorf("invalid user setting memo visibility value")
}
} else {
return fmt.Errorf("invalid user setting key")
}
return nil
}
type UserSettingFind struct {
UserID int