[Spring Core]주병주 미션 제출합니다.

build.gradle

@@ -15,9 +15,9 @@ repositories {
 dependencies {
     implementation 'org.springframework.boot:spring-boot-starter-web'
     implementation 'org.springframework.boot:spring-boot-starter-thymeleaf'
-    implementation 'org.springframework.boot:spring-boot-starter-jdbc'
-
+    implementation 'org.springframework.boot:spring-boot-starter-data-jpa'
     implementation 'dev.akkinoc.spring.boot:logback-access-spring-boot-starter:4.0.0'
+    implementation 'org.springframework.boot:spring-boot-starter'
 
     implementation 'io.jsonwebtoken:jjwt-api:0.11.2'
     implementation 'io.jsonwebtoken:jjwt-impl:0.11.2'

src/main/java/auth/AuthConfig.java

@@ -0,0 +1,14 @@
+package auth;
+
+import org.springframework.beans.factory.annotation.Value;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+
+@Configuration
+public class AuthConfig {
+
+    @Bean
+    public JwtUtils jwtUtils(@Value("${roomescape.auth.jwt.secret}") String secretKey) {
+        return new JwtUtils(secretKey);
+    }
+}

src/main/java/auth/JwtTokenMember.java

@@ -0,0 +1,7 @@
+package auth;
+
+public record JwtTokenMember(
+        String name,
+        String role
+) {
+}

src/main/java/auth/JwtUtils.java

@@ -0,0 +1,44 @@
+package auth;
+
+import io.jsonwebtoken.Claims;
+import io.jsonwebtoken.Jwts;
+import io.jsonwebtoken.security.Keys;
+import roomescape.member.Member;
+
+public class JwtUtils {
+
+    private final String secretKey;
+
+    public JwtUtils(String secretKey) {
+        this.secretKey = secretKey;
+    }
+
+    public String createToken(Member member){
+        return Jwts.builder()
+                .setSubject(member.getId().toString())
+                .claim("name", member.getName())
+                .claim("role", member.getRole())
+                .signWith(Keys.hmacShaKeyFor(secretKey.getBytes()))
+                .compact();
+    }
+
+    public JwtTokenMember extractToken(String token) {
+        Claims claims = Jwts.parserBuilder()
+                .setSigningKey(Keys.hmacShaKeyFor(secretKey.getBytes()))
+                .build()
+                .parseClaimsJws(token)
+                .getBody();
+        String name = claims.get("name", String.class);
+        String role = claims.get("role", String.class);
+        return new JwtTokenMember(name,role);
+    }
+    public Long decodeToken(String token) {
+        if (token == null) return null;
+        return Long.valueOf(Jwts.parserBuilder()
+                .setSigningKey(Keys.hmacShaKeyFor(secretKey.getBytes()))
+                .build()
+                .parseClaimsJws(token)
+                .getBody().getSubject());
+    }
+
+}

src/main/java/roomescape/AppConfig.java

@@ -0,0 +1,33 @@
+package roomescape;
+
+import org.springframework.context.annotation.Configuration;
+import org.springframework.web.method.support.HandlerMethodArgumentResolver;
+import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
+import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
+import roomescape.argumentResolver.LoginMemberArgumentResolver;
+import roomescape.interceptor.AuthInterceptor;
+
+import java.util.List;
+
+@Configuration
+public class AppConfig implements WebMvcConfigurer {
+
+    private final LoginMemberArgumentResolver loginMemberArgumentResolver;
+    private final AuthInterceptor authInterceptor;
+
+    public AppConfig(LoginMemberArgumentResolver loginMemberArgumentResolver, AuthInterceptor authInterceptor) {
+        this.loginMemberArgumentResolver = loginMemberArgumentResolver;
+        this.authInterceptor = authInterceptor;
+    }
+
+    @Override
+    public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
+        argumentResolvers.add(loginMemberArgumentResolver);
+    }
+
+    @Override
+    public void addInterceptors(InterceptorRegistry registry) {
+        registry.addInterceptor(authInterceptor)
+                .addPathPatterns("/admin/**");
+    }
+}

src/main/java/roomescape/Login/LoginController.java

@@ -0,0 +1,31 @@
+package roomescape.Login;
+
+import jakarta.servlet.http.Cookie;
+import jakarta.servlet.http.HttpServletResponse;
+import org.springframework.stereotype.Controller;
+import org.springframework.web.bind.annotation.*;
+import roomescape.member.MemberResponse;
+
+@RestController
+public class LoginController {
+    private final LoginService loginService;
+
+    public LoginController(LoginService loginService) {
+        this.loginService = loginService;
+    }
+
+    @PostMapping("/login")
+    public void login(@RequestBody LoginRequest loginRequest, HttpServletResponse response) {
+        String token = loginService.login(loginRequest); //토큰 받음
+        Cookie cookie = new Cookie("token", token);
+        cookie.setHttpOnly(true);
+        cookie.setPath("/");
+        response.addCookie(cookie);
+    }
+
+    @GetMapping("/login/check")
+    public MemberResponse loginCheck(
+            @CookieValue("token") String token) {
+        return loginService.checkLogin(token);
+    }
+}

src/main/java/roomescape/Login/LoginMember.java

@@ -0,0 +1,6 @@
+package roomescape.Login;
+
+public record LoginMember(
+        Long id, String name, String email, String role
+) {
+}

src/main/java/roomescape/Login/LoginRequest.java

@@ -0,0 +1,7 @@
+package roomescape.Login;
+
+public record LoginRequest(
+        String email,
+        String password
+) {
+}

src/main/java/roomescape/Login/LoginService.java

@@ -0,0 +1,44 @@
+package roomescape.Login;
+
+import org.springframework.dao.DataAccessException;
+import org.springframework.stereotype.Service;
+import auth.JwtUtils;
+import roomescape.member.Member;
+import roomescape.member.MemberRepository;
+import roomescape.member.MemberResponse;
+
+import java.util.Optional;
+
+@Service
+public class LoginService {
+    private final MemberRepository memberRepository;
+    private final JwtUtils jwtUtils;
+
+    public LoginService(MemberRepository memberRepository, JwtUtils jwtUtils) {
+        this.memberRepository = memberRepository;
+        this.jwtUtils = jwtUtils;
+    }
+
+    public String login(LoginRequest loginRequest) {
+        try {
+            Optional<Member> member = memberRepository.findByEmailAndPassword(loginRequest.email(), loginRequest.password());
+            return jwtUtils.createToken(member.get());
+        } catch (DataAccessException e){
+            throw new IllegalArgumentException("로그인 정보가 일치하지 않습니다.");
+        }
+    }
+
+    public MemberResponse checkLogin(String token) {
+        Long userId = jwtUtils.decodeToken(token);
+        try {
+            Optional<Member> member = memberRepository.findById(userId);
+            return new MemberResponse(member.get().getId(), member.get().getName(), member.get().getEmail());
+        } catch (DataAccessException exception){
+            throw new IllegalArgumentException("존재하지 않는 회원입니다.");
+        }
+    }
+
+    public Member findByName(String name) {
+        return memberRepository.findByName(name).stream().findFirst().get();
+    }
+}

src/main/java/roomescape/RoomescapeApplication.java

@@ -1,9 +1,12 @@
 package roomescape;
 
+import auth.AuthConfig;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.context.annotation.Import;
 
 @SpringBootApplication
+@Import(AuthConfig.class)
 public class RoomescapeApplication {
     public static void main(String[] args) {
         SpringApplication.run(RoomescapeApplication.class, args);

src/main/java/roomescape/argumentResolver/LoginMemberArgumentResolver.java

@@ -0,0 +1,47 @@
+package roomescape.argumentResolver;
+
+import jakarta.servlet.http.HttpServletRequest;
+import org.springframework.core.MethodParameter;
+import org.springframework.stereotype.Component;
+import org.springframework.web.bind.support.WebDataBinderFactory;
+import org.springframework.web.context.request.NativeWebRequest;
+import org.springframework.web.method.support.HandlerMethodArgumentResolver;
+import org.springframework.web.method.support.ModelAndViewContainer;
+import roomescape.Login.LoginMember;
+import roomescape.Login.LoginService;
+import roomescape.member.Member;
+import roomescape.member.MemberResponse;
+import roomescape.member.MemberService;
+import jakarta.servlet.http.Cookie;
+
+import java.util.Arrays;
+
+@Component
+public class LoginMemberArgumentResolver implements HandlerMethodArgumentResolver {
+    private final LoginService loginService;
+
+    public LoginMemberArgumentResolver(LoginService loginService) {
+        this.loginService = loginService;
+    }
+
+    @Override
+    public boolean supportsParameter(MethodParameter parameter) {
+        return parameter.getParameterType().equals(LoginMember.class);
+    }
+
+    @Override
+    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer, NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
+        HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);
+
+        String token = Arrays.stream(request.getCookies())
+                .filter(cookie -> cookie.getName().equals("token"))
+                .findFirst()
+                .map(Cookie::getValue).orElseThrow(() -> new IllegalArgumentException("토큰이 없습니다."));
+
+        MemberResponse memberResponse = loginService.checkLogin(token);
+        Member member = loginService.findByName(memberResponse.getName());
+
+        return new LoginMember(member.getId(), member.getName(), member.getEmail(), member.getRole());
+    }
+
+}

src/main/java/roomescape/data/DataLoader.java

@@ -0,0 +1,24 @@
+package roomescape.data;
+
+import org.springframework.boot.CommandLineRunner;
+import org.springframework.context.annotation.Profile;
+import org.springframework.stereotype.Component;
+import roomescape.member.Member;
+import roomescape.member.MemberRepository;
+
+@Profile("default")
+@Component
+public class DataLoader implements CommandLineRunner {
+
+    private final MemberRepository memberRepository;
+
+    public DataLoader(final MemberRepository memberRepository) {
+        this.memberRepository = memberRepository;
+    }
+
+    @Override
+    public void run(final String... args) throws Exception {
+        final Member member1 = memberRepository.save(new Member("어드민", "[email protected]", "password", "ADMIN"));
+        final Member member2 = memberRepository.save(new Member("브라운", "[email protected]", "password", "USER"));
+    }
+}

src/main/java/roomescape/data/TestDataLoader.java

@@ -0,0 +1,59 @@
+package roomescape.data;
+
+import org.springframework.boot.CommandLineRunner;
+import org.springframework.context.annotation.Profile;
+import org.springframework.stereotype.Component;
+import roomescape.member.Member;
+import roomescape.member.MemberRepository;
+import roomescape.reservation.Reservation;
+import roomescape.reservation.ReservationRepository;
+import roomescape.theme.Theme;
+import roomescape.theme.ThemeRepository;
+import roomescape.time.Time;
+import roomescape.time.TimeRepository;
+
+@Profile("test")
+@Component
+public class TestDataLoader implements CommandLineRunner {
+
+    private final MemberRepository memberRepository;
+
+    private final ThemeRepository themeRepository;
+
+    private final TimeRepository timeRepository;
+
+    private final ReservationRepository reservationRepository;
+
+    public TestDataLoader(final MemberRepository memberRepository,
+                          final ThemeRepository themeRepository,
+                          final TimeRepository timeRepository,
+                          final ReservationRepository reservationRepository) {
+        this.memberRepository = memberRepository;
+        this.themeRepository = themeRepository;
+        this.timeRepository = timeRepository;
+        this.reservationRepository = reservationRepository;
+    }
+
+    @Override
+    public void run(final String... args) throws Exception {
+        final Member member1 = memberRepository.save(new Member("어드민", "[email protected]", "password", "ADMIN"));
+        final Member member2 = memberRepository.save(new Member("브라운", "[email protected]", "password", "USER"));
+
+        final Theme theme1 = themeRepository.save(new Theme("테마1", "테마1입니다."));
+        final Theme theme2 = themeRepository.save(new Theme("테마2", "테마2입니다."));
+        final Theme theme3 = themeRepository.save(new Theme("테마3", "테마3입니다."));
+
+        final Time time1 = timeRepository.save(new Time("10:00"));
+        final Time time2 = timeRepository.save(new Time("12:00"));
+        final Time time3 = timeRepository.save(new Time("14:00"));
+        final Time time4 = timeRepository.save(new Time("16:00"));
+        final Time time5 = timeRepository.save(new Time("18:00"));
+        final Time time6 = timeRepository.save(new Time("20:00"));
+
+        reservationRepository.save(new Reservation("어드민", "2024-03-01", member1, time1, theme1));
+        reservationRepository.save(new Reservation("어드민", "2024-03-01", member1, time2, theme2));
+        reservationRepository.save(new Reservation("어드민", "2024-03-01", member1, time3, theme3));
+
+        reservationRepository.save(new Reservation("브라운", "2024-03-01", member2, time4, theme1));
+    }
+}