如何使用 spring-authorization-server 在 JWT 中创建自定义声明

Question

我正在基于实验性 Spring 项目Spring Authorization Server构建 OAuth2 授权服务器

我的用例非常简单，从数据库中获取用户，并根据用户的某些属性，在正在生成的 JWT 中设置一些自定义声明。我还没有找到使用 Spring Authorization Server 的方法，我能解决的唯一方法是注入一个jwtCustomizer对象作为JwtEncoderbean 定义的一部分：

  @Bean
  public JwtEncoder jwtEncoder(CryptoKeySource keySource) {
    NimbusJwsEncoder jwtEncoder = new NimbusJwsEncoder(keySource);
    jwtEncoder.setJwtCustomizer((headersBuilder, claimsBuilder) -> {
      // Inject some headers and claims...
    });
    return jwtEncoder;
  }

这显然不允许我访问用户信息，因此我目前无法设置我需要的声明。有人设法解决这个问题吗？

Answer 1

解决方案是在库的测试中

    @Bean
    OAuth2TokenCustomizer<JwtEncodingContext> jwtCustomizer() {
        return context -> {
            if (context.getTokenType().getValue().equals(OidcParameterNames.ID_TOKEN)) {
                Authentication principal = context.getPrincipal();
                Set<String> authorities = principal.getAuthorities().stream()
                        .map(GrantedAuthority::getAuthority)
                        .collect(Collectors.toSet());
                context.getClaims().claim(AUTHORITIES_CLAIM, authorities);
            }
        };
    }

Answer 2

你可以尝试下面的方法。虽然它是 Kotlin 代码，而不是 Java，但方法应该很清楚：

import org.springframework.security.oauth2.provider.token.TokenEnhancer

class UserTokenEnhancer : TokenEnhancer {
    
    override fun enhance(accessToken: OAuth2AccessToken,
                         authentication: OAuth2Authentication): OAuth2AccessToken {

        val username = authentication.userAuthentication.name
        val additionalInfo = mapOf( /* populate with some data for given username */ )

        (accessToken as DefaultOAuth2AccessToken).additionalInformation = additionalInfo
        return accessToken
    }
}

然后只需注册bean：

@Bean
fun userTokenEnhancer(): TokenEnhancer {
    return UserTokenEnhancer()
}