PortfolioSet.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2011 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.financials.portfolio;
 
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.TreeSet;
import org.turro.action.IAcceptances;
import org.turro.action.Plugins;
import org.turro.elephant.db.WhereClause;
import org.turro.financials.db.FinancialsPU;
import org.turro.financials.entity.Document;
import org.turro.financials.entity.DocumentDefinition;
import org.turro.financials.entity.DocumentWorkflow;
import org.turro.jpa.Dao;
import org.turro.plugin.filter.IFilterValue;
import org.turro.zkoss.grid.GroupExtended;
import org.zkoss.zul.Grid;
import org.zkoss.zul.Group;
 
public class PortfolioSet extends TreeSet<PortfolioItem> {
 
  private DocumentDefinition documentDefinition;
  private long lastContract = 0, lastDocDef = 0;
 
  public void setDocumentDefinition(DocumentDefinition documentDefinition) {
    this.documentDefinition = documentDefinition;
  }
 
  public Collection<Document> getDocuments() {
    ArrayList<Document> docs = new ArrayList<Document>();
    for(PortfolioItem item : this) {
      docs.add(item.getDocument());
    }
    return docs;
  }
 
  public void load(RelationMode mode, List<IFilterValue> values) {
    clear();
    lastContract = 0;
    lastDocDef = 0;
    Dao dao = new FinancialsPU();
    IAcceptances acceptances = Plugins.loadImplementation(IAcceptances.class);
    WhereClause wc = createCriteria(mode, values);
    for(Object[] o : (List<Object[]>) dao.getResultList(wc)) {
      if(acceptances.getStatusFor(FinancialsPU.getObjectPath(o[0])).isPass()) {
        addDocument((Document) o[0], (DocumentWorkflow) o[1]);
      }
    }
  }
 
  public void addDocument(Document document, DocumentWorkflow workflow) {
    PortfolioItem item = getItem(document);
    if(item == null) {
      item = new PortfolioItem(document);
      add(item);
    }
    if(workflow.getAncestor().getId() == document.getDocumentDefinition().getId()) {
      item.addForward(workflow);
    } else {
      item.addBackward(workflow);
    }
  }
 
  public void checkGroup(Grid grid, Document document) {
    if((lastContract == 0 || lastContract != document.getContract().getId()) ||
      (lastDocDef == 0 || lastDocDef != document.getDocumentDefinition().getId())) {
      lastContract = document.getContract().getId();
      lastDocDef = document.getDocumentDefinition().getId();
      Group group = new GroupExtended(
              document.getContract().getFullDescription() +
              " - " +
              document.getDocumentDefinition().getName());
      group.setOpen(true);
      grid.getRows().appendChild(group);
    }
  }
 
  private WhereClause createCriteria(RelationMode mode, List<IFilterValue> values) {
    WhereClause wc = new WhereClause();
    wc.addClause("select doc, dw from Document as doc, DocumentWorkflow as dw");
    wc.addClause("where 1=1");
    for(IFilterValue v : values) {
      if(!v.getLabel().equals("Document model")) {
        v.addConstraint(wc);
      }
    }
    wc.addClause("and doc.contract.contractDefinition = dw.contractDefinition");
 
    String sep = "";
    wc.addClause("and (");
    if(mode.equals(RelationMode.RELATE_FORWARD) || mode.equals(RelationMode.RELATE_BOTH)) {
      wc.addClause(sep);
      sep = "or";
      wc.addClause("(");
      wc.addClause("doc.documentDefinition = dw.ancestor");
      for(IFilterValue v : values) {
        if(v.getLabel().equals("Document model")) {
          wc.getAttributes().put("mode", RelationMode.RELATE_FORWARD);
          v.addConstraint(wc);
        }
      }
      if(documentDefinition != null) {
        wc.addClause("and dw.descendant = :docDefDes");
        wc.addNamedValue("docDefDes", documentDefinition);
      }
      wc.addClause("and doc.noDescendants = FALSE");
      wc.addClause("and not exists (");
      wc.addClause("select dr from DocumentRelation as dr");
      wc.addClause("where dr.ancestor = doc");
      wc.addClause("))");
    }
    if(mode.equals(RelationMode.RELATE_BACKWARD) || mode.equals(RelationMode.RELATE_BOTH)) {
      wc.addClause(sep);
      wc.addClause("(");
      wc.addClause("doc.documentDefinition = dw.descendant");
      wc.addClause("and dw.allowNewDescendants = TRUE");
      for(IFilterValue v : values) {
        if(v.getLabel().equals("Document model")) {
          wc.getAttributes().put("mode", RelationMode.RELATE_BACKWARD);
          v.addConstraint(wc);
        }
      }
      if(documentDefinition != null) {
        wc.addClause("and dw.ancestor = :docDefAnc");
        wc.addNamedValue("docDefAnc", documentDefinition);
      }
      wc.addClause("and doc.noAncestors = FALSE");
      wc.addClause("and not exists (");
      wc.addClause("select dr from DocumentRelation as dr");
      wc.addClause("where dr.descendant = doc");
      wc.addClause("))");
    }
    wc.addClause(")");
 
    wc.addClause("order by doc.contract.id, doc.documentDefinition.id, doc.documentDate");
    return wc;
  }
 
  private PortfolioItem getItem(Document document) {
    for(PortfolioItem item : this) {
      if(item.getDocument().getId() == document.getId()) {
        return item;
      }
    }
    return null;
  }
 
}