Implemented refersh for session and more fields for session

Author: PythonHacker24

cmd/laclm/main.go

@@ -47,7 +47,6 @@ func exec() error {
 			Use:   "laclm <command> <subcommand>",
 			Short: "Backend server for linux acl management",
 			Example: heredoc.Doc(`
-				$ laclm
 				$ laclm --config /path/to/config.yaml
 			`),
 			Run: func(cmd *cobra.Command, args []string) {
@@ -187,8 +186,6 @@ func run(ctx context.Context) error {
 		return err
 	}
 
-	// <-ctx.Done()
-
 	/*
 		after this, exit signal is triggered
 		following code must be executed to shutdown graceful shutdown

internal/auth/handler.go

@@ -44,7 +44,7 @@ func LoginHandler(sessionManager *session.Manager) http.HandlerFunc {
 		}
 
 		/* after building session manager */
-		err = sessionManager.CreateSession(user.Username)
+		err = sessionManager.CreateSession(user.Username, r.RemoteAddr, r.UserAgent())
 		if err != nil {
 			zap.L().Error("Error creating session",
 				zap.Error(err),

internal/session/manager.go

@@ -6,6 +6,8 @@ import (
 	"sync"
 	"time"
 
+	"github.com/google/uuid"
+
 	"github.com/PythonHacker24/linux-acl-management-backend/config"
 )
 
@@ -24,7 +26,7 @@ type Manager struct {
 }
 
 /* for creating a session for user - used by HTTP HANDLERS */
-func (m *Manager) CreateSession(username string) error {
+func (m *Manager) CreateSession(username, ipAddress, userAgent string) error {
 
 	/* lock the ActiveSessions mutex till the function ends */
 	m.mutex.Lock()
@@ -35,13 +37,24 @@ func (m *Manager) CreateSession(username string) error {
 		return fmt.Errorf("user already exists in active sessions")
 	}
 
+	/* Generate session metadata */
+	sessionID := uuid.New().String()
+	now := time.Now()
+
 	/* create the session */
 	session := &Session{
+		ID: sessionID,
 		Username: username,
+		IP: ipAddress,
+		UserAgent: userAgent,
 		Expiry:   time.Now().Add(time.Duration(config.BackendConfig.AppInfo.SessionTimeout) * time.Hour),
-		Timer: time.AfterFunc(time.Duration(config.BackendConfig.AppInfo.SessionTimeout)*time.Hour,
+		CreatedAt:        now,
+		LastActiveAt:     now,
+		Timer: time.AfterFunc(time.Duration(config.BackendConfig.AppInfo.SessionTimeout) * time.Hour,
 			func() { m.ExpireSession(username) },
 		),
+		CompletedCount:   0,
+		FailedCount:      0,
 		TransactionQueue:  list.New(),
 	}
 
@@ -76,7 +89,7 @@ func (m *Manager) ExpireSession(username string) {
 
 /* add transaction to a session */
 func (m *Manager) AddTransaction(username string, txn interface{}) error {
-	/* thread safety the manager mutex */
+	/* thread safety for the manager */
 	m.mutex.Lock()
 	defer m.mutex.Unlock()
 
@@ -95,3 +108,32 @@ func (m *Manager) AddTransaction(username string, txn interface{}) error {
 
 	return nil
 }
+
+/* refresh the session timer */
+func (m *Manager) RefreshTimer(username string) error {
+	/* thread safety for the manager */
+	m.mutex.Lock()
+	defer m.mutex.Unlock()
+
+	/* get session from sessionMap */
+	session, exists := m.sessionsMap[username]
+	if !exists {
+		return fmt.Errorf("Session not found")
+	}
+
+	/* reset the expiry time and last active time */
+	session.Expiry = time.Now().Add(time.Duration(config.BackendConfig.AppInfo.SessionTimeout) * time.Hour)
+	session.LastActiveAt = time.Now()
+
+	/* stop the session timer */
+	if session.Timer != nil {
+		session.Timer.Stop()
+	}
+	
+	/* reset the session timer */
+	session.Timer = time.AfterFunc(time.Duration(config.BackendConfig.AppInfo.SessionTimeout) * time.Hour,
+			func() { m.ExpireSession(username) },
+		) 
+
+	return nil
+}

internal/session/model.go

@@ -6,12 +6,39 @@ import (
 	"time"
 )
 
-/* session struct for a user */
+/* 
+	session struct for a user 
+	appropriate fields must always be updated when any request is made
+*/
 type Session struct {
+	/* keep count of completed and failed transactions */
+	CompletedCount    int
+	FailedCount       int
+
+	/* unique ID of session [will be associated with the user forever in logs] */
+	ID 				  string
+
+	/* username of the user */
 	Username          string
+
+	/* 
+		IP and UserAgent for security logs
+		also can be used for blacklisting and whitelistings 
+		illegal useragents can be caught as well as unauthorized IP addresses
+	*/
+	IP 				  string
+	UserAgent		  string
+
+	/* for logging user activity */
+	CreatedAt		  time.Time
+	LastActiveAt      time.Time
 	Expiry            time.Time
 	Timer             *time.Timer
+
+	/* transactions issued by the user */
 	TransactionQueue  *list.List
+
+	/* mutex for thread safety */
 	Mutex             sync.Mutex
 
 	/*