ui: Groups with link count resource

src/main/java/com/scality/keycloak/groupFederationLink/GroupFederationLinkEntity.java

@@ -19,6 +19,8 @@
         @NamedQuery(name = "findByGroupId", query = "select t from GroupFederationLinkEntity t where t.groupId = :groupId"),
         @NamedQuery(name = "findGroupsByFederationLinkAndName", query = "select g from GroupFederationLinkEntity gl, GroupEntity g where gl.groupId = g.id and g.name like concat('%',:name,'%') and gl.federationLink = :federationLink"),
         @NamedQuery(name = "findGroupsByFederationLink", query = "select g from GroupFederationLinkEntity gl, GroupEntity g where gl.groupId = g.id and gl.federationLink = :federationLink"),
+        @NamedQuery(name = "countGroupsByFederationLinkAndName", query = "select count(g) from GroupFederationLinkEntity gl, GroupEntity g where gl.groupId = g.id and g.name like concat('%',:name,'%') and gl.federationLink = :federationLink"),
+        @NamedQuery(name = "countGroupsByFederationLink", query = "select count(g) from GroupFederationLinkEntity gl, GroupEntity g where gl.groupId = g.id and gl.federationLink = :federationLink"),
 })
 public class GroupFederationLinkEntity {
 

src/main/java/com/scality/keycloak/groupFederationLink/GroupWithLinkAdminResource.java

@@ -2,7 +2,9 @@
 
 import static org.keycloak.models.jpa.PaginationUtils.paginateQuery;
 
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 import java.util.Objects;
 import java.util.stream.Stream;
 
@@ -27,6 +29,7 @@
 import jakarta.persistence.TypedQuery;
 import jakarta.ws.rs.DefaultValue;
 import jakarta.ws.rs.GET;
+import jakarta.ws.rs.Path;
 import jakarta.ws.rs.Produces;
 import jakarta.ws.rs.QueryParam;
 import jakarta.ws.rs.core.MediaType;
@@ -70,7 +73,7 @@ public Stream<GroupWithLinkRepresentation> getGroupsWithLink(@QueryParam("search
         logger.info("getGroupsWithLink");
 
         if (Objects.isNull(federationLink) || federationLink.isEmpty()) {
-            Stream<GroupRepresentation> groups = groupsResource.getGroups(search, search, exact, firstResult,
+            Stream<GroupRepresentation> groups = groupsResource.getGroups(search, null, exact, firstResult,
                     maxResults, true, true);
 
             return groups.map(group -> {
@@ -125,4 +128,49 @@ public Stream<GroupWithLinkRepresentation> getGroupsWithLink(@QueryParam("search
         }
     }
 
+    @GET
+    @Path("count")
+    @NoCache
+    @Produces(MediaType.APPLICATION_JSON)
+    @Tag(name = KeycloakOpenAPI.Admin.Tags.GROUPS)
+    @Operation(summary = "Get groups")
+    public Map<String, Long> countGroupsWithLink(@QueryParam("search") String search,
+            @QueryParam("link") String federationLink,
+            @QueryParam("exact") @DefaultValue("false") Boolean exact,
+            @QueryParam("first") Integer firstResult,
+            @QueryParam("max") Integer maxResults) {
+
+        GroupPermissionEvaluator groupsEvaluator = auth.groups();
+        groupsEvaluator.requireList();
+        logger.info("getGroupsWithLink");
+
+        if (Objects.isNull(federationLink) || federationLink.isEmpty()) {
+            Map<String, Long> count = groupsResource.getGroupCount(search, false);
+
+            return count;
+        }
+
+        try {
+            TypedQuery<Long> query = getEntityManager()
+                    .createNamedQuery("countGroupsByFederationLinkAndName", Long.class)
+                    .setParameter("federationLink", federationLink)
+                    .setParameter("name", search);
+            if (Objects.isNull(search) || search.isEmpty()) {
+                query = getEntityManager()
+                        .createNamedQuery("findGroupsByFederationLink", Long.class)
+                        .setParameter("federationLink", federationLink);
+            }
+
+            Long count = query.getSingleResult();
+            Map<String, Long> result = new HashMap<>();
+            result.put("count", count);
+            return result;
+        } catch (NoResultException e) {
+            logger.trace("No group found for federation link " + federationLink, e);
+            Map<String, Long> result = new HashMap<>();
+            result.put("count", 0L);
+            return result;
+        }
+    }
+
 }

src/test/java/com/scality/keycloak/GroupWithLinkTest.java

@@ -8,6 +8,8 @@
 import java.net.HttpURLConnection;
 import java.net.URL;
 import java.time.Duration;
+import java.util.HashMap;
+import java.util.Map;
 import java.util.stream.Stream;
 
 import org.apache.commons.io.IOUtils;
@@ -19,6 +21,7 @@
 import org.testcontainers.containers.output.Slf4jLogConsumer;
 import org.testcontainers.utility.MountableFile;
 
+import com.fasterxml.jackson.core.type.TypeReference;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.scality.keycloak.groupFederationLink.GroupWithLinkRepresentation;
 
@@ -262,6 +265,37 @@ private void syncLdapGroups(KeycloakContainer keycloak, ProviderAndMapper provid
         }
     }
 
+    private Long countGroupsWithLink(KeycloakContainer keycloak, String federationLink) throws IOException {
+        URL url = new URL(
+                keycloak.getAuthServerUrl() + "/admin/realms/master/groups-with-link/count?link="
+                        + federationLink);
+        HttpURLConnection conn = (HttpURLConnection) url.openConnection();
+        conn.setRequestMethod("GET");
+        conn.setRequestProperty("Authorization", "Bearer " + tokenProvider.getToken(keycloak));
+
+        int responseCode = conn.getResponseCode();
+
+        if (responseCode != 200) {
+            System.out.println("responseCode = " + responseCode);
+            InputStream errorStream = conn.getErrorStream();
+            if (errorStream != null) {
+                byte[] buffer = new byte[1024];
+                int bytesRead;
+                while ((bytesRead = errorStream.read(buffer)) != -1) {
+                    System.out.write(buffer, 0, bytesRead);
+                }
+            }
+            return 0L;
+        }
+
+        String responsePayload = IOUtils.toString(conn.getInputStream(), "UTF-8");
+        ObjectMapper objectMapper = new ObjectMapper();
+        TypeReference<HashMap<String, Long>> typeRef = new TypeReference<HashMap<String, Long>>() {
+        };
+        Map<String, Long> result = objectMapper.readValue(responsePayload, typeRef);
+        return result.get("count");
+    }
+
     private Stream<GroupWithLinkRepresentation> getGroupsWithLink(KeycloakContainer keycloak, String federationLink,
             String search)
             throws IOException {
@@ -358,6 +392,10 @@ public void groups_with_links_should_be_returned_when_listing_groups()
                 groupsWithLink = getGroupsWithLink(keycloak, providerAndMapper.providerID(), "myGroup1");
                 System.out.println(groupsWithLink);
                 assertEquals(0, groupsWithLink.count());
+
+                Long count = countGroupsWithLink(keycloak, providerAndMapper.providerID());
+                Long expectedLong = 1L;
+                assertEquals(expectedLong, count);
             }
         }