[시온] step6. 페이지 나누기

build.gradle

@@ -21,6 +21,9 @@ dependencies {
 
 	// https://mvnrepository.com/artifact/org.springframework.boot/spring-boot-starter-data-jpa
 	implementation group: 'org.springframework.boot', name: 'spring-boot-starter-data-jpa', version: '2.4.3'
+
+	compile 'io.springfox:springfox-swagger2:2.6.1'
+	compile 'io.springfox:springfox-swagger-ui:2.6.1'
 }
 
 test {

src/main/java/com/codessquad/qna/QnaApplication.java

@@ -2,12 +2,39 @@
 
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.Bean;
+import org.springframework.data.jpa.repository.config.EnableJpaAuditing;
+import springfox.documentation.builders.ApiInfoBuilder;
+import springfox.documentation.builders.PathSelectors;
+import springfox.documentation.builders.RequestHandlerSelectors;
+import springfox.documentation.service.ApiInfo;
+import springfox.documentation.spi.DocumentationType;
+import springfox.documentation.spring.web.plugins.Docket;
+import springfox.documentation.swagger2.annotations.EnableSwagger2;
 
+@EnableSwagger2
+@EnableJpaAuditing
 @SpringBootApplication
 public class QnaApplication {
 
 	public static void main(String[] args) {
 		SpringApplication.run(QnaApplication.class, args);
 	}
 
+	private ApiInfo apiInfo() {
+		return new ApiInfoBuilder()
+				.title("spring-boot-qna")
+				.description("spring-boot-qna")
+				.build();
+	}
+
+	@Bean
+	public Docket api() {
+		return new Docket(DocumentationType.SWAGGER_2)
+				.groupName("spring-boot-qna")
+				.apiInfo(this.apiInfo())
+				.select()
+				.paths(PathSelectors.ant("/api/**"))
+				.build();
+	}
 }

src/main/java/com/codessquad/qna/controller/AnswerController.java

@@ -3,6 +3,7 @@
 import com.codessquad.qna.domain.answer.Answer;
 import com.codessquad.qna.domain.question.Question;
 import com.codessquad.qna.domain.user.User;
+import com.codessquad.qna.exception.NotAuthorizationException;
 import com.codessquad.qna.service.AnswerService;
 import com.codessquad.qna.service.QuestionService;
 import com.codessquad.qna.utils.HttpSessionUtils;
@@ -26,27 +27,17 @@ public AnswerController(AnswerService answerService, QuestionService questionSer
 
     @PostMapping("/")
     public String createAnswer(@PathVariable Long questionId, Answer answer, HttpSession session) {
-        if (!HttpSessionUtils.isLoginUser(session)) {
-            return "redirect:/users/login";
-        }
         User sessionedUser = HttpSessionUtils.getUserFromSession(session);
-        answer.setWriter(sessionedUser);
-        answerService.create(questionId, answer);
+        answerService.create(questionId, answer, sessionedUser);
         return "redirect:/questions/" + questionId;
     }
 
     @GetMapping("/{id}/form")
     public String getUpdateForm(@PathVariable Long questionId, @PathVariable Long id, Model model, HttpSession session) {
-        if (!HttpSessionUtils.isLoginUser(session)) {
-            return "redirect:/users/login";
-        }
         Question question = questionService.findById(questionId);
         Answer answer = answerService.findById(id);
         User sessionedUser = HttpSessionUtils.getUserFromSession(session);
-        if (!answer.isWrittenBy(sessionedUser)) {
-            throw new IllegalStateException("자신이 작성한 답변만 수정할 수 있습니다.");
-        }
-
+        answer.isWrittenBy(sessionedUser);
         model.addAttribute("question", question);
         model.addAttribute("answer", answer);
 
@@ -62,15 +53,8 @@ public String updateAnswer(@PathVariable Long questionId, @PathVariable Long id,
 
     @DeleteMapping("/{id}")
     public String deleteAnswer(@PathVariable Long questionId, @PathVariable Long id, HttpSession session) {
-        if (!HttpSessionUtils.isLoginUser(session)) {
-            return "redirect:/users/login";
-        }
-        Answer answer = answerService.findById(id);
         User sessionedUser = HttpSessionUtils.getUserFromSession(session);
-        if (!answer.isWrittenBy(sessionedUser)) {
-            throw new IllegalStateException("자신이 작성한 답변만 삭제할 수 있습니다.");
-        }
-        answerService.deleteById(id);
+        answerService.deleteById(id, sessionedUser);
 
         return "redirect:/questions/" + questionId;
     }

src/main/java/com/codessquad/qna/controller/ApiAnswerController.java

@@ -0,0 +1,34 @@
+package com.codessquad.qna.controller;
+
+import com.codessquad.qna.domain.answer.Answer;
+import com.codessquad.qna.domain.user.User;
+import com.codessquad.qna.service.AnswerService;
+import com.codessquad.qna.utils.HttpSessionUtils;
+import org.springframework.web.bind.annotation.*;
+
+import javax.servlet.http.HttpSession;
+
+@RequestMapping("/api/questions/{questionId}/answers")
+@RestController
+public class ApiAnswerController {
+
+    private final AnswerService answerService;
+
+    public ApiAnswerController(AnswerService answerService) {
+        this.answerService = answerService;
+    }
+
+    @PostMapping("/")
+    public Answer createAnswer(@PathVariable Long questionId, Answer answer, HttpSession session) {
+        User sessionedUser = HttpSessionUtils.getUserFromSession(session);
+        return answerService.create(questionId, answer, sessionedUser);
+    }
+
+    @DeleteMapping("/{id}")
+    public Answer delete(@PathVariable Long id, HttpSession session) {
+        Answer answer = answerService.findById(id);
+        User loginUser = HttpSessionUtils.getUserFromSession(session);
+        answerService.deleteById(id, loginUser);
+        return answer;
+    }
+}

src/main/java/com/codessquad/qna/controller/ApiQuestionController.java

@@ -0,0 +1,25 @@
+package com.codessquad.qna.controller;
+
+import com.codessquad.qna.domain.pagination.Criteria;
+import com.codessquad.qna.domain.question.Question;
+import com.codessquad.qna.service.QuestionService;
+import org.springframework.data.domain.Page;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RequestMapping("/api/questions")
+@RestController
+public class ApiQuestionController {
+
+    private final QuestionService questionService;
+
+    public ApiQuestionController(QuestionService questionService) {
+        this.questionService = questionService;
+    }
+
+    @GetMapping("/")
+    public Page<Question> pagingList(Criteria criteria) {
+        return questionService.pagingList(criteria);
+    }
+}

src/main/java/com/codessquad/qna/controller/ApiUserController.java

@@ -0,0 +1,24 @@
+package com.codessquad.qna.controller;
+
+import com.codessquad.qna.domain.user.User;
+import com.codessquad.qna.service.UserService;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.PathVariable;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+@RequestMapping("/api/users")
+@RestController
+public class ApiUserController {
+
+    private final UserService userService;
+
+    public ApiUserController(UserService userService) {
+        this.userService = userService;
+    }
+
+    @GetMapping("{id}")
+    public User show(@PathVariable Long id) {
+        return userService.findById(id);
+    }
+}

src/main/java/com/codessquad/qna/controller/IndexController.java

@@ -1,22 +1,31 @@
 package com.codessquad.qna.controller;
 
+import com.codessquad.qna.domain.pagination.Criteria;
+import com.codessquad.qna.domain.pagination.PageDTO;
+import com.codessquad.qna.domain.question.Question;
 import com.codessquad.qna.service.QuestionService;
+import org.springframework.data.domain.Page;
 import org.springframework.stereotype.Controller;
 import org.springframework.ui.Model;
 import org.springframework.web.bind.annotation.GetMapping;
 
+import java.util.Optional;
+
 @Controller
 public class IndexController {
 
     private final QuestionService questionService;
 
-    public IndexController (QuestionService questionService) {
+    public IndexController(QuestionService questionService) {
         this.questionService = questionService;
     }
 
     @GetMapping("/")
-    public String goMain(Model model) {
-        model.addAttribute("questions", questionService.findAll());
+    public String goMain(Optional<Integer> pageNum, Model model) { // 메인화면(Url 매핑 -> "/") 진입 시 1페이지로 설정
+        Criteria criteria = new Criteria(pageNum.orElse(1));
+        Page<Question> questions = questionService.pagingList(criteria);
+        model.addAttribute("pageMaker", new PageDTO(questions));
+        model.addAttribute("questions", questions);
         return "index";
     }
 }

src/main/java/com/codessquad/qna/controller/QuestionController.java

@@ -3,6 +3,7 @@
 import com.codessquad.qna.domain.answer.Answer;
 import com.codessquad.qna.domain.question.Question;
 import com.codessquad.qna.domain.user.User;
+import com.codessquad.qna.exception.NotAuthorizationException;
 import com.codessquad.qna.service.AnswerService;
 import com.codessquad.qna.service.QuestionService;
 import com.codessquad.qna.utils.HttpSessionUtils;
@@ -27,9 +28,7 @@ public QuestionController(QuestionService questionService, AnswerService answerS
 
     @GetMapping("/form")
     public String getQuestionForm(HttpSession session) {
-        if (!HttpSessionUtils.isLoginUser(session)) {
-            return "redirect:/users/login";
-        }
+        HttpSessionUtils.isLoginUser(session);
         return "/qna/form";
     }
 
@@ -45,23 +44,16 @@ public String createQuestion(Question question, HttpSession session) {
     public String getQuestion(@PathVariable Long id, Model model) {
         Question question = questionService.findById(id);
         List<Answer> answers = answerService.findAllByQuestionId(id);
-
         model.addAttribute("question", question);
         model.addAttribute("answers", answers);
-
         return "/qna/show";
     }
 
     @GetMapping("/{id}/form")
     public String getUpdateForm(@PathVariable Long id, HttpSession session, Model model) {
-        if (!HttpSessionUtils.isLoginUser(session)) {
-            return "redirect:/users/login";
-        }
         Question question = questionService.findById(id);
         User sessionedUser = HttpSessionUtils.getUserFromSession(session);
-        if (!question.isWrittenBy(sessionedUser)) {
-            throw new IllegalStateException("자신이 작성한 글만 수정할 수 있습니다.");
-        }
+        question.isWrittenBy(sessionedUser);
         model.addAttribute("question", question);
         return "/qna/update_form";
     }
@@ -74,14 +66,9 @@ public String updateQuestion(@PathVariable Long id, Question questionWithUpdated
 
     @DeleteMapping("/{id}")
     public String deleteQuestion(@PathVariable Long id, HttpSession session) {
-        if (!HttpSessionUtils.isLoginUser(session)) {
-            return "redirect:/users/login";
-        }
         Question question = questionService.findById(id);
         User sessionedUser = HttpSessionUtils.getUserFromSession(session);
-        if (!question.isWrittenBy(sessionedUser)) {
-            throw new IllegalStateException("자신이 작성한 글만 삭제할 수 있습니다.");
-        }
+        question.isWrittenBy(sessionedUser);
         questionService.deleteById(id);
         return "redirect:/";
     }

src/main/java/com/codessquad/qna/controller/UserController.java

@@ -1,6 +1,7 @@
 package com.codessquad.qna.controller;
 
 import com.codessquad.qna.domain.user.User;
+import com.codessquad.qna.exception.NotAuthorizationException;
 import com.codessquad.qna.service.UserService;
 import com.codessquad.qna.utils.HttpSessionUtils;
 import org.springframework.stereotype.Controller;
@@ -40,13 +41,8 @@ public String getProfile(@PathVariable Long id, Model model) {
 
     @GetMapping("/{id}/form")
     public String getUpdateForm(@PathVariable Long id, Model model, HttpSession session) {
-        if (!HttpSessionUtils.isLoginUser(session)) {
-            return "redirect:/users/login";
-        }
         User sessionedUser = HttpSessionUtils.getUserFromSession(session);
-        if (!sessionedUser.isYourId(id)) {
-            throw new IllegalStateException("자신의 정보만 수정할 수 있습니다.");
-        }
+        sessionedUser.isYourId(id);
         User user = userService.findById(id);
         model.addAttribute("user", user);
         return "/users/updateForm";

src/main/java/com/codessquad/qna/controller/advice/ControllerExceptionHandler.java

@@ -0,0 +1,56 @@
+package com.codessquad.qna.controller.advice;
+
+import com.codessquad.qna.exception.*;
+import org.springframework.http.HttpStatus;
+import org.springframework.ui.Model;
+import org.springframework.web.bind.annotation.ControllerAdvice;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+import org.springframework.web.bind.annotation.ResponseStatus;
+
+@ControllerAdvice
+public class ControllerExceptionHandler {
+
+    @ResponseStatus(HttpStatus.UNAUTHORIZED)
+    @ExceptionHandler(NotAuthorizationException.class)
+    public String handleNotAuthorization(
+            NotAuthorizationException notAuthorizationException, Model model) {
+        model.addAttribute("errorMessage", notAuthorizationException.getMessage());
+        return "error/401";
+    }
+
+    @ResponseStatus(HttpStatus.BAD_REQUEST)
+    @ExceptionHandler(AnotherAnswerException.class)
+    public String HandleAnotherAnswer(AnotherAnswerException anotherAnswerException, Model model) {
+        model.addAttribute("errorMessage", anotherAnswerException.getMessage());
+        return "/error/400";
+    }
+
+    @ResponseStatus(HttpStatus.BAD_REQUEST)
+    @ExceptionHandler(WrongPasswordException.class)
+    public String handleWrongPassword(WrongPasswordException wrongPasswordException, Model model) {
+        model.addAttribute("errorMessage", wrongPasswordException.getMessage());
+        return "/error/400";
+    }
+
+    @ResponseStatus(HttpStatus.NOT_FOUND)
+    @ExceptionHandler(UserNotFoundException.class)
+    public String handleUserNotFound(UserNotFoundException userNotFoundException, Model model) {
+        model.addAttribute("errorMessage", userNotFoundException.getMessage());
+        return "/error/404";
+    }
+
+    @ResponseStatus(HttpStatus.NOT_FOUND)
+    @ExceptionHandler(AnswerNotFoundException.class)
+    public String handleAnswerNotFound(AnswerNotFoundException answerNotFoundException, Model model) {
+        model.addAttribute("errorMessage", answerNotFoundException.getMessage());
+        return "/error/404";
+    }
+
+    @ResponseStatus(HttpStatus.NOT_FOUND)
+    @ExceptionHandler(QuestionNotFoundException.class)
+    public String handleNotFound(QuestionNotFoundException QuestionNotFoundException, Model model) {
+        model.addAttribute("errorMessage", QuestionNotFoundException.getMessage());
+        return "/error/404";
+    }
+
+}