neko/internal/config/member.go

package config

import (
	"github.com/spf13/cobra"
	"github.com/spf13/viper"
)

type Member struct {
	Provider      string
	FilePath      string
	Password      string
	AdminPassword string
}

func (Member) Init(cmd *cobra.Command) error {
	cmd.PersistentFlags().String("members_provider", "object", "choose members provider")
	if err := viper.BindPFlag("members_provider", cmd.PersistentFlags().Lookup("members_provider")); err != nil {
		return err
	}

	cmd.PersistentFlags().String("members_file_path", "", "mebmer file provider path")
	if err := viper.BindPFlag("members_file_path", cmd.PersistentFlags().Lookup("members_file_path")); err != nil {
		return err
	}

	cmd.PersistentFlags().String("password", "neko", "password for connecting to stream")
	if err := viper.BindPFlag("password", cmd.PersistentFlags().Lookup("password")); err != nil {
		return err
	}

	cmd.PersistentFlags().String("password_admin", "admin", "admin password for connecting to stream")
	if err := viper.BindPFlag("password_admin", cmd.PersistentFlags().Lookup("password_admin")); err != nil {
		return err
	}

	return nil
}

func (s *Member) Set() {
	s.Provider = viper.GetString("members_provider")
	s.FilePath = viper.GetString("members_file_path")
	s.Password = viper.GetString("password")
	s.AdminPassword = viper.GetString("password_admin")
}