Implement login resolver (#15)

* add sign_up_method to users table

* add session store

* implement login resolver

server/session/inMemoryStore.go

@@ -0,0 +1,41 @@
+package session
+
+import "sync"
+
+type InMemoryStore struct {
+	mu    sync.Mutex
+	store map[string]string
+}
+
+func (c *InMemoryStore) AddToken(userId, token string) {
+	c.mu.Lock()
+	// delete sessions > 500 // not recommended for production
+	if len(c.store) >= 500 {
+		c.store = make(map[string]string)
+	}
+	c.store[userId] = token
+	c.mu.Unlock()
+}
+
+func (c *InMemoryStore) DeleteToken(userId string) {
+	c.mu.Lock()
+	delete(c.store, userId)
+	c.mu.Unlock()
+}
+
+func (c *InMemoryStore) ClearStore() {
+	c.mu.Lock()
+	c.store = make(map[string]string)
+	c.mu.Unlock()
+}
+
+func (c *InMemoryStore) GetToken(userId string) string {
+	token := ""
+	c.mu.Lock()
+	if val, ok := c.store[userId]; ok {
+		token = val
+	}
+	c.mu.Unlock()
+
+	return token
+}

server/session/redisStore.go

@@ -0,0 +1,43 @@
+package session
+
+import (
+	"context"
+	"log"
+
+	"github.com/go-redis/redis/v8"
+)
+
+type RedisStore struct {
+	ctx   context.Context
+	store *redis.Client
+}
+
+func (c *RedisStore) AddToken(userId, token string) {
+	err := c.store.Set(c.ctx, "yauth_"+userId, token, 0).Err()
+	if err != nil {
+		log.Fatalln("Error saving redis token:", err)
+	}
+}
+
+func (c *RedisStore) DeleteToken(userId string) {
+	err := c.store.Del(c.ctx, "yauth_"+userId).Err()
+	if err != nil {
+		log.Fatalln("Error deleting redis token:", err)
+	}
+}
+
+func (c *RedisStore) ClearStore() {
+	err := c.store.Del(c.ctx, "yauth_*").Err()
+	if err != nil {
+		log.Fatalln("Error clearing redis store:", err)
+	}
+}
+
+func (c *RedisStore) GetToken(userId string) string {
+	token := ""
+	token, err := c.store.Get(c.ctx, "yauth_"+userId).Result()
+	if err != nil {
+		log.Println("Error getting token from redis store:", err)
+	}
+	return token
+}

server/session/session.go

@@ -0,0 +1,81 @@
+package session
+
+import (
+	"context"
+	"log"
+
+	"github.com/go-redis/redis/v8"
+	"github.com/yauthdev/yauth/server/constants"
+)
+
+type SessionStore struct {
+	inMemoryStoreObj    *InMemoryStore
+	redisMemoryStoreObj *RedisStore
+}
+
+var SessionStoreObj SessionStore
+
+func SetToken(userId, token string) {
+	if SessionStoreObj.redisMemoryStoreObj != nil {
+		SessionStoreObj.redisMemoryStoreObj.AddToken(userId, token)
+	}
+	if SessionStoreObj.inMemoryStoreObj != nil {
+		SessionStoreObj.inMemoryStoreObj.AddToken(userId, token)
+	}
+}
+
+func DeleteToken(userId string) {
+	if SessionStoreObj.redisMemoryStoreObj != nil {
+		SessionStoreObj.redisMemoryStoreObj.DeleteToken(userId)
+	}
+	if SessionStoreObj.inMemoryStoreObj != nil {
+		SessionStoreObj.inMemoryStoreObj.DeleteToken(userId)
+	}
+}
+
+func GetToken(userId string) string {
+	if SessionStoreObj.redisMemoryStoreObj != nil {
+		return SessionStoreObj.redisMemoryStoreObj.GetToken(userId)
+	}
+	if SessionStoreObj.inMemoryStoreObj != nil {
+		return SessionStoreObj.inMemoryStoreObj.GetToken(userId)
+	}
+
+	return ""
+}
+
+func ClearStore() {
+	if SessionStoreObj.redisMemoryStoreObj != nil {
+		SessionStoreObj.redisMemoryStoreObj.ClearStore()
+	}
+	if SessionStoreObj.inMemoryStoreObj != nil {
+		SessionStoreObj.inMemoryStoreObj.ClearStore()
+	}
+}
+
+func init() {
+	if constants.REDIS_URL != "" {
+		log.Println("Using redis store to save sessions")
+		opt, err := redis.ParseURL(constants.REDIS_URL)
+		if err != nil {
+			log.Fatalln("Error parsing redis url:", err)
+		}
+		rdb := redis.NewClient(opt)
+		ctx := context.Background()
+		_, err = rdb.Ping(ctx).Result()
+
+		if err != nil {
+			log.Fatalln("Error connecting to redis server", err)
+		}
+		SessionStoreObj.redisMemoryStoreObj = &RedisStore{
+			ctx:   ctx,
+			store: rdb,
+		}
+
+	} else {
+		log.Println("Using in memory store to save sessions")
+		SessionStoreObj.inMemoryStoreObj = &InMemoryStore{
+			store: make(map[string]string),
+		}
+	}
+}