Merge pull request #16 from whyphi/feat/new-user-flag

feat: add flag for isNewUser
whyphi · Mar 7, 2024 · 204b086 · 204b086
2 parents a1c3d9c + cbb3256
commit 204b086
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -13,8 +13,9 @@ import (
 )
 
 type User struct {
-	Name  string `yaml:"name"`
-	Email string `yaml:"email"`
+	Name      string `yaml:"name"`
+	Email     string `yaml:"email"`
+	IsNewUser bool   `json:"isNewUser" bson:"isNewUser"`
 }
 
 type UsersData struct {

diff --git a/mongodb.go b/mongodb.go
@@ -11,6 +11,7 @@ import (
 // Function to insert a user into MongoDB
 func insertUser(client *mongo.Client, user User) error {
 	collection := client.Database("vault").Collection("users")
+	user.IsNewUser = true
 	_, err := collection.InsertOne(context.Background(), user)
 	if err != nil {
 		log.Printf("Error inserting user: %v", err)