Fix isAuthorizedPartial due to rust interface changes

CedarJava/config/checkstyle/suppressions.xml

@@ -9,5 +9,5 @@
     <suppress checks="HiddenField" files="PartialAuthorizationRequest.java$"/>
     <suppress checks="LineLength" files="JSONTests.java$"/>
     <suppress checks="LineLength" files="IntegrationTests.java$"/>
-    <suppress checks="ParameterNumber" files="PartialAuthorizationResponse.java$"/>
+    <suppress checks="ParameterNumber" files="PartialAuthorizationSuccessResponse.java$"/>
 </suppressions>

CedarJava/src/main/java/com/cedarpolicy/BasicAuthorizationEngine.java

@@ -169,13 +169,18 @@ private static class AuthorizationRequest extends com.cedarpolicy.model.Authoriz
     }
 
     @JsonInclude(JsonInclude.Include.NON_ABSENT)
-    private static final class PartialAuthorizationRequest {
+    private static final class PartialAuthorizationRequest extends com.cedarpolicy.model.PartialAuthorizationRequest {
         @JsonProperty private final PolicySet policies;
         @JsonProperty private final Set<Entity> entities;
-        @JsonProperty public final com.cedarpolicy.model.PartialAuthorizationRequest request;
 
         PartialAuthorizationRequest(com.cedarpolicy.model.PartialAuthorizationRequest request, PolicySet policySet, Set<Entity> entities) {
-            this.request = request;
+            super(
+                request.principal,
+                request.action,
+                request.resource,
+                request.context,
+                request.schema,
+                request.enableRequestValidation);
             this.policies = policySet;
             this.entities = entities;
         }

CedarJava/src/main/java/com/cedarpolicy/model/PartialAuthorizationResponse.java

@@ -2,137 +2,71 @@
 
 import com.cedarpolicy.Experimental;
 import com.cedarpolicy.ExperimentalFeature;
-import com.cedarpolicy.model.AuthorizationSuccessResponse.Decision;
 import com.fasterxml.jackson.annotation.JsonCreator;
 import com.fasterxml.jackson.annotation.JsonProperty;
-import com.fasterxml.jackson.databind.JsonNode;
-import com.google.common.collect.ImmutableMap;
-import com.google.common.collect.ImmutableSet;
+import com.google.common.collect.ImmutableList;
 
-import java.util.Map;
-import java.util.Set;
+import java.util.ArrayList;
+import java.util.Optional;
 
 @Experimental(ExperimentalFeature.PARTIAL_EVALUATION)
 public class PartialAuthorizationResponse {
-    private final Decision decision;
-    private final ImmutableSet<String> satisfied;
-    private final ImmutableSet<String> errored;
-    private final ImmutableSet<String> mayBeDetermining;
-    private final ImmutableSet<String> mustBeDetermining;
-    private final ImmutableMap<String, JsonNode> residuals;
-    private final ImmutableSet<String> nontrivialResiduals;
-    private final ImmutableSet<String> warnings;
-
-    public PartialAuthorizationResponse(Decision decision, Set<String> satisfied, Set<String> errored,
-                                        Set<String> mayBeDetermining, Set<String> mustBeDetermining, Map<String, JsonNode> residuals,
-                                        Set<String> nontrivialResiduals, Set<String> warnings) {
-        this.decision = decision;
-        // note that ImmutableSet.copyOf() attempts to avoid a full copy when possible
-        // see https://github.com/google/guava/wiki/ImmutableCollectionsExplained
-        this.satisfied = ImmutableSet.copyOf(satisfied);
-        this.errored = ImmutableSet.copyOf(errored);
-        this.mayBeDetermining = ImmutableSet.copyOf(mayBeDetermining);
-        this.mustBeDetermining = ImmutableSet.copyOf(mustBeDetermining);
-        this.residuals = ImmutableMap.copyOf(residuals);
-        this.nontrivialResiduals = ImmutableSet.copyOf(nontrivialResiduals);
-        if (warnings == null) {
-            this.warnings = ImmutableSet.of(); // empty
-        } else {
-            this.warnings = ImmutableSet.copyOf(warnings);
-        }
-    }
-
     /**
-     * The optional decision returned by partial authorization
-     *
-     * @return a nullable reference to the decision (null means that no conclusive decision can be made)
+     * Is this a success or a failure response
      */
-    public Decision getDecision() {
-        return this.decision;
-    }
-
+    @JsonProperty("type")
+    public final SuccessOrFailure type;
     /**
-     * The map from policy ids to residuals
-     *
-     * @return map of residuals
+     * This will be present if and only if `type` is `Success`.
      */
-    public Map<String, JsonNode> getResiduals() {
-            return this.residuals;
-    }
-
+    @JsonProperty("response")
+    public final Optional<PartialAuthorizationSuccessResponse> success;
     /**
-     * Set of policies that are satisfied by the partial request
-     *
-     * @return set of policy ids
+     * This will be present if and only if `type` is `Failure`.
      */
-    public Set<String> getSatisfied() {
-        return this.satisfied;
-    }
-
+    @JsonProperty("errors")
+    public final Optional<ImmutableList<DetailedError>> errors;
     /**
-     * Set of policies that errored during the partial authorization
-     *
-     * @return set of policy ids
+     * Warnings can be produced regardless of whether we have a `Success` or `Failure`.
      */
-    public Set<String> getErrored() {
-        return this.errored;
-    }
+    @JsonProperty("warnings")
+    public final ImmutableList<String> warnings;
 
     /**
-     * Over approximation of policies that determine the auth decision
-     *
-     * @return set of policy ids
+     * If `type` is `Success`, `success` should be present and `errors` empty.
+     * If `type` is `Failure`, `errors` should be present and `success` empty.
      */
-    public Set<String> getMayBeDetermining() {
-        return this.mayBeDetermining;
-    }
-
-    /**
-     * Under approximation of policies that determine the auth decision
-     *
-     * @return set of policy ids
-     */
-    public Set<String> getMustBeDetermining() {
-        return this.mustBeDetermining;
+    @JsonCreator
+    public PartialAuthorizationResponse(
+        @JsonProperty("type") SuccessOrFailure type,
+        @JsonProperty("response") Optional<PartialAuthorizationSuccessResponse> success,
+        @JsonProperty("errors") Optional<ArrayList<DetailedError>> errors,
+        @JsonProperty("warnings") ArrayList<String> warnings
+    ) {
+        this.type = type;
+        this.success = success;
+        this.errors = errors.map((list) -> ImmutableList.copyOf(list));
+        if (warnings == null) {
+            this.warnings = ImmutableList.of(); // empty
+        } else {
+            this.warnings = ImmutableList.copyOf(warnings);
+        }
     }
 
-    /**
-     * Set of non-trivial residual policies
-     *
-     * @return set of policy ids
-     */
-    public Set<String> getNontrivialResiduals() {
-        return this.nontrivialResiduals;
+    @Override
+    public String toString() {
+        final String warningsString = warnings.isEmpty() ? "" : "\nwith warnings: " + warnings;
+        if (type == SuccessOrFailure.Success) {
+            return "SUCCESS: " + success.get() + warningsString;
+        } else {
+            return "FAILURE: " + errors.get() + warningsString;
+        }
     }
 
-    /**
-     * Deserializer factory method for PartialAuthorizationResponse.
-     * @param nested Deserialized object for nested JSON object.
-     * @param decision Deserialized `decision` attribute of nested JSON object.
-     * @param satisfied Deserialized `satisfied` attribute of nested JSON object.
-     * @param errored Deserialized `errored` attribute of nested JSON object.
-     * @param mayBeDetermining Deserialized `mayBeDetermining` attribute of nested JSON object.
-     * @param mustBeDetermining Deserialized `mustBeDetermining` attribute of nested JSON object.
-     * @param residuals Deserialized `residual` attribute of nested JSON object.
-     * @param nontrivialResiduals Deserialized `nontrivialResiduals` attribute of nested JSON object.
-     * @param warnings Deserialized `warnings` attribute of nested JSON object.
-     * @return
-     */
-    @JsonCreator
-    public static PartialAuthorizationResponse createPartialAuthorizationResponse(
-        @JsonProperty("response") PartialAuthorizationResponse nested,
-        @JsonProperty("decision") Decision decision,
-        @JsonProperty("satisfied") Set<String> satisfied,
-        @JsonProperty("errored") Set<String> errored,
-        @JsonProperty("mayBeDetermining") Set<String> mayBeDetermining,
-        @JsonProperty("mustBeDetermining") Set<String> mustBeDetermining,
-        @JsonProperty("residuals") Map<String, JsonNode> residuals,
-        @JsonProperty("nontrivialResiduals") Set<String> nontrivialResiduals,
-        @JsonProperty("warnings") Set<String> warnings) {
-        if (nested != null) {
-            return nested;
-        }
-        return new PartialAuthorizationResponse(decision, satisfied, errored, mayBeDetermining, mustBeDetermining,
-                residuals, nontrivialResiduals, warnings);
+    public enum SuccessOrFailure {
+        @JsonProperty("residuals")
+        Success,
+        @JsonProperty("failure")
+        Failure,
     }
 }

CedarJava/src/main/java/com/cedarpolicy/model/PartialAuthorizationSuccessResponse.java

@@ -0,0 +1,159 @@
+/*
+ * Copyright Cedar Contributors
+ *
+ * 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
+ *
+ *      https://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 com.cedarpolicy.model;
+
+import java.util.Map;
+import java.util.Set;
+
+import com.cedarpolicy.Experimental;
+import com.cedarpolicy.ExperimentalFeature;
+import com.fasterxml.jackson.annotation.JsonCreator;
+import com.fasterxml.jackson.annotation.JsonProperty;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.google.common.collect.ImmutableMap;
+import com.google.common.collect.ImmutableSet;
+
+/**
+ * Successful partial authorization response
+ */
+@Experimental(ExperimentalFeature.PARTIAL_EVALUATION)
+public final class PartialAuthorizationSuccessResponse {
+    private final AuthorizationSuccessResponse.Decision decision;
+    private final ImmutableSet<String> satisfied;
+    private final ImmutableSet<String> errored;
+    private final ImmutableSet<String> mayBeDetermining;
+    private final ImmutableSet<String> mustBeDetermining;
+    private final ImmutableMap<String, JsonNode> residuals;
+    private final ImmutableSet<String> nontrivialResiduals;
+    private final ImmutableSet<String> warnings;
+
+    public PartialAuthorizationSuccessResponse(
+        AuthorizationSuccessResponse.Decision decision, Set<String> satisfied, Set<String> errored,
+        Set<String> mayBeDetermining, Set<String> mustBeDetermining, Map<String, JsonNode> residuals,
+        Set<String> nontrivialResiduals, Set<String> warnings) {
+        this.decision = decision;
+        // note that ImmutableSet.copyOf() attempts to avoid a full copy when possible
+        // see https://github.com/google/guava/wiki/ImmutableCollectionsExplained
+        this.satisfied = ImmutableSet.copyOf(satisfied);
+        this.errored = ImmutableSet.copyOf(errored);
+        this.mayBeDetermining = ImmutableSet.copyOf(mayBeDetermining);
+        this.mustBeDetermining = ImmutableSet.copyOf(mustBeDetermining);
+        this.residuals = ImmutableMap.copyOf(residuals);
+        this.nontrivialResiduals = ImmutableSet.copyOf(nontrivialResiduals);
+        if (warnings == null) {
+            this.warnings = ImmutableSet.of(); // empty
+        } else {
+            this.warnings = ImmutableSet.copyOf(warnings);
+        }
+    }
+
+    /**
+     * Deserializer factory method for PartialAuthorizationResponse.
+     *
+     * @param nested              Deserialized object for nested JSON object.
+     * @param decision            Deserialized `decision` attribute of nested JSON object.
+     * @param satisfied           Deserialized `satisfied` attribute of nested JSON object.
+     * @param errored             Deserialized `errored` attribute of nested JSON object.
+     * @param mayBeDetermining    Deserialized `mayBeDetermining` attribute of nested JSON object.
+     * @param mustBeDetermining   Deserialized `mustBeDetermining` attribute of nested JSON object.
+     * @param residuals           Deserialized `residual` attribute of nested JSON object.
+     * @param nontrivialResiduals Deserialized `nontrivialResiduals` attribute of nested JSON object.
+     * @param warnings            Deserialized `warnings` attribute of nested JSON object.
+     * @return
+     */
+    @JsonCreator
+    public static PartialAuthorizationSuccessResponse createPartialAuthorizationSuccessResponse(
+        @JsonProperty("response") PartialAuthorizationSuccessResponse nested,
+        @JsonProperty("decision") AuthorizationSuccessResponse.Decision decision,
+        @JsonProperty("satisfied") Set<String> satisfied,
+        @JsonProperty("errored") Set<String> errored,
+        @JsonProperty("mayBeDetermining") Set<String> mayBeDetermining,
+        @JsonProperty("mustBeDetermining") Set<String> mustBeDetermining,
+        @JsonProperty("residuals") Map<String, JsonNode> residuals,
+        @JsonProperty("nontrivialResiduals") Set<String> nontrivialResiduals,
+        @JsonProperty("warnings") Set<String> warnings) {
+        if (nested != null) {
+            return nested;
+        }
+        return new PartialAuthorizationSuccessResponse(decision, satisfied, errored, mayBeDetermining,
+            mustBeDetermining,
+            residuals, nontrivialResiduals, warnings);
+    }
+
+    /**
+     * The optional decision returned by partial authorization
+     *
+     * @return a nullable reference to the decision (null means that no conclusive decision can be made)
+     */
+    public AuthorizationSuccessResponse.Decision getDecision() {
+        return this.decision;
+    }
+
+    /**
+     * The map from policy ids to residuals
+     *
+     * @return map of residuals
+     */
+    public Map<String, JsonNode> getResiduals() {
+        return this.residuals;
+    }
+
+    /**
+     * Set of policies that are satisfied by the partial request
+     *
+     * @return set of policy ids
+     */
+    public Set<String> getSatisfied() {
+        return this.satisfied;
+    }
+
+    /**
+     * Set of policies that errored during the partial authorization
+     *
+     * @return set of policy ids
+     */
+    public Set<String> getErrored() {
+        return this.errored;
+    }
+
+    /**
+     * Over approximation of policies that determine the auth decision
+     *
+     * @return set of policy ids
+     */
+    public Set<String> getMayBeDetermining() {
+        return this.mayBeDetermining;
+    }
+
+    /**
+     * Under approximation of policies that determine the auth decision
+     *
+     * @return set of policy ids
+     */
+    public Set<String> getMustBeDetermining() {
+        return this.mustBeDetermining;
+    }
+
+    /**
+     * Set of non-trivial residual policies
+     *
+     * @return set of policy ids
+     */
+    public Set<String> getNontrivialResiduals() {
+        return this.nontrivialResiduals;
+    }
+}