session save & load from a file. (#38)

internal/session/manager.go

@@ -49,6 +49,9 @@ func New(config *config.Session) *SessionManagerCtx {
 		}
 	}
 
+	// try to load sessions from file
+	manager.load()
+
 	return manager
 }
 
@@ -102,6 +105,8 @@ func (manager *SessionManagerCtx) Create(id string, profile types.MemberProfile)
 	manager.sessionsMu.Unlock()
 
 	manager.emmiter.Emit("created", session)
+	manager.save()
+
 	return session, token, nil
 }
 
@@ -118,6 +123,8 @@ func (manager *SessionManagerCtx) Update(id string, profile types.MemberProfile)
 	manager.sessionsMu.Unlock()
 
 	manager.emmiter.Emit("profile_changed", session)
+	manager.save()
+
 	session.profileChanged()
 	return nil
 }
@@ -143,6 +150,8 @@ func (manager *SessionManagerCtx) Delete(id string) error {
 	}
 
 	manager.emmiter.Emit("deleted", session)
+	manager.save()
+
 	return nil
 }
 

internal/session/serialize.go

@@ -0,0 +1,97 @@
+package session
+
+import (
+	"encoding/json"
+	"errors"
+	"os"
+
+	"github.com/demodesk/neko/pkg/types"
+)
+
+func (manager *SessionManagerCtx) save() {
+	if manager.config.File == "" {
+		return
+	}
+
+	// serialize sessions
+	sessions := make([]types.SessionProfile, 0, len(manager.sessions))
+	for _, session := range manager.sessions {
+		sessions = append(sessions, types.SessionProfile{
+			Id:      session.id,
+			Token:   session.token,
+			Profile: session.profile,
+		})
+	}
+
+	// convert to json
+	data, err := json.Marshal(sessions)
+	if err != nil {
+		manager.logger.Error().Err(err).Msg("failed to marshal sessions")
+		return
+	}
+
+	// write to file
+	err = os.WriteFile(manager.config.File, data, 0644)
+	if err != nil {
+		manager.logger.Error().Err(err).
+			Str("file", manager.config.File).
+			Msg("failed to write sessions to a file")
+	}
+}
+
+func (manager *SessionManagerCtx) load() {
+	if manager.config.File == "" {
+		return
+	}
+
+	// read file
+	data, err := os.ReadFile(manager.config.File)
+	if err != nil {
+		// if file does not exist
+		if errors.Is(err, os.ErrNotExist) {
+			manager.logger.Info().
+				Str("file", manager.config.File).
+				Msg("sessions file does not exist")
+			return
+		}
+		manager.logger.Error().Err(err).
+			Str("file", manager.config.File).
+			Msg("failed to read sessions from a file")
+		return
+	}
+
+	// if file is empty
+	if len(data) == 0 {
+		manager.logger.Info().
+			Str("file", manager.config.File).
+			Msg("sessions file is empty")
+		return
+	}
+
+	// deserialize sessions
+	sessions := make([]types.SessionProfile, 0)
+	err = json.Unmarshal(data, &sessions)
+	if err != nil {
+		manager.logger.Error().Err(err).Msg("failed to unmarshal sessions")
+		return
+	}
+
+	// create sessions
+	manager.sessionsMu.Lock()
+	for _, session := range sessions {
+		manager.tokens[session.Token] = session.Id
+		manager.sessions[session.Id] = &SessionCtx{
+			id:      session.Id,
+			token:   session.Token,
+			manager: manager,
+			logger:  manager.logger.With().Str("session_id", session.Id).Logger(),
+			profile: session.Profile,
+		}
+	}
+	manager.sessionsMu.Unlock()
+
+	manager.logger.Info().
+		Int("sessions", len(sessions)).
+		Str("file", manager.config.File).
+		Msg("loaded sessions from a file")
+}