add validator to PATCH

traPtitech · Oct 24, 2023 · cc7b5dd · cc7b5dd
1 parent fa55f46
commit cc7b5dd
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/router/questions.go b/router/questions.go
@@ -61,6 +61,18 @@ func (q *Question) EditQuestion(c echo.Context) error {
 		return echo.NewHTTPError(http.StatusBadRequest)
 	}
 
+	validate, err := getValidator(c)
+	if err != nil {
+		c.Logger().Errorf("failed to get validator: %+v", err)
+		return echo.NewHTTPError(http.StatusInternalServerError, err)
+	}
+
+	err = validate.Struct(req)
+	if err != nil {
+		c.Logger().Infof("validation failed: %+v", err)
+		return echo.NewHTTPError(http.StatusBadRequest, err)
+	}
+
 	switch req.QuestionType {
 	case "Text":
 		//正規表現のチェック

diff --git a/router/responses.go b/router/responses.go
@@ -242,6 +242,18 @@ func (r *Response) EditResponse(c echo.Context) error {
 		return echo.NewHTTPError(http.StatusBadRequest)
 	}
 
+	validate, err := getValidator(c)
+	if err != nil {
+		c.Logger().Errorf("failed to get validator: %+v", err)
+		return echo.NewHTTPError(http.StatusInternalServerError, err)
+	}
+
+	err = validate.Struct(req)
+	if err != nil {
+		c.Logger().Infof("validation failed: %+v", err)
+		return echo.NewHTTPError(http.StatusBadRequest, err)
+	}
+
 	limit, err := r.GetQuestionnaireLimit(c.Request().Context(), req.ID)
 	if err != nil {
 		if errors.Is(err, model.ErrRecordNotFound) {