From 251ab85702f324e49b6f57fcf7219ea9b338d9bb Mon Sep 17 00:00:00 2001
From: "valentin.daviot" <valentin.daviot@alterway.fr>
Date: Thu, 8 Nov 2018 16:53:44 +0100
Subject: [PATCH] listing active session

---
 shell.go | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/shell.go b/shell.go
index d82e1a3c..253414c3 100644
--- a/shell.go
+++ b/shell.go
@@ -2066,6 +2066,7 @@ GLOBAL OPTIONS:
 					Usage: "Lists sessions",
 					Flags: []cli.Flag{
 						cli.BoolFlag{Name: "latest, l", Usage: "Show the latest session"},
+						cli.BoolFlag{Name: "active, a", Usage: "Show only active session"},
 						cli.BoolFlag{Name: "quiet, q", Usage: "Only display IDs"},
 					},
 					Action: func(c *cli.Context) error {
@@ -2074,7 +2075,14 @@ GLOBAL OPTIONS:
 						}
 
 						var sessions []*Session
-						query := db.Order("created_at desc").Preload("User").Preload("Host")
+
+						limit, offset, status := 60000, -1, []string{string(SessionStatusActive), string(SessionStatusClosed), string(SessionStatusUnknown)}
+						if c.Bool("active") {
+							status = status[:1]
+						}
+
+						query := db.Order("created_at desc").Limit(limit).Offset(offset).Where("status in (?)", status).Preload("User").Preload("Host")
+
 						if c.Bool("latest") {
 							var session Session
 							if err := query.First(&session).Error; err != nil {
@@ -2085,6 +2093,20 @@ GLOBAL OPTIONS:
 							if err := query.Find(&sessions).Error; err != nil {
 								return err
 							}
+
+							factor := 1
+							for len(sessions) >= limit*factor {
+
+								var additionnalSessions []*Session
+
+								offset = limit * factor
+								query := db.Order("created_at desc").Limit(limit).Offset(offset).Where("status in (?)", status).Preload("User").Preload("Host")
+								if err := query.Find(&additionnalSessions).Error; err != nil {
+									return err
+								}
+								sessions = append(sessions, additionnalSessions...)
+								factor++
+							}
 						}
 						if c.Bool("quiet") {
 							for _, session := range sessions {