Skip to content

Commit

Permalink
Merge pull request #17 from permitio/asaf/cto-121-java-sdk-more-checks
Browse files Browse the repository at this point in the history 
Advanced permission checks: bulk check, check in all tenants, get user permissions
  • Loading branch information
Asaf Cohen authored Oct 17, 2023
2 parents fe3ce2c + 9fee5b5 commit c388a80
Show file tree
Hide file tree
Showing 10 changed files with 577 additions and 8 deletions.
26 changes: 22 additions & 4 deletions src/main/java/io/permit/sdk/Permit.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -4,16 +4,14 @@
import com.google.gson.GsonBuilder;
import io.permit.sdk.api.ApiClient;
import io.permit.sdk.api.ElementsApi;
import io.permit.sdk.enforcement.Enforcer;
import io.permit.sdk.enforcement.IEnforcerApi;
import io.permit.sdk.enforcement.Resource;
import io.permit.sdk.enforcement.User;
import io.permit.sdk.enforcement.*;
import io.permit.sdk.util.Context;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.util.List;

/**
* The {@code Permit} class represents the main entry point for interacting with the Permit.io SDK.
Expand Down Expand Up @@ -137,4 +135,24 @@ public boolean checkUrl(User user, String httpMethod, String url, String tenant,
public boolean checkUrl(User user, String httpMethod, String url, String tenant) throws IOException {
return this.enforcer.checkUrl(user, httpMethod, url, tenant);
}

@Override
public boolean[] bulkCheck(List<CheckQuery> checks) throws IOException {
return this.enforcer.bulkCheck(checks);
}

@Override
public List<TenantDetails> checkInAllTenants(User user, String action, Resource resource, Context context) throws IOException {
return this.enforcer.checkInAllTenants(user, action, resource, context);
}

@Override
public List<TenantDetails> checkInAllTenants(User user, String action, Resource resource) throws IOException {
return this.enforcer.checkInAllTenants(user, action, resource);
}

@Override
public UserPermissions getUserPermissions(GetUserPermissionsQuery input) throws IOException {
return this.enforcer.getUserPermissions(input);
}
}
19 changes: 19 additions & 0 deletions src/main/java/io/permit/sdk/enforcement/CheckQuery.java
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,19 @@
package io.permit.sdk.enforcement;

import io.permit.sdk.util.Context;

import java.util.HashMap;

/**
* The {@code CheckQuery} class represents a single permit.check() request (query)
* It is used by the bulk APIs to call many checks at once.
*/
public final class CheckQuery extends EnforcerInput {
public CheckQuery(User user, String action, Resource resource, Context context) {
super(user, action, resource, context);
}

public CheckQuery(User user, String action, Resource resource) {
this(user, action, resource, new Context());
}
}
258 changes: 256 additions & 2 deletions src/main/java/io/permit/sdk/enforcement/Enforcer.java
View file
Original file line number Diff line number Diff line change
@@ -1,13 +1,18 @@
package io.permit.sdk.enforcement;

import com.google.common.primitives.Booleans;
import com.google.gson.Gson;
import io.permit.sdk.PermitConfig;
import io.permit.sdk.api.HttpLoggingInterceptor;
import io.permit.sdk.util.Context;
import io.permit.sdk.util.ContextStore;

import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.stream.Collectors;

import okhttp3.MediaType;
import okhttp3.OkHttpClient;
Expand All @@ -26,7 +31,7 @@ class EnforcerInput {
public final User user;
public final String action;
public final Resource resource;
public final HashMap<String, Object> context;
public final Context context;

/**
* Constructs a new instance of the {@code EnforcerInput} class with the specified data.
Expand All @@ -36,7 +41,7 @@ class EnforcerInput {
* @param resource The resource on which the action is performed.
* @param context The context for the authorization check.
*/
EnforcerInput(User user, String action, Resource resource, HashMap<String, Object> context) {
EnforcerInput(User user, String action, Resource resource, Context context) {
this.user = user;
this.action = action;
this.resource = resource;
Expand Down Expand Up @@ -76,6 +81,48 @@ class OpaResult {
}
}


/**
* The {@code TenantResult} class represents a single tenant returned by the checkInAllTenants query.
*/
class TenantResult {
public final Boolean allow;

public final TenantDetails tenant;

public TenantResult(Boolean allow, TenantDetails tenant) {
this.allow = allow;
this.tenant = tenant;
}
}

/**
* The {@code AllTenantsResult} class represents the result of the checkInAllTenants query.
*/
class AllTenantsResult {
public final TenantResult[] allowed_tenants;

public AllTenantsResult(TenantResult[] allowed_tenants) {
this.allowed_tenants = allowed_tenants;
}
}

/**
* The {@code OpaBulkResult} class represents the result of a Permit bulk enforcement check returned by the policy agent.
*/
class OpaBulkResult {
public final List<OpaResult> allow;

/**
* Constructs a new instance of the {@code OpaResult} class with the specified result.
*
* @param allow {@code true} if the action is allowed, {@code false} otherwise.
*/
OpaBulkResult(List<OpaResult> allow) {
this.allow = allow;
}
}

/**
* The {@code Enforcer} class is responsible for performing permission checks against the PDP.
* It implements the {@link IEnforcerApi} interface.
Expand Down Expand Up @@ -257,8 +304,215 @@ public boolean checkUrl(User user, String httpMethod, String url, String tenant,
}
}

@Override
public boolean[] bulkCheck(List<CheckQuery> checks) throws IOException {
List<EnforcerInput> inputs = new ArrayList<>();

for (CheckQuery check: checks) {
Resource normalizedResource = check.resource.normalize(this.config);
inputs.add(new EnforcerInput(check.user, check.action, normalizedResource, check.context));
}

// request body
Gson gson = new Gson();
String requestBody = gson.toJson(inputs);
RequestBody body = RequestBody.create(requestBody, MediaType.parse("application/json"));

// create the request
String url = String.format("%s/allowed/bulk", this.config.getPdpAddress());
Request request = new Request.Builder()
.url(url)
.post(body)
.addHeader("Content-Type", "application/json")
.addHeader("Authorization", String.format("Bearer %s", this.config.getToken()))
.addHeader("X-Permit-SDK-Version", String.format("java:%s", this.config.version))
.build();

try (Response response = client.newCall(request).execute()) {
if (!response.isSuccessful()) {
String errorMessage = String.format(
"Error in %s: got unexpected status code %d",
bulkCheckRepr(inputs),
response.code()
);
logger.error(errorMessage);
throw new IOException(errorMessage);
}
ResponseBody responseBody = response.body();
if (responseBody == null) {
String errorMessage = String.format(
"Error in %s: got empty response",
bulkCheckRepr(inputs)
);
logger.error(errorMessage);
throw new IOException(errorMessage);
}

String responseString = responseBody.string();
OpaBulkResult result = gson.fromJson(responseString, OpaBulkResult.class);
if (this.config.isDebugMode()) {
for (int i = 0; i < result.allow.size(); i++) {
logger.info(String.format(
"permit.bulkCheck[%d/%d](%s, %s, %s) = %s",
i + 1,
result.allow.size(),
inputs.get(i).user,
inputs.get(i).action,
inputs.get(i).resource,
result.allow.get(i).allow
));
}

}
return Booleans.toArray(result.allow.stream().map(r -> r.allow).collect(Collectors.toList()));
}
}

@Override
public List<TenantDetails> checkInAllTenants(User user, String action, Resource resource, Context context) throws IOException {
Resource normalizedResource = resource.normalize(this.config);
Context queryContext = this.contextStore.getDerivedContext(context);

EnforcerInput input = new EnforcerInput(
user,
action,
normalizedResource,
queryContext
);

// request body
Gson gson = new Gson();
String requestBody = gson.toJson(input);
RequestBody body = RequestBody.create(requestBody, MediaType.parse("application/json"));

// create the request
String url = String.format("%s/allowed/all-tenants", this.config.getPdpAddress());
Request request = new Request.Builder()
.url(url)
.post(body)
.addHeader("Content-Type", "application/json")
.addHeader("Authorization", String.format("Bearer %s", this.config.getToken()))
.addHeader("X-Permit-SDK-Version", String.format("java:%s", this.config.version))
.build();

try (Response response = client.newCall(request).execute()) {
if (!response.isSuccessful()) {
String errorMessage = String.format(
"Error in permit.checkInAllTenants(%s, %s, %s): got unexpected status code %d",
user.toString(),
action,
resource,
response.code()
);
logger.error(errorMessage);
throw new IOException(errorMessage);
}
ResponseBody responseBody = response.body();
if (responseBody == null) {
String errorMessage = String.format(
"Error in permit.checkInAllTenants(%s, %s, %s): got empty response",
user,
action,
resource
);
logger.error(errorMessage);
throw new IOException(errorMessage);
}
String responseString = responseBody.string();
AllTenantsResult result = gson.fromJson(responseString, AllTenantsResult.class);
List<TenantDetails> tenants = Arrays.stream(result.allowed_tenants).map(r -> r.tenant).collect(Collectors.toList());
if (this.config.isDebugMode()) {
logger.info(String.format(
"permit.checkInAllTenants(%s, %s, %s) => allowed in: [%s]",
user,
action,
resource,
tenants.stream().map(t -> t.key).collect(Collectors.joining(", "))
));
}
return tenants;
}
}

@Override
public List<TenantDetails> checkInAllTenants(User user, String action, Resource resource) throws IOException {
return checkInAllTenants(user, action, resource, new Context());
}

@Override
public UserPermissions getUserPermissions(GetUserPermissionsQuery input) throws IOException {
// request body
Gson gson = new Gson();
String requestBody = gson.toJson(input);
RequestBody body = RequestBody.create(requestBody, MediaType.parse("application/json"));

// create the request
String url = String.format("%s/user-permissions", this.config.getPdpAddress());
Request request = new Request.Builder()
.url(url)
.post(body)
.addHeader("Content-Type", "application/json")
.addHeader("Authorization", String.format("Bearer %s", this.config.getToken()))
.addHeader("X-Permit-SDK-Version", String.format("java:%s", this.config.version))
.build();

try (Response response = client.newCall(request).execute()) {
if (!response.isSuccessful()) {
String errorMessage = String.format(
"Error in permit.getUserPermissions(%s, %s, %s, %s): got unexpected status code %d",
input.user.toString(),
input.tenants.toString(),
input.resource_types.toString(),
input.resources.toString(),
response.code()
);
logger.error(errorMessage);
throw new IOException(errorMessage);
}
ResponseBody responseBody = response.body();
if (responseBody == null) {
String errorMessage = String.format(
"Error in permit.getUserPermissions(%s, %s, %s, %s): got empty response",
input.user.toString(),
input.tenants.toString(),
input.resource_types.toString(),
input.resources.toString()
);
logger.error(errorMessage);
throw new IOException(errorMessage);
}
String responseString = responseBody.string();
UserPermissions result = gson.fromJson(responseString, UserPermissions.class);
if (this.config.isDebugMode()) {
logger.info(String.format(
"permit.getUserPermissions(%s, %s, %s, %s) => returned %d permissions on %d objects",
input.user.toString(),
input.tenants != null ? input.tenants.toString() : "null",
input.resource_types != null ? input.resource_types.toString() : "null",
input.resources != null ? input.resources.toString() : "null",
result.values().stream().map(obj -> obj.permissions.size()).reduce(0, Integer::sum),
result.keySet().size()
));
}
return result;
}
}

@Override
public boolean checkUrl(User user, String httpMethod, String url, String tenant) throws IOException {
return this.checkUrl(user, httpMethod, url, tenant, new Context());
}

private String bulkCheckRepr(List<EnforcerInput> inputs) {
return String.format(
"permit.bulkCheck(%s)",
inputs.stream().map(i -> String.format(
"%s, %s, %s, %s",
i.user,
i.action,
i.resource,
i.context
)).collect(Collectors.toList())
);
}
}
Loading

0 comments on commit c388a80

Please sign in to comment.