Merge pull request #185 from kshchepanovskyi/fix/do-not-fail-without-oauth2-deps

176: Fix exception GrpcSecurityConfigurerAdapter initialization

Author: jvmlet

grpc-spring-boot-starter/src/main/java/org/lognet/springboot/grpc/security/GrpcSecurityConfiguration.java

@@ -4,6 +4,7 @@
 import org.lognet.springboot.grpc.GRpcGlobalInterceptor;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnClass;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
 import org.springframework.security.config.annotation.ObjectPostProcessor;
@@ -38,6 +39,18 @@ public ServerInterceptor springGrpcSecurityInterceptor() throws Exception   {
 
     }
 
+    @Bean
+    public BasicAuthSchemeSelector basicAuthSchemeSelector() {
+        return new BasicAuthSchemeSelector();
+    }
+
+    @Bean
+    @ConditionalOnClass(name = {
+            "org.springframework.security.oauth2.server.resource.BearerTokenAuthenticationToken",
+            "org.springframework.security.oauth2.core.OAuth2AuthenticationException"})
+    public BearerTokenAuthSchemeSelector bearerTokenAuthSchemeSelector() {
+        return new BearerTokenAuthSchemeSelector();
+    }
 
     @Autowired(required = false)
     @SuppressWarnings({ "rawtypes", "unchecked" })

grpc-spring-boot-starter/src/main/java/org/lognet/springboot/grpc/security/GrpcSecurityConfigurerAdapter.java

@@ -9,6 +9,8 @@
 import org.springframework.security.config.annotation.authentication.configuration.AuthenticationConfiguration;
 import org.springframework.security.oauth2.jwt.JwtDecoder;
 
+import java.util.Map;
+
 public abstract class GrpcSecurityConfigurerAdapter extends GrpcSecurityConfigurer<GrpcSecurity> {
 
     private AuthenticationConfiguration authenticationConfiguration;
@@ -29,7 +31,7 @@ public void setApplicationContext(ApplicationContext context) throws Exception {
         this.authenticationConfiguration = context.getBean(AuthenticationConfiguration.class);
 
         authenticationManagerBuilder = authenticationConfiguration
-                .authenticationManagerBuilder(objectPostProcessor,context)
+                .authenticationManagerBuilder(objectPostProcessor, context)
                 .parentAuthenticationManager(authenticationConfiguration.getAuthenticationManager());
 
         this.context = context;
@@ -38,23 +40,29 @@ public void setApplicationContext(ApplicationContext context) throws Exception {
     @Override
     public void init(GrpcSecurity builder) throws Exception {
         builder.apply(new GrpcServiceAuthorizationConfigurer(builder.getApplicationContext()));
-        builder.setSharedObject(AuthenticationManagerBuilder.class,authenticationManagerBuilder);
+        builder.setSharedObject(AuthenticationManagerBuilder.class, authenticationManagerBuilder);
         final AuthenticationSchemeService authenticationSchemeService = new AuthenticationSchemeService();
-        authenticationSchemeService.register(new BasicAuthSchemeSelector());
-        authenticationSchemeService.register(new BearerTokenAuthSchemeSelector());
+        registerSchemaSelectors(authenticationSchemeService);
         builder.setSharedObject(AuthenticationSchemeService.class, authenticationSchemeService);
 
     }
 
+    protected void registerSchemaSelectors(AuthenticationSchemeService authenticationSchemeService) {
+        Map<String, AuthenticationSchemeSelector> schemeSelectorMap = context.getBeansOfType(AuthenticationSchemeSelector.class);
+        for (AuthenticationSchemeSelector selector : schemeSelectorMap.values()) {
+            authenticationSchemeService.register(selector);
+        }
+    }
+
     @Override
     public void configure(GrpcSecurity builder) throws Exception {
         try {
             final Class<?> jwtDecoderClass = Class.forName("org.springframework.security.oauth2.jwt.JwtDecoder");
             final String[] beanNames = context.getBeanNamesForType(jwtDecoderClass);
-            if (1==beanNames.length){
-                builder.authenticationProvider(JwtAuthProviderFactory.forAuthorities(context.getBean(beanNames[0],JwtDecoder.class)));
+            if (1 == beanNames.length) {
+                builder.authenticationProvider(JwtAuthProviderFactory.forAuthorities(context.getBean(beanNames[0], JwtDecoder.class)));
             }
-        }catch (ClassNotFoundException e){
+        } catch (ClassNotFoundException e) {
             //swallow
         }
         builder.authorizeRequests()