[장익환] WAS 과제 7단계 - 이미지 업로드 구현

.gitignore

@@ -197,3 +197,5 @@ gradle-app.setting
 *.hprof
 
 # End of https://www.toptal.com/developers/gitignore/api/gradle,intellij,java,macos
+
+**/uploads

src/main/java/db/ArticleRepositoryImpl.java

@@ -15,14 +15,15 @@ public class ArticleRepositoryImpl implements ArticleRepository {
     private static final Logger logger = LoggerFactory.getLogger(ArticleRepositoryImpl.class);
 
     public Article save(Article article) {
-        String sql = "insert into article_tbl(creatorId, title, content) values(?, ?, ?)";
+        String sql = "insert into article_tbl(creatorId, title, content, image_url) values(?, ?, ?, ?)";
         try (
                 Connection con = DriverManager.getConnection(JDBC_URL, USER, PASSWORD);
                 PreparedStatement pstmt = con.prepareStatement(sql, Statement.RETURN_GENERATED_KEYS);
         ) {
             pstmt.setLong(1, article.getCreatorId());
             pstmt.setString(2, article.getTitle());
             pstmt.setString(3, article.getContent());
+            pstmt.setString(4, article.getImageUrl());
 
             pstmt.executeUpdate();
 
@@ -41,7 +42,7 @@ public Article save(Article article) {
 
     @Override
     public List<Article> findTopNLessThanByIdDecreasingOrder(int limit, long id) {
-        String sql = "select id, creatorId, title, content from article_tbl where id < ? order by id desc limit ?";
+        String sql = "select id, creatorId, title, content, image_url from article_tbl where id < ? order by id desc limit ?";
 
         try (
                 Connection con = DriverManager.getConnection(JDBC_URL, USER, PASSWORD);
@@ -58,7 +59,8 @@ public List<Article> findTopNLessThanByIdDecreasingOrder(int limit, long id) {
                     long creatorId = rs.getLong("creatorId");
                     String title = rs.getString("title");
                     String content = rs.getString("content");
-                    articles.add(new Article(articleId, creatorId, title, content));
+                    String imageUrl = rs.getString("image_url");
+                    articles.add(new Article(articleId, creatorId, title, content, imageUrl));
                 }
             }
             return articles;

src/main/java/db/UserRepository.java

@@ -9,4 +9,5 @@ public interface UserRepository {
     User save(User user);
     Optional<User> findByEmail(String email);
     Optional<User> findById(Long id);
+    User update(User user);
 }

src/main/java/db/UserRepositoryImpl.java

@@ -14,7 +14,7 @@ public class UserRepositoryImpl implements UserRepository {
     private static final Logger logger = LoggerFactory.getLogger(UserRepository.class);
 
     public User save(User user) {
-        String sql = "insert into user_tbl(password, nickname, email) values(?, ?, ?)";
+        String sql = "insert into user_tbl(password, nickname, email, image_url) values(?, ?, ?)";
 
         try (
                 Connection con = DriverManager.getConnection(JDBC_URL, USER, PASSWORD);
@@ -25,6 +25,7 @@ public User save(User user) {
             pstmt.setString(1, user.getPassword());
             pstmt.setString(2, user.getName());
             pstmt.setString(3, user.getEmail());
+            pstmt.setString(4, user.getImageUrl());
 
             pstmt.executeUpdate();
 
@@ -43,7 +44,7 @@ public User save(User user) {
     }
 
     public Optional<User> findByEmail(String email) {
-        String sql = "select id, password, nickname from user_tbl where email = ?";
+        String sql = "select id, password, nickname, image_url from user_tbl where email = ?";
 
         try (
                 Connection con = DriverManager.getConnection(JDBC_URL, USER, PASSWORD);
@@ -61,7 +62,8 @@ public Optional<User> findByEmail(String email) {
                 long id = rs.getLong("id");
                 String password = rs.getString("password");
                 String nickname = rs.getString("nickname");
-                user = new User(id, password, nickname, email);
+                String imageUrl = rs.getString("image_url");
+                user = new User(id, password, nickname, email, imageUrl);
             }
             return Optional.of(user);
 
@@ -73,7 +75,7 @@ public Optional<User> findByEmail(String email) {
 
     @Override
     public Optional<User> findById(Long id) {
-        String sql = "select password, nickname, email from user_tbl where id = ?";
+        String sql = "select password, nickname, email, image_url from user_tbl where id = ?";
 
         try (
                 Connection con = DriverManager.getConnection(JDBC_URL, USER, PASSWORD);
@@ -91,7 +93,8 @@ public Optional<User> findById(Long id) {
                 String password = rs.getString("password");
                 String nickname = rs.getString("nickname");
                 String email = rs.getString("email");
-                user = new User(id, password, nickname, email);
+                String imageUrl = rs.getString("image_url");
+                user = new User(id, password, nickname, email, imageUrl);
             }
             return Optional.of(user);
 
@@ -100,4 +103,34 @@ public Optional<User> findById(Long id) {
             throw new RuntimeException(e);
         }
     }
+
+    @Override
+    public User update(User user) {
+        String sql = "update user_tbl set password = ?, nickname = ?, email = ?, image_url = ? where id = ?";
+
+        try (
+                Connection con = DriverManager.getConnection(JDBC_URL, USER, PASSWORD);
+                PreparedStatement pstmt = con.prepareStatement(sql);
+        ) {
+            logger.debug("DB 연결 성공");
+
+            pstmt.setString(1, user.getPassword());
+            pstmt.setString(2, user.getName());
+            pstmt.setString(3, user.getEmail());
+            pstmt.setString(4, user.getImageUrl());
+            pstmt.setLong(5, user.getId());
+
+            int updatedRows = pstmt.executeUpdate();
+
+            if (updatedRows == 0) {
+                throw new SQLException("업데이트 실패: 해당 id 없음");
+            }
+
+            return user;
+        } catch (SQLException e) {
+            logger.error("DB 업데이트 실패", e);
+            throw new RuntimeException(e);
+        }
+    }
+
 }

src/main/java/model/Article.java

@@ -6,18 +6,21 @@ public class Article {
     private String writerName;
     private String title;
     private String content;
+    private String imageUrl;
 
-    public Article(Long id, Long creatorId, String title, String content) {
+    public Article(Long id, Long creatorId, String title, String content, String imageUrl) {
         this.id = id;
         this.creatorId = creatorId;
         this.title = title;
         this.content = content;
+        this.imageUrl = imageUrl;
     }
 
-    public Article(Long creatorId, String title, String content) {
+    public Article(Long creatorId, String title, String content, String imageUrl) {
         this.creatorId = creatorId;
         this.title = title;
         this.content = content;
+        this.imageUrl = imageUrl;
     }
 
     public Long getId() {
@@ -40,6 +43,10 @@ public String getContent() {
         return content;
     }
 
+    public String getImageUrl() {
+        return imageUrl;
+    }
+
     public void setId(Long id) {
         this.id = id;
     }

src/main/java/model/User.java

@@ -7,12 +7,14 @@ public class User {
     private String password;
     private String name;
     private String email;
+    private String imageUrl;
 
-    public User(Long id, String password, String name, String email) {
+    public User(Long id, String password, String name, String email, String imageUrl) {
         this.id = id;
         this.password = password;
         this.name = name;
         this.email = email;
+        this.imageUrl = imageUrl;
     }
 
     public User(String password, String name, String email) {
@@ -51,4 +53,15 @@ public void setId(Long id) {
     public String toString() {
         return "User [userId=" + id + ", password=" + password + ", name=" + name + ", email=" + email + "]";
     }
+
+    public void changeProfileImage(String imageUrl) {
+        if(imageUrl == null){
+            throw new BusinessException();
+        }
+        this.imageUrl = imageUrl;
+    }
+
+    public String getImageUrl() {
+        return imageUrl;
+    }
 }

src/main/java/webserver/handler/CreateArticleFormHandler.java

@@ -4,8 +4,8 @@
 import webserver.http.Response;
 import webserver.mvc.Handler;
 import webserver.mvc.ModelAndView;
-import webserver.mvc.RedirectView;
-import webserver.mvc.StaticResourceView;
+import webserver.view.RedirectView;
+import webserver.view.StaticResourceView;
 import webserver.util.AuthUtil;
 
 public class CreateArticleFormHandler implements Handler {

src/main/java/webserver/handler/CreateArticleHandler.java

@@ -5,10 +5,14 @@
 import model.User;
 import webserver.http.Request;
 import webserver.http.Response;
-import webserver.mvc.Handler;
-import webserver.mvc.ModelAndView;
-import webserver.mvc.RedirectView;
+import webserver.mvc.*;
 import webserver.util.AuthUtil;
+import webserver.mvc.ModelAndView;
+import webserver.util.MultipartFileUtil;
+import webserver.view.RedirectView;
+import webserver.view.StaticResourceView;
+
+import java.io.IOException;
 
 public class CreateArticleHandler implements Handler {
     private final ArticleRepository articleRepository;
@@ -26,7 +30,20 @@ public ModelAndView handle(Request request, Response response) {
         String title = request.getParameter("title");
         String content = request.getParameter("content");
 
-        articleRepository.save(new Article(user.getId(), title, content));
+        if(request.getMultipartFiles().size() != 1){
+            return new StaticResourceView("/article/error.html");
+        }
+
+        MultipartFile multipartFile = request.getMultipartFiles().get(0);
+
+        String imageUrl = null;
+        try {
+            imageUrl = MultipartFileUtil.saveFile("uploads/images", multipartFile);
+        } catch (IOException e) {
+            throw new RuntimeException(e);
+        }
+
+        articleRepository.save(new Article(user.getId(), title, content, imageUrl));
 
         return new RedirectView("/");
     }

src/main/java/webserver/handler/CreateUserHandler.java

@@ -9,7 +9,7 @@
 import webserver.http.Response;
 import webserver.mvc.Handler;
 import webserver.mvc.ModelAndView;
-import webserver.mvc.RedirectView;
+import webserver.view.RedirectView;
 
 public class CreateUserHandler implements Handler {
     private static final Logger logger = LoggerFactory.getLogger(CreateUserHandler.class);

src/main/java/webserver/handler/GetImageHandler.java

@@ -0,0 +1,19 @@
+package webserver.handler;
+
+import webserver.exception.StaticResourceNotFoundException;
+import webserver.http.Request;
+import webserver.http.Response;
+import webserver.mvc.Handler;
+import webserver.view.UploadedFileView;
+import webserver.mvc.ModelAndView;
+
+public class GetImageHandler implements Handler {
+    @Override
+    public ModelAndView handle(Request request, Response response) {
+        String imageFileUrl = request.getParameter("imageUrl");
+        if(imageFileUrl == null){
+            throw new StaticResourceNotFoundException();
+        }
+        return new UploadedFileView("/uploads/images/" + imageFileUrl);
+    }
+}

src/main/java/webserver/handler/LoginFormHandler.java

@@ -6,7 +6,7 @@
 import webserver.http.Response;
 import webserver.mvc.Handler;
 import webserver.mvc.ModelAndView;
-import webserver.mvc.StaticResourceView;
+import webserver.view.StaticResourceView;
 
 public class LoginFormHandler implements Handler {
     private static final Logger logger = LoggerFactory.getLogger(LoginFormHandler.class);

src/main/java/webserver/handler/LoginHandler.java

@@ -1,15 +1,14 @@
 package webserver.handler;
 
-import db.Database;
 import db.UserRepository;
 import model.User;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import webserver.http.Request;
 import webserver.http.Response;
 import webserver.mvc.Handler;
-import webserver.mvc.RedirectView;
-import webserver.mvc.StaticResourceView;
+import webserver.view.RedirectView;
+import webserver.view.StaticResourceView;
 import webserver.session.SessionStore;
 import webserver.mvc.ModelAndView;
 

src/main/java/webserver/handler/LogoutHandler.java

@@ -5,7 +5,7 @@
 import webserver.http.Request;
 import webserver.http.Response;
 import webserver.mvc.Handler;
-import webserver.mvc.RedirectView;
+import webserver.view.RedirectView;
 import webserver.util.AuthUtil;
 import webserver.session.SessionStore;
 import webserver.mvc.ModelAndView;

src/main/java/webserver/handler/MainHandler.java

@@ -10,6 +10,8 @@
 import webserver.http.Response;
 import webserver.mvc.*;
 import webserver.util.AuthUtil;
+import webserver.view.MainPageDynamicView;
+import webserver.mvc.ModelAndView;
 
 import java.util.HashMap;
 import java.util.List;

src/main/java/webserver/handler/MyPageHandler.java

@@ -1,19 +1,42 @@
 package webserver.handler;
 
+import db.UserRepository;
+import model.User;
+import webserver.exception.BusinessException;
 import webserver.http.Request;
 import webserver.http.Response;
-import webserver.mvc.Handler;
-import webserver.mvc.RedirectView;
-import webserver.mvc.StaticResourceView;
-import webserver.util.AuthUtil;
+import webserver.mvc.*;
 import webserver.mvc.ModelAndView;
+import webserver.view.MyPageDynamicView;
+import webserver.util.AuthUtil;
+import webserver.view.RedirectView;
+
+import java.util.HashMap;
 
 public class MyPageHandler implements Handler {
+    private final UserRepository userRepository;
+
+    public MyPageHandler(UserRepository userRepository) {
+        this.userRepository = userRepository;
+    }
+
     @Override
     public ModelAndView handle(Request request, Response response) {
         if(!AuthUtil.isAuthenticatedUser(request)){
             return new RedirectView("/login");
         }
-        return new StaticResourceView("/mypage/index.html");
+
+        User cachedUser = AuthUtil.getAuthenticatedUser(request);
+
+        User user = userRepository.findByEmail(cachedUser.getEmail())
+                .orElseThrow(() -> new BusinessException());
+
+        HashMap<String, Object> model = new HashMap<>();
+        model.put("name", user.getName());
+        String imageUrl = user.getImageUrl();
+        if(imageUrl != null){
+            model.put("profile_image", imageUrl);
+        }
+        return new MyPageDynamicView(model, "/mypage/index.html");
     }
 }