[HTTP 서버 구현하기 - 2, 3단계] 배럴(김나은) 미션 제출합니다.

app/src/main/java/nextstep/jwp/HttpServer.java

@@ -24,15 +24,15 @@ public class HttpServer {
     private static final Logger log = LoggerFactory.getLogger(HttpServer.class);
 
     private final BufferedReader reader;
-    private final RequestLine requestLine;
-    private final RequestHeader headers;
-    private final RequestBody requestBody;
+    private final RequestHeader requestHeader;
+    private final MappingHandler mappingHandler;
 
     public HttpServer(InputStream inputStream) throws IOException {
         this.reader = new BufferedReader(new InputStreamReader(inputStream));
-        this.requestLine = new RequestLine(extractRequestLine());
-        this.headers = new RequestHeader(extractHeaders());
-        this.requestBody = new RequestBody(extractRequestBody());
+        final RequestLine requestLine = new RequestLine(extractRequestLine());
+        this.requestHeader = new RequestHeader(extractHeaders());
+        final RequestBody requestBody = new RequestBody(extractRequestBody());
+        this.mappingHandler = new MappingHandler(requestLine, requestHeader, requestBody);
     }
 
     private String extractRequestLine() throws IOException {
@@ -54,8 +54,8 @@ private String extractHeaders() throws IOException {
     }
 
     private String extractRequestBody() throws IOException {
-        if (headers.contains(Header.CONTENT_LENGTH.getKey())) {
-            int contentLength = Integer.parseInt(headers.get(Header.CONTENT_LENGTH.getKey()));
+        if (requestHeader.contains(Header.CONTENT_LENGTH.getKey())) {
+            int contentLength = Integer.parseInt(requestHeader.get(Header.CONTENT_LENGTH.getKey()));
             char[] buffer = new char[contentLength];
             reader.read(buffer, 0, contentLength);
             return new String(buffer);
@@ -64,11 +64,10 @@ private String extractRequestBody() throws IOException {
     }
 
     public String getResponse() throws IOException {
-        final MappingHandler mappingHandler = new MappingHandler(this.requestLine, this.requestBody);
         try {
             return mappingHandler.response();
         } catch (InvocationTargetException e) {
-            log.info(e.getMessage());
+            log.info("Reflection related controller  Error {}", e.toString());
             return handleReflectionException(e);
         } catch (PageNotFoundException e) {
             log.info(e.getMessage());

app/src/main/java/nextstep/jwp/constants/ContentType.java

@@ -7,7 +7,8 @@ public enum ContentType {
     HTML("html", "text/html"),
     CSS("css", "text/css"),
     JS("js", "text/js"),
-    SVG("svg", "image/svg+xml");
+    SVG("svg", "image/svg+xml"),
+    ICO("ico", "image/svg+xml");
 
     private final String fileType;
     private final String contentType;

app/src/main/java/nextstep/jwp/constants/Header.java

@@ -3,7 +3,9 @@
 public enum Header {
     CONTENT_LENGTH("Content-Length"),
     LOCATION("Location"),
-    CONTENT_TYPE("Content-Type");
+    CONTENT_TYPE("Content-Type"),
+    SET_COOKIE("Set-Cookie"),
+    COOKIE("Cookie");
 
     private final String key;
 

app/src/main/java/nextstep/jwp/constants/Http.java

@@ -13,6 +13,8 @@ public class Http {
     public static final String AND_PERCENT_SEPARATOR = "&";
     public static final String EQUAL_SEPARATOR = "=";
     public static final String COLUMN_SEPARATOR = ":";
+    public static final String SEMI_COLUMN_SEPARATOR = ";";
+    public static final String JSESSIONID = "JSESSIONID";
 
     private Http() {
     }

app/src/main/java/nextstep/jwp/controller/Controller.java

@@ -2,45 +2,57 @@
 
 import java.io.IOException;
 import java.util.Map;
+import java.util.Objects;
+import java.util.UUID;
 import nextstep.jwp.constants.Header;
+import nextstep.jwp.constants.Http;
 import nextstep.jwp.constants.StatusCode;
+import nextstep.jwp.constants.UserParams;
 import nextstep.jwp.exception.UnauthorizedException;
+import nextstep.jwp.model.User;
+import nextstep.jwp.request.HttpCookie;
+import nextstep.jwp.request.HttpSession;
+import nextstep.jwp.request.HttpSessions;
 import nextstep.jwp.request.RequestBody;
+import nextstep.jwp.request.RequestHeader;
 import nextstep.jwp.response.ResponseEntity;
 import nextstep.jwp.service.HttpService;
 
 public class Controller {
 
     @GetMapping(path = "/")
-    public String basic() {
+    public String basic(RequestHeader header) {
         return ResponseEntity
                 .responseBody("Hello world!")
                 .build();
     }
 
     @GetMapping(path = "/login")
-    public String login() throws IOException {
+    public String login(RequestHeader header) throws IOException {
+        if (header.contains(Header.COOKIE.getKey())) {
+            return redirectToIndex(header);
+        }
         return ResponseEntity
                 .responseResource("/login.html")
                 .build();
     }
 
     @GetMapping(path = "/register")
-    public String register() throws IOException {
+    public String register(RequestHeader header) throws IOException {
         return ResponseEntity
                 .responseResource("/register.html")
                 .build();
     }
 
     @GetMapping(path = "/index")
-    public String index() throws IOException {
+    public String index(RequestHeader header) throws IOException {
         return ResponseEntity
                 .responseResource("/index.html")
                 .build();
     }
 
     @PostMapping(path = "/register")
-    public String register(RequestBody body) throws IOException {
+    public String register(RequestHeader header, RequestBody body) throws IOException {
         Map<String, String> params = body.getParams();
         HttpService.register(params);
         return ResponseEntity
@@ -51,11 +63,46 @@ public String register(RequestBody body) throws IOException {
     }
 
     @PostMapping(path = "/login")
-    public String login(RequestBody body) throws IOException {
+    public String login(RequestHeader header, RequestBody body) throws IOException {
         Map<String, String> params = body.getParams();
+        if (header.contains(Header.COOKIE.getKey())) {
+            return redirectToIndex(header);
+        }
         if (!HttpService.isAuthorized(params)) {
             throw new UnauthorizedException("인증되지 않은 사용자 입니다.");
         }
+        final UUID sessionId = UUID.randomUUID();
+        setSession(params, sessionId);
+        return ResponseEntity
+                .statusCode(StatusCode.FOUND)
+                .addHeaders(Header.LOCATION, "/index.html")
+                .addHeaders(Header.SET_COOKIE, Http.JSESSIONID + Http.EQUAL_SEPARATOR + sessionId)
+                .responseResource("/index.html")
+                .build();
+    }
+
+    private void setSession(Map<String, String> params, UUID sessionId) {
+        HttpSessions.addSession(sessionId.toString());
+        HttpSession session = HttpSessions.getSession(sessionId.toString());
+        User user = HttpService.findUser(params.get(UserParams.ACCOUNT));
+        session.setAttribute("user", user);
+    }
+
+    private void checkCookieSessionId(RequestHeader header) {
+        HttpCookie cookie = header.getCookie();
+        if (!cookie.contains(Http.JSESSIONID)) {
+            throw new UnauthorizedException("인증되지 않은 사용자 입니다.");
+        }
+        String sessionId = cookie.get(Http.JSESSIONID);
+        HttpSession session = HttpSessions.getSession(sessionId);
+        Object user = session.getAttribute("user");
+        if (Objects.isNull(user)) {
+            throw new UnauthorizedException("인증되지 않은 사용자 입니다.");
+        }
+    }
+
+    private String redirectToIndex(RequestHeader header) throws IOException {
+        checkCookieSessionId(header);
         return ResponseEntity
                 .statusCode(StatusCode.FOUND)
                 .addHeaders(Header.LOCATION, "/index.html")

app/src/main/java/nextstep/jwp/controller/GetHandler.java

@@ -6,6 +6,7 @@
 import java.util.stream.Collectors;
 import nextstep.jwp.constants.HttpMethod;
 import nextstep.jwp.exception.HttpException;
+import nextstep.jwp.request.RequestHeader;
 import nextstep.jwp.response.ResponseEntity;
 
 public class GetHandler implements Handler {
@@ -14,14 +15,16 @@ public class GetHandler implements Handler {
     public GetHandler() {
     }
 
+    @Override
     public boolean matchHttpMethod(HttpMethod httpMethod) {
         return this.httpMethod == httpMethod;
     }
 
-    public String runController(String uri, Controller controller) throws Exception {
+    @Override
+    public String runController(String uri, RequestHeader requestHeader, Controller controller) throws Exception {
         for (Method method : Controller.class.getDeclaredMethods()) {
             if (matchGetMapping(method, uri)) {
-                return (String) method.invoke(controller);
+                return (String) method.invoke(controller, requestHeader);
             }
         }
 

app/src/main/java/nextstep/jwp/controller/Handler.java

@@ -1,10 +1,11 @@
 package nextstep.jwp.controller;
 
 import nextstep.jwp.constants.HttpMethod;
+import nextstep.jwp.request.RequestHeader;
 
 public interface Handler {
 
     boolean matchHttpMethod(HttpMethod httpMethod);
 
-    String runController(String uri, Controller controller) throws Exception;
+    String runController(String uri, RequestHeader requestHeader, Controller controller) throws Exception;
 }

app/src/main/java/nextstep/jwp/controller/MappingHandler.java

@@ -5,15 +5,18 @@
 import nextstep.jwp.constants.HttpMethod;
 import nextstep.jwp.exception.HttpException;
 import nextstep.jwp.request.RequestBody;
+import nextstep.jwp.request.RequestHeader;
 import nextstep.jwp.request.RequestLine;
 
 public class MappingHandler {
     private final RequestLine requestLine;
+    private final RequestHeader requestHeader;
     private final RequestBody requestBody;
     private final Controller controller;
 
-    public MappingHandler(RequestLine requestLine, RequestBody requestBody) {
+    public MappingHandler(RequestLine requestLine, RequestHeader requestHeader, RequestBody requestBody) {
         this.requestLine = requestLine;
+        this.requestHeader = requestHeader;
         this.requestBody = requestBody;
         this.controller = new Controller();
     }
@@ -22,7 +25,7 @@ public String response() throws Exception {
         final HttpMethod httpMethod = requestLine.getHttpMethod();
         final String uri = requestLine.getUri();
         final Handler handler = findHandler(httpMethod);
-        return handler.runController(uri, controller);
+        return handler.runController(uri, requestHeader, controller);
     }
 
     private Handler findHandler(HttpMethod httpMethod) {

app/src/main/java/nextstep/jwp/controller/PostHandler.java

@@ -4,6 +4,7 @@
 import nextstep.jwp.constants.HttpMethod;
 import nextstep.jwp.exception.HttpException;
 import nextstep.jwp.request.RequestBody;
+import nextstep.jwp.request.RequestHeader;
 
 public class PostHandler implements Handler {
     private final HttpMethod httpMethod = HttpMethod.POST;
@@ -13,14 +14,16 @@ public PostHandler(RequestBody requestBody) {
         this.requestBody = requestBody;
     }
 
+    @Override
     public boolean matchHttpMethod(HttpMethod httpMethod) {
         return this.httpMethod == httpMethod;
     }
 
-    public String runController(String uri, Controller controller) throws Exception {
+    @Override
+    public String runController(String uri, RequestHeader requestHeader, Controller controller) throws Exception {
         for (Method method : Controller.class.getDeclaredMethods()) {
             if (matchPostMapping(method, uri)) {
-                return (String) method.invoke(controller, requestBody);
+                return (String) method.invoke(controller, requestHeader, requestBody);
             }
         }
         throw new HttpException("잘못된 http post 요청입니다.");

app/src/main/java/nextstep/jwp/model/User.java

@@ -1,5 +1,7 @@
 package nextstep.jwp.model;
 
+import java.util.Objects;
+
 public class User {
     private final long id;
     private final String account;
@@ -21,6 +23,24 @@ public String getAccount() {
         return account;
     }
 
+    @Override
+    public boolean equals(Object o) {
+        if (this == o) {
+            return true;
+        }
+        if (o == null || getClass() != o.getClass()) {
+            return false;
+        }
+        User user = (User) o;
+        return id == user.id && Objects.equals(account, user.account) && Objects.equals(password,
+                user.password) && Objects.equals(email, user.email);
+    }
+
+    @Override
+    public int hashCode() {
+        return Objects.hash(id, account, password, email);
+    }
+
     @Override
     public String toString() {
         return "User{" +

app/src/main/java/nextstep/jwp/request/HttpCookie.java

@@ -0,0 +1,28 @@
+package nextstep.jwp.request;
+
+import java.util.Map;
+import java.util.stream.Collectors;
+import java.util.stream.Stream;
+import nextstep.jwp.constants.Http;
+
+public class HttpCookie {
+    private final Map<String, String> cookies;
+
+    public HttpCookie(String cookie) {
+        this.cookies = parseCookie(cookie);
+    }
+
+    private Map<String, String> parseCookie(String cookie) {
+        return Stream.of(cookie.split(Http.SEMI_COLUMN_SEPARATOR))
+                .map(item -> item.trim().split(Http.EQUAL_SEPARATOR))
+                .collect(Collectors.toMap(x -> x[Http.KEY].trim(), x -> x[Http.VALUE].trim()));
+    }
+
+    public boolean contains(String key) {
+        return cookies.containsKey(key);
+    }
+
+    public String get(String key) {
+        return cookies.get(key);
+    }
+}

app/src/main/java/nextstep/jwp/request/HttpSession.java

@@ -0,0 +1,25 @@
+package nextstep.jwp.request;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class HttpSession {
+    private final String id;
+    private final Map<String, Object> values = new HashMap<>();
+
+    HttpSession(String id) {
+        this.id = id;
+    }
+
+    public String getId() {
+        return id;
+    }
+
+    public void setAttribute(String name, Object value) {
+        values.put(name, value);
+    }
+
+    public Object getAttribute(String name) {
+        return values.get(name);
+    }
+}

app/src/main/java/nextstep/jwp/request/HttpSessions.java

@@ -0,0 +1,19 @@
+package nextstep.jwp.request;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class HttpSessions {
+    private static final Map<String, HttpSession> SESSIONS = new HashMap<>();
+
+    public static void addSession(String id) {
+        SESSIONS.put(id, new HttpSession(id));
+    }
+
+    public static HttpSession getSession(String id) {
+        return SESSIONS.get(id);
+    }
+
+    private HttpSessions() {
+    }
+}