Updated api_handlers.go, database.go, and utils.go

COS301-SE-2024 · Jun 9, 2024 · 1f55fb5 · 1f55fb5
1 parent 24c4985
commit 1f55fb5
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 2 deletions.
diff --git a/occupi-backend/pkg/handlers/api_handlers.go b/occupi-backend/pkg/handlers/api_handlers.go
@@ -49,7 +49,7 @@ func BookRoom(ctx *gin.Context, appsession *models.AppSession) {
 
 	// Generate a unique ID for the booking
 	booking.ID = primitive.NewObjectID().Hex()
-	booking.OccupiID = 1
+	booking.OccupiID = utils.GenerateOccupiBookID()
 	booking.CheckedIn = false
 
 	// Save the booking to the database

diff --git a/occupi-backend/pkg/models/database.go b/occupi-backend/pkg/models/database.go
@@ -17,7 +17,7 @@ type User struct {
 // structure of booking
 type Booking struct {
 	ID        string   `json:"_id" bson:"_id,omitempty"`
-	OccupiID  int      `json:"occupiId" bson:"occupiId,omitempty"`
+	OccupiID  string   `json:"occupiId" bson:"occupiId,omitempty"`
 	RoomID    string   `json:"roomId" bson:"roomId"`
 	Slot      int      `json:"slot" bson:"slot"`
 	Emails    []string `json:"emails" bson:"emails"`

diff --git a/occupi-backend/pkg/utils/utils.go b/occupi-backend/pkg/utils/utils.go
@@ -64,6 +64,17 @@ func GenerateEmployeeID() string {
 	return employeeID
 }
 
+// Function to generate a Booking ID with the structure BOOKYYYYMMDDXXXX
+func GenerateOccupiBookID() string {
+	currentYear := time.Now().Year()
+	randomNum, err := generateRandomNumber()
+	if err != nil {
+		return "BOOK00000000"
+	}
+	BookID := fmt.Sprintf("BOOK%d%04d", currentYear, randomNum)
+	return BookID
+}
+
 // generates a random auth0 state
 func GenerateRandomState() (string, error) {
 	b := make([]byte, 32)