feat(annotation): userId, roleId 추출 어노테이션 추가

Author: m-a-king

src/main/java/com/somemore/global/auth/annotation/RoleId.java

@@ -0,0 +1,11 @@
+package com.somemore.global.auth.annotation;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Retention(RetentionPolicy.RUNTIME)
+@Target(ElementType.PARAMETER)
+public @interface RoleId {
+}

src/main/java/com/somemore/global/auth/annotation/RoleIdArgumentResolver.java

@@ -0,0 +1,40 @@
+package com.somemore.global.auth.annotation;
+
+import com.somemore.global.auth.authentication.UserIdentity;
+import com.somemore.global.exception.InvalidAuthenticationException;
+import org.springframework.core.MethodParameter;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+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 static com.somemore.global.exception.ExceptionMessage.AUTHENTICATION_MISSING;
+import static com.somemore.global.exception.ExceptionMessage.INVALID_PRINCIPAL_TYPE;
+
+@Component
+public class RoleIdArgumentResolver implements HandlerMethodArgumentResolver {
+
+    @Override
+    public boolean supportsParameter(MethodParameter parameter) {
+        return parameter.hasParameterAnnotation(RoleId.class);
+    }
+
+    @Override
+    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,
+                                  NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
+        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+
+        if (authentication == null) {
+            throw new InvalidAuthenticationException(AUTHENTICATION_MISSING);
+        }
+
+        if (authentication.getPrincipal() instanceof UserIdentity principal) {
+            return principal.roleId();
+        }
+
+        throw new InvalidAuthenticationException(INVALID_PRINCIPAL_TYPE);
+    }
+}

src/main/java/com/somemore/global/auth/annotation/UserId.java

@@ -0,0 +1,11 @@
+package com.somemore.global.auth.annotation;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+@Retention(RetentionPolicy.RUNTIME)
+@Target(ElementType.PARAMETER)
+public @interface UserId {
+}

src/main/java/com/somemore/global/auth/annotation/UserIdArgumentResolver.java

@@ -0,0 +1,40 @@
+package com.somemore.global.auth.annotation;
+
+import com.somemore.global.auth.authentication.UserIdentity;
+import com.somemore.global.exception.InvalidAuthenticationException;
+import org.springframework.core.MethodParameter;
+import org.springframework.security.core.Authentication;
+import org.springframework.security.core.context.SecurityContextHolder;
+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 static com.somemore.global.exception.ExceptionMessage.AUTHENTICATION_MISSING;
+import static com.somemore.global.exception.ExceptionMessage.INVALID_PRINCIPAL_TYPE;
+
+@Component
+public class UserIdArgumentResolver implements HandlerMethodArgumentResolver {
+
+    @Override
+    public boolean supportsParameter(MethodParameter parameter) {
+        return parameter.hasParameterAnnotation(UserId.class);
+    }
+
+    @Override
+    public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,
+                                  NativeWebRequest webRequest, WebDataBinderFactory binderFactory) throws Exception {
+        Authentication authentication = SecurityContextHolder.getContext().getAuthentication();
+
+        if (authentication == null) {
+            throw new InvalidAuthenticationException(AUTHENTICATION_MISSING);
+        }
+
+        if (authentication.getPrincipal() instanceof UserIdentity principal) {
+            return principal.userId();
+        }
+
+        return new InvalidAuthenticationException(INVALID_PRINCIPAL_TYPE);
+    }
+}