[톰캣 구현하기 1, 2단계] 애쉬(정설희) 미션 제출합니다.

tomcat/src/main/java/nextstep/Application.java

@@ -1,11 +1,28 @@
 package nextstep;
 
+import java.util.Set;
+
+import org.apache.catalina.HandlerMapping;
 import org.apache.catalina.startup.Tomcat;
 
+import nextstep.jwp.handler.BaseRequestHandler;
+import nextstep.jwp.handler.LoginRequestHandler;
+import nextstep.jwp.handler.RegisterRequestHandler;
+import nextstep.jwp.handler.StaticContentRequestHandler;
+
 public class Application {
 
-    public static void main(String[] args) {
-        final var tomcat = new Tomcat();
-        tomcat.start();
-    }
+	public static void main(String[] args) {
+
+		final var handlers = Set.of(
+			new BaseRequestHandler(),
+			new LoginRequestHandler(),
+			new RegisterRequestHandler()
+		);
+		final var defaultHandler = new StaticContentRequestHandler();
+		final var handlerMapping = new HandlerMapping(handlers, defaultHandler);
+
+		final var tomcat = new Tomcat(handlerMapping);
+		tomcat.start();
+	}
 }

tomcat/src/main/java/nextstep/jwp/db/InMemoryUserRepository.java

@@ -1,27 +1,30 @@
 package nextstep.jwp.db;
 
-import nextstep.jwp.model.User;
-
 import java.util.Map;
 import java.util.Optional;
 import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.atomic.AtomicLong;
+
+import nextstep.jwp.model.User;
 
 public class InMemoryUserRepository {
 
-    private static final Map<String, User> database = new ConcurrentHashMap<>();
+	private static final Map<String, User> database = new ConcurrentHashMap<>();
+	private static final AtomicLong idCounter = new AtomicLong();
 
-    static {
-        final User user = new User(1L, "gugu", "password", "[email protected]");
-        database.put(user.getAccount(), user);
-    }
+	static {
+		final User user = new User(idCounter.incrementAndGet(), "gugu", "password", "[email protected]");
+		database.put(user.getAccount(), user);
+	}
 
-    public static void save(User user) {
-        database.put(user.getAccount(), user);
-    }
+	private InMemoryUserRepository() {
+	}
 
-    public static Optional<User> findByAccount(String account) {
-        return Optional.ofNullable(database.get(account));
-    }
+	public static void save(User user) {
+		database.put(user.getAccount(), user.addId(idCounter.incrementAndGet()));
+	}
 
-    private InMemoryUserRepository() {}
+	public static Optional<User> findByAccount(String account) {
+		return Optional.ofNullable(database.get(account));
+	}
 }

tomcat/src/main/java/nextstep/jwp/handler/BaseRequestHandler.java

@@ -0,0 +1,21 @@
+package nextstep.jwp.handler;
+
+import org.apache.catalina.RequestHandler;
+import org.apache.coyote.MimeType;
+import org.apache.coyote.request.Request;
+import org.apache.coyote.response.Response;
+
+public class BaseRequestHandler extends RequestHandler {
+
+	private static final String REQUEST_PATH = "/";
+	private static final String RESPONSE_BODY = "Hello world!";
+
+	public BaseRequestHandler() {
+		super(REQUEST_PATH);
+	}
+
+	@Override
+	protected Response doGet(final Request request) {
+		return Response.ok(RESPONSE_BODY, MimeType.HTML);
+	}
+}

tomcat/src/main/java/nextstep/jwp/handler/LoginRequestHandler.java

@@ -0,0 +1,83 @@
+package nextstep.jwp.handler;
+
+import org.apache.catalina.RequestHandler;
+import org.apache.catalina.session.Session;
+import org.apache.catalina.session.SessionManager;
+import org.apache.coyote.Cookie;
+import org.apache.coyote.MimeType;
+import org.apache.coyote.request.Request;
+import org.apache.coyote.response.Response;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import nextstep.jwp.db.InMemoryUserRepository;
+import nextstep.jwp.model.User;
+
+public class LoginRequestHandler extends RequestHandler {
+
+	private static final Logger log = LoggerFactory.getLogger(LoginRequestHandler.class);
+
+	private static final String REQUEST_PATH = "/login";
+	private static final String LOGIN_PAGE_PATH = "/login.html";
+	private static final String REDIRECT_LOCATION = "/index.html";
+
+	public LoginRequestHandler() {
+		super(LoginRequestHandler.REQUEST_PATH);
+	}
+
+	@Override
+	protected Response doGet(final Request request) {
+		if (isSessionExist(request)) {
+			return Response.redirect(REDIRECT_LOCATION);
+		}
+		return Response.ok(ResourceProvider.provide(LOGIN_PAGE_PATH), MimeType.fromPath(LOGIN_PAGE_PATH));
+	}
+
+	private boolean isSessionExist(final Request request) {
+		final var sessionId = request.findSession();
+		if (sessionId == null) {
+			return false;
+		}
+		return SessionManager.findById(sessionId) != null;
+	}
+
+	@Override
+	protected Response doPost(final Request request) {
+		final var account = request.findBodyField("account");
+		final var password = request.findBodyField("password");
+		return login(account, password);
+	}
+
+	private Response login(final String account, final String password) {
+		if (isInvalidInput(account, password)) {
+			return Response.badRequest();
+		}
+
+		final var user = InMemoryUserRepository.findByAccount(account);
+		if (user.isEmpty() || !user.get().checkPassword(password)) {
+			return Response.unauthorized();
+		}
+
+		final var session = createSession(user.get());
+		final var cookie = Cookie.session(session.getId());
+
+		log.info("[LOGIN SUCCESS] account: {}", account);
+		return Response.redirect(REDIRECT_LOCATION)
+			.addCookie(cookie);
+	}
+
+	private boolean isInvalidInput(final String account, final String password) {
+		return isBlank(account) || isBlank(password);
+	}
+
+	private boolean isBlank(final String value) {
+		return value == null || value.isBlank();
+	}
+
+	private Session createSession(final User user) {
+		final var session = Session.create();
+		session.setAttribute("user", user);
+		SessionManager.add(session);
+		return session;
+	}
+}

tomcat/src/main/java/nextstep/jwp/handler/RegisterRequestHandler.java

@@ -0,0 +1,59 @@
+package nextstep.jwp.handler;
+
+import org.apache.catalina.RequestHandler;
+import org.apache.coyote.MimeType;
+import org.apache.coyote.request.Request;
+import org.apache.coyote.response.Response;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import nextstep.jwp.db.InMemoryUserRepository;
+import nextstep.jwp.model.User;
+
+public class RegisterRequestHandler extends RequestHandler {
+
+	private static final Logger log = LoggerFactory.getLogger(RegisterRequestHandler.class);
+
+	private static final String REQUEST_PATH = "/register";
+	private static final String PAGE_PATH = "/register.html";
+
+	public RegisterRequestHandler() {
+		super(REQUEST_PATH);
+	}
+
+	@Override
+	protected Response doGet(final Request request) {
+		return Response.ok(ResourceProvider.provide(PAGE_PATH), MimeType.fromPath(PAGE_PATH));
+	}
+
+	@Override
+	protected Response doPost(final Request request) {
+		final var account = request.findBodyField("account");
+		final var password = request.findBodyField("password");
+		final var email = request.findBodyField("email");
+
+		return register(account, password, email);
+	}
+
+	private Response register(final String account, final String password, final String email) {
+		if (isInvalidInput(account, password, email) || isDuplicatedAccount(account)) {
+			return Response.badRequest();
+		}
+
+		InMemoryUserRepository.save(new User(account, password, email));
+		log.info("[REGISTER SUCCESS] account: {}", account);
+		return Response.redirect("/index.html");
+	}
+
+	private boolean isInvalidInput(final String account, final String password, final String email) {
+		return isBlank(account) || isBlank(password) || isBlank(email);
+	}
+
+	private boolean isBlank(final String value) {
+		return value == null || value.isBlank();
+	}
+
+	private boolean isDuplicatedAccount(final String account) {
+		return InMemoryUserRepository.findByAccount(account).isPresent();
+	}
+}

tomcat/src/main/java/nextstep/jwp/handler/ResourceProvider.java

@@ -0,0 +1,31 @@
+package nextstep.jwp.handler;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.file.Files;
+
+public class ResourceProvider {
+
+	private static final String RESOURCE_ROOT_PATH = "static";
+
+	public static String provide(final String path) {
+		final var resource = ResourceProvider.class.getClassLoader()
+			.getResource(RESOURCE_ROOT_PATH + path);
+
+		if (resource == null) {
+			throw new IllegalArgumentException("해당하는 자원을 찾을 수 없습니다.");
+		}
+
+		final var file = new File(resource.getPath()).toPath();
+
+		if (Files.isRegularFile(file)) {
+			try {
+				return new String(Files.readAllBytes(file));
+			} catch (IOException e) {
+				throw new IllegalArgumentException("파일을 읽을 수 없습니다.");
+			}
+		}
+
+		throw new IllegalArgumentException("올바르지 않은 파일입니다.");
+	}
+}

tomcat/src/main/java/nextstep/jwp/handler/StaticContentRequestHandler.java

@@ -0,0 +1,34 @@
+package nextstep.jwp.handler;
+
+import org.apache.catalina.RequestHandler;
+import org.apache.coyote.MimeType;
+import org.apache.coyote.request.Request;
+import org.apache.coyote.response.Response;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+public class StaticContentRequestHandler extends RequestHandler {
+
+	private static final Logger log = LoggerFactory.getLogger(StaticContentRequestHandler.class);
+	private static final String REQUEST_PATH = "/*";
+
+	public StaticContentRequestHandler() {
+		super(REQUEST_PATH);
+	}
+
+	@Override
+	public Response handle(final Request request) {
+		final String requestPath = request.getPath();
+		if (requestPath == null) {
+			return Response.notFound();
+		}
+		try {
+			final var responseBody = ResourceProvider.provide(requestPath);
+			final var mimeType = MimeType.fromPath(requestPath);
+			return Response.ok(responseBody, mimeType);
+		} catch (IllegalArgumentException e) {
+			log.warn("{}: {}", request.getPath(), e.getMessage());
+			return Response.notFound();
+		}
+	}
+}

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

@@ -26,13 +26,17 @@ public String getAccount() {
         return account;
     }
 
+    public User addId(Long id) {
+        return new User(id, account, password, email);
+    }
+
     @Override
     public String toString() {
         return "User{" +
-                "id=" + id +
-                ", account='" + account + '\'' +
-                ", email='" + email + '\'' +
-                ", password='" + password + '\'' +
-                '}';
+            "id=" + id +
+            ", account='" + account + '\'' +
+            ", email='" + email + '\'' +
+            ", password='" + password + '\'' +
+            '}';
     }
 }

tomcat/src/main/java/org/apache/catalina/HandlerMapping.java

@@ -0,0 +1,35 @@
+package org.apache.catalina;
+
+import java.util.Map;
+import java.util.Set;
+import java.util.stream.Collectors;
+
+import org.apache.coyote.request.Request;
+import org.apache.coyote.response.Response;
+
+public class HandlerMapping {
+
+	private final Map<String, RequestHandler> handlers;
+	private final RequestHandler defaultHandler;
+
+	public HandlerMapping(final Set<RequestHandler> handlers, final RequestHandler defaultHandler) {
+		this.handlers = handlers.stream()
+			.collect(Collectors.toMap(
+				RequestHandler::getRequestPath,
+				handler -> handler
+			));
+		this.defaultHandler = defaultHandler;
+	}
+
+	public Response handle(final Request request) {
+		final String requestPath = request.getPath();
+		if (requestPath == null) {
+			return Response.notFound();
+		}
+		final var handler = handlers.get(requestPath);
+		if (handler == null) {
+			return defaultHandler.handle(request);
+		}
+		return handler.handle(request);
+	}
+}