iridium/neko
2
2
Fork 0
mirror of https://github.com/m1k1o/neko.git synced 2024-07-24 14:40:50 +12:00
Code Issues Packages Projects Releases Wiki Activity

move cookies to session + config.

Browse Source
This commit is contained in:
Miroslav Šedivý
2021-03-17 14:09:10 +01:00
parent d06a5a2ac7
commit 4abe0a5dba
4 changed files with 66 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
internal/config/session.go
View File

@ -1,6 +1,8 @@
package config
import (
"time"
"github.com/spf13/cobra"
"github.com/spf13/viper"
)
@ -8,6 +10,10 @@ import (
type Session struct {
ImplicitHosting bool
APIToken string
CookieName string
CookieExpiration time.Time
CookieSecure bool
}
func (Session) Init(cmd *cobra.Command) error {
@ -21,10 +27,30 @@ func (Session) Init(cmd *cobra.Command) error {
return err
}
// cookie
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
}
func (s *Session) Set() {
s.ImplicitHosting = viper.GetBool("session.implicit_hosting")
s.APIToken = viper.GetString("session.api_token")
s.CookieName = viper.GetString("session.cookie.name")
s.CookieExpiration = time.Now().Add(time.Duration(viper.GetInt("session.cookie.expiration")) * time.Hour)
s.CookieSecure = viper.GetBool("session.cookie.secure")
}