Capstone/ui/db/models/user.go

package models

import (
	"context"
	"errors"
	"time"

	"git.preston-baxter.com/Preston_PLB/capstone/frontend-service/config"
	"github.com/google/uuid"
	"go.mongodb.org/mongo-driver/bson"
	"go.mongodb.org/mongo-driver/bson/primitive"
	"go.mongodb.org/mongo-driver/mongo"
	"go.mongodb.org/mongo-driver/mongo/options"
)

const USER_TYPE string = "user"

type User struct {
	CommonFields `bson:"obj_info"`
	mongoId      primitive.ObjectID `bson:"_id,omitempty"`
	UserId       string             `bson:"user_id,omitempty"`
	Email        string             `bson:"email,omitempty"`
	PassowrdHash string             `bson:"password_hash,omitempty"`
}

func (user *User) Save(client *mongo.Client) error {
	conf := config.Config()

	if user.mongoId.IsZero() {
		now := time.Now()
		user.EntityType = USER_TYPE
		user.CreatedAt = now
		user.UpdatedAt = now
		user.UserId = uuid.New().String()
		user.mongoId = primitive.NewObjectIDFromTimestamp(now)
	}

	opts := options.Update().SetUpsert(true)

	res, err := client.Database(conf.Mongo.EntDb).Collection(conf.Mongo.EntCol).UpdateOne(context.Background(),  bson.M{"user_id": user.UserId}, bson.M{"$set": user},opts)
	if err != nil {
		return err
	}

	if res.MatchedCount == 0 && res.ModifiedCount == 0 && res.UpsertedCount == 0 {
		return errors.New("Failed to update client properly")
	}

	return nil
}

func (user *User) Delete(client *mongo.Client) error {
	conf := config.Config()
	opts := options.Delete()

	res, err := client.Database(conf.Mongo.EntDb).Collection(conf.Mongo.EntCol).DeleteOne(context.Background(), bson.M{"_id": user.mongoId}, opts)
	if err != nil {
		return err
	}

	if res.DeletedCount == 0 {
		return errors.New("There was no item to delete")
	}

	return nil
}