feat: more info when end exam

Author: ppolariss

backend/apis/exam/driver.go

@@ -74,8 +74,8 @@ func AddDriverPunishment(c *fiber.Ctx) (err error) {
 		if err != nil {
 			return
 		}
-		if !user.IsPassed {
-			return common.BadRequest("You don't have the driver's license")
+		if !user.ValidateDriver() {
+			return common.Forbidden("You don't have the driver's license")
 		}
 		if user.Point <= addPunishmentRequest.Score {
 			user.Point = 0

backend/apis/exam/exam.go

@@ -1,11 +1,13 @@
 package exam
 
 import (
+	"fmt"
+	. "src/models"
+	"time"
+
 	"github.com/gofiber/fiber/v2"
 	"github.com/opentreehole/go-common"
 	"gorm.io/gorm"
-	. "src/models"
-	"time"
 )
 
 // ListExams @ListExams
@@ -146,6 +148,8 @@ func EndExam(ctx *fiber.Ctx) (err error) {
 	if err != nil {
 		return
 	}
+	isDriver := false
+	info := "Oops, something went wrong"
 	var endExamRequest EndExamRequest
 	if err = ctx.BodyParser(&endExamRequest); err != nil {
 		return common.BadRequest("Invalid request body")
@@ -169,6 +173,7 @@ func EndExam(ctx *fiber.Ctx) (err error) {
 			return
 		}
 		if exam.Score != 100 {
+			info = "Sorry, you didn't pass the exam"
 			return
 		}
 		var user User
@@ -177,14 +182,27 @@ func EndExam(ctx *fiber.Ctx) (err error) {
 			return
 		}
 		if user.IsPassed {
+			info = "Congratulations! You have passed the exam"
 			return
 		}
+		if user.Age < 18 {
+			info = "Congratulations! You have passed the exam. However, you are not eligible for a driver's license because you are " + fmt.Sprintf("%d", user.Age) + " years old"
+			return
+		}
+		info = "Congratulations! You have passed the exam and obtained the driver's license"
+		isDriver = true
 		user.IsPassed = true
 		user.Point = 12
 		return DB.Model(&user).Select("IsPassed", "Point").UpdateColumns(&user).Error
 	})
+	if err != nil {
+		return
+	}
 	var endExamResponse = EndExamResponse{
-		Score: exam.Score,
+		Score:    exam.Score,
+		IsPassed: exam.Score == 100,
+		IsDriver: isDriver,
+		Info:     info,
 	}
 	return ctx.JSON(endExamResponse)
 }

backend/apis/exam/schemas.go

@@ -24,7 +24,10 @@ type EndExamRequest struct {
 }
 
 type EndExamResponse struct {
-	Score int `json:"score"`
+	Score    int    `json:"score"`
+	IsPassed bool   `json:"is_passed"`
+	IsDriver bool   `json:"is_driver"`
+	Info     string `json:"info"`
 }
 
 type AddExamRequest struct {

backend/docs/docs.go

@@ -1602,6 +1602,15 @@ const docTemplate = `{
         "exam.EndExamResponse": {
             "type": "object",
             "properties": {
+                "info": {
+                    "type": "string"
+                },
+                "is_driver": {
+                    "type": "boolean"
+                },
+                "is_passed": {
+                    "type": "boolean"
+                },
                 "score": {
                     "type": "integer"
                 }

backend/docs/swagger.json

@@ -1594,6 +1594,15 @@
         "exam.EndExamResponse": {
             "type": "object",
             "properties": {
+                "info": {
+                    "type": "string"
+                },
+                "is_driver": {
+                    "type": "boolean"
+                },
+                "is_passed": {
+                    "type": "boolean"
+                },
                 "score": {
                     "type": "integer"
                 }

backend/docs/swagger.yaml

@@ -120,6 +120,12 @@ definitions:
     type: object
   exam.EndExamResponse:
     properties:
+      info:
+        type: string
+      is_driver:
+        type: boolean
+      is_passed:
+        type: boolean
       score:
         type: integer
     type: object

backend/models/exam.go

@@ -7,8 +7,8 @@ import (
 
 type Exam struct {
 	ID          int            `json:"id" gorm:"primaryKey"`
-	Title       string         `json:"title" gorm:"default:'exam'"`
-	Description string         `json:"description" gorm:"default:'description'"`
+	Title       string         `json:"title" gorm:"default:'default exam title'"`
+	Description string         `json:"description" gorm:"default:'default exam description'"`
 	StartTime   *MyTime        `json:"start_time"`
 	EndTime     *MyTime        `json:"end_time"`
 	Duration    time.Duration  `json:"duration"`

backend/models/user.go

@@ -100,3 +100,33 @@ func DeleteUserByID(id int) error {
 		return nil
 	})
 }
+
+func ValidateDriver(id int) (bool, error) {
+	var user User
+	err := DB.Take(&user, id).Error
+	if err != nil {
+		return false, err
+	}
+	return user.Age >= 18 && user.IsPassed, nil
+}
+
+func (user *User) AddPunishment(score int) error {
+	return DB.Transaction(func(tx *gorm.DB) error {
+		var user User
+		err := tx.Take(&user, user.ID).Error
+		if err != nil {
+			return err
+		}
+		if user.Point <= score {
+			user.Point = 0
+			user.IsPassed = false
+		} else {
+			user.Point -= score
+		}
+		return tx.Model(&user).Select("Point", "IsPassed").Updates(&user).Error
+	})
+}
+
+func (user *User) ValidateDriver() bool {
+	return user.Age >= 18 && user.IsPassed
+}