Add password validation

session/session.go

@@ -0,0 +1,100 @@
+package session
+
+import (
+	"github.com/fossyy/filekeeper/utils"
+	"net/http"
+	"strconv"
+	"sync"
+)
+
+type Session struct {
+	ID     string
+	Values map[string]interface{}
+}
+
+type StoreSession struct {
+	Sessions map[string]*Session
+	mu       sync.Mutex
+}
+
+var Store = StoreSession{Sessions: make(map[string]*Session)}
+var userSessions = make(map[string][]string)
+
+type SessionNotFound struct{}
+
+func (e *SessionNotFound) Error() string {
+	return "session not found"
+}
+
+func (s *StoreSession) Get(id string) (*Session, error) {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	if session, ok := s.Sessions[id]; ok {
+		return session, nil
+	}
+	return nil, &SessionNotFound{}
+}
+
+func (s *StoreSession) Create() *Session {
+	id := utils.GenerateRandomString(128)
+	session := &Session{
+		ID:     id,
+		Values: make(map[string]interface{}),
+	}
+	s.Sessions[id] = session
+	return session
+}
+
+func (s *StoreSession) Delete(id string) {
+	s.mu.Lock()
+	defer s.mu.Unlock()
+	delete(s.Sessions, id)
+}
+
+func (s *Session) Save(w http.ResponseWriter) {
+	maxAge, _ := strconv.Atoi(utils.Getenv("SESSION_MAX_AGE"))
+	http.SetCookie(w, &http.Cookie{
+		Name:   utils.Getenv("SESSION_NAME"),
+		Value:  s.ID,
+		MaxAge: maxAge,
+	})
+}
+
+func (s *Session) Destroy(w http.ResponseWriter) {
+	http.SetCookie(w, &http.Cookie{
+		Name:   utils.Getenv("SESSION_NAME"),
+		Value:  "",
+		MaxAge: -1,
+	})
+}
+
+func AppendSession(email string, session *Session) {
+	userSessions[email] = append(userSessions[email], session.ID)
+}
+
+func RemoveSession(email string, id string) {
+	sessions := userSessions[email]
+	var updatedSessions []string
+	for _, userSession := range sessions {
+		if userSession != id {
+			updatedSessions = append(updatedSessions, userSession)
+		}
+	}
+	if len(updatedSessions) > 0 {
+		userSessions[email] = updatedSessions
+		return
+	}
+	delete(userSessions, email)
+}
+
+func RemoveAllSession(email string) {
+	sessions := userSessions[email]
+	for _, session := range sessions {
+		delete(Store.Sessions, session)
+	}
+	delete(userSessions, email)
+}
+
+func Getses() map[string][]string {
+	return userSessions
+}