[3단계 Tomcat 구현하기] 마크(김승현) 미션 제출합니다.

tomcat/src/main/java/com/techcourse/controller/DashBoardController.java

@@ -0,0 +1,17 @@
+package com.techcourse.controller;
+
+import org.apache.catalina.connector.HttpRequest;
+import org.apache.catalina.connector.HttpResponse;
+import com.techcourse.http.MimeType;
+import java.io.IOException;
+import org.apache.catalina.StaticResourceProvider;
+import jakarta.servlet.http.AbstractController;
+
+public class DashBoardController extends AbstractController {
+
+    @Override
+    protected void doGet(HttpRequest request, HttpResponse response) throws IOException {
+        response.setBody(StaticResourceProvider.getStaticResource("/index.html"))
+                .setContentType(MimeType.HTML.getMimeType());
+    }
+}

tomcat/src/main/java/com/techcourse/controller/LoginController.java

@@ -0,0 +1,73 @@
+package com.techcourse.controller;
+
+import org.apache.catalina.session.Session;
+import org.apache.catalina.session.SessionManager;
+import com.techcourse.db.InMemoryUserRepository;
+import org.apache.catalina.connector.HttpRequest;
+import org.apache.catalina.connector.HttpResponse;
+import com.techcourse.http.MimeType;
+import com.techcourse.model.User;
+import java.io.IOException;
+import java.util.UUID;
+import org.apache.catalina.StaticResourceProvider;
+import jakarta.servlet.http.AbstractController;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class LoginController extends AbstractController {
+
+    private static final Logger log = LoggerFactory.getLogger(LoginController.class);
+    private static final String JSESSIONID = "JSESSIONID";
+    private static final String ACCOUNT = "account";
+    private static final String PASSWORD = "password";
+    private static final String INDEX_HTML_PATH = "/index.html";
+    private static final String LOGIN_HTML_PATH = "/login.html";
+    private static final String ERROR_401_HTML_PATH = "/401.html";
+
+    private final InMemoryUserRepository inMemoryUserRepository = InMemoryUserRepository.getInstance();
+    private final SessionManager sessionManager = SessionManager.getInstance();
+
+    @Override
+    public void doGet(HttpRequest request, HttpResponse response) throws IOException {
+        if (request.getCookie(JSESSIONID) != null) {
+            response.found(INDEX_HTML_PATH);
+            return;
+        }
+
+        response.setBody(StaticResourceProvider.getStaticResource(LOGIN_HTML_PATH))
+                .setContentType(MimeType.HTML.getMimeType());
+    }
+
+    @Override
+    protected void doPost(HttpRequest request, HttpResponse response) {
+        if (!request.hasParameter(ACCOUNT) || !request.hasParameter(PASSWORD)) {
+            response.found(ERROR_401_HTML_PATH);
+            return;
+        }
+
+        String account = request.getParameter(ACCOUNT);
+        String password = request.getParameter(PASSWORD);
+        inMemoryUserRepository.findByAccount(account).ifPresentOrElse(
+                user -> login(user, request, response, password),
+                () -> response.found(ERROR_401_HTML_PATH)
+        );
+    }
+
+    private void login(User user, HttpRequest request, HttpResponse response, String password) {
+        if (!user.checkPassword(password)) {
+            response.found(ERROR_401_HTML_PATH);
+            return;
+        }
+
+        log.info("user : {}", user);
+
+        if (request.getCookie(JSESSIONID) == null) {
+            Session session = new Session(UUID.randomUUID().toString());
+            session.setAttribute("user", user);
+            sessionManager.add(session);
+            response.setCookie(JSESSIONID, session.getId())
+                    .setHttpOnly(true);
+        }
+        response.found(INDEX_HTML_PATH);
+    }
+}

tomcat/src/main/java/com/techcourse/controller/RegisterController.java

@@ -0,0 +1,42 @@
+package com.techcourse.controller;
+
+import com.techcourse.db.InMemoryUserRepository;
+import org.apache.catalina.connector.HttpRequest;
+import org.apache.catalina.connector.HttpResponse;
+import com.techcourse.http.MimeType;
+import com.techcourse.model.User;
+import java.io.IOException;
+import org.apache.catalina.StaticResourceProvider;
+import jakarta.servlet.http.AbstractController;
+
+public class RegisterController extends AbstractController {
+
+    private static final String ACCOUNT = "account";
+    private static final String PASSWORD = "password";
+    private static final String EMAIL = "email";
+
+    private final InMemoryUserRepository inMemoryUserRepository = InMemoryUserRepository.getInstance();
+
+    @Override
+    protected void doGet(HttpRequest request, HttpResponse response) throws IOException {
+        String registerHtml = StaticResourceProvider.getStaticResource("/register.html");
+        response.setBody(registerHtml)
+                .setContentType(MimeType.HTML.getMimeType());
+    }
+
+    @Override
+    protected void doPost(HttpRequest request, HttpResponse response) {
+        if (!request.hasParameter(ACCOUNT) || !request.hasParameter(PASSWORD) || !request.hasParameter(EMAIL)) {
+            response.badRequest();
+            return;
+        }
+
+        inMemoryUserRepository.save(new User(
+                request.getParameter(ACCOUNT),
+                request.getParameter(PASSWORD),
+                request.getParameter(EMAIL)
+        ));
+
+        response.found("/index.html");
+    }
+}

tomcat/src/main/java/com/techcourse/controller/StaticResourceController.java

@@ -0,0 +1,16 @@
+package com.techcourse.controller;
+
+import org.apache.catalina.connector.HttpRequest;
+import org.apache.catalina.connector.HttpResponse;
+import org.apache.catalina.StaticResourceProvider;
+import jakarta.servlet.http.AbstractController;
+
+public class StaticResourceController extends AbstractController {
+
+    @Override
+    protected void doGet(HttpRequest request, HttpResponse response) throws Exception {
+        String path = request.getPath();
+        response.setBody(StaticResourceProvider.getStaticResource(path))
+                .setContentType(StaticResourceProvider.probeContentType(path));
+    }
+}

tomcat/src/main/java/com/techcourse/db/InMemoryUserRepository.java

@@ -1,27 +1,39 @@
 package com.techcourse.db;
 
 import com.techcourse.model.User;
-
 import java.util.Map;
 import java.util.Optional;
 import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.atomic.AtomicLong;
 
 public class InMemoryUserRepository {
 
+    private static final InMemoryUserRepository INSTANCE = new InMemoryUserRepository();
     private static final Map<String, User> database = new ConcurrentHashMap<>();
 
+    private final AtomicLong id = new AtomicLong();
+
     static {
-        final User user = new User(1L, "gugu", "password", "hkkang@woowahan.com");
-        database.put(user.getAccount(), user);
+        INSTANCE.save(new User("gugu", "password", "hkkang@woowahan.com"));
+    }
+
+    public static InMemoryUserRepository getInstance() {
+        return INSTANCE;
     }
 
-    public static void save(User user) {
+    public void save(User user) {
+        user.setId(id.getAndIncrement());
         database.put(user.getAccount(), user);
     }
 
-    public static Optional<User> findByAccount(String account) {
+    public Optional<User> findByAccount(String account) {
         return Optional.ofNullable(database.get(account));
     }
 
-    private InMemoryUserRepository() {}
+    private InMemoryUserRepository() {
+    }
+
+    public void clear() {
+        database.clear();
+    }
 }

tomcat/src/main/java/com/techcourse/http/HttpHeaders.java

@@ -0,0 +1,84 @@
+package com.techcourse.http;
+
+import jakarta.servlet.http.HttpCookie;
+import java.util.HashMap;
+import java.util.Map;
+import lombok.AllArgsConstructor;
+
+@AllArgsConstructor
+public class HttpHeaders {
+
+    private static final String HEADER_SEPARATOR = ": ";
+    private static final String CRLF = "\r\n";
+    public static final String CONTENT_TYPE = "Content-Type";
+    public static final String CONTENT_LENGTH = "Content-Length";
+    public static final String SET_COOKIE_HEADER = "Set-Cookie";
+    public static final String COOKIE = "Cookie";
+    public static final String LOCATION = "Location";
+    public static final String APPLICATION_X_WWW_FORM_URLENCODED = "application/x-www-form-urlencoded";
+
+    private final Map<String, String> headers;
+    private final HttpCookie cookie;
+
+    public HttpHeaders() {
+        this.headers = new HashMap<>();
+        this.cookie = new HttpCookie();
+    }
+
+    public void clear() {
+        headers.clear();
+        cookie.clear();
+    }
+
+    public String getHeadersString() {
+        if (cookie.isExist()) {
+            headers.put(SET_COOKIE_HEADER, cookie.toMessage());
+        }
+        StringBuilder headersString = new StringBuilder();
+        for (Map.Entry<String, String> entry : headers.entrySet()) {
+            headersString.append(entry.getKey())
+                    .append(HEADER_SEPARATOR)
+                    .append(entry.getValue())
+                    .append(" ")
+                    .append(CRLF);
+        }
+        headersString.append(cookie.toMessage());
+        return headersString.toString();
+    }
+
+    public String get(String key) {
+        return headers.get(key);
+    }
+
+    public void set(String key, String value) {
+        headers.put(key, value);
+    }
+
+    public void setContentType(String contentType) {
+        set(CONTENT_TYPE, contentType);
+    }
+
+    public void setContentLength(int contentLength) {
+        set(CONTENT_LENGTH, String.valueOf(contentLength));
+    }
+
+    public void setCookie(String cookieKey, String cookieValue) {
+        cookie.setCookie(cookieKey, cookieValue);
+    }
+
+    public void setLocation(String location) {
+        set(LOCATION, location);
+    }
+
+    public String getCookie(String key) {
+        return cookie.getCookie(key);
+    }
+
+    public void setHttpOnly(boolean httpOnly) {
+        cookie.setHttpOnly(httpOnly);
+    }
+
+    public void setMaxAge(int maxAge) {
+        cookie.setMaxAge(maxAge);
+    }
+}