chore: move populate.go to utils

Author: tituschewxj

apps/question-service/README.md

@@ -60,5 +60,5 @@ The server will be available at http://localhost:8080.
 To reset and repopulate the database, run the following command:
 
 ```bash
-go run populate.go
+go run main.go -populate
 ```

apps/question-service/main.go

@@ -2,11 +2,13 @@ package main
 
 import (
 	"context"
+	"flag"
 	"fmt"
 	"log"
 	"net/http"
 	"os"
 	"question-service/handlers"
+	"question-service/utils"
 	"time"
 
 	"cloud.google.com/go/firestore"
@@ -51,6 +53,14 @@ func main() {
 
 	service := &handlers.Service{Client: client}
 
+	// Check flags if should populate instead.
+	shouldPopulate := flag.Bool("populate", false, "Populate database")
+	flag.Parse()
+	if *shouldPopulate {
+		utils.Populate(client)
+		return
+	}
+
 	// Set up chi router
 	r := chi.NewRouter()
 	r.Use(middleware.Logger)

apps/question-service/populate.go renamed to apps/question-service/utils/populate.go

@@ -1,15 +1,13 @@
-package main
+package utils
 
 import (
-	"cloud.google.com/go/firestore"
 	"context"
-	firebase "firebase.google.com/go/v4"
-	"fmt"
-	"google.golang.org/api/iterator"
-	"google.golang.org/api/option"
 	"log"
 	"question-service/models"
 	"time"
+
+	"cloud.google.com/go/firestore"
+	"google.golang.org/api/iterator"
 )
 
 // PopulateSampleQuestionsInTransaction deletes all existing questions and then adds new ones in a single transaction
@@ -295,32 +293,11 @@ Return the result table in any order.`,
 	})
 }
 
-// initFirestore initializes the Firestore client
-func initFirestore(ctx context.Context, credentialsPath string) (*firestore.Client, error) {
-	opt := option.WithCredentialsFile(credentialsPath)
-	app, err := firebase.NewApp(ctx, nil, opt)
-	if err != nil {
-		return nil, fmt.Errorf("failed to initialize Firebase App: %v", err)
-	}
-
-	client, err := app.Firestore(ctx)
-	if err != nil {
-		return nil, fmt.Errorf("failed to get Firestore client: %v", err)
-	}
-	return client, nil
-}
-
-func main() {
-	// Initialize Firestore client
+func Populate(client *firestore.Client) {
 	ctx := context.Background()
-	client, err := initFirestore(ctx, "cs3219-g24-firebase-adminsdk-9cm7h-b1675603ab.json")
-	if err != nil {
-		log.Fatalf("Failed to initialize Firestore client: %v", err)
-	}
-	defer client.Close()
 
 	// Run the transaction to delete all questions and add new ones
-	err = populateSampleQuestionsInTransaction(ctx, client)
+	err := populateSampleQuestionsInTransaction(ctx, client)
 	if err != nil {
 		log.Fatalf("Failed to populate sample questions in transaction: %v", err)
 	}