feat: add session token

server/sessionstore/in_memory_session.go

@@ -6,9 +6,9 @@ import (
 
 // InMemoryStore is a simple in-memory store for sessions.
 type InMemoryStore struct {
-	mutex            sync.Mutex
-	store            map[string]map[string]string
-	socialLoginState map[string]string
+	mutex        sync.Mutex
+	sessionStore map[string]map[string]string
+	stateStore   map[string]string
 }
 
 // AddUserSession adds a user session to the in-memory store.
@@ -16,20 +16,20 @@ func (c *InMemoryStore) AddUserSession(userId, accessToken, refreshToken string)
 	c.mutex.Lock()
 	defer c.mutex.Unlock()
 	// delete sessions > 500 // not recommended for production
-	if len(c.store) >= 500 {
-		c.store = map[string]map[string]string{}
+	if len(c.sessionStore) >= 500 {
+		c.sessionStore = map[string]map[string]string{}
 	}
 	// check if entry exists in map
-	_, exists := c.store[userId]
+	_, exists := c.sessionStore[userId]
 	if exists {
-		tempMap := c.store[userId]
+		tempMap := c.sessionStore[userId]
 		tempMap[accessToken] = refreshToken
-		c.store[userId] = tempMap
+		c.sessionStore[userId] = tempMap
 	} else {
 		tempMap := map[string]string{
 			accessToken: refreshToken,
 		}
-		c.store[userId] = tempMap
+		c.sessionStore[userId] = tempMap
 	}
 }
 
@@ -37,21 +37,21 @@ func (c *InMemoryStore) AddUserSession(userId, accessToken, refreshToken string)
 func (c *InMemoryStore) DeleteAllUserSession(userId string) {
 	c.mutex.Lock()
 	defer c.mutex.Unlock()
-	delete(c.store, userId)
+	delete(c.sessionStore, userId)
 }
 
 // DeleteUserSession deletes the particular user session from in-memory store.
 func (c *InMemoryStore) DeleteUserSession(userId, accessToken string) {
 	c.mutex.Lock()
 	defer c.mutex.Unlock()
-	delete(c.store[userId], accessToken)
+	delete(c.sessionStore[userId], accessToken)
 }
 
 // ClearStore clears the in-memory store.
 func (c *InMemoryStore) ClearStore() {
 	c.mutex.Lock()
 	defer c.mutex.Unlock()
-	c.store = map[string]map[string]string{}
+	c.sessionStore = map[string]map[string]string{}
 }
 
 // GetUserSession returns the user session token from the in-memory store.
@@ -60,7 +60,7 @@ func (c *InMemoryStore) GetUserSession(userId, accessToken string) string {
 	// defer c.mutex.Unlock()
 
 	token := ""
-	if sessionMap, ok := c.store[userId]; ok {
+	if sessionMap, ok := c.sessionStore[userId]; ok {
 		if val, ok := sessionMap[accessToken]; ok {
 			token = val
 		}
@@ -74,7 +74,7 @@ func (c *InMemoryStore) GetUserSessions(userId string) map[string]string {
 	// c.mutex.Lock()
 	// defer c.mutex.Unlock()
 
-	sessionMap, ok := c.store[userId]
+	sessionMap, ok := c.sessionStore[userId]
 	if !ok {
 		return nil
 	}
@@ -82,31 +82,31 @@ func (c *InMemoryStore) GetUserSessions(userId string) map[string]string {
 	return sessionMap
 }
 
-// SetSocialLoginState sets the social login state in the in-memory store.
-func (c *InMemoryStore) SetSocialLoginState(key, state string) {
+// SetState sets the state in the in-memory store.
+func (c *InMemoryStore) SetState(key, state string) {
 	c.mutex.Lock()
 	defer c.mutex.Unlock()
 
-	c.socialLoginState[key] = state
+	c.stateStore[key] = state
 }
 
-// GetSocialLoginState gets the social login state from the in-memory store.
-func (c *InMemoryStore) GetSocialLoginState(key string) string {
+// GetState gets the state from the in-memory store.
+func (c *InMemoryStore) GetState(key string) string {
 	c.mutex.Lock()
 	defer c.mutex.Unlock()
 
 	state := ""
-	if stateVal, ok := c.socialLoginState[key]; ok {
+	if stateVal, ok := c.stateStore[key]; ok {
 		state = stateVal
 	}
 
 	return state
 }
 
-// RemoveSocialLoginState removes the social login state from the in-memory store.
-func (c *InMemoryStore) RemoveSocialLoginState(key string) {
+// RemoveState removes the state from the in-memory store.
+func (c *InMemoryStore) RemoveState(key string) {
 	c.mutex.Lock()
 	defer c.mutex.Unlock()
 
-	delete(c.socialLoginState, key)
+	delete(c.stateStore, key)
 }