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

Example 11 with TenantEntity

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

the class AccessPolicyEntityMerger method mergeDtos.

private static void mergeDtos(final AccessPolicyDTO clientDto, final Map<NodeIdentifier, AccessPolicyDTO> dtoMap) {
    // if unauthorized for the client dto, simple return
    if (clientDto == null) {
        return;
    }
    final Set<TenantEntity> users = new HashSet<>(clientDto.getUsers());
    final Set<TenantEntity> userGroups = new HashSet<>(clientDto.getUserGroups());
    for (final Map.Entry<NodeIdentifier, AccessPolicyDTO> nodeEntry : dtoMap.entrySet()) {
        final AccessPolicyDTO nodeAccessPolicy = nodeEntry.getValue();
        if (nodeAccessPolicy != null) {
            users.retainAll(nodeAccessPolicy.getUsers());
            userGroups.retainAll(nodeAccessPolicy.getUserGroups());
        }
    }
    clientDto.setUsers(users);
    clientDto.setUserGroups(userGroups);
}

Example 12 with TenantEntity

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

the class UserEntityMerger method mergeDtos.

private static void mergeDtos(final UserDTO clientDto, final Map<NodeIdentifier, UserDTO> dtoMap) {
    // if unauthorized for the client dto, simple return
    if (clientDto == null) {
        return;
    }
    final Set<AccessPolicySummaryEntity> accessPolicyEntities = new HashSet<>(clientDto.getAccessPolicies());
    final Set<TenantEntity> tenantEntities = new HashSet<>(clientDto.getUserGroups());
    for (final Map.Entry<NodeIdentifier, UserDTO> nodeEntry : dtoMap.entrySet()) {
        final UserDTO nodeUser = nodeEntry.getValue();
        if (nodeUser != null) {
            accessPolicyEntities.retainAll(nodeUser.getAccessPolicies());
            tenantEntities.retainAll(nodeUser.getUserGroups());
        }
    }
    clientDto.setAccessPolicies(accessPolicyEntities);
    clientDto.setUserGroups(tenantEntities);
}

Example 13 with TenantEntity

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

the class AccessPolicyEntityMergerTest method testMergeAccessPolicy.

@Test
public void testMergeAccessPolicy() throws Exception {
    final NodeIdentifier node1 = new NodeIdentifier("node-1", "host-1", 8080, "host-1", 19998, null, null, null, false);
    final NodeIdentifier node2 = new NodeIdentifier("node-2", "host-2", 8081, "host-2", 19999, null, null, null, false);
    final PermissionsDTO permissed = new PermissionsDTO();
    permissed.setCanRead(true);
    permissed.setCanWrite(true);
    final TenantDTO user1DTO = new TenantDTO();
    user1DTO.setId("user-1");
    final TenantEntity user1Entity = new TenantEntity();
    user1Entity.setPermissions(permissed);
    user1Entity.setId(user1DTO.getId());
    user1Entity.setComponent(user1DTO);
    final TenantDTO user2DTO = new TenantDTO();
    user1DTO.setId("user-2");
    final TenantEntity user2Entity = new TenantEntity();
    user2Entity.setPermissions(permissed);
    user2Entity.setId(user2DTO.getId());
    user2Entity.setComponent(user2DTO);
    final AccessPolicyDTO accessPolicy1DTO = new AccessPolicyDTO();
    accessPolicy1DTO.setId("policy-1");
    accessPolicy1DTO.setUsers(Stream.of(user1Entity, user2Entity).collect(Collectors.toSet()));
    accessPolicy1DTO.setUserGroups(Stream.of(user2Entity).collect(Collectors.toSet()));
    final AccessPolicyEntity accessPolicy1Entity = new AccessPolicyEntity();
    accessPolicy1Entity.setPermissions(permissed);
    accessPolicy1Entity.setId(accessPolicy1DTO.getId());
    accessPolicy1Entity.setComponent(accessPolicy1DTO);
    final AccessPolicyDTO accessPolicy2DTO = new AccessPolicyDTO();
    accessPolicy2DTO.setId("policy-2");
    accessPolicy2DTO.setUsers(Stream.of(user1Entity).collect(Collectors.toSet()));
    accessPolicy2DTO.setUserGroups(Stream.of(user1Entity, user2Entity).collect(Collectors.toSet()));
    final AccessPolicyEntity accessPolicy2Entity = new AccessPolicyEntity();
    accessPolicy2Entity.setPermissions(permissed);
    accessPolicy2Entity.setId(accessPolicy2DTO.getId());
    accessPolicy2Entity.setComponent(accessPolicy2DTO);
    final Map<NodeIdentifier, AccessPolicyEntity> nodeMap = new HashMap<>();
    nodeMap.put(node1, accessPolicy1Entity);
    nodeMap.put(node2, accessPolicy2Entity);
    final AccessPolicyEntityMerger merger = new AccessPolicyEntityMerger();
    merger.merge(accessPolicy1Entity, nodeMap);
    assertEquals(1, accessPolicy1DTO.getUserGroups().size());
    assertTrue(accessPolicy1DTO.getUsers().contains(user1Entity));
    assertEquals(1, accessPolicy1DTO.getUserGroups().size());
    assertTrue(accessPolicy1DTO.getUserGroups().contains(user2Entity));
}

Example 14 with TenantEntity

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

the class UserEntityMergerTest method testMergeAccessPolicy.

@Test
public void testMergeAccessPolicy() throws Exception {
    final NodeIdentifier node1 = new NodeIdentifier("node-1", "host-1", 8080, "host-1", 19998, null, null, null, false);
    final NodeIdentifier node2 = new NodeIdentifier("node-2", "host-2", 8081, "host-2", 19999, null, null, null, false);
    final PermissionsDTO permissed = new PermissionsDTO();
    permissed.setCanRead(true);
    permissed.setCanWrite(true);
    final TenantDTO userGroup1DTO = new TenantDTO();
    userGroup1DTO.setId("user-group-1");
    final TenantEntity userGroup1Entity = new TenantEntity();
    userGroup1Entity.setPermissions(permissed);
    userGroup1Entity.setId(userGroup1DTO.getId());
    userGroup1Entity.setComponent(userGroup1DTO);
    final TenantDTO userGroup2DTO = new TenantDTO();
    userGroup1DTO.setId("user-group-2");
    final TenantEntity userGroup2Entity = new TenantEntity();
    userGroup2Entity.setPermissions(permissed);
    userGroup2Entity.setId(userGroup2DTO.getId());
    userGroup2Entity.setComponent(userGroup2DTO);
    final AccessPolicySummaryDTO policy1DTO = new AccessPolicySummaryDTO();
    policy1DTO.setId("policy-1");
    final AccessPolicySummaryEntity policy1Entity = new AccessPolicySummaryEntity();
    policy1Entity.setPermissions(permissed);
    policy1Entity.setId(policy1DTO.getId());
    policy1Entity.setComponent(policy1DTO);
    final AccessPolicySummaryDTO policy2DTO = new AccessPolicySummaryDTO();
    policy2DTO.setId("policy-2");
    final AccessPolicySummaryEntity policy2Entity = new AccessPolicySummaryEntity();
    policy2Entity.setPermissions(permissed);
    policy2Entity.setId(policy2DTO.getId());
    policy2Entity.setComponent(policy2DTO);
    final UserDTO user1DTO = new UserDTO();
    user1DTO.setId("user-1");
    user1DTO.setAccessPolicies(Stream.of(policy1Entity, policy2Entity).collect(Collectors.toSet()));
    user1DTO.setUserGroups(Stream.of(userGroup2Entity).collect(Collectors.toSet()));
    final UserEntity user1Entity = new UserEntity();
    user1Entity.setPermissions(permissed);
    user1Entity.setId(user1DTO.getId());
    user1Entity.setComponent(user1DTO);
    final UserDTO user2DTO = new UserDTO();
    user2DTO.setId("user-2");
    user2DTO.setAccessPolicies(Stream.of(policy1Entity).collect(Collectors.toSet()));
    user2DTO.setUserGroups(Stream.of(userGroup1Entity, userGroup2Entity).collect(Collectors.toSet()));
    final UserEntity user2Entity = new UserEntity();
    user2Entity.setPermissions(permissed);
    user2Entity.setId(user2DTO.getId());
    user2Entity.setComponent(user2DTO);
    final Map<NodeIdentifier, UserEntity> nodeMap = new HashMap<>();
    nodeMap.put(node1, user1Entity);
    nodeMap.put(node2, user2Entity);
    final UserEntityMerger merger = new UserEntityMerger();
    merger.merge(user1Entity, nodeMap);
    assertEquals(1, user1DTO.getUserGroups().size());
    assertTrue(user1DTO.getAccessPolicies().contains(policy1Entity));
    assertEquals(1, user1DTO.getUserGroups().size());
    assertTrue(user1DTO.getUserGroups().contains(userGroup2Entity));
}

Example 15 with TenantEntity

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

the class StandardNiFiServiceFacade method deleteUserGroup.

@Override
public UserGroupEntity deleteUserGroup(final Revision revision, final String userGroupId) {
    final Group userGroup = userGroupDAO.getUserGroup(userGroupId);
    final PermissionsDTO permissions = dtoFactory.createPermissionsDto(authorizableLookup.getTenant());
    final Set<TenantEntity> users = userGroup != null ? userGroup.getUsers().stream().map(mapUserIdToTenantEntity()).collect(Collectors.toSet()) : null;
    final Set<AccessPolicySummaryEntity> policyEntities = userGroupDAO.getAccessPoliciesForUserGroup(userGroup.getIdentifier()).stream().map(ap -> createAccessPolicySummaryEntity(ap)).collect(Collectors.toSet());
    final String resourceIdentifier = ResourceFactory.getTenantResource().getIdentifier() + "/" + userGroupId;
    final UserGroupDTO snapshot = deleteComponent(revision, new Resource() {

        @Override
        public String getIdentifier() {
            return resourceIdentifier;
        }

        @Override
        public String getName() {
            return resourceIdentifier;
        }

        @Override
        public String getSafeDescription() {
            return "User Group " + userGroupId;
        }
    }, () -> userGroupDAO.deleteUserGroup(userGroupId), // no user group specific policies to remove
    false, dtoFactory.createUserGroupDto(userGroup, users, policyEntities));
    return entityFactory.createUserGroupEntity(snapshot, null, permissions);
}