RoleAdapter.java

// /*
//  * Copyright 2016 Red Hat, Inc. and/or its affiliates
//  * and other contributors as indicated by the @author tags.
//  *
//  * 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 org.keycloak.models.cache.infinispan;

// import org.keycloak.models.RealmModel;
// import org.keycloak.models.RoleContainerModel;
// import org.keycloak.models.RoleModel;
// import org.keycloak.models.cache.infinispan.entities.CachedClientRole;
// import org.keycloak.models.cache.infinispan.entities.CachedRealmRole;
// import org.keycloak.models.cache.infinispan.entities.CachedRole;
// import org.keycloak.models.utils.KeycloakModelUtils;

// import java.util.Collection;
// import java.util.Collections;
// import java.util.HashSet;
// import java.util.List;
// import java.util.Map;
// import java.util.Set;
// import java.util.function.Supplier;

// /**
//  * @author <a href="mailto:bill@burkecentral.com">Bill Burke</a>
//  * @version $Revision: 1 $
//  */
// public class RoleAdapter implements RoleModel {

//     protected RoleModel updated;
//     protected CachedRole cached;
//     protected RealmCacheSession cacheSession;
//     protected RealmModel realm;
//     protected Set<RoleModel> composites;
//     private final Supplier<RoleModel> modelSupplier;

//     public RoleAdapter(CachedRole cached, RealmCacheSession session, RealmModel realm) {
//         this.cached = cached;
//         this.cacheSession = session;
//         this.realm = realm;
//         this.modelSupplier = this::getRoleModel;
//     }

//     protected void getDelegateForUpdate() {
//         if (updated == null) {
//             cacheSession.registerRoleInvalidation(cached.getId(), cached.getName(), getContainerId());
//             updated = modelSupplier.get();
//             if (updated == null) throw new IllegalStateException("Not found in database");
//         }
//     }

//     protected boolean invalidated;

//     public void invalidate() {
//         invalidated = true;
//     }

//     protected boolean isUpdated() {
//         if (updated != null) return true;
//         if (!invalidated) return false;
//         updated = cacheSession.getRealmDelegate().getRoleById(cached.getId(), realm);
//         if (updated == null) throw new IllegalStateException("Not found in database");
//         return true;
//     }


//     @Override
//     public String getName() {
//         if (isUpdated()) return updated.getName();
//         return cached.getName();
//     }

//     @Override
//     public String getDescription() {
//         if (isUpdated()) return updated.getDescription();
//         return cached.getDescription();
//     }

//     @Override
//     public void setDescription(String description) {
//         getDelegateForUpdate();
//         updated.setDescription(description);
//     }

//     @Override
//     public String getId() {
//         if (isUpdated()) return updated.getId();
//         return cached.getId();
//     }

//     @Override
//     public void setName(String name) {
//         getDelegateForUpdate();
//         updated.setName(name);
//     }

//     @Override
//     public boolean isComposite() {
//         if (isUpdated()) return updated.isComposite();
//         return cached.isComposite();
//     }

//     @Override
//     public void addCompositeRole(RoleModel role) {
//         getDelegateForUpdate();
//         updated.addCompositeRole(role);
//     }

//     @Override
//     public void removeCompositeRole(RoleModel role) {
//         getDelegateForUpdate();
//         updated.removeCompositeRole(role);
//     }

//     @Override
//     public Set<RoleModel> getComposites() {
//         if (isUpdated()) return updated.getComposites();

//         if (composites == null) {
//             composites = new HashSet<RoleModel>();
//             for (String id : cached.getComposites()) {
//                 RoleModel role = realm.getRoleById(id);
//                 if (role == null) {
//                     throw new IllegalStateException("Could not find composite in role " + getName() + ": " + id);
//                 }
//                 composites.add(role);
//             }
//         }

//         return composites;
//     }

//     @Override
//     public boolean isClientRole() {
//         return cached instanceof CachedClientRole;
//     }

//     @Override
//     public String getContainerId() {
//         if (isClientRole()) {
//             CachedClientRole appRole = (CachedClientRole) cached;
//             return appRole.getClientId();
//         } else {
//             return realm.getId();
//         }
//     }


//     @Override
//     public RoleContainerModel getContainer() {
//         if (cached instanceof CachedRealmRole) {
//             return realm;
//         } else {
//             CachedClientRole appRole = (CachedClientRole) cached;
//             return realm.getClientById(appRole.getClientId());
//         }
//     }

//     @Override
//     public boolean hasRole(RoleModel role) {
//         return this.equals(role) || KeycloakModelUtils.searchFor(role, this, new HashSet<>());
//     }

//     @Override
//     public void setSingleAttribute(String name, String value) {
//         getDelegateForUpdate();
//         updated.setSingleAttribute(name, value);
//     }

//     @Override
//     public void setAttribute(String name, Collection<String> values) {
//         getDelegateForUpdate();
//         updated.setAttribute(name, values);
//     }

//     @Override
//     public void removeAttribute(String name) {
//         getDelegateForUpdate();
//         updated.removeAttribute(name);
//     }

//     @Override
//     public String getFirstAttribute(String name) {
//         if (updated != null) {
//             return updated.getFirstAttribute(name);
//         }

//         return cached.getAttributes(modelSupplier).getFirst(name);
//     }

//     @Override
//     public List<String> getAttribute(String name) {
//         if (updated != null) {
//             return updated.getAttribute(name);
//         }

//         List<String> result = cached.getAttributes(modelSupplier).get(name);
//         if (result == null) {
//             result = Collections.emptyList();
//         }
//         return result;
//     }

//     @Override
//     public Map<String, List<String>> getAttributes() {
//         if (updated != null) {
//             return updated.getAttributes();
//         }

//         return cached.getAttributes(modelSupplier);
//     }

//     private RoleModel getRoleModel() {
//         return cacheSession.getRealmDelegate().getRoleById(cached.getId(), realm);
//     }

//     @Override
//     public boolean equals(Object o) {
//         if (this == o) return true;
//         if (o == null || !(o instanceof RoleModel)) return false;

//         RoleModel that = (RoleModel) o;
//         return that.getId().equals(getId());
//     }

//     @Override
//     public int hashCode() {
//         return getId().hashCode();
//     }

// }