Retrieving questions by user id and function to get user id

Author: BrainNotFoundException

pkg/controllers/questions.go

@@ -4,6 +4,7 @@ import (
 	"net/http"
 
 	"github.com/CodeChefVIT/cookoff-10.0-be/pkg/db"
+	"github.com/CodeChefVIT/cookoff-10.0-be/pkg/helpers/auth"
 	"github.com/CodeChefVIT/cookoff-10.0-be/pkg/utils"
 	"github.com/google/uuid"
 	"github.com/labstack/echo/v4"
@@ -67,6 +68,70 @@ func GetAllQuestions(c echo.Context) error {
 	})
 }
 
+func GetQuestionsByRound(c echo.Context) error {
+
+	userID, err := auth.GetUserID(c)
+
+	if err != nil {
+		return c.JSON(http.StatusBadRequest, echo.Map{
+			"status":  "Failed",
+			"message": "Error Getting user_id",
+			"error":   err.Error(),
+		})
+	}
+
+	round, err := utils.Queries.GetUserRound(c.Request().Context(), userID)
+
+	if err != nil {
+		return c.JSON(http.StatusInternalServerError, echo.Map{
+			"status":  "Failed",
+			"message": "could not get the users current round",
+			"error":   err.Error(),
+		})
+	}
+
+	if round < 0 || round > 3 {
+		return c.JSON(http.StatusBadRequest, echo.Map{
+			"status":  "Failed",
+			"message": "round number is invalid",
+		})
+	}
+
+	questions, err := utils.Queries.GetQuestionsByRound(c.Request().Context(), int32(round))
+
+	if err != nil {
+		return c.JSON(http.StatusInternalServerError, echo.Map{
+			"status":  "Failed",
+			"message": "unable to fetch the questions",
+			"error":   err.Error(),
+		})
+	}
+
+	result := []echo.Map{}
+
+	for _, q := range questions {
+		testcases, err := utils.Queries.GetTestCasesByQuestion(c.Request().Context(), q.ID)
+		if err != nil {
+			return c.JSON(http.StatusInternalServerError, echo.Map{
+				"status":      "Failed",
+				"message":     "Could not get the testcases for question",
+				"question_id": q.ID,
+				"error":       err.Error(),
+			})
+		}
+		result = append(result, echo.Map{
+			"question":  q,
+			"testcases": testcases,
+		})
+	}
+
+	return c.JSON(http.StatusOK, echo.Map{
+		"status":              "success",
+		"round":               round,
+		"questions_testcases": result,
+	})
+}
+
 func UpdateQuestion(c echo.Context) error {
 	id, err := uuid.Parse(c.Param("id"))
 	if err != nil {

pkg/helpers/auth/jwt.go

@@ -1,31 +1,34 @@
 package auth
 
 import (
+	"fmt"
 	"time"
 
 	"github.com/CodeChefVIT/cookoff-10.0-be/pkg/db"
 	"github.com/CodeChefVIT/cookoff-10.0-be/pkg/utils"
 	"github.com/golang-jwt/jwt/v5"
+	"github.com/google/uuid"
+	"github.com/labstack/echo/v4"
 )
 
 var jwtSecret = []byte(utils.Config.JwtSecret)
 
 type AccessTokenClaims struct {
 	Username string `json:"username"`
-	UserID   string  `json:"user_id"`
+	UserID   string `json:"user_id"`
 	Role     string `json:"role"`
 	Type     string `json:"type"`
 	jwt.RegisteredClaims
 }
 
 type RefreshTokenClaims struct {
-	UserID   string  `json:"user_id"`
-	Type     string `json:"type"`
+	UserID string `json:"user_id"`
+	Type   string `json:"type"`
 	jwt.RegisteredClaims
 }
 
 func CreateAccessToken(user *db.User) (string, error) {
-	expirationTime := time.Now().Add(1*time.Minute)
+	expirationTime := time.Now().Add(1 * time.Minute)
 	claims := &AccessTokenClaims{
 		Username: user.Name,
 		UserID:   user.ID.String(),
@@ -53,3 +56,20 @@ func CreateRefreshToken(user *db.User) (string, error) {
 	token := jwt.NewWithClaims(jwt.SigningMethodHS256, claims)
 	return token.SignedString(jwtSecret)
 }
+
+func GetUserID(c echo.Context) (uuid.UUID, error) {
+	user := c.Get("user").(*jwt.Token)
+	claims := user.Claims.(jwt.MapClaims)
+
+	userIDStr, ok := claims["user_id"].(string)
+	if !ok {
+		return uuid.UUID{}, fmt.Errorf("user_id not found in token")
+	}
+
+	uid, err := uuid.Parse(userIDStr)
+	if err != nil {
+		return uuid.UUID{}, fmt.Errorf("invalid user_id format: %v", err)
+	}
+
+	return uid, nil
+}