feat: Support for setting session path

neticdk · Aug 29, 2023 · 62d0f53 · 62d0f53
1 parent 2abdcd3
commit 62d0f53
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pkg/cmd/root.go b/pkg/cmd/root.go
@@ -28,6 +28,7 @@ const (
 	redirectURL = "redirectURL"
 	providers   = "providers"
 	upstreams   = "upstreams"
+	sessionPath = "sessionPath"
 )
 
 type provider struct {
@@ -66,7 +67,7 @@ var (
 			if err != nil {
 				return err
 			}
-			fs := sessions.NewFilesystemStore("", hashKey, blockKey)
+			fs := sessions.NewFilesystemStore(viper.GetString(sessionPath), hashKey, blockKey)
 			fs.MaxLength(0)
 			e.Use(session.Middleware(fs))
 
@@ -162,6 +163,11 @@ func init() {
 	_ = viper.BindPFlag(redirectURL, rootCmd.Flags().Lookup("redirect-url"))
 	_ = viper.BindEnv(redirectURL, "REDIRECT_URL")
 
+	rootCmd.Flags().String("session-path", "", "Path to store encrypted session data (SESSION_PATH)")
+	_ = viper.BindPFlag(sessionPath, rootCmd.Flags().Lookup("session-path"))
+	_ = viper.BindEnv(redirectURL, "SESSION_PATH")
+	viper.SetDefault(sessionPath, "")
+
 	viper.SetConfigName("config")
 	viper.SetConfigType("yaml")
 	viper.AddConfigPath(".")