fix: refactor schema for open id claim standards

server/session/in_memory_session.go

@@ -0,0 +1,85 @@
+package session
+
+import (
+	"sync"
+)
+
+type InMemoryStore struct {
+	mu               sync.Mutex
+	store            map[string]map[string]string
+	socialLoginState map[string]string
+}
+
+func (c *InMemoryStore) AddToken(userId, accessToken, refreshToken string) {
+	c.mu.Lock()
+	// delete sessions > 500 // not recommended for production
+	if len(c.store) >= 500 {
+		c.store = map[string]map[string]string{}
+	}
+	// check if entry exists in map
+	_, exists := c.store[userId]
+	if exists {
+		tempMap := c.store[userId]
+		tempMap[accessToken] = refreshToken
+		c.store[userId] = tempMap
+	} else {
+		tempMap := map[string]string{
+			accessToken: refreshToken,
+		}
+		c.store[userId] = tempMap
+	}
+
+	c.mu.Unlock()
+}
+
+func (c *InMemoryStore) DeleteUserSession(userId string) {
+	c.mu.Lock()
+	delete(c.store, userId)
+	c.mu.Unlock()
+}
+
+func (c *InMemoryStore) DeleteVerificationRequest(userId, accessToken string) {
+	c.mu.Lock()
+	delete(c.store[userId], accessToken)
+	c.mu.Unlock()
+}
+
+func (c *InMemoryStore) ClearStore() {
+	c.mu.Lock()
+	c.store = map[string]map[string]string{}
+	c.mu.Unlock()
+}
+
+func (c *InMemoryStore) GetToken(userId, accessToken string) string {
+	token := ""
+	c.mu.Lock()
+	if sessionMap, ok := c.store[userId]; ok {
+		if val, ok := sessionMap[accessToken]; ok {
+			token = val
+		}
+	}
+	c.mu.Unlock()
+
+	return token
+}
+
+func (c *InMemoryStore) SetSocialLoginState(key, state string) {
+	c.mu.Lock()
+	c.socialLoginState[key] = state
+	c.mu.Unlock()
+}
+
+func (c *InMemoryStore) GetSocialLoginState(key string) string {
+	state := ""
+	if stateVal, ok := c.socialLoginState[key]; ok {
+		state = stateVal
+	}
+
+	return state
+}
+
+func (c *InMemoryStore) RemoveSocialLoginState(key string) {
+	c.mu.Lock()
+	delete(c.socialLoginState, key)
+	c.mu.Unlock()
+}