mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
allow multiple users to be set in object provider.
This commit is contained in:
@ -1,12 +1,14 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"github.com/rs/zerolog/log"
|
||||
"github.com/spf13/cobra"
|
||||
"github.com/spf13/viper"
|
||||
|
||||
"github.com/demodesk/neko/internal/member/file"
|
||||
"github.com/demodesk/neko/internal/member/multiuser"
|
||||
"github.com/demodesk/neko/internal/member/object"
|
||||
"github.com/demodesk/neko/pkg/utils"
|
||||
)
|
||||
|
||||
type Member struct {
|
||||
@ -31,13 +33,8 @@ func (Member) Init(cmd *cobra.Command) error {
|
||||
}
|
||||
|
||||
// object provider
|
||||
cmd.PersistentFlags().String("member.object.user_password", "", "member object provider: user password")
|
||||
if err := viper.BindPFlag("member.object.user_password", cmd.PersistentFlags().Lookup("member.object.user_password")); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
cmd.PersistentFlags().String("member.object.admin_password", "", "member object provider: admin password")
|
||||
if err := viper.BindPFlag("member.object.admin_password", cmd.PersistentFlags().Lookup("member.object.admin_password")); err != nil {
|
||||
cmd.PersistentFlags().String("member.object.users", "[]", "member object provider: users in JSON format")
|
||||
if err := viper.BindPFlag("member.object.users", cmd.PersistentFlags().Lookup("member.object.users")); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
@ -62,8 +59,11 @@ func (s *Member) Set() {
|
||||
s.File.Path = viper.GetString("member.file.path")
|
||||
|
||||
// object provider
|
||||
s.Object.UserPassword = viper.GetString("member.object.user_password")
|
||||
s.Object.AdminPassword = viper.GetString("member.object.admin_password")
|
||||
if err := viper.UnmarshalKey("member.object.users", &s.Object.Users, viper.DecodeHook(
|
||||
utils.JsonStringAutoDecode(s.Object.Users),
|
||||
)); err != nil {
|
||||
log.Warn().Err(err).Msgf("unable to parse member object users")
|
||||
}
|
||||
|
||||
// multiuser provider
|
||||
s.Multiuser.UserPassword = viper.GetString("member.multiuser.user_password")
|
||||
|
Reference in New Issue
Block a user