fix: user + verification requests to new db format

2022-01-21 12:53:30 +05:30
parent 13c038effd
commit 8a4b2feffe
7 changed files with 673 additions and 1 deletions
--- a/server/db/providers/mongodb/user.go
+++ b/server/db/providers/mongodb/user.go
@@ -0,0 +1,108 @@
+package mongodb
+
+import (
+	"log"
+	"strings"
+	"time"
+
+	"github.com/authorizerdev/authorizer/server/constants"
+	"github.com/authorizerdev/authorizer/server/db/models"
+	"github.com/authorizerdev/authorizer/server/envstore"
+	"github.com/google/uuid"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+// AddUser to save user information in database
+func (p *provider) AddUser(user models.User) (models.User, error) {
+	if user.ID == "" {
+		user.ID = uuid.New().String()
+	}
+
+	if user.Roles == "" {
+		user.Roles = strings.Join(envstore.EnvInMemoryStoreObj.GetSliceStoreEnvVariable(constants.EnvKeyDefaultRoles), ",")
+	}
+	user.CreatedAt = time.Now().Unix()
+	user.UpdatedAt = time.Now().Unix()
+	user.Key = user.ID
+	userCollection := p.db.Collection(models.Collections.User, options.Collection())
+	_, err := userCollection.InsertOne(nil, user)
+	if err != nil {
+		log.Println("error adding user:", err)
+		return user, err
+	}
+
+	return user, nil
+}
+
+// UpdateUser to update user information in database
+func (p *provider) UpdateUser(user models.User) (models.User, error) {
+	user.UpdatedAt = time.Now().Unix()
+	userCollection := p.db.Collection(models.Collections.User, options.Collection())
+	_, err := userCollection.UpdateOne(nil, bson.M{"_id": bson.M{"$eq": user.ID}}, bson.M{"$set": user}, options.MergeUpdateOptions())
+	if err != nil {
+		log.Println("error updating user:", err)
+		return user, err
+	}
+	return user, nil
+}
+
+// DeleteUser to delete user information from database
+func (p *provider) DeleteUser(user models.User) error {
+	userCollection := p.db.Collection(models.Collections.User, options.Collection())
+	_, err := userCollection.DeleteOne(nil, bson.M{"_id": user.ID}, options.Delete())
+	if err != nil {
+		log.Println("error deleting user:", err)
+		return err
+	}
+
+	return nil
+}
+
+// ListUsers to get list of users from database
+func (p *provider) ListUsers() ([]models.User, error) {
+	var users []models.User
+	userCollection := p.db.Collection(models.Collections.User, options.Collection())
+	cursor, err := userCollection.Find(nil, bson.M{}, options.Find())
+	if err != nil {
+		log.Println("error getting users:", err)
+		return users, err
+	}
+	defer cursor.Close(nil)
+
+	for cursor.Next(nil) {
+		var user models.User
+		err := cursor.Decode(&user)
+		if err != nil {
+			return users, err
+		}
+		users = append(users, user)
+	}
+
+	return users, nil
+}
+
+// GetUserByEmail to get user information from database using email address
+func (p *provider) GetUserByEmail(email string) (models.User, error) {
+	var user models.User
+	userCollection := p.db.Collection(models.Collections.User, options.Collection())
+	err := userCollection.FindOne(nil, bson.M{"email": email}).Decode(&user)
+	if err != nil {
+		return user, err
+	}
+
+	return user, nil
+}
+
+// GetUserByID to get user information from database using user ID
+func (p *provider) GetUserByID(id string) (models.User, error) {
+	var user models.User
+
+	userCollection := p.db.Collection(models.Collections.User, options.Collection())
+	err := userCollection.FindOne(nil, bson.M{"_id": id}).Decode(&user)
+	if err != nil {
+		return user, err
+	}
+
+	return user, nil
+}
--- a/server/db/providers/mongodb/verification_requests.go
+++ b/server/db/providers/mongodb/verification_requests.go
@@ -0,0 +1,91 @@
+package mongodb
+
+import (
+	"log"
+	"time"
+
+	"github.com/authorizerdev/authorizer/server/db/models"
+	"github.com/google/uuid"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+// AddVerification to save verification request in database
+func (p *provider) AddVerificationRequest(verificationRequest models.VerificationRequest) (models.VerificationRequest, error) {
+	if verificationRequest.ID == "" {
+		verificationRequest.ID = uuid.New().String()
+
+		verificationRequest.CreatedAt = time.Now().Unix()
+		verificationRequest.UpdatedAt = time.Now().Unix()
+		verificationRequest.Key = verificationRequest.ID
+		verificationRequestCollection := p.db.Collection(models.Collections.VerificationRequest, options.Collection())
+		_, err := verificationRequestCollection.InsertOne(nil, verificationRequest)
+		if err != nil {
+			log.Println("error saving verification record:", err)
+			return verificationRequest, err
+		}
+	}
+
+	return verificationRequest, nil
+}
+
+// GetVerificationRequestByToken to get verification request from database using token
+func (p *provider) GetVerificationRequestByToken(token string) (models.VerificationRequest, error) {
+	var verificationRequest models.VerificationRequest
+
+	verificationRequestCollection := p.db.Collection(models.Collections.VerificationRequest, options.Collection())
+	err := verificationRequestCollection.FindOne(nil, bson.M{"token": token}).Decode(&verificationRequest)
+	if err != nil {
+		return verificationRequest, err
+	}
+
+	return verificationRequest, nil
+}
+
+// GetVerificationRequestByEmail to get verification request by email from database
+func (p *provider) GetVerificationRequestByEmail(email string, identifier string) (models.VerificationRequest, error) {
+	var verificationRequest models.VerificationRequest
+
+	verificationRequestCollection := p.db.Collection(models.Collections.VerificationRequest, options.Collection())
+	err := verificationRequestCollection.FindOne(nil, bson.M{"email": email, "identifier": identifier}).Decode(&verificationRequest)
+	if err != nil {
+		return verificationRequest, err
+	}
+
+	return verificationRequest, nil
+}
+
+// ListVerificationRequests to get list of verification requests from database
+func (p *provider) ListVerificationRequests() ([]models.VerificationRequest, error) {
+	var verificationRequests []models.VerificationRequest
+	verificationRequestCollection := p.db.Collection(models.Collections.VerificationRequest, options.Collection())
+	cursor, err := verificationRequestCollection.Find(nil, bson.M{}, options.Find())
+	if err != nil {
+		log.Println("error getting verification requests:", err)
+		return verificationRequests, err
+	}
+	defer cursor.Close(nil)
+
+	for cursor.Next(nil) {
+		var verificationRequest models.VerificationRequest
+		err := cursor.Decode(&verificationRequest)
+		if err != nil {
+			return verificationRequests, err
+		}
+		verificationRequests = append(verificationRequests, verificationRequest)
+	}
+
+	return verificationRequests, nil
+}
+
+// DeleteVerificationRequest to delete verification request from database
+func (p *provider) DeleteVerificationRequest(verificationRequest models.VerificationRequest) error {
+	verificationRequestCollection := p.db.Collection(models.Collections.VerificationRequest, options.Collection())
+	_, err := verificationRequestCollection.DeleteOne(nil, bson.M{"_id": verificationRequest.ID}, options.Delete())
+	if err != nil {
+		log.Println("error deleting verification request::", err)
+		return err
+	}
+
+	return nil
+}