[Spring Core] 한명수 미션 제출합니다.

build.gradle

@@ -15,7 +15,7 @@ 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'
 

src/main/java/roomescape/DataLoader.java

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

src/main/java/roomescape/TestDataLoader.java

@@ -0,0 +1,66 @@
+package roomescape;
+
+import java.util.Arrays;
+import java.util.List;
+
+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;
+
+@Component
+@Profile("test")
+public class TestDataLoader implements CommandLineRunner {
+
+	private final MemberRepository memberRepository;
+	private final ThemeRepository themeRepository;
+	private final TimeRepository timeRepository;
+	private final ReservationRepository reservationRepository;
+
+	public TestDataLoader(MemberRepository memberRepository, ThemeRepository themeRepository,
+		TimeRepository timeRepository, ReservationRepository reservationRepository) {
+		this.memberRepository = memberRepository;
+		this.themeRepository = themeRepository;
+		this.timeRepository = timeRepository;
+		this.reservationRepository = reservationRepository;
+	}
+
+	@Override
+	public void run(String... args) {
+		Member admin = new Member("어드민", "[email protected]", "password", "ADMIN");
+		Member brown = new Member("브라운", "[email protected]", "password", "USER");
+		memberRepository.saveAll(Arrays.asList(admin, brown));
+
+		Theme theme1 = new Theme("테마1", "테마1입니다.");
+		Theme theme2 = new Theme("테마2", "테마2입니다.");
+		Theme theme3 = new Theme("테마3", "테마3입니다.");
+		themeRepository.saveAll(Arrays.asList(theme1, theme2, theme3));
+
+		List<Time> times = Arrays.asList(
+			new Time("10:00"),
+			new Time("12:00"),
+			new Time("14:00"),
+			new Time("16:00"),
+			new Time("18:00"),
+			new Time("20:00")
+		);
+		timeRepository.saveAll(times);
+
+		Reservation reservation1 = new Reservation("admin", "2024-03-01", times.get(0), theme1);
+		Reservation reservation2 = new Reservation("admin", "2024-03-01", times.get(1), theme2);
+		Reservation reservation3 = new Reservation("admin", "2024-03-01", times.get(2), theme3);
+		Reservation reservation4 = new Reservation("brown", "2024-03-01", times.get(0), theme2);
+
+		reservationRepository.saveAll(Arrays.asList(reservation1, reservation2, reservation3, reservation4));
+	}
+}
+
+

src/main/java/roomescape/auth/AdminInterceptor.java

@@ -0,0 +1,33 @@
+package roomescape.auth;
+
+import org.springframework.stereotype.Component;
+import org.springframework.web.servlet.HandlerInterceptor;
+
+import jakarta.servlet.http.HttpServletRequest;
+import jakarta.servlet.http.HttpServletResponse;
+import roomescape.member.Member;
+import roomescape.member.MemberService;
+
+@Component
+public class AdminInterceptor implements HandlerInterceptor {
+
+	private MemberService memberService;
+	private final JwtUtils jwtUtils;
+
+	public AdminInterceptor(JwtUtils jwtUtils, MemberService memberService) {
+		this.jwtUtils = jwtUtils;
+		this.memberService = memberService;
+	}
+
+	@Override
+	public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws
+		Exception {
+		String token = jwtUtils.extractTokenFromCookie(request.getCookies());
+		Member member = memberService.getMemberFromToken(token);
+		if (member == null || !member.getRole().equals("ADMIN")) {
+			response.setStatus(401);
+			return false;
+		}
+		return true;
+	}
+}

src/main/java/roomescape/auth/AuthWebConfiguration.java

@@ -0,0 +1,34 @@
+package roomescape.auth;
+
+import java.util.List;
+
+import org.springframework.beans.factory.annotation.Autowired;
+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;
+
+@Configuration
+public class AuthWebConfiguration implements WebMvcConfigurer {
+
+	private final LoginMemberArgumentResolver loginMemberArgumentResolver;
+	private final AdminInterceptor adminInterceptor;
+
+	@Autowired
+	public AuthWebConfiguration(LoginMemberArgumentResolver loginMemberArgumentResolver,
+		AdminInterceptor adminInterceptor) {
+		this.loginMemberArgumentResolver = loginMemberArgumentResolver;
+		this.adminInterceptor = adminInterceptor;
+	}
+
+	@Override
+	public void addArgumentResolvers(List<HandlerMethodArgumentResolver> argumentResolvers) {
+		argumentResolvers.add(loginMemberArgumentResolver);
+	}
+
+	@Override
+	public void addInterceptors(InterceptorRegistry registry) {
+		registry.addInterceptor(adminInterceptor)
+			.addPathPatterns("/admin/**");
+	}
+}

src/main/java/roomescape/auth/JwtConfiguration.java

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

src/main/java/roomescape/auth/JwtUtils.java

@@ -0,0 +1,49 @@
+package roomescape.auth;
+
+import java.util.Arrays;
+
+import org.springframework.beans.factory.annotation.Value;
+
+import io.jsonwebtoken.Jwts;
+import io.jsonwebtoken.security.Keys;
+import jakarta.servlet.http.Cookie;
+import roomescape.member.Member;
+
+public class JwtUtils {
+
+	private String secretKey;
+
+	@Value("${roomescape.auth.jwt.secret}")
+	public void setSecretKey(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 String extractTokenFromCookie(Cookie[] cookies) {
+		if (cookies == null) {
+			throw new IllegalArgumentException("Cookies array is null");
+		}
+
+		return Arrays.stream(cookies)
+			.filter(cookie -> "token".equals(cookie.getName()))
+			.map(Cookie::getValue)
+			.findFirst()
+			.orElseThrow(() -> new IllegalArgumentException("Token cookie not found"));
+	}
+
+	public Long getIdFromToken(String token) {
+		return Long.valueOf(Jwts.parserBuilder()
+			.setSigningKey(Keys.hmacShaKeyFor(secretKey.getBytes()))
+			.build()
+			.parseClaimsJws(token)
+			.getBody().getSubject());
+	}
+}

src/main/java/roomescape/auth/LoginMember.java

@@ -0,0 +1,8 @@
+package roomescape.auth;
+
+public record LoginMember(
+	Long id,
+	String name,
+	String email,
+	String password) {
+}

src/main/java/roomescape/auth/LoginMemberArgumentResolver.java

@@ -0,0 +1,43 @@
+package roomescape.auth;
+
+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 jakarta.servlet.http.Cookie;
+import jakarta.servlet.http.HttpServletRequest;
+import roomescape.member.Member;
+import roomescape.member.MemberService;
+
+@Component
+public class LoginMemberArgumentResolver implements HandlerMethodArgumentResolver {
+
+	private final MemberService memberService;
+	private final JwtUtils jwtUtils;
+
+	public LoginMemberArgumentResolver(MemberService memberService, JwtUtils jwtUtils) {
+		this.memberService = memberService;
+		this.jwtUtils = jwtUtils;
+	}
+
+	@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);
+		Cookie[] cookies = request.getCookies();
+		String token = jwtUtils.extractTokenFromCookie(cookies);
+		if (token == null) {
+			return null;
+		}
+		Member member = memberService.getMemberFromToken(token);
+		return new LoginMember(member.getId(), member.getName(), member.getEmail(), member.getRole());
+	}
+}

src/main/java/roomescape/member/Member.java

@@ -1,43 +1,64 @@
 package roomescape.member;
 
+import jakarta.persistence.Column;
+import jakarta.persistence.Entity;
+import jakarta.persistence.GeneratedValue;
+import jakarta.persistence.GenerationType;
+import jakarta.persistence.Id;
+
+@Entity
 public class Member {
-    private Long id;
-    private String name;
-    private String email;
-    private String password;
-    private String role;
-
-    public Member(Long id, String name, String email, String role) {
-        this.id = id;
-        this.name = name;
-        this.email = email;
-        this.role = role;
-    }
-
-    public Member(String name, String email, String password, String role) {
-        this.name = name;
-        this.email = email;
-        this.password = password;
-        this.role = role;
-    }
-
-    public Long getId() {
-        return id;
-    }
-
-    public String getName() {
-        return name;
-    }
-
-    public String getEmail() {
-        return email;
-    }
-
-    public String getPassword() {
-        return password;
-    }
-
-    public String getRole() {
-        return role;
-    }
+
+	@Id
+	@GeneratedValue(strategy = GenerationType.IDENTITY)
+	private Long id;
+
+	@Column(nullable = false)
+	private String name;
+
+	@Column(nullable = false, unique = true)
+	private String email;
+
+	@Column(nullable = false)
+	private String password;
+
+	@Column(nullable = false)
+	private String role;
+
+	public Member() {
+	}
+
+	public Member(Long id, String name, String email, String role) {
+		this.id = id;
+		this.name = name;
+		this.email = email;
+		this.role = role;
+	}
+
+	public Member(String name, String email, String password, String role) {
+		this.name = name;
+		this.email = email;
+		this.password = password;
+		this.role = role;
+	}
+
+	public Long getId() {
+		return id;
+	}
+
+	public String getName() {
+		return name;
+	}
+
+	public String getEmail() {
+		return email;
+	}
+
+	public String getPassword() {
+		return password;
+	}
+
+	public String getRole() {
+		return role;
+	}
 }