mirror of
https://github.com/m1k1o/neko.git
synced 2024-07-24 14:40:50 +12:00
52 lines
1.6 KiB
Go
52 lines
1.6 KiB
Go
package config
|
|
|
|
import (
|
|
"github.com/spf13/cobra"
|
|
"github.com/spf13/viper"
|
|
)
|
|
|
|
type Session struct {
|
|
Password string
|
|
AdminPassword string
|
|
ImplicitHosting bool
|
|
DatabaseAdapter string
|
|
FilePath string
|
|
}
|
|
|
|
func (Session) Init(cmd *cobra.Command) error {
|
|
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
|
|
}
|
|
|
|
cmd.PersistentFlags().Bool("implicit_hosting", true, "allow implicit control switching")
|
|
if err := viper.BindPFlag("implicit_hosting", cmd.PersistentFlags().Lookup("implicit_hosting")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().String("database_adapter", "file", "choose database adapter for members")
|
|
if err := viper.BindPFlag("database_adapter", cmd.PersistentFlags().Lookup("database_adapter")); err != nil {
|
|
return err
|
|
}
|
|
|
|
cmd.PersistentFlags().String("file_path", "/home/neko/members.json", "file adapter: specify file path")
|
|
if err := viper.BindPFlag("file_path", cmd.PersistentFlags().Lookup("file_path")); err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (s *Session) Set() {
|
|
s.Password = viper.GetString("password")
|
|
s.AdminPassword = viper.GetString("password_admin")
|
|
s.ImplicitHosting = viper.GetBool("implicit_hosting")
|
|
s.DatabaseAdapter = viper.GetString("database_adapter")
|
|
s.FilePath = viper.GetString("file_path")
|
|
}
|