de.arbeitsagentur.opdt.keycloak.cassandra.role.persistence.entities.Roles Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of keycloak-cassandra-extension Show documentation
There is a newer version: 5.1.0-26.1
Show newest version
/*
 * Copyright 2022 IT-Systemhaus der Bundesagentur fuer Arbeit
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package de.arbeitsagentur.opdt.keycloak.cassandra.role.persistence.entities;

import com.datastax.oss.driver.api.mapper.annotations.CqlName;
import com.datastax.oss.driver.api.mapper.annotations.Entity;
import com.datastax.oss.driver.api.mapper.annotations.PartitionKey;
import de.arbeitsagentur.opdt.keycloak.cassandra.transaction.TransactionalEntity;
import java.util.*;
import java.util.stream.Collectors;
import lombok.*;

@EqualsAndHashCode(of = "realmId")
@Builder
@Data
@NoArgsConstructor
@AllArgsConstructor
@Entity
@CqlName("roles")
public class Roles implements TransactionalEntity {
  @PartitionKey private String realmId;

  private Long version;

  @Builder.Default private Set realmRoles = new HashSet<>();

  @Builder.Default private Map> clientRoles = new HashMap<>();

  @Override
  public String getId() {
    return realmId;
  }

  public Set getRealmRoles() {
    if (realmRoles == null) {
      realmRoles = new HashSet<>();
    }
    return realmRoles;
  }

  public Map> getClientRoles() {
    if (clientRoles == null) {
      clientRoles = new HashMap<>();
    }
    return clientRoles;
  }

  public RoleValue getRoleById(String id) {
    RoleValue realmRole =
        realmRoles.stream().filter(r -> r.getId().equals(id)).findFirst().orElse(null);
    if (realmRole == null) {
      return clientRoles.entrySet().stream()
          .flatMap(e -> e.getValue().stream().filter(r -> r.getId().equals(id)))
          .findFirst()
          .orElse(null);
    }
    return realmRole;
  }

  public void addRealmRole(RoleValue role) {
    realmRoles.add(role);
  }

  public List getRealmRoles(Integer first, Integer max) {
    return realmRoles.stream()
        .skip(first == null || first < 0 ? 0 : first)
        .limit(max == null || max < 0 ? Long.MAX_VALUE : max)
        .collect(Collectors.toList());
  }

  public boolean removeClientRole(String clientId, String id) {
    return clientRoles.get(clientId).remove(RoleValue.builder().id(id).build());
  }

  public boolean removeRealmRole(String id) {
    return realmRoles.remove(RoleValue.builder().id(id).build());
  }

  public void addClientRole(String clientId, RoleValue role) {
    Set concreteClientRoles = clientRoles.getOrDefault(clientId, new HashSet<>());
    concreteClientRoles.add(role);
    clientRoles.put(clientId, concreteClientRoles);
  }

  public Collection getClientRoles(String clientId, Integer first, Integer max) {
    return clientRoles.getOrDefault(clientId, new HashSet<>()).stream()
        .skip(first == null || first < 0 ? 0 : first)
        .limit(max == null || max < 0 ? Long.MAX_VALUE : max)
        .collect(Collectors.toList());
  }

  @Override
  public Map> getAttributes() {
    return Collections.emptyMap();
  }
}