PollsUtil.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2018 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.polls;
 
import java.util.Date;
import java.util.List;
import java.util.Set;
import java.util.TreeSet;
import org.turro.action.Contacts;
import org.turro.elephant.db.ElephantPU;
import org.turro.elephant.db.WhereClause;
import org.turro.elephant.entities.db.Poll;
import org.turro.elephant.entities.db.PollOption;
import org.turro.elephant.entities.db.PollOptionPK;
import org.turro.elephant.entities.db.PollVote;
import org.turro.elephant.entities.db.PollVotePK;
import org.turro.jpa.Dao;
import org.turro.plugin.contacts.IContact;
 
public class PollsUtil {
 
  private final IContact contact;
 
  public PollsUtil(IContact contact) {
    this.contact = contact;
  }
  
  public PollsUtil(String idContact) {
    this.contact = Contacts.getContactById(idContact);
  }
 
  public IContact getContact() {
    return contact;
  }
 
  public boolean vote(Long pollId, String pollData, Integer vote) {
    return vote(getPollOption(pollId, pollData), vote);
  }
 
  public synchronized boolean vote(PollOption pollOption, Integer vote) {
    if(pollOption != null) {
      PollVote pollVote = getPollVote(pollOption);
      if(pollVote != null && pollVote.getVote() == vote) {
        // Only 0/1, delete 0
        getDao().deleteObject(pollVote);
        return true;
      }
      if(pollVote == null) {
        if(canVote(pollOption)) {
          pollVote = new PollVote();
          pollVote.setIdContact(contact.getId());
          pollVote.setIdPoll(pollOption.getIdPoll());
          pollVote.setPollData(pollOption.getPollData());
        }
      }
      if(pollVote != null) {
        pollVote.setVoteDate(new Date());
        pollVote.setVote(vote);
        getDao().saveObject(pollVote);
        return true;
      }
    }
    return false;
  }
 
  public boolean isAllowedToVote(PollOption pollOption) {
    if(pollOption != null) {
      return getPollVote(pollOption) != null || canVote(pollOption);
    }
    return false;
  }
 
  private boolean canVote(PollOption pollOption) {
    if(!pollOption.getPoll().isParticipant(contact)) {
      return false;
    }
    boolean canVote = true;
    Poll poll = pollOption.getPoll();
    if(poll.getDeadline() != null && poll.getDeadline().before(new Date())) {
      canVote = false;
    }
    if(canVote) {
      List<PollVote> votes = pollOption.getVotes();
      if(poll.getMaxPerOption() > 0 && votes.size() >= poll.getMaxPerOption()) {
        canVote = false;
      }
    }
    if(canVote) {
      List<PollVote> contactVotes = poll.getVotes(contact);
      if(poll.getMaxVotedOptions() > 0 && contactVotes.size() >= poll.getMaxVotedOptions()) {
        canVote = false;
      }
    }
    return canVote;
  }
 
  public PollOption getPollOption(Long pollId, String pollData) {
    PollOptionPK pk = new PollOptionPK();
    pk.setIdPoll(pollId);
    pk.setPollData(pollData);
    return getDao().find(PollOption.class, pk);
  }
 
  public PollVote getPollVote(PollOption pollOption) {
    return getPollVote(pollOption, contact);
  }
  
  public PollVote getPollVote(PollOption pollOption, IContact contact) {
    PollVotePK pk = new PollVotePK();
    pk.setIdPoll(pollOption.getIdPoll());
    pk.setPollData(pollOption.getPollData());
    pk.setIdContact(contact.getId());
    return getDao().find(PollVote.class, pk, null);
  }
  
  public long getPollOptionCount(PollOption pollOption, int vote) {
    WhereClause wc = new WhereClause();
    wc.addClause("select count(vote) from PollVote vote");
    wc.addClause("where vote.idPoll = :idPoll");
    wc.addNamedValue("idPoll", pollOption.getIdPoll());
    wc.addClause("and vote.pollData = :pollData");
    wc.addNamedValue("pollData", pollOption.getPollData());
    wc.addClause("and vote.vote = :vote");
    wc.addNamedValue("vote", vote);
    return (long) getDao().getSingleResultOrNull(wc);
  }
 
  public static List<Poll> getPolls(String entityPath) {
    Dao dao = new ElephantPU();
    WhereClause wc = new WhereClause();
    wc.addClause("select poll from Poll as poll");
    wc.addClause("where poll.entityPath = :path");
    wc.addNamedValue("path", entityPath);
    return dao.getResultList(wc);
  }
  
  public static Poll savePoll(Poll poll, List<PollOption> pos) {
    Dao dao = new ElephantPU();
    if(!poll.isEmpty()) {
      poll = dao.saveObject(poll);
      for(PollOption po : pos) {
        if(!po.isEmpty()) {
          dao.saveObject(po);
        }
      }
    }
    return poll;
  }
 
  public static void removePoll(Poll poll) {
    if(poll == null) return;
    Dao dao = new ElephantPU();
    WhereClause wc = new WhereClause();
    wc.addClause("delete from Poll");
    wc.addClause("where id = :idPoll");
    wc.addNamedValue("idPoll", poll.getId());
    dao.executeUpdate(wc);
    wc = new WhereClause();
    wc.addClause("delete from PollOption");
    wc.addClause("where idPoll = :idPoll");
    wc.addNamedValue("idPoll", poll.getId());
    dao.executeUpdate(wc);
    wc = new WhereClause();
    wc.addClause("delete from PollVote");
    wc.addClause("where idPoll = :idPoll");
    wc.addNamedValue("idPoll", poll.getId());
    dao.executeUpdate(wc);
  }
 
  public static void removeOption(PollOption po) {
    if(po == null) return;
    Dao dao = new ElephantPU();
    WhereClause wc = new WhereClause();
    wc.addClause("delete from PollOption");
    wc.addClause("where idPoll = :idPoll");
    wc.addNamedValue("idPoll", po.getIdPoll());
    wc.addClause("and pollData = :pollData");
    wc.addNamedValue("pollData", po.getPollData());
    dao.executeUpdate(wc);
    wc = new WhereClause();
    wc.addClause("delete from PollVote");
    wc.addClause("where idPoll = :idPoll");
    wc.addNamedValue("idPoll", po.getIdPoll());
    wc.addClause("and pollData = :pollData");
    wc.addNamedValue("pollData", po.getPollData());
    dao.executeUpdate(wc);
  }
 
  /* My polls */
  
  public static Set<Poll> getMyPolls(IContact contact) {
    return getPolls(contact, false, true, 0);
  }
  
  public static Set<Poll> getPolls(IContact contact, boolean active, boolean restricted, int maxResults) {
    Dao dao = new ElephantPU();
    TreeSet<Poll> polls = new TreeSet<>();
    if(!active) {
      WhereClause wc = new WhereClause();
      wc.addClause("select distinct poll from Poll poll");
      if(restricted) {
        wc.addClause("join PollVote vote on vote.idPoll = poll.id");
        wc.addClause("where vote.idContact = :contact");
        wc.addNamedValue("contact", contact.getId());
      }
      wc.addClause("order by poll.deadline desc");
      polls.addAll(dao.getResultList(wc, maxResults));
    }
    WhereClause wc = new WhereClause();
    wc.addClause("select poll from Poll poll");
    wc.addClause("where poll.deadline >= current_date");
    if(restricted) {
      for(Poll poll : (List<Poll>) dao.getResultList(wc)) {
        if(poll.isParticipant(contact)) {
          polls.add(poll);
        }
      }
    }
    return polls;
  }
  
  /* Dao */
  
  private Dao _dao;
 
  private Dao getDao() {
    if (_dao == null) {
      _dao = new ElephantPU();
    }
    return _dao;
  }
 
}