neko/internal/config/session.go

78 lines
2.8 KiB
Go
Raw Normal View History

package config
import (
2021-03-18 02:09:10 +13:00
"time"
"github.com/spf13/cobra"
"github.com/spf13/viper"
)
2020-11-02 06:39:12 +13:00
type Session struct {
2021-08-24 04:41:19 +12:00
ImplicitHosting bool
2021-11-02 05:49:02 +13:00
InactiveCursors bool
2021-08-24 04:41:19 +12:00
MercifulReconnect bool
APIToken string
2021-03-18 02:09:10 +13:00
2021-04-25 06:53:37 +12:00
CookieEnabled bool
2021-03-18 02:09:10 +13:00
CookieName string
2022-01-17 09:03:25 +13:00
CookieExpiration time.Duration
2021-03-18 02:09:10 +13:00
CookieSecure bool
}
2020-11-02 06:39:12 +13:00
func (Session) Init(cmd *cobra.Command) error {
2021-03-17 03:24:58 +13:00
cmd.PersistentFlags().Bool("session.implicit_hosting", true, "allow implicit control switching")
if err := viper.BindPFlag("session.implicit_hosting", cmd.PersistentFlags().Lookup("session.implicit_hosting")); err != nil {
2020-12-02 23:24:20 +13:00
return err
}
2021-11-02 09:35:15 +13:00
cmd.PersistentFlags().Bool("session.inactive_cursors", false, "show inactive cursors on the screen")
2021-11-02 05:49:02 +13:00
if err := viper.BindPFlag("session.inactive_cursors", cmd.PersistentFlags().Lookup("session.inactive_cursors")); err != nil {
return err
}
2021-08-24 04:41:19 +12:00
cmd.PersistentFlags().Bool("session.merciful_reconnect", true, "allow reconnecting to websocket even if previous connection was not closed")
if err := viper.BindPFlag("session.merciful_reconnect", cmd.PersistentFlags().Lookup("session.merciful_reconnect")); err != nil {
return err
}
2021-03-17 03:24:58 +13:00
cmd.PersistentFlags().String("session.api_token", "", "API token for interacting with external services")
if err := viper.BindPFlag("session.api_token", cmd.PersistentFlags().Lookup("session.api_token")); err != nil {
2021-03-16 01:34:14 +13:00
return err
}
2021-03-18 02:09:10 +13:00
// cookie
2021-04-25 06:53:37 +12:00
cmd.PersistentFlags().Bool("session.cookie.enabled", true, "whether cookies authentication should be enabled")
if err := viper.BindPFlag("session.cookie.enabled", cmd.PersistentFlags().Lookup("session.cookie.enabled")); err != nil {
return err
}
2021-03-18 02:09:10 +13:00
cmd.PersistentFlags().String("session.cookie.name", "NEKO_SESSION", "name of the cookie that holds token")
if err := viper.BindPFlag("session.cookie.name", cmd.PersistentFlags().Lookup("session.cookie.name")); err != nil {
return err
}
cmd.PersistentFlags().Int("session.cookie.expiration", 365*24, "expiration of the cookie in hours")
if err := viper.BindPFlag("session.cookie.expiration", cmd.PersistentFlags().Lookup("session.cookie.expiration")); err != nil {
return err
}
cmd.PersistentFlags().Bool("session.cookie.secure", true, "use secure cookies")
if err := viper.BindPFlag("session.cookie.secure", cmd.PersistentFlags().Lookup("session.cookie.secure")); err != nil {
return err
}
return nil
}
2020-11-02 06:39:12 +13:00
func (s *Session) Set() {
2021-03-17 03:24:58 +13:00
s.ImplicitHosting = viper.GetBool("session.implicit_hosting")
2021-11-02 05:49:02 +13:00
s.InactiveCursors = viper.GetBool("session.inactive_cursors")
2021-08-24 04:41:19 +12:00
s.MercifulReconnect = viper.GetBool("session.merciful_reconnect")
2021-03-17 03:24:58 +13:00
s.APIToken = viper.GetString("session.api_token")
2021-03-18 02:09:10 +13:00
2021-04-25 06:53:37 +12:00
s.CookieEnabled = viper.GetBool("session.cookie.enabled")
2021-03-18 02:09:10 +13:00
s.CookieName = viper.GetString("session.cookie.name")
2022-01-17 09:03:25 +13:00
s.CookieExpiration = time.Duration(viper.GetInt("session.cookie.expiration")) * time.Hour
2021-03-18 02:09:10 +13:00
s.CookieSecure = viper.GetBool("session.cookie.secure")
}