Examples with StoreFactory org.keycloak.authorization.store.StoreFactory used on opensource projects

Example 61 with StoreFactory

use of org.keycloak.authorization.store.StoreFactory in project keycloak by keycloak.

the class RepresentationToModel method importPolicies.

private static Policy importPolicies(AuthorizationProvider authorization, ResourceServer resourceServer, List<PolicyRepresentation> policiesToImport, String parentPolicyName) {
    StoreFactory storeFactory = authorization.getStoreFactory();
    for (PolicyRepresentation policyRepresentation : policiesToImport) {
        if (parentPolicyName != null && !parentPolicyName.equals(policyRepresentation.getName())) {
            continue;
        }
        Map<String, String> config = policyRepresentation.getConfig();
        String applyPolicies = config.get("applyPolicies");
        if (applyPolicies != null && !applyPolicies.isEmpty()) {
            PolicyStore policyStore = storeFactory.getPolicyStore();
            try {
                List<String> policies = (List<String>) JsonSerialization.readValue(applyPolicies, List.class);
                Set<String> policyIds = new HashSet<>();
                for (String policyName : policies) {
                    Policy policy = policyStore.findByName(policyName, resourceServer.getId());
                    if (policy == null) {
                        policy = policyStore.findById(policyName, resourceServer.getId());
                    }
                    if (policy == null) {
                        policy = importPolicies(authorization, resourceServer, policiesToImport, policyName);
                        if (policy == null) {
                            throw new RuntimeException("Policy with name [" + policyName + "] not defined.");
                        }
                    }
                    policyIds.add(policy.getId());
                }
                config.put("applyPolicies", JsonSerialization.writeValueAsString(policyIds));
            } catch (Exception e) {
                throw new RuntimeException("Error while importing policy [" + policyRepresentation.getName() + "].", e);
            }
        }
        PolicyStore policyStore = storeFactory.getPolicyStore();
        Policy policy = policyStore.findById(policyRepresentation.getId(), resourceServer.getId());
        if (policy == null) {
            policy = policyStore.findByName(policyRepresentation.getName(), resourceServer.getId());
        }
        if (policy == null) {
            policy = policyStore.create(policyRepresentation, resourceServer);
        } else {
            policy = toModel(policyRepresentation, authorization, policy);
        }
        if (parentPolicyName != null && parentPolicyName.equals(policyRepresentation.getName())) {
            return policy;
        }
    }
    return null;
}