Examples with SearchProvider fi.otavanopisto.muikku.search.SearchProvider used on opensource projects

Example 11 with SearchProvider

use of fi.otavanopisto.muikku.search.SearchProvider in project muikku by otavanopisto.

the class WorkspaceRESTService method listWorkspaceStudents.

@GET
@Path("/workspaces/{ID}/students")
@RESTPermit(handling = Handling.INLINE)
public Response listWorkspaceStudents(@PathParam("ID") Long workspaceEntityId, @QueryParam("active") Boolean active, @QueryParam("requestedAssessment") Boolean requestedAssessment, @QueryParam("assessed") Boolean assessed, @QueryParam("studentIdentifier") String studentId, @QueryParam("search") String searchString, @QueryParam("flags") Long[] flagIds, @QueryParam("maxResults") Integer maxResults, @QueryParam("orderBy") String orderBy) {
    List<SchoolDataIdentifier> studentIdentifiers = null;
    if (StringUtils.isNotBlank(studentId)) {
        SchoolDataIdentifier studentIdentifier = SchoolDataIdentifier.fromId(studentId);
        if (studentIdentifier == null) {
            return Response.status(Status.BAD_REQUEST).entity(String.format("Malformed student identifier %s", studentId)).build();
        }
        studentIdentifiers = Collections.singletonList(studentIdentifier);
    }
    // Workspace
    WorkspaceEntity workspaceEntity = workspaceController.findWorkspaceEntityById(workspaceEntityId);
    if (workspaceEntity == null) {
        return Response.status(Status.NOT_FOUND).build();
    }
    // Access check
    if (!sessionController.hasWorkspacePermission(MuikkuPermissions.LIST_WORKSPACE_MEMBERS, workspaceEntity)) {
        if (studentIdentifiers == null || studentIdentifiers.size() != 1 || !studentIdentifiers.get(0).equals(sessionController.getLoggedUser())) {
            return Response.status(Status.FORBIDDEN).build();
        }
    }
    List<fi.otavanopisto.muikku.schooldata.entity.WorkspaceUser> workspaceUsers = null;
    if (searchString != null) {
        studentIdentifiers = new ArrayList<>();
        Iterator<SearchProvider> searchProviderIterator = searchProviders.iterator();
        if (!searchProviderIterator.hasNext()) {
            return Response.status(Status.INTERNAL_SERVER_ERROR).entity("No search provider found").build();
        }
        SearchProvider elasticSearchProvider = searchProviderIterator.next();
        if (elasticSearchProvider != null) {
            String[] fields = new String[] { "firstName", "lastName", "nickName", "email" };
            SearchResult result = elasticSearchProvider.searchUsers(searchString, fields, Arrays.asList(EnvironmentRoleArchetype.STUDENT), (Collection<Long>) null, Collections.singletonList(workspaceEntityId), (Collection<SchoolDataIdentifier>) null, Boolean.FALSE, Boolean.FALSE, false, 0, maxResults != null ? maxResults : Integer.MAX_VALUE);
            List<Map<String, Object>> results = result.getResults();
            if (results != null && !results.isEmpty()) {
                for (Map<String, Object> o : results) {
                    String foundStudentId = (String) o.get("id");
                    if (StringUtils.isBlank(foundStudentId)) {
                        logger.severe("Could not process user found from search index because it had a null id");
                        continue;
                    }
                    String[] studentIdParts = foundStudentId.split("/", 2);
                    SchoolDataIdentifier foundStudentIdentifier = studentIdParts.length == 2 ? new SchoolDataIdentifier(studentIdParts[0], studentIdParts[1]) : null;
                    if (foundStudentIdentifier == null) {
                        logger.severe(String.format("Could not process user found from search index with id %s", studentId));
                        continue;
                    }
                    studentIdentifiers.add(foundStudentIdentifier);
                }
            }
        }
    }
    List<Flag> flags = null;
    if (flagIds != null && flagIds.length > 0) {
        flags = new ArrayList<>(flagIds.length);
        for (Long flagId : flagIds) {
            Flag flag = flagController.findFlagById(flagId);
            if (flag == null) {
                return Response.status(Status.BAD_REQUEST).entity(String.format("Invalid flag id %d", flagId)).build();
            }
            if (!flagController.hasFlagPermission(flag, sessionController.getLoggedUser())) {
                return Response.status(Status.FORBIDDEN).entity(String.format("You don't have permission to use flag %d", flagId)).build();
            }
            flags.add(flag);
        }
    }
    if (flags != null) {
        List<SchoolDataIdentifier> flaggedStudents = flagController.getFlaggedStudents(flags);
        if (studentIdentifiers != null) {
            studentIdentifiers.retainAll(flaggedStudents);
        } else {
            studentIdentifiers = flaggedStudents;
        }
    }
    List<WorkspaceStudent> result = new ArrayList<>();
    if (studentIdentifiers != null) {
        workspaceUsers = new ArrayList<>();
        for (SchoolDataIdentifier studentIdentifier : studentIdentifiers) {
            WorkspaceUserEntity wue = workspaceUserEntityController.findWorkspaceUserByWorkspaceEntityAndUserIdentifier(workspaceEntity, studentIdentifier);
            if (wue == null) {
                continue;
            }
            if (active != null && !active.equals(wue.getActive())) {
                continue;
            }
            WorkspaceUser workspaceUser = workspaceController.findWorkspaceUser(wue);
            if (workspaceUser == null) {
                continue;
            }
            workspaceUsers.add(workspaceUser);
        }
    } else {
        // Students via WorkspaceSchoolDataBridge
        workspaceUsers = workspaceController.listWorkspaceStudents(workspaceEntity);
    }
    if (workspaceUsers == null || workspaceUsers.isEmpty()) {
        return Response.noContent().build();
    }
    Map<String, WorkspaceUserEntity> workspaceUserEntityMap = new HashMap<>();
    List<WorkspaceUserEntity> workspaceUserEntities = workspaceUserEntityController.listWorkspaceUserEntities(workspaceEntity);
    for (WorkspaceUserEntity workspaceUserEntity : workspaceUserEntities) {
        workspaceUserEntityMap.put(new SchoolDataIdentifier(workspaceUserEntity.getIdentifier(), workspaceUserEntity.getUserSchoolDataIdentifier().getDataSource().getIdentifier()).toId(), workspaceUserEntity);
    }
    if (maxResults != null && workspaceUsers.size() > maxResults) {
        workspaceUsers.subList(maxResults, workspaceUsers.size() - 1).clear();
    }
    for (fi.otavanopisto.muikku.schooldata.entity.WorkspaceUser workspaceUser : workspaceUsers) {
        SchoolDataIdentifier workspaceUserIdentifier = workspaceUser.getIdentifier();
        WorkspaceUserEntity workspaceUserEntity = workspaceUserEntityMap.get(workspaceUserIdentifier.toId());
        if (workspaceUserEntity == null) {
            logger.log(Level.WARNING, String.format("Workspace student %s does not exist in Muikku", workspaceUserIdentifier.toId()));
            continue;
        }
        if (active == null || active.equals(workspaceUserEntity.getActive())) {
            if (requestedAssessment != null) {
                boolean hasAssessmentRequest = workspaceUserEntity != null && !assessmentRequestController.listByWorkspaceUser(workspaceUserEntity).isEmpty();
                if (requestedAssessment != hasAssessmentRequest) {
                    continue;
                }
            }
            if (assessed != null) {
                boolean isAssessed = !gradingController.listWorkspaceAssessments(workspaceUser.getWorkspaceIdentifier(), workspaceUser.getUserIdentifier()).isEmpty();
                if (assessed != isAssessed) {
                    continue;
                }
            }
            SchoolDataIdentifier userIdentifier = workspaceUser.getUserIdentifier();
            User user = userController.findUserByIdentifier(userIdentifier);
            if (user != null) {
                UserEntity userEntity = null;
                if (workspaceUserEntity != null) {
                    userEntity = workspaceUserEntity.getUserSchoolDataIdentifier().getUserEntity();
                } else {
                    userEntity = userEntityController.findUserEntityByDataSourceAndIdentifier(user.getSchoolDataSource(), user.getIdentifier());
                }
                result.add(createRestModel(userEntity, user, workspaceUser, workspaceUserEntity != null && workspaceUserEntity.getActive()));
            } else {
                logger.log(Level.SEVERE, String.format("Could not find user for identifier %s", userIdentifier));
            }
        }
    }
    // Sorting
    if (StringUtils.equals(orderBy, "name")) {
        Collections.sort(result, new Comparator<WorkspaceStudent>() {

            @Override
            public int compare(WorkspaceStudent o1, WorkspaceStudent o2) {
                String s1 = String.format("%s, %s", StringUtils.defaultString(o1.getLastName(), ""), StringUtils.defaultString(o1.getFirstName(), ""));
                String s2 = String.format("%s, %s", StringUtils.defaultString(o2.getLastName(), ""), StringUtils.defaultString(o2.getFirstName(), ""));
                return s1.compareTo(s2);
            }
        });
    }
    // Response
    return Response.ok(result).build();
}