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

add merciful reconnect option.

Browse Source
This commit is contained in:
Miroslav Šedivý
2021-08-23 18:41:19 +02:00
parent 01c0f269ee
commit b2ae808233
4 changed files with 35 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
internal/config/session.go
View File

@ -8,8 +8,9 @@ import (
)
type Session struct {
ImplicitHosting bool
APIToken string
ImplicitHosting bool
MercifulReconnect bool
APIToken string
CookieEnabled bool
CookieName string
@ -23,6 +24,11 @@ func (Session) Init(cmd *cobra.Command) error {
return err
}
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
}
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 {
return err
@ -54,6 +60,7 @@ func (Session) Init(cmd *cobra.Command) error {
func (s *Session) Set() {
s.ImplicitHosting = viper.GetBool("session.implicit_hosting")
s.MercifulReconnect = viper.GetBool("session.merciful_reconnect")
s.APIToken = viper.GetString("session.api_token")
s.CookieEnabled = viper.GetBool("session.cookie.enabled")