feat(client): add contextual tuples to ClientCheckRequest

openfga · Nov 21, 2023 · 861b0c5 · 861b0c5
1 parent 6cb5468
commit 861b0c5
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/src/main/java/dev/openfga/sdk/api/client/ClientCheckRequest.java b/src/main/java/dev/openfga/sdk/api/client/ClientCheckRequest.java
@@ -12,11 +12,13 @@
 
 package dev.openfga.sdk.api.client;
 
+import java.util.List;
+
 public class ClientCheckRequest {
     private String user;
     private String relation;
     private String _object;
-    // TODO: Add "contextual tuples"
+    private List<ClientTupleKey> contextualTuples;
 
     public ClientCheckRequest _object(String _object) {
         this._object = _object;
@@ -56,4 +58,13 @@ public ClientCheckRequest user(String user) {
     public String getUser() {
         return user;
     }
+
+    public ClientCheckRequest contextualTuples(List<ClientTupleKey> contextualTuples) {
+        this.contextualTuples = contextualTuples;
+        return this;
+    }
+
+    public List<ClientTupleKey> getContextualTuples() {
+        return contextualTuples;
+    }
 }
diff --git a/src/main/java/dev/openfga/sdk/api/client/OpenFgaClient.java b/src/main/java/dev/openfga/sdk/api/client/OpenFgaClient.java
@@ -417,6 +417,11 @@ public CompletableFuture<ClientCheckResponse> check(ClientCheckRequest request,
                     .user(request.getUser())
                     .relation(request.getRelation())
                     ._object(request.getObject()));
+
+            var contextualTuples = request.getContextualTuples();
+            if (contextualTuples != null && !contextualTuples.isEmpty()) {
+                body.contextualTuples(ClientTupleKey.asContextualTupleKeys(contextualTuples));
+            }
         }
 
         if (options != null && !isNullOrWhitespace(options.getAuthorizationModelId())) {