de.arbeitsagentur.opdt.keycloak.filestore.group.FileGroupAdapter Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2024. 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.filestore.group;

import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.stream.Stream;
import org.keycloak.models.*;
import org.keycloak.models.utils.RoleUtils;

public abstract class FileGroupAdapter extends AbstractGroupModel {
  public FileGroupAdapter(KeycloakSession session, RealmModel realm, FileGroupEntity entity) {
    super(session, realm, entity);
  }

  @Override
  public String getId() {
    return entity.getName();
  }

  @Override
  public String getName() {
    return entity.getName();
  }

  @Override
  public void setName(String name) {
    entity.setName(name);
  }

  @Override
  public void setSingleAttribute(String name, String value) {
    entity.setAttribute(name, Collections.singletonList(value));
  }

  @Override
  public void setAttribute(String name, List values) {
    entity.setAttribute(name, values);
  }

  @Override
  public void removeAttribute(String name) {
    entity.removeAttribute(name);
  }

  @Override
  public String getFirstAttribute(String name) {
    return getAttributeStream(name).findFirst().orElse(null);
  }

  @Override
  public Stream getAttributeStream(String name) {
    List attributes = entity.getAttribute(name);
    if (attributes == null || attributes.isEmpty()) return Stream.empty();
    return attributes.stream();
  }

  @Override
  public Map> getAttributes() {
    Map> attrs = entity.getMultiValuedAttributes();
    return attrs == null ? Collections.emptyMap() : attrs;
  }

  @Override
  public GroupModel getParent() {
    String parentId = getParentId();
    if (parentId == null) {
      return null;
    }
    return session.groups().getGroupById(realm, parentId);
  }

  @Override
  public String getParentId() {
    return entity.getParentId();
  }

  @Override
  public void setParent(GroupModel group) {
    if (group == null) {
      entity.setParentId(null);
      return;
    }
    if (!getId().equals(group.getId())) {
      entity.setParentId(group.getId());
    }
  }

  @Override
  public void addChild(GroupModel subGroup) {
    subGroup.setParent(this);
  }

  @Override
  public void removeChild(GroupModel subGroup) {
    if (getId().equals(subGroup.getParentId())) {
      subGroup.setParent(null);
    }
  }

  @Override
  public Stream getRealmRoleMappingsStream() {
    return getRoleMappingsStream()
        .filter(roleModel -> roleModel.getContainer() instanceof RealmModel);
  }

  @Override
  public Stream getClientRoleMappingsStream(ClientModel app) {
    final String clientId = app.getId();
    return getRoleMappingsStream()
        .filter(roleModel -> roleModel.getContainer() instanceof ClientModel)
        .filter(roleModel -> roleModel.getContainer().getId().equals(clientId));
  }

  @Override
  public boolean hasDirectRole(RoleModel role) {
    List grantedRoles = entity.getGrantedRoles();
    return grantedRoles != null && grantedRoles.contains(role.getId());
  }

  @Override
  public boolean hasRole(RoleModel role) {
    if (RoleUtils.hasRole(getRoleMappingsStream(), role)) return true;
    GroupModel parent = getParent();
    return parent != null && parent.hasRole(role);
  }

  @Override
  public void grantRole(RoleModel role) {
    entity.addGrantedRole(role.getId());
  }

  @Override
  public Stream getRoleMappingsStream() {
    List grantedRoles = entity.getGrantedRoles();
    return grantedRoles == null
        ? Stream.empty()
        : grantedRoles.stream().map(roleId -> session.roles().getRoleById(realm, roleId));
  }

  @Override
  public void deleteRoleMapping(RoleModel role) {
    entity.removeGrantedRole(role.getId());
  }
}