user crud added

server/db/providers/couchbase/user.go

@@ -2,12 +2,16 @@ package couchbase
 
 import (
 	"context"
+	"fmt"
+	"log"
+	"reflect"
 	"time"
 
 	"github.com/authorizerdev/authorizer/server/constants"
 	"github.com/authorizerdev/authorizer/server/db/models"
 	"github.com/authorizerdev/authorizer/server/graph/model"
 	"github.com/authorizerdev/authorizer/server/memorystore"
+	"github.com/couchbase/gocb/v2"
 	"github.com/google/uuid"
 )
 
@@ -27,29 +31,108 @@ func (p *provider) AddUser(ctx context.Context, user models.User) (models.User,
 
 	user.CreatedAt = time.Now().Unix()
 	user.UpdatedAt = time.Now().Unix()
-
+	insertOpt := gocb.InsertOptions{
+		Context: ctx,
+	}
+	_, err := p.db.Collection(models.Collections.User).Insert(user.ID, user, &insertOpt)
+	if err != nil {
+		return user, err
+	}
 	return user, nil
 }
 
 // UpdateUser to update user information in database
 func (p *provider) UpdateUser(ctx context.Context, user models.User) (models.User, error) {
 	user.UpdatedAt = time.Now().Unix()
+	unsertOpt := gocb.UpsertOptions{
+		Context: ctx,
+	}
+	_, err := p.db.Collection(models.Collections.User).Upsert(user.ID, user, &unsertOpt)
+	if err != nil {
+		return user, err
+	}
 	return user, nil
 }
 
 // DeleteUser to delete user information from database
 func (p *provider) DeleteUser(ctx context.Context, user models.User) error {
+	removeOpt := gocb.RemoveOptions{
+		Context: ctx,
+	}
+	_, err := p.db.Collection(models.Collections.User).Remove(user.ID, &removeOpt)
+	// query := fmt.Sprintf("INSERT INTO %s %s VALUES %s IF NOT EXISTS", KeySpace+"."+models.Collections.User, fields, values)
+
+	// sessionCollection := p.db.Collection(models.Collections.Session).Queue()
+	// _, err = sessionCollection.DeleteMany(ctx, bson.M{"user_id": user.ID}, options.Delete())
+	// if err != nil {
+	// 	return err
+	// }
+	if err != nil {
+		return err
+	}
 	return nil
 }
 
 // ListUsers to get list of users from database
 func (p *provider) ListUsers(ctx context.Context, pagination model.Pagination) (*model.Users, error) {
-	return nil, nil
+	users := []*model.User{}
+	// r := p.db.Collection(models.Collections.User).
+	paginationClone := pagination
+
+	inventoryScope := p.db.Scope("_default")
+	userQuery := fmt.Sprintf("SELECT * FROM auth._default.%s ORDER BY id OFFSET %d LIMIT %d", models.Collections.User, paginationClone.Offset, paginationClone.Limit)
+
+	queryResult, err := inventoryScope.Query(userQuery, &gocb.QueryOptions{})
+
+	if err != nil {
+		return nil, err
+	}
+
+	for queryResult.Next() {
+		var user models.User
+		err := queryResult.Row(&user)
+		if err != nil {
+			log.Fatal(err)
+		}
+		users = append(users, user.AsAPIUser())
+	}
+
+	if err := queryResult.Err(); err != nil {
+		return nil, err
+
+	}
+
+	return &model.Users{
+		Pagination: &paginationClone,
+		Users:      users,
+	}, nil
 }
 
 // GetUserByEmail to get user information from database using email address
 func (p *provider) GetUserByEmail(ctx context.Context, email string) (models.User, error) {
 	var user models.User
+	// inventoryScope := p.db.Scope("_default")
+	// userQuery := fmt.Sprintf("SELECT * FROM auth._default.%s WHERE email = '%s' LIMIT %d", models.Collections.User, email, 1)
+	queryResult, err := p.db.Collection(models.Collections.User).Get("email:"+email, &gocb.GetOptions{})
+	if err != nil {
+		return user, err
+	}
+
+	// queryResult, err := inventoryScope.Query(userQuery, &gocb.QueryOptions{})
+	err = queryResult.Content(&user)
+
+	if err != nil {
+		return user, err
+	}
+
+	// for queryResult.Next() {
+	// 	var user models.User
+	// 	err := queryResult.Row(&user)
+	// 	if err != nil {
+	// 		log.Fatal(err)
+	// 	}
+	// 	return user, nil
+	// }
 
 	return user, nil
 }
@@ -57,7 +140,15 @@ func (p *provider) GetUserByEmail(ctx context.Context, email string) (models.Use
 // GetUserByID to get user information from database using user ID
 func (p *provider) GetUserByID(ctx context.Context, id string) (models.User, error) {
 	var user models.User
+	queryResult, err := p.db.Collection(models.Collections.User).Get("id:"+id, &gocb.GetOptions{})
+	if err != nil {
+		return user, err
+	}
 
+	err = queryResult.Content(&user)
+	if err != nil {
+		return user, err
+	}
 	return user, nil
 }
 
@@ -66,6 +157,46 @@ func (p *provider) GetUserByID(ctx context.Context, id string) (models.User, err
 func (p *provider) UpdateUsers(ctx context.Context, data map[string]interface{}, ids []string) error {
 	// set updated_at time for all users
 	data["updated_at"] = time.Now().Unix()
+	inventoryScope := p.db.Scope("_default")
+
+	updateFields := ""
+	for key, value := range data {
+		if key == "_id" {
+			continue
+		}
+
+		if key == "_key" {
+			continue
+		}
+
+		if value == nil {
+			updateFields += fmt.Sprintf("%s = null,", key)
+			continue
+		}
+
+		valueType := reflect.TypeOf(value)
+		if valueType.Name() == "string" {
+			updateFields += fmt.Sprintf("%s = '%s', ", key, value.(string))
+		} else {
+			updateFields += fmt.Sprintf("%s = %v, ", key, value)
+		}
+	}
+
+	if ids != nil && len(ids) > 0 {
+		for _, v := range ids {
+			userQuery := fmt.Sprintf("UPDATE * FROM auth._default.%s SET %s WHERE id = '%s'", models.Collections.User, updateFields, v)
+			_, err := inventoryScope.Query(userQuery, &gocb.QueryOptions{})
+			if err != nil {
+				return err
+			}
+		}
+	} else {
+		userQuery := fmt.Sprintf("UPDATE * FROM auth._default.%s SET %s WHERE id IS NOT NULL", models.Collections.User, updateFields)
+		_, err := inventoryScope.Query(userQuery, &gocb.QueryOptions{})
+		if err != nil {
+			return err
+		}
+	}
 
 	return nil
 }