Examples with FederatedUserCredentialEntity org.keycloak.storage.jpa.entity.FederatedUserCredentialEntity used on opensource projects

Example 1 with FederatedUserCredentialEntity

use of org.keycloak.storage.jpa.entity.FederatedUserCredentialEntity in project keycloak by keycloak.

the class JpaUserFederatedStorageProvider method createCredential.

@Override
public CredentialModel createCredential(RealmModel realm, String userId, CredentialModel cred) {
    createIndex(realm, userId);
    FederatedUserCredentialEntity entity = new FederatedUserCredentialEntity();
    String id = cred.getId() == null ? KeycloakModelUtils.generateId() : cred.getId();
    entity.setId(id);
    entity.setCreatedDate(cred.getCreatedDate());
    entity.setType(cred.getType());
    entity.setCredentialData(cred.getCredentialData());
    entity.setSecretData(cred.getSecretData());
    entity.setUserLabel(cred.getUserLabel());
    entity.setUserId(userId);
    entity.setRealmId(realm.getId());
    entity.setStorageProviderId(new StorageId(userId).getProviderId());
    // add in linkedlist to last position
    List<FederatedUserCredentialEntity> credentials = getStoredCredentialEntitiesStream(userId).collect(Collectors.toList());
    int priority = credentials.isEmpty() ? JpaUserCredentialStore.PRIORITY_DIFFERENCE : credentials.get(credentials.size() - 1).getPriority() + JpaUserCredentialStore.PRIORITY_DIFFERENCE;
    entity.setPriority(priority);
    em.persist(entity);
    return toModel(entity);
}

Example 2 with FederatedUserCredentialEntity

use of org.keycloak.storage.jpa.entity.FederatedUserCredentialEntity in project keycloak by keycloak.

the class JpaUserFederatedStorageProvider method getStoredCredentialById.

@Override
public CredentialModel getStoredCredentialById(RealmModel realm, String userId, String id) {
    FederatedUserCredentialEntity entity = em.find(FederatedUserCredentialEntity.class, id);
    if (!checkCredentialEntity(entity, userId))
        return null;
    CredentialModel model = toModel(entity);
    return model;
}

Example 3 with FederatedUserCredentialEntity

use of org.keycloak.storage.jpa.entity.FederatedUserCredentialEntity in project keycloak by keycloak.

the class JpaUserFederatedStorageProvider method moveCredentialTo.

@Override
public boolean moveCredentialTo(RealmModel realm, UserModel user, String id, String newPreviousCredentialId) {
    // 1 - Create new list and move everything to it.
    List<FederatedUserCredentialEntity> newList = this.getStoredCredentialEntitiesStream(user.getId()).collect(Collectors.toList());
    // 2 - Find indexes of our and newPrevious credential
    int ourCredentialIndex = -1;
    int newPreviousCredentialIndex = -1;
    FederatedUserCredentialEntity ourCredential = null;
    int i = 0;
    for (FederatedUserCredentialEntity credential : newList) {
        if (id.equals(credential.getId())) {
            ourCredentialIndex = i;
            ourCredential = credential;
        } else if (newPreviousCredentialId != null && newPreviousCredentialId.equals(credential.getId())) {
            newPreviousCredentialIndex = i;
        }
        i++;
    }
    if (ourCredentialIndex == -1) {
        logger.warnf("Not found credential with id [%s] of user [%s]", id, user.getUsername());
        return false;
    }
    if (newPreviousCredentialId != null && newPreviousCredentialIndex == -1) {
        logger.warnf("Can't move up credential with id [%s] of user [%s]", id, user.getUsername());
        return false;
    }
    // 3 - Compute index where we move our credential
    int toMoveIndex = newPreviousCredentialId == null ? 0 : newPreviousCredentialIndex + 1;
    // 4 - Insert our credential to new position, remove it from the old position
    newList.add(toMoveIndex, ourCredential);
    int indexToRemove = toMoveIndex < ourCredentialIndex ? ourCredentialIndex + 1 : ourCredentialIndex;
    newList.remove(indexToRemove);
    // 5 - newList contains credentials in requested order now. Iterate through whole list and change priorities accordingly.
    int expectedPriority = 0;
    for (FederatedUserCredentialEntity credential : newList) {
        expectedPriority += JpaUserCredentialStore.PRIORITY_DIFFERENCE;
        if (credential.getPriority() != expectedPriority) {
            credential.setPriority(expectedPriority);
            logger.tracef("Priority of credential [%s] of user [%s] changed to [%d]", credential.getId(), user.getUsername(), expectedPriority);
        }
    }
    return true;
}

Example 4 with FederatedUserCredentialEntity

use of org.keycloak.storage.jpa.entity.FederatedUserCredentialEntity in project keycloak by keycloak.

the class JpaUserFederatedStorageProvider method removeStoredCredential.

@Override
public boolean removeStoredCredential(RealmModel realm, String userId, String id) {
    FederatedUserCredentialEntity entity = em.find(FederatedUserCredentialEntity.class, id, LockModeType.PESSIMISTIC_WRITE);
    if (!checkCredentialEntity(entity, userId))
        return false;
    int currentPriority = entity.getPriority();
    this.getStoredCredentialEntitiesStream(userId).filter(credentialEntity -> credentialEntity.getPriority() > currentPriority).forEach(credentialEntity -> credentialEntity.setPriority(credentialEntity.getPriority() - JpaUserCredentialStore.PRIORITY_DIFFERENCE));
    em.remove(entity);
    return true;
}

Example 5 with FederatedUserCredentialEntity

use of org.keycloak.storage.jpa.entity.FederatedUserCredentialEntity in project keycloak by keycloak.

the class JpaUserFederatedStorageProvider method updateCredential.

@Override
public void updateCredential(RealmModel realm, String userId, CredentialModel cred) {
    FederatedUserCredentialEntity entity = em.find(FederatedUserCredentialEntity.class, cred.getId());
    if (!checkCredentialEntity(entity, userId))
        return;
    createIndex(realm, userId);
    entity.setCreatedDate(cred.getCreatedDate());
    entity.setType(cred.getType());
    entity.setCredentialData(cred.getCredentialData());
    entity.setSecretData(cred.getSecretData());
    cred.setUserLabel(entity.getUserLabel());
}