edu.stanford.protege.webprotege.issues.GetCommentedEntitiesActionHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of webprotege-backend-service Show documentation
The newest version!
package edu.stanford.protege.webprotege.issues;

import edu.stanford.protege.webprotege.access.AccessManager;
import edu.stanford.protege.webprotege.common.PageRequest;
import edu.stanford.protege.webprotege.common.Pager;
import edu.stanford.protege.webprotege.common.UserId;
import edu.stanford.protege.webprotege.dispatch.AbstractProjectActionHandler;
import edu.stanford.protege.webprotege.entity.CommentedEntityData;
import edu.stanford.protege.webprotege.index.EntitiesInProjectSignatureIndex;
import edu.stanford.protege.webprotege.ipc.ExecutionContext;
import edu.stanford.protege.webprotege.mansyntax.render.HasGetRendering;
import org.semanticweb.owlapi.model.OWLEntity;

import javax.annotation.Nonnull;
import javax.inject.Inject;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import static com.google.common.base.Preconditions.checkNotNull;
import static edu.stanford.protege.webprotege.entity.CommentedEntityData.byEntity;
import static edu.stanford.protege.webprotege.entity.CommentedEntityData.byLastModified;
import static java.util.Comparator.comparing;
import static java.util.stream.Collectors.groupingBy;
import static java.util.stream.Collectors.toList;

/**
 * Matthew Horridge
 * Stanford Center for Biomedical Informatics Research
 * 7 Mar 2017
 */
public class GetCommentedEntitiesActionHandler extends AbstractProjectActionHandler {

    @Nonnull
    private final EntityDiscussionThreadRepository repository;

    @Nonnull
    private final HasGetRendering renderer;

    @Nonnull
    private final EntitiesInProjectSignatureIndex entitiesInSignature;

    @Inject
    public GetCommentedEntitiesActionHandler(@Nonnull AccessManager accessManager,
                                             @Nonnull EntityDiscussionThreadRepository repository,
                                             @Nonnull HasGetRendering renderer,
                                             @Nonnull EntitiesInProjectSignatureIndex entitiesInSignature) {
        super(accessManager);
        this.repository = checkNotNull(repository);
        this.renderer = checkNotNull(renderer);
        this.entitiesInSignature = checkNotNull(entitiesInSignature);
    }

    @Nonnull
    @Override
    public Class getActionClass() {
        return GetCommentedEntitiesAction.class;
    }

    @Nonnull
    @Override
    public GetCommentedEntitiesResult execute(@Nonnull GetCommentedEntitiesAction action,
                                              @Nonnull ExecutionContext executionContext) {
        PageRequest request = action.pageRequest();
        List allThreads = repository.getThreadsInProject(action.projectId());


        Map> commentsByEntity = allThreads.stream()
                .collect(groupingBy(
                        EntityDiscussionThread::getEntity));

        List result = new ArrayList<>();
        commentsByEntity.forEach((entity, threads) -> {
            if (entitiesInSignature.containsEntityInSignature(entity)) {
                int totalThreadCount = threads.size();
                int openThreadCount = (int) threads.stream()
                        .filter(thread -> thread.getStatus().isOpen())
                        .count();
                List entityComments = threads.stream()
                        .flatMap(thread -> thread.getComments()
                                .stream()).toList();
                Comment lastComment = entityComments.stream()
                        .max(comparing(c -> c.getUpdatedAt().orElse(c.getCreatedAt()))).get();

                List participants = entityComments.stream()
                        .map(Comment::getCreatedBy)
                        .collect(toList());
                result.add(new CommentedEntityData(
                        renderer.getRendering(entity),
                        totalThreadCount,
                        openThreadCount,
                        entityComments.size(),
                        lastComment.getUpdatedAt().orElse(lastComment.getCreatedAt()),
                        lastComment.getCreatedBy(),
                        participants
                ));
            }
        });
        if (action.sortingKey() == SortingKey.SORT_BY_ENTITY) {
            result.sort(byEntity);
        } else {
            result.sort(byLastModified);
        }
        Pager pager = Pager.getPagerForPageSize(result, request.getPageSize());
        return new GetCommentedEntitiesResult(action.projectId(), pager.getPage(request.getPageNumber()));
    }
}