[케이] Step 3: HTTP 웹서버 구현

src/main/java/webserver/RequestHandler.java

@@ -1,13 +1,21 @@
 package webserver;
 
-import java.io.DataOutputStream;
-import java.io.IOException;
-import java.io.InputStream;
-import java.io.OutputStream;
+import java.io.*;
 import java.net.Socket;
+import java.net.URLDecoder;
+import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
+import java.util.Collection;
+import java.util.HashMap;
+import java.util.Map;
 
+import db.DataBase;
+import model.User;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import util.HttpRequestUtils;
+import util.IOUtils;
+
 
 public class RequestHandler extends Thread {
     private static final Logger log = LoggerFactory.getLogger(RequestHandler.class);
@@ -23,11 +31,97 @@ public void run() {
                 connection.getPort());
 
         try (InputStream in = connection.getInputStream(); OutputStream out = connection.getOutputStream()) {
-            // TODO 사용자 요청에 대한 처리는 이 곳에 구현하면 된다.
-            DataOutputStream dos = new DataOutputStream(out);
-            byte[] body = "Hello World".getBytes();
-            response200Header(dos, body.length);
-            responseBody(dos, body);
+            // 3.4.3.1 1단계: 사용자 요청에 대한 처리는 이 곳에 구현하면 된다.
+            BufferedReader br = new BufferedReader(new InputStreamReader(in, "UTF-8"));
+            String line = br.readLine();
+            log.debug("request line: {}", line);
+            String url = parseLine(line);
+
+            int contentLength = 0;
+            Map<String, String> cookies = new HashMap<>();
+            //2단계 + 3단계
+            if (line == null) {
+                return;
+            }
+            while (!line.equals("")){
+                line = br.readLine();
+                if (line.contains("Content-Length")){
+                    contentLength = getContentLength(line);
+                }
+                if (line.contains("Cookie")){
+                    cookies = HttpRequestUtils.parseCookies(line);
+                    log.debug("쿠키: {}", cookies);
+                }
+                if(line.contains("Accept: text/css")){
+                    responseCss(out, url);
+                }
+                log.debug("header: {}", line);
+            }
+
+            if (url.startsWith("/user/create")){
+
+                String body = URLDecoder.decode(IOUtils.readData(br, contentLength), StandardCharsets.UTF_8);
+                log.debug("바디: {}", body);
+                Map<String, String> params = HttpRequestUtils.parseQueryString(body);
+                User user = new User(params.get("userId"), params.get("password"), params.get("name"),
+                        params.get("email"));
+                DataBase.addUser(user);
+                log.debug("user: {}", user);
+
+                DataOutputStream dos = new DataOutputStream(out);
+                response302Header(dos, "/index.html");
+
+            } else if (url.equals("/user/login")){
+                String body = IOUtils.readData(br, contentLength);
+                Map<String, String> params = HttpRequestUtils.parseQueryString(body);
+                User user = DataBase.findUserById(params.get("userId"));
+                log.debug("로그인유저: {}", user );
+
+                if (user == null) {
+                    response(out, "/user/login_failed.html");
+                    return;
+                }
+                if (user.getPassword().equals(params.get("password"))){
+                    DataOutputStream dos = new DataOutputStream(out);
+                    String cookie = "logined=true";
+                    String redirect ="/index.html";
+                    responseLogin302Header(dos, cookie, redirect);
+                } else {
+                    DataOutputStream dos = new DataOutputStream(out);
+                    String cookie = "logined=false";
+                    String redirect = "/user/login_failed.html";
+                    responseLogin302Header(dos, cookie, redirect);
+                }
+
+            } else if (url.equals("/user/list")){
+                if (cookies.get("logined").equals("true")) {
+                    log.debug("쿠키확인: {}" , cookies.get("logined"));
+                    Collection<User> users = DataBase.findAll();
+                    StringBuilder sb = new StringBuilder();
+                    sb.append("<table>");
+                    sb.append("<th>#</th> <th>사용자 아이디</th> <th>이름</th> <th>이메일</th><th></th>");
+                    sb.append("<tbody>");
+                    for (User u : users) {
+                        sb.append("<tr>");
+                        sb.append("<td>#</td>");
+                        sb.append("<td>" + u.getUserId() + "</td>");
+                        sb.append("<td>" + u.getName() + "</td>");
+                        sb.append("<td>" + u.getEmail() + "</td>");
+                        sb.append("</tr>");
+                    }
+                    sb.append("</tbody>");
+                    sb.append("</table>");
+                    byte[] body = sb.toString().getBytes(StandardCharsets.UTF_8);
+                    DataOutputStream dos = new DataOutputStream(out);
+                    response200Header(dos, body.length);
+                    responseBody(dos, body);
+                } else {
+                    response(out, "/user/login.html");
+                }
+
+            } else {
+                response(out, url);
+            }
         } catch (IOException e) {
             log.error(e.getMessage());
         }
@@ -42,6 +136,36 @@ private void response200Header(DataOutputStream dos, int lengthOfBodyContent) {
         } catch (IOException e) {
             log.error(e.getMessage());
         }
+    }private void responseCss200Header(DataOutputStream dos, int lengthOfBodyContent) {
+        try {
+            dos.writeBytes("HTTP/1.1 200 OK \r\n");
+            dos.writeBytes("Content-Type: text/css;charset=utf-8\r\n");
+            dos.writeBytes("Content-Length: " + lengthOfBodyContent + "\r\n");
+            dos.writeBytes("\r\n");
+        } catch (IOException e) {
+            log.error(e.getMessage());
+        }
+    }
+
+    private void responseLogin302Header(DataOutputStream dos, String cookie, String url) {
+        try {
+            dos.writeBytes("HTTP/1.1 302 Redirect \r\n");
+            dos.writeBytes("Set-Cookie: " + cookie + "\r\n");
+            dos.writeBytes("Location: " + url + "\r\n");
+            dos.writeBytes("\r\n");
+        } catch (IOException e) {
+            log.error(e.getMessage());
+        }
+    }
+
+    private void response302Header(DataOutputStream dos, String url) {
+        try {
+            dos.writeBytes("HTTP/1.1 302 Found \r\n");
+            dos.writeBytes("Location: " + url + "\r\n");
+            dos.writeBytes("\r\n");
+        } catch (IOException e) {
+            log.error(e.getMessage());
+        }
     }
 
     private void responseBody(DataOutputStream dos, byte[] body) {
@@ -52,4 +176,28 @@ private void responseBody(DataOutputStream dos, byte[] body) {
             log.error(e.getMessage());
         }
     }
+
+    private void response(OutputStream out, String url) throws IOException {
+        DataOutputStream dos = new DataOutputStream(out);
+        byte[] body = Files.readAllBytes(new File("./webapp" + url).toPath());
+        response200Header(dos, body.length);
+        responseBody(dos, body);
+    }
+
+    private void responseCss(OutputStream out, String url) throws IOException {
+        DataOutputStream dos = new DataOutputStream(out);
+        byte[] body = Files.readAllBytes(new File("./webapp" + url).toPath());
+        responseCss200Header(dos, body.length);
+        responseBody(dos, body);
+    }
+
+    private int getContentLength(String line) {
+        String[] split = line.split(":");
+        return Integer.parseInt(split[1].trim());
+    }
+
+    private String parseLine(String line){
+        String[] parsedLines = line.split(" ");
+        return parsedLines[1];
+    }
 }

webapp/user/form.html

@@ -75,7 +75,7 @@
 <div class="container" id="main">
    <div class="col-md-6 col-md-offset-3">
       <div class="panel panel-default content-main">
-          <form name="question" method="get" action="/user/create">
+          <form name="question" method="post" action="/user/create">
               <div class="form-group">
                   <label for="userId">사용자 아이디</label>
                   <input class="form-control" id="userId" name="userId" placeholder="User ID">
@@ -104,4 +104,4 @@
 <script src="../js/bootstrap.min.js"></script>
 <script src="../js/scripts.js"></script>
 	</body>
-</html>
+</html>