TaskModel.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2020 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.dossier.task;
 
import java.util.ArrayList;
import java.util.Collection;
import java.util.function.Consumer;
import java.util.stream.Collectors;
import org.turro.dossier.db.DossierPU;
import org.turro.dossier.entity.Dossier;
import org.turro.dossier.entity.Issue;
import org.turro.dossier.entity.IssuePredecessor;
import org.turro.dossier.entity.IssueStatus;
import org.turro.elephant.db.WhereClause;
import org.turro.jpa.Dao;
 
public class TaskModel {
  
  private final Dao dao;
  private final Dossier dossier;
  private final boolean onlyOpen;
 
  public TaskModel(Dossier dossier, boolean onlyOpen) {
    this.dao = new DossierPU();
    this.dossier = dossier;
    this.onlyOpen = onlyOpen;
  }
 
  public Collection<TaskItem> getRoots() {
    WhereClause wc = new WhereClause();
    wc.addClause("select issue from Issue issue");
    wc.addClause("where issue.dossier = :dossier");
    wc.addNamedValue("dossier", dossier);
    if(onlyOpen) {
      wc.addClause("and issue.status in (:status)");
      wc.addNamedValue("status", IssueStatus.selectBy(null, null, null, false));
    }
    wc.addClause("and issue.sources is empty");
    return dao.getResultList(Issue.class, wc).stream().map((issue) -> {
      return new TaskItem(issue, null, 1);
    }).collect(Collectors.toList());
  }
  
  public Collection<TaskItem> getSubtasks(TaskItem item) {
    WhereClause wc = new WhereClause();
    wc.addClause("select predecessor from IssuePredecessor predecessor");
    wc.addClause("where predecessor.source = :issue");
    wc.addNamedValue("issue", item.getIssue());
    wc.addClause("and predecessor.source.dossier = :dossier");
    wc.addNamedValue("dossier", dossier);
    if(onlyOpen) {
      wc.addClause("and predecessor.target.status in (:status)");
      wc.addNamedValue("status", IssueStatus.selectBy(null, null, null, false));
    }
    return dao.getResultList(IssuePredecessor.class, wc).stream().map((predecessor) -> {
      return new TaskItem(predecessor.getTarget(), predecessor.getType(), item.getLevel() + 1);
    }).collect(Collectors.toList());
  }
  
  public Collection<TaskItem> getFlat(Collection<TaskItem> items) {
    ArrayList<TaskItem> list = new ArrayList<>();
    Consumer<TaskItem> addItem = item -> { list.add(item); item.getSubtasks().forEach((subtask) -> list.add(subtask));};
    items.forEach((item) -> {
      addItem.accept(item);
    });
    return list;
  }
 
}