IdpPolicyRuleActionProvider needs the ability to set the provider id to be useful for creating IDP Routing Rules about okta-sdk-java HOT 3 OPEN

private PolicyRule createOktaIdpDiscoveryRuleForOrg(OrganizationRecord orgRec) {
    final Policy idpDiscoveryPolicy = getOktaIdpDiscoveryPolicy();

    HashMap<String, Object> idpPolicyRuleMap = buildIdpPolicyRuleBody(orgRec);

    PolicyRule createdIdpPolicyRule = oktaPolicyApi.getApiClient()
        .invokeAPI(String.format("/api/v1/policies/%s/rules", idpDiscoveryPolicy.getId()),
                HttpMethod.POST.name(),
                null,
                null,
                null,
                idpPolicyRuleMap,
                new HashMap<>(),
                new HashMap<>(),
                new HashMap<>(),
                MediaType.APPLICATION_JSON_VALUE,
                MediaType.APPLICATION_JSON_VALUE,
                new String[]{"apiToken", "oauth2"},
                new TypeReference<>() {}
        );

    if (createdIdpPolicyRule == null || createdIdpPolicyRule.getId() == null) {
        throw new IllegalStateException("Failed to create Okta IDP Discovery Rule, no PolicyRule returned.");
    }

    orgRec.setOktaIdpDiscoveryRuleId(createdIdpPolicyRule.getId());

    return createdIdpPolicyRule;
}

private Policy getOktaIdpDiscoveryPolicy() {
    final List<Policy> idpDiscoveryPolicies = oktaPolicyApi.listPolicies(PolicyType.IDP_DISCOVERY.getValue(), null, null);

    // There should only be one IDP Discovery Policy. Each Organization will have its own Policy Rule
    if (idpDiscoveryPolicies.size() != 1) {
        throw new IllegalArgumentException("Unexpected IDP Discovery policy size of " + idpDiscoveryPolicies.size());
    }

    return idpDiscoveryPolicies.get(0);
}

private static HashMap<String, Object> buildIdpPolicyRuleBody(OrganizationRecord orgRec) {
    Map<String, Object> conditionsMap = buildRoutingRuleConditions(orgRec);
    Map<String, Object> actionsMap = buildRoutingRuleActions(orgRec);

    HashMap<String, Object> idpRoutingRuleMap = new HashMap<>();
    idpRoutingRuleMap.put("type", PolicyRuleType.IDP_DISCOVERY);
    idpRoutingRuleMap.put("name", "routingRuleNameLimit50SpecificToEachIDP");
    idpRoutingRuleMap.put("actions", actionsMap);
    idpRoutingRuleMap.put("conditions", conditionsMap);

    return idpRoutingRuleMap;
}

private static Map<String, Object> buildRoutingRuleActions(OrganizationRecord orgRec) {
    Map<String, Object> providerMap = new HashMap<>();
    providerMap.put("id", orgRec.getOktaIdpId());

    Map<String, Object> idpMap = new HashMap<>();
    idpMap.put("providers", List.of(providerMap));
    idpMap.put("idpSelectionType", "SPECIFIC");

    return Map.of("idp", idpMap);
}

private static Map<String, Object> buildRoutingRuleConditions(OrganizationRecord orgRec) {
    HashMap<String, String> patternMap = new HashMap<>();
    patternMap.put("matchType", "SUFFIX");
    patternMap.put("value", orgRec.getSpDomain());

    HashMap<String, Object> userIdentifierMap = new HashMap<>();
    userIdentifierMap.put("type", "IDENTIFIER");
    userIdentifierMap.put("patterns", List.of(patternMap));

    return Map.of("userIdentifier", userIdentifierMap);
}