feat: Extract user roles from a string property in user claims #496 (#…

…497)

src/main/java/com/epam/aidial/core/security/IdentityProvider.java

@@ -152,6 +152,8 @@ private List<String> extractUserRoles(Map<String, Object> map) {
             if (i == rolePath.length - 1) {
                 if (next instanceof List) {
                     return (List<String>) next;
+                } else if (next instanceof String) {
+                    return List.of((String) next);
                 }
             } else {
                 if (next instanceof Map) {

src/test/java/com/epam/aidial/core/security/IdentityProviderTest.java

@@ -360,6 +360,51 @@ public void testExtractClaims_13() {
         });
     }
 
+    @Test
+    public void testExtractClaims_14() {
+        settings.put("disableJwtVerification", Boolean.TRUE);
+        IdentityProvider identityProvider = new IdentityProvider(settings, vertx, client, url -> jwkProvider, factory);
+        Algorithm algorithm = Algorithm.RSA256((RSAPublicKey) keyPair.getPublic(), (RSAPrivateKey) keyPair.getPrivate());
+
+        String token = JWT.create().withHeader(Map.of("kid", "kid1"))
+                // test with a single role as a string field
+                .withClaim("roles", "role")
+                .withClaim("email", "[email protected]")
+                .withClaim("id", 15)
+                .withClaim("title", "title")
+                .withClaim("access", List.of("read", "write"))
+                .withClaim("expire", new Date(1713355825858L))
+                .withClaim("numberList", List.of("15", "17", "34"))
+                .withClaim("map", Map.of("a", List.of("b")))
+                .withClaim("sub", "sub").sign(algorithm);
+
+        Future<ExtractedClaims> result = identityProvider.extractClaimsFromJwt(JWT.decode(token));
+
+        verifyNoInteractions(jwkProvider);
+
+        assertNotNull(result);
+        result.onComplete(res -> {
+            assertTrue(res.succeeded());
+            ExtractedClaims claims = res.result();
+            assertNotNull(claims);
+            assertEquals(List.of("role"), claims.userRoles());
+            assertEquals("sub", claims.sub());
+            assertNotNull(claims.userHash());
+            Map<String, List<String>> userClaims = claims.userClaims();
+            // assert user claim
+            assertEquals(9, userClaims.size());
+            assertEquals(List.of("sub"), userClaims.get("sub"));
+            assertEquals(List.of("read", "write"), userClaims.get("access"));
+            assertEquals(List.of("role"), userClaims.get("roles"));
+            assertEquals(List.of(), userClaims.get("expire"));
+            assertEquals(List.of("15", "17", "34"), userClaims.get("numberList"));
+            assertEquals(List.of(), userClaims.get("id"));
+            assertEquals(List.of("title"), userClaims.get("title"));
+            assertEquals(List.of(), userClaims.get("map"));
+            assertEquals(List.of("[email protected]"), userClaims.get("email"));
+        });
+    }
+
     @Test
     public void testExtractClaims_FromUserInfo_01() {
         settings.remove("jwksUrl");