add question to contest and create contest completed

Author: krishkumar84

cmd/portal-api/main.go

@@ -76,6 +76,7 @@ router.Handle("GET /api/admin/test",
 	router.HandleFunc("GET /api/contest",contest.GetAllContests(storage))
 	router.HandleFunc("GET /api/contest/{id}",contest.GetContestById(storage))
 	router.HandleFunc("GET /api/question/{id}",question.GetQuestionById(storage))
+	router.HandleFunc("POST /api/contest/{id}/question", contest.AddQuestionToContest(storage))
 
 	//start server
 

pkg/http/handler/contest/contest.go

@@ -63,4 +63,37 @@ func GetContestById(storage storage.Storage) http.HandlerFunc {
 		}
 		response.WriteJson(w, http.StatusOK, contest)
 	}
+}
+
+func AddQuestionToContest(storage storage.Storage) http.HandlerFunc {
+	return func(w http.ResponseWriter, r *http.Request) {
+		path := r.URL.Path
+		parts := strings.Split(path, "/")
+		if len(parts) < 5 {
+			response.WriteJson(w, http.StatusBadRequest, response.GeneralError(fmt.Errorf("invalid URL format")))
+			return
+		}
+		contestId := parts[3]
+		
+		fmt.Printf("Extracted contest ID: %s\n", contestId)
+		
+		if contestId == "" {
+			response.WriteJson(w, http.StatusBadRequest, response.GeneralError(fmt.Errorf("contest id is required")))
+			return
+		}
+
+		var question types.Question
+		if err := json.NewDecoder(r.Body).Decode(&question); err != nil {
+			response.WriteJson(w, http.StatusBadRequest, response.GeneralError(err))
+			return
+		}
+
+		questionId, err := storage.AddQuestionToContest(contestId, question)
+		if err != nil {
+			response.WriteJson(w, http.StatusInternalServerError, response.GeneralError(err))
+			return
+		}
+
+		response.WriteJson(w, http.StatusCreated, map[string]string{"question_id": questionId})
+	}
 }

pkg/storage/mongodb/mongodb.go

@@ -313,4 +313,39 @@ func (m *MongoDB) GetQuestionById(id string) ([]bson.M, error) {
     }
 
     return results, nil
+}
+
+func (m *MongoDB) AddQuestionToContest(contestId string, question types.Question) (string, error) {
+    ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
+    defer cancel()
+
+    fmt.Printf("Received contest ID: %s\n", contestId)
+
+    question.TestCaseIDs = []string{}
+
+    //Create question
+    questionResult, err := m.db.Collection("questions").InsertOne(ctx, question)
+    if err != nil {
+        return "", fmt.Errorf("failed to create question: %v", err)
+    }
+    questionId := questionResult.InsertedID.(primitive.ObjectID).Hex()
+
+    fmt.Printf("Created question with ID: %s\n", questionId)
+
+    contestObjID, err := primitive.ObjectIDFromHex(contestId)
+    if err != nil {
+        return "", fmt.Errorf("invalid contest id format: %v", err)
+    }
+
+    filter := bson.M{"_id": contestObjID}
+    update := bson.M{"$push": bson.M{"question_ids": questionId}}
+    
+    result, err := m.db.Collection("contests").UpdateOne(ctx, filter, update)
+    if err != nil {
+        return "", fmt.Errorf("failed to update contest: %v", err)
+    }
+
+    fmt.Printf("Updated contest. Modified count: %d\n", result.ModifiedCount)
+
+    return questionId, nil
 }

pkg/storage/storage.go

@@ -18,4 +18,5 @@ type Storage interface {
 	GetAllContests() ([]types.ContestBasicInfo, error)
 	GetContestById(id string) ([]bson.M, error)
 	GetQuestionById(id string) ([]bson.M, error)
+	AddQuestionToContest(contestId string, question types.Question) (string, error)
 }