3주차 미션 / 서버 3조 - 함형주

src/main/java/webserver/CustomHandler/CssHandler.java

@@ -0,0 +1,18 @@
+package webserver.CustomHandler;
+
+import webserver.HttpRequest;
+import webserver.HttpResponse;
+import webserver.constant.Http;
+
+import java.io.IOException;
+
+import static webserver.constant.Http.*;
+
+public class CssHandler implements CustomHandler{
+
+    @Override
+    public byte[] process(HttpRequest request, HttpResponse response) throws IOException {
+        response.setHeader(CONTENT_TYPE.getValue(), TEXT_CSS.getValue());
+        return "Content-Type".getBytes();
+    }
+}

src/main/java/webserver/CustomHandler/CustomHandler.java

@@ -0,0 +1,11 @@
+package webserver.CustomHandler;
+
+import webserver.HttpRequest;
+import webserver.HttpResponse;
+
+import java.io.IOException;
+
+public interface CustomHandler {
+
+    byte[] process(HttpRequest request, HttpResponse response) throws IOException;
+}

src/main/java/webserver/CustomHandler/IndexHandler.java

@@ -0,0 +1,25 @@
+package webserver.CustomHandler;
+
+import http.util.IOUtils;
+import webserver.HttpRequest;
+import webserver.HttpResponse;
+import webserver.constant.Http;
+
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
+
+import static webserver.constant.Http.*;
+
+public class IndexHandler implements CustomHandler{
+
+    private final File file = new File(WEBAPP.getValue() + INDEX.getValue());
+
+    @Override
+    public byte[] process(HttpRequest request, HttpResponse response) throws IOException {
+        BufferedReader fr = new BufferedReader(new FileReader(file));
+        String fileData = IOUtils.readData(fr, (int) file.length());
+        return fileData.getBytes();
+    }
+}

src/main/java/webserver/CustomHandler/LoginFailFormHandler.java

@@ -0,0 +1,25 @@
+package webserver.CustomHandler;
+
+import http.util.IOUtils;
+import webserver.HttpRequest;
+import webserver.HttpResponse;
+import webserver.constant.Http;
+
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
+
+import static webserver.constant.Http.*;
+
+public class LoginFailFormHandler implements CustomHandler{
+
+    private final File file = new File(WEBAPP.getValue() + USER_LOGIN_FAILED.getValue());
+
+    @Override
+    public byte[] process(HttpRequest request, HttpResponse response) throws IOException {
+        BufferedReader fr = new BufferedReader(new FileReader(file));
+        String fileData = IOUtils.readData(fr, (int) file.length());
+        return fileData.getBytes();
+    }
+}

src/main/java/webserver/CustomHandler/LoginFormHandler.java

@@ -0,0 +1,24 @@
+package webserver.CustomHandler;
+
+import http.util.IOUtils;
+import webserver.HttpRequest;
+import webserver.HttpResponse;
+
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
+
+import static webserver.constant.Http.*;
+
+public class LoginFormHandler implements CustomHandler{
+
+    private final File file = new File(WEBAPP.getValue() + LOGIN_FORM.getValue());
+
+    @Override
+    public byte[] process(HttpRequest request, HttpResponse response) throws IOException {
+        BufferedReader fr = new BufferedReader(new FileReader(file));
+        String fileData = IOUtils.readData(fr, (int) file.length());
+        return fileData.getBytes();
+    }
+}

src/main/java/webserver/CustomHandler/LoginHandler.java

@@ -0,0 +1,47 @@
+package webserver.CustomHandler;
+
+import db.MemoryUserRepository;
+import model.User;
+import webserver.HttpRequest;
+import webserver.HttpResponse;
+import webserver.constant.Http;
+
+import java.io.IOException;
+import java.util.Map;
+
+import static webserver.constant.Http.*;
+
+public class LoginHandler implements CustomHandler{
+
+    private final MemoryUserRepository repository;
+
+    public LoginHandler() {
+        repository = MemoryUserRepository.getInstance();
+    }
+
+    @Override
+    public byte[] process(HttpRequest request, HttpResponse response) throws IOException {
+        try {
+            Map<String, String> paramMap = request.getParamMap();
+            User findById = repository.findUserById(paramMap.get("userId"));
+            return passwordCheck(paramMap, findById, response);
+        } catch (NullPointerException e) {
+            setStatusCodeAndLocation(USER_LOGIN_FAILED.getValue(), response);
+            return USER_LOGIN_FAILED.getValue().getBytes();
+        }
+    }
+
+    private static byte[] passwordCheck(Map<String, String> paramMap, User findById, HttpResponse response) {
+        if (!findById.getPassword().equals(paramMap.get("password"))) {
+            return USER_LOGIN_FAILED.getValue().getBytes();
+        }
+        setStatusCodeAndLocation(INDEX.getValue(), response);
+        response.setHeader(SET_COOKIE.getValue(), LOGINED_TRUE.getValue());
+        return INDEX.getValue().getBytes();
+    }
+
+    private static void setStatusCodeAndLocation(String location, HttpResponse response) {
+        response.setStatusCode(FOUND.getValue());
+        response.setHeader(LOCATION.getValue(), location);
+    }
+}

src/main/java/webserver/CustomHandler/SignUpFormHandler.java

@@ -0,0 +1,24 @@
+package webserver.CustomHandler;
+
+import http.util.IOUtils;
+import webserver.HttpRequest;
+import webserver.HttpResponse;
+
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
+
+import static webserver.constant.Http.*;
+
+public class SignUpFormHandler implements CustomHandler{
+
+    private final File file = new File(WEBAPP.getValue() + SIGNUP_FORM.getValue());
+
+    @Override
+    public byte[] process(HttpRequest request, HttpResponse response) throws IOException {
+        BufferedReader fr = new BufferedReader(new FileReader(file));
+        String fileData = IOUtils.readData(fr, (int) file.length());
+        return fileData.getBytes();
+    }
+}

src/main/java/webserver/CustomHandler/SignUpHandler.java

@@ -0,0 +1,44 @@
+package webserver.CustomHandler;
+
+import db.MemoryUserRepository;
+import model.User;
+import webserver.HttpRequest;
+import webserver.HttpResponse;
+import webserver.constant.Http;
+
+import java.io.IOException;
+import java.util.Map;
+
+import static webserver.constant.Http.*;
+
+public class SignUpHandler implements CustomHandler {
+
+    private final MemoryUserRepository repository;
+
+    public SignUpHandler() {
+        repository = MemoryUserRepository.getInstance();
+    }
+
+    @Override
+    public byte[] process(HttpRequest request, HttpResponse response) throws IOException {
+        Map<String, String> paramMap = request.getParamMap();
+
+        User user = new User(
+                paramMap.get("userId"),
+                paramMap.get("password"),
+                paramMap.get("name"),
+                paramMap.get("email")
+        );
+
+        repository.addUser(user);
+
+        setStatusCodeAndLocation(INDEX.getValue(), response);
+
+        return INDEX.getValue().getBytes();
+    }
+
+    private static void setStatusCodeAndLocation(String location, HttpResponse response) {
+        response.setStatusCode(FOUND.getValue());
+        response.setHeader(LOCATION.getValue(), location);
+    }
+}

src/main/java/webserver/CustomHandler/UserListHandler.java

@@ -0,0 +1,40 @@
+package webserver.CustomHandler;
+
+import http.util.IOUtils;
+import webserver.HttpRequest;
+import webserver.HttpResponse;
+
+import java.io.BufferedReader;
+import java.io.File;
+import java.io.FileReader;
+import java.io.IOException;
+import java.util.Map;
+
+import static webserver.constant.Http.*;
+
+public class UserListHandler implements CustomHandler{
+
+    private final File file = new File(WEBAPP.getValue() + LIST.getValue());
+
+    @Override
+    public byte[] process(HttpRequest request, HttpResponse response) throws IOException {
+        System.out.println("진입");
+        Map<String, String> headerMap = request.getHeaderMap();
+        String cookieValue = headerMap.get(COOKIE.getValue());
+        String[] split = cookieValue.split(";");
+        System.out.println("split[0] = " + split[0]);
+        if (split[0].equals(LOGINED_TRUE.getValue())) {
+            BufferedReader fr = new BufferedReader(new FileReader(file));
+            String fileData = IOUtils.readData(fr, (int) file.length());
+            return fileData.getBytes();
+        }
+
+        setStatusCodeAndLocation(response);
+        return INDEX.getValue().getBytes();
+    }
+
+    private static void setStatusCodeAndLocation(HttpResponse response) {
+        response.setStatusCode(FOUND.getValue());
+        response.setHeader(LOCATION.getValue(), INDEX.getValue());
+    }
+}

src/main/java/webserver/FrontHandler.java

@@ -0,0 +1,54 @@
+package webserver;
+
+import webserver.CustomHandler.*;
+
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.Map;
+
+import static webserver.constant.Http.*;
+
+public class FrontHandler {
+
+    private static final Map<String, CustomHandler> handlerMappingMap = new HashMap<>();
+    private static FrontHandler frontHandler;
+
+    private FrontHandler() {
+    }
+
+    public static FrontHandler getInstance() {
+        if (handlerMappingMap.isEmpty()) {
+            frontHandler = new FrontHandler();
+            handlerMapping();
+            return frontHandler;
+        }
+        return frontHandler;
+    }
+
+    private static void handlerMapping() {
+        handlerMappingMap.put(INDEX.getValue(), new IndexHandler());
+        handlerMappingMap.put("/", new IndexHandler());
+        handlerMappingMap.put(SIGNUP_FORM.getValue(), new SignUpFormHandler());
+        handlerMappingMap.put(SIGNUP.getValue(), new SignUpHandler());
+        handlerMappingMap.put(LOGIN_FORM.getValue(), new LoginFormHandler());
+        handlerMappingMap.put(USER_LOGIN_FAILED.getValue(), new LoginFailFormHandler());
+        handlerMappingMap.put(LOGIN.getValue(), new LoginHandler());
+        handlerMappingMap.put(USER_LIST.getValue(), new UserListHandler());
+        handlerMappingMap.put(CSS.getValue(), new CssHandler());
+    }
+
+    public void service(HttpRequest request, HttpResponse response) throws IOException {
+
+        String requestUri = request.getRequestUri();
+        CustomHandler handler;
+        if (requestUri.contains(CSS.getValue())) handler = handlerMappingMap.get(CSS.getValue());
+        else handler = handlerMappingMap.get(request.getRequestUri());
+
+        byte[] bytes = handler.process(request, response);
+
+        response.createStartLine();
+        response.createHeader();
+        response.responseBody(bytes);
+
+    }
+}