io.quarkiverse.openfga.client.AuthorizationModelClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of quarkus-openfga-client Show documentation
There is a newer version: 2.1.0
Show newest version
package io.quarkiverse.openfga.client;

import static io.quarkiverse.openfga.client.utils.PaginatedList.collectAllPages;

import java.util.List;
import java.util.Map;

import javax.annotation.Nullable;

import io.quarkiverse.openfga.client.api.API;
import io.quarkiverse.openfga.client.model.*;
import io.quarkiverse.openfga.client.model.dto.*;
import io.quarkiverse.openfga.client.utils.PaginatedList;
import io.smallrye.mutiny.Uni;

public class AuthorizationModelClient {

    private final API api;
    private final Uni storeId;
    private final String authorizationModelId;

    public AuthorizationModelClient(API api, Uni storeId, @Nullable String authorizationModelId) {
        this.api = api;
        this.storeId = storeId;
        this.authorizationModelId = authorizationModelId;
    }

    public Uni get() {
        return storeId.flatMap(storeId -> api.readAuthorizationModel(storeId, authorizationModelId))
                .map(ReadAuthorizationModelResponse::getAuthorizationModel);
    }

    public Uni check(TupleKey tupleKey, @Nullable ContextualTupleKeys contextualTupleKeys) {
        return storeId
                .flatMap(
                        storeId -> api.check(storeId, new CheckBody(tupleKey, contextualTupleKeys, authorizationModelId, null)))
                .map(CheckResponse::getAllowed);
    }

    public Uni expand(TupleKey tupleKey) {
        return storeId.flatMap(storeId -> api.expand(storeId, new ExpandBody(tupleKey, authorizationModelId)))
                .map(ExpandResponse::getTree);
    }

    public Uni> listObjects(String type, @Nullable String relation, String user,
            @Nullable List contextualTupleKeys) {
        return storeId.flatMap(storeId -> api.listObjects(storeId,
                new ListObjectsBody(authorizationModelId, type, relation, user,
                        contextualTupleKeys != null ? new ContextualTupleKeys(contextualTupleKeys) : null)))
                .map(ListObjectsResponse::getObjects);
    }

    public Uni> queryTuples(PartialTupleKey tupleKey, @Nullable Integer pageSize,
            @Nullable String pagingToken) {
        return storeId
                .flatMap(storeId -> api.read(storeId, new ReadBody(tupleKey, authorizationModelId, pageSize, pagingToken)))
                .map(res -> new PaginatedList<>(res.getTuples(), res.getContinuationToken()));
    }

    public Uni> queryAllTuples(PartialTupleKey tupleKey) {
        return queryAllTuples(tupleKey, null);
    }

    public Uni> queryAllTuples(PartialTupleKey tupleKey, @Nullable Integer pageSize) {
        return collectAllPages(pageSize, (currentPageSize, currentToken) -> {
            return queryTuples(tupleKey, currentPageSize, currentToken);
        });
    }

    public Uni> readTuples(@Nullable Integer pageSize, @Nullable String pagingToken) {
        return storeId.flatMap(storeId -> api.read(storeId, new ReadBody(null, authorizationModelId, pageSize, pagingToken)))
                .map(res -> new PaginatedList<>(res.getTuples(), res.getContinuationToken()));
    }

    public Uni> readAllTuples() {
        return readAllTuples(null);
    }

    public Uni> readAllTuples(@Nullable Integer pageSize) {
        return collectAllPages(pageSize, this::readTuples);
    }

    public Uni write(TupleKey tupleKey) {
        return write(List.of(tupleKey), null)
                .replaceWithVoid();
    }

    public Uni> write(@Nullable List writes, @Nullable List deletes) {
        var writeKeys = TupleKeys.of(writes);
        var deleteKeys = TupleKeys.of(deletes);
        return storeId.flatMap(storeId -> api.write(storeId, new WriteBody(writeKeys, deleteKeys, authorizationModelId)))
                .map(WriteResponse::getValues);
    }

}