3주차 미션 / 서버 2조 박서연

src/main/java/http/HttpHeader.java

@@ -0,0 +1,41 @@
+package http;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+
+public class HttpHeader {
+    private final List<String> lines;
+    private final int lengthOfContent;
+
+    public HttpHeader(List<String> lines, int lengthOfContent) {
+        this.lines = lines;
+        this.lengthOfContent = lengthOfContent;
+    }
+
+    public static HttpHeader from(BufferedReader br) throws IOException {
+        int requestContentLength = 0;
+        List<String> lines = new ArrayList<>();
+        while (true) {
+            final String line = br.readLine();
+            if (line.equals("")) {
+                break;
+            }
+            lines.add(line);
+            // header info
+            if (line.startsWith("Content-Length")) {
+                requestContentLength = Integer.parseInt(line.split(": ")[1]);
+            }
+        }
+        return new HttpHeader(lines, requestContentLength);
+    }
+
+    public int getLengthOfContent() {
+        return lengthOfContent;
+    }
+
+    public List<String> getLines() {
+        return lines;
+    }
+}

src/main/java/http/HttpMethod.java

@@ -0,0 +1,16 @@
+package http;
+
+public enum HttpMethod {
+    GET("get"),
+    POST("post");
+
+    private final String method;
+
+    public String getMethod() {
+        return method;
+    }
+
+    HttpMethod(String method) {
+        this.method = method;
+    }
+}

src/main/java/http/HttpRequest.java

@@ -0,0 +1,44 @@
+package http;
+
+import http.util.IOUtils;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+
+public class HttpRequest {
+    private final HttpStartLine startLine;
+    private final HttpHeader header;
+    private final String body;
+
+    private HttpRequest(HttpStartLine startLine, HttpHeader header, String body) {
+        this.startLine = startLine;
+        this.header = header;
+        this.body = body;
+    }
+
+    public static HttpRequest from(BufferedReader br) throws IOException {
+        String requestLine = br.readLine();
+        if (requestLine == null || requestLine.isEmpty()) return null;
+
+        HttpHeader httpHeader = HttpHeader.from(br);
+        String body = IOUtils.readData(br, httpHeader.getLengthOfContent());
+
+        return new HttpRequest(HttpStartLine.from(requestLine), httpHeader, body);
+    }
+
+    public String getMethod(){
+        return startLine.getMethod();
+    }
+
+    public String getUrl(){
+        return startLine.getTarget();
+    }
+
+    public String getBody() {
+        return body;
+    }
+
+    public HttpHeader getHeader(){
+        return header;
+    }
+}

src/main/java/http/HttpResponse.java

@@ -0,0 +1,83 @@
+package http;
+
+import java.io.DataOutputStream;
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Paths;
+
+public class HttpResponse {
+    private final DataOutputStream dos;
+
+    public HttpResponse(DataOutputStream dos) {
+        this.dos = dos;
+    }
+
+    public void forward(String path) throws IOException {
+        File file = new File("webapp" + path);
+        int fileLength = (int) file.length();
+        if(path.equals("/user/login_failed.html")){
+            responseLoginFailedHeader(dos, fileLength);
+        }
+        else response200Header(dos, fileLength);
+        responseBody(dos, Files.readAllBytes(Paths.get(file.getPath())));
+    }
+
+    public void redirectLogin(String path) {
+        responseLoginHeader(dos, "/index.html");
+    }
+
+    public void redirectLoginFail(String path) {
+        response302Header(dos, "/user/login_failed.html");    }
+
+    private void response200Header(DataOutputStream dos, int lengthOfBodyContent) {
+        try {
+            dos.writeBytes("HTTP/1.1 200 OK \r\n");
+            dos.writeBytes("Content-Type: text/html;charset=utf-8\r\n");
+            dos.writeBytes("Content-Length: " + lengthOfBodyContent + "\r\n");
+            dos.writeBytes("\r\n");
+        } catch (IOException e) {
+
+        }
+    }
+
+    private void responseLoginHeader(DataOutputStream dos, String path) {
+        try {
+            dos.writeBytes("HTTP/1.1 302 Found \r\n");
+            dos.writeBytes("Location: " + path + "\r\n");
+            dos.writeBytes("Set-Cookie: logined=true; Path=/; HttpOnly; SameSite=Lax\r\n");
+            dos.writeBytes("\r\n");
+        } catch (IOException e) {
+
+        }
+    }
+
+    private void response302Header(DataOutputStream dos, String path) {
+        try {
+            dos.writeBytes("HTTP/1.1 302 Found \r\n");
+            dos.writeBytes("Location: " + path);
+            dos.writeBytes("\r\n");
+        } catch (IOException e) {
+        }
+    }
+
+    private void responseLoginFailedHeader(DataOutputStream dos, int lengthOfBodyContent) {
+        try {
+            dos.writeBytes("HTTP/1.1 200 OK \r\n");
+            dos.writeBytes("Content-Type: text/html;charset=utf-8\r\n");
+            dos.writeBytes("Set-Cookie: logined=; Path=/; Max-Age=0; HttpOnly; SameSite=Lax\r\n");
+            dos.writeBytes("Content-Length: " + lengthOfBodyContent + "\r\n");
+            dos.writeBytes("\r\n");
+        } catch (IOException e) {
+        }
+    }
+
+    private void responseBody(DataOutputStream dos, byte[] body) {
+        try {
+            dos.write(body, 0, body.length);
+            dos.flush();
+        } catch (IOException e) {
+
+        }
+    }
+}

src/main/java/http/HttpStartLine.java

@@ -0,0 +1,30 @@
+package http;
+
+import webserver.URL;
+
+import java.util.Optional;
+
+public class HttpStartLine {
+    private final HttpMethod method;
+    private final URL target;
+    private final String version;
+
+    public HttpStartLine(HttpMethod method, Optional<URL> target, String version) {
+        this.method = method;
+        this.target = target.orElse(null);
+        this.version = version;
+    }
+
+    public static HttpStartLine from(String startLine){
+        String[] tokens = startLine.split(" ");
+        return new HttpStartLine(HttpMethod.valueOf(tokens[0]), URL.fromPath(tokens[1]), tokens[2]);
+    }
+
+    public String getMethod() {
+        return method.getMethod();
+    }
+
+    public String getTarget(){
+        return target.getUrl();
+    }
+}

src/main/java/http/RequestMapper.java

@@ -0,0 +1,37 @@
+package http;
+
+import http.controller.*;
+import webserver.URL;
+
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.Map;
+
+public class RequestMapper {
+    private static final Map<String, Controller> controllerMap = new HashMap<>();
+    private final Controller controller;
+    private final HttpRequest request;
+    private final HttpResponse response;
+
+    public RequestMapper(HttpRequest request, HttpResponse response) {
+        initializeMap();
+        this.request = request;
+        this.response = response;
+        if(request.getMethod().equals("get") && request.getUrl().endsWith(".html")) {
+           this.controller = controllerMap.get("forward");
+        }else this.controller = controllerMap.get(request.getUrl());
+    }
+
+    private void initializeMap(){
+        controllerMap.put("forward",new ForwardController());
+        controllerMap.put(URL.DEFAULT.getUrl(),new HomeController());
+        controllerMap.put(URL.INDEX.getUrl(),new HomeController());
+        controllerMap.put(URL.USER_SIGNUP.getUrl(),new SignUpController());
+        controllerMap.put(URL.USER_LOGIN.getUrl(),new LoginController());
+        controllerMap.put(URL.USER_USERLIST.getUrl(),new ListController());
+    }
+
+    public void proceed() throws IOException {
+        controller.execute(request, response);
+    }
+}

src/main/java/http/controller/Controller.java

@@ -0,0 +1,10 @@
+package http.controller;
+
+import http.HttpRequest;
+import http.HttpResponse;
+
+import java.io.IOException;
+
+public interface Controller {
+    public void execute(HttpRequest request, HttpResponse response) throws IOException;
+}

src/main/java/http/controller/ForwardController.java

@@ -0,0 +1,13 @@
+package http.controller;
+
+import http.HttpRequest;
+import http.HttpResponse;
+
+import java.io.IOException;
+
+public class ForwardController implements Controller{
+    @Override
+    public void execute(HttpRequest request, HttpResponse response) throws IOException {
+        response.forward(request.getUrl());
+    }
+}

src/main/java/http/controller/HomeController.java

@@ -0,0 +1,13 @@
+package http.controller;
+
+import http.HttpRequest;
+import http.HttpResponse;
+
+import java.io.IOException;
+
+public class HomeController implements Controller{
+    @Override
+    public void execute(HttpRequest request, HttpResponse response) throws IOException {
+        response.forward("/index.html");
+    }
+}

src/main/java/http/controller/ListController.java

@@ -0,0 +1,46 @@
+package http.controller;
+
+import http.HttpRequest;
+import http.HttpResponse;
+import webserver.URL;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+import java.nio.file.Paths;
+import java.util.List;
+
+public class ListController implements Controller{
+    @Override
+    public void execute(HttpRequest request, HttpResponse response) throws IOException {
+        String logined = "";
+        int i = 0;
+        List<String> lines = request.getHeader().getLines();
+        while (true) {
+            if( i== lines.size()) {
+                break;
+            }
+            final String line = lines.get(i++);
+            if (line.isEmpty()) {
+                break;
+            }
+            if (line.startsWith("Cookie: ")) {
+                String cookieHeader = line.substring("Cookie:".length()).trim();
+                String[] pairs = cookieHeader.split(";");
+                for (String pair : pairs) {
+                    String[] kv = pair.trim().split("=", 2); // 최대 2개만 split
+                    if (kv.length == 2) {
+                        String name = kv[0].trim();
+                        String value = kv[1].trim();
+                        if (name.equals("logined")) {
+                            logined = value;
+                        }
+                    }
+                }
+            }
+        }
+        if(logined.equals("true")){
+            response.forward( "/user/list.html");
+        }
+    }
+}

src/main/java/http/controller/LoginController.java

@@ -0,0 +1,29 @@
+package http.controller;
+
+import db.MemoryUserRepository;
+import http.HttpRequest;
+import http.HttpResponse;
+import http.util.HttpRequestUtils;
+import http.util.IOUtils;
+import model.User;
+
+import java.io.IOException;
+import java.util.List;
+import java.util.Map;
+
+public class LoginController implements Controller {
+    @Override
+    public void execute(HttpRequest request, HttpResponse response) throws IOException {
+        MemoryUserRepository db = MemoryUserRepository.getInstance();
+        Map<String, String> userInfos = HttpRequestUtils
+                .parseQueryParameter(request.getBody());
+        User user = db.findUserById(userInfos.get("userId"));
+        if (user == null) {
+            response.redirectLoginFail("/user/login_failed.html");
+            return;
+        }
+        if (user.getPassword().equals(userInfos.get("password"))) {
+            response.redirectLogin("/index.html");
+        } else response.redirectLoginFail("/user/login_failed.html");
+    }
+}