Examples with TenantEntity org.apache.nifi.web.api.entity.TenantEntity used on opensource projects

Example 6 with TenantEntity

use of org.apache.nifi.web.api.entity.TenantEntity in project nifi by apache.

the class SearchUsersEndpointMerger method merge.

@Override
public final NodeResponse merge(final URI uri, final String method, final Set<NodeResponse> successfulResponses, final Set<NodeResponse> problematicResponses, final NodeResponse clientResponse) {
    if (!canHandle(uri, method)) {
        throw new IllegalArgumentException("Cannot use Endpoint Mapper of type " + getClass().getSimpleName() + " to map responses for URI " + uri + ", HTTP Method " + method);
    }
    final TenantsEntity responseEntity = clientResponse.getClientResponse().readEntity(TenantsEntity.class);
    final Collection<TenantEntity> userEntities = responseEntity.getUsers();
    final Collection<TenantEntity> userGroupEntities = responseEntity.getUserGroups();
    for (final NodeResponse nodeResponse : successfulResponses) {
        final TenantsEntity nodeResponseEntity = nodeResponse == clientResponse ? responseEntity : nodeResponse.getClientResponse().readEntity(TenantsEntity.class);
        // only retain users/groups that all nodes agree on
        userEntities.retainAll(nodeResponseEntity.getUsers());
        userGroupEntities.retainAll(nodeResponseEntity.getUserGroups());
    }
    // create a new client response
    return new NodeResponse(clientResponse, responseEntity);
}

Example 7 with TenantEntity

use of org.apache.nifi.web.api.entity.TenantEntity in project nifi by apache.

the class StandardNiFiServiceFacade method createUserGroupEntity.

private UserGroupEntity createUserGroupEntity(final Group userGroup) {
    final RevisionDTO userGroupRevision = dtoFactory.createRevisionDTO(revisionManager.getRevision(userGroup.getIdentifier()));
    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(authorizableLookup.getTenant());
    final Set<TenantEntity> users = userGroup.getUsers().stream().map(mapUserIdToTenantEntity()).collect(Collectors.toSet());
    final Set<AccessPolicySummaryEntity> policyEntities = userGroupDAO.getAccessPoliciesForUserGroup(userGroup.getIdentifier()).stream().map(ap -> createAccessPolicySummaryEntity(ap)).collect(Collectors.toSet());
    return entityFactory.createUserGroupEntity(dtoFactory.createUserGroupDto(userGroup, users, policyEntities), userGroupRevision, permissions);
}

Example 8 with TenantEntity

use of org.apache.nifi.web.api.entity.TenantEntity in project nifi by apache.

the class StandardNiFiServiceFacade method deleteAccessPolicy.

@Override
public AccessPolicyEntity deleteAccessPolicy(final Revision revision, final String accessPolicyId) {
    final AccessPolicy accessPolicy = accessPolicyDAO.getAccessPolicy(accessPolicyId);
    final ComponentReferenceEntity componentReference = createComponentReferenceEntity(accessPolicy.getResource());
    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(authorizableLookup.getAccessPolicyById(accessPolicyId));
    final Set<TenantEntity> userGroups = accessPolicy != null ? accessPolicy.getGroups().stream().map(mapUserGroupIdToTenantEntity()).collect(Collectors.toSet()) : null;
    final Set<TenantEntity> users = accessPolicy != null ? accessPolicy.getUsers().stream().map(mapUserIdToTenantEntity()).collect(Collectors.toSet()) : null;
    final AccessPolicyDTO snapshot = deleteComponent(revision, new Resource() {

        @Override
        public String getIdentifier() {
            return accessPolicy.getResource();
        }

        @Override
        public String getName() {
            return accessPolicy.getResource();
        }

        @Override
        public String getSafeDescription() {
            return "Policy " + accessPolicyId;
        }
    }, () -> accessPolicyDAO.deleteAccessPolicy(accessPolicyId), // no need to clean up any policies as it's already been removed above
    false, dtoFactory.createAccessPolicyDto(accessPolicy, userGroups, users, componentReference));
    return entityFactory.createAccessPolicyEntity(snapshot, null, permissions);
}

Example 9 with TenantEntity

use of org.apache.nifi.web.api.entity.TenantEntity in project nifi by apache.

the class StandardNiFiServiceFacade method createUser.

@Override
public UserEntity createUser(final Revision revision, final UserDTO userDTO) {
    final String creator = NiFiUserUtils.getNiFiUserIdentity();
    final User newUser = userDAO.createUser(userDTO);
    final Set<TenantEntity> tenantEntities = userGroupDAO.getUserGroupsForUser(newUser.getIdentifier()).stream().map(g -> g.getIdentifier()).map(mapUserGroupIdToTenantEntity()).collect(Collectors.toSet());
    final Set<AccessPolicySummaryEntity> policyEntities = userGroupDAO.getAccessPoliciesForUser(newUser.getIdentifier()).stream().map(ap -> createAccessPolicySummaryEntity(ap)).collect(Collectors.toSet());
    final UserDTO newUserDto = dtoFactory.createUserDto(newUser, tenantEntities, policyEntities);
    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(authorizableLookup.getTenant());
    return entityFactory.createUserEntity(newUserDto, dtoFactory.createRevisionDTO(new FlowModification(revision, creator)), permissions);
}

Example 10 with TenantEntity

use of org.apache.nifi.web.api.entity.TenantEntity in project nifi by apache.

the class StandardNiFiServiceFacade method createUserGroup.

@Override
public UserGroupEntity createUserGroup(final Revision revision, final UserGroupDTO userGroupDTO) {
    final String creator = NiFiUserUtils.getNiFiUserIdentity();
    final Group newUserGroup = userGroupDAO.createUserGroup(userGroupDTO);
    final Set<TenantEntity> tenantEntities = newUserGroup.getUsers().stream().map(mapUserIdToTenantEntity()).collect(Collectors.toSet());
    final Set<AccessPolicySummaryEntity> policyEntities = userGroupDAO.getAccessPoliciesForUserGroup(newUserGroup.getIdentifier()).stream().map(ap -> createAccessPolicySummaryEntity(ap)).collect(Collectors.toSet());
    final UserGroupDTO newUserGroupDto = dtoFactory.createUserGroupDto(newUserGroup, tenantEntities, policyEntities);
    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(authorizableLookup.getTenant());
    return entityFactory.createUserGroupEntity(newUserGroupDto, dtoFactory.createRevisionDTO(new FlowModification(revision, creator)), permissions);
}