HashtagsXRep/hashtagsxrep

package cat.xarxarepublicana.hashtagsxrep.application.poll;

import cat.xarxarepublicana.hashtagsxrep.domain.core.error.EntityNotFoundException;
import cat.xarxarepublicana.hashtagsxrep.domain.invite.InviteGroup;
import cat.xarxarepublicana.hashtagsxrep.domain.invite.InviteRepository;
import cat.xarxarepublicana.hashtagsxrep.domain.monitor.Monitor;
import cat.xarxarepublicana.hashtagsxrep.domain.monitor.MonitorRepository;
import cat.xarxarepublicana.hashtagsxrep.domain.poll.Poll;
import cat.xarxarepublicana.hashtagsxrep.domain.poll.PollRepository;
import cat.xarxarepublicana.hashtagsxrep.domain.poll.Proposal;
import cat.xarxarepublicana.hashtagsxrep.domain.user.User;
import java.util.List;

public class LoadPollUseCase {

  private final PollRepository pollRepository;
  private final InviteRepository inviteRepository;
  private final MonitorRepository monitorRepository;

  public LoadPollUseCase(
      PollRepository pollRepository,
      InviteRepository inviteRepository,
      MonitorRepository monitorRepository) {
    this.pollRepository = pollRepository;
    this.inviteRepository = inviteRepository;
    this.monitorRepository = monitorRepository;
  }

  public LoadPollResponse loadPoll(String pollId, User user) {
    Poll poll = pollRepository.findById(pollId);
    if (poll == null) {
      throw new EntityNotFoundException("Enquesta no trobada: " + pollId);
    }
    Proposal userProposal = pollRepository.findProposal(poll.getId(), user.getId());
    Proposal userVote = pollRepository.findUserVote(poll, user.getId());
    List<Proposal> proposalList = pollRepository.findPollProposals(poll);
    InviteGroup inviteGroup = inviteRepository.loadInvitesForPoll(poll.getId());

    Monitor monitor = null;
    if (poll.isVotingClosed()) {
      monitor = monitorRepository.findById(poll.getId());
    }
    return new LoadPollResponse(poll, userProposal, userVote, proposalList, inviteGroup, monitor);
  }

  public static class LoadPollResponse {
    private final Poll poll;
    private final Proposal userProposal;
    private final Proposal userVote;
    private final List<Proposal> pollProposals;
    private final InviteGroup inviteGroup;
    private final Monitor monitor;

    public LoadPollResponse(
        Poll poll,
        Proposal userProposal,
        Proposal userVote,
        List<Proposal> pollProposals,
        InviteGroup inviteGroup,
        Monitor monitor) {
      this.poll = poll;
      this.userProposal = userProposal;
      this.userVote = userVote;
      this.pollProposals = pollProposals;
      this.inviteGroup = inviteGroup;
      this.monitor = monitor;
    }

    public Poll getPoll() {
      return poll;
    }

    public Proposal getUserProposal() {
      return userProposal;
    }

    public Proposal getUserVote() {
      return userVote;
    }

    public List<Proposal> getPollProposals() {
      return pollProposals;
    }

    public InviteGroup getInviteGroup() {
      return inviteGroup;
    }

    public Monitor getMonitor() {
      return monitor;
    }

    public Integer getPollVoteCount() {
      if (getPollProposals() == null) {
        return 0;
      }
      int count = 0;
      for (Proposal proposal : getPollProposals()) {
        count += proposal.getVotes();
      }
      return count;
    }

    public boolean isDraw() {
      return pollProposals.size() > 1 && pollProposals.get(0).getVotes() == pollProposals.get(1).getVotes();
    }
  }
}