bsfinancials-core/src/main/java/org/turro/financials/model/document/DocumentWrapper.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.model.document;
 
import java.util.*;
import org.turro.financials.db.FinancialsPU;
import org.turro.financials.entity.*;
import org.turro.financials.model.contract.ContractWrapper;
import org.turro.financials.model.register.RegisterWrapper;
import org.turro.financials.model.register.ViewWrapper;
import org.turro.jpa.Dao;
import org.turro.jpa.entity.DaoEntity;
import org.turro.usual.UsualUsages;
 
public class DocumentWrapper extends DaoEntity<Document, Long> {
 
  private List<DocumentLine> lines;
  private List<DocumentRelation> descendants;
  
  public DocumentWrapper(Document entity) {
    super(entity);
    for(Register r : entity.getRegisters()) {
      entity.setRegularizeVAT(r.isRegularizeVAT());
      entity.setRegularizeIRPF(r.isRegularizeIRPF());
    }
  }
 
  @Override
  protected Dao createDao() {
    return new FinancialsPU();
  }
 
  public Set<RegisterWrapper> getRegisters() {
    Set<RegisterWrapper> s = new HashSet<RegisterWrapper>();
    for(RegisterView rv : ViewWrapper.getViews()) {
      boolean done = false;
      for(Register r : entity.getRegisters()) {
        if(r.getView().getId() == rv.getId()) {
          r.setRegisterDate(entity.getReceiptDate());
          s.add(new RegisterWrapper(r));
          done = true;
          break;
        }
      }
      if(!done) {
        Register r = new Register();
        r.setView(rv);
        r.setRegisterDate(entity.getReceiptDate());
        r.setDocument(entity);
        entity.getRegisters().add(r);
        s.add(new RegisterWrapper(r));
      }
    }
    return s;
  }
 
  public void setPriorSavingValues(List<DocumentLine> lines, List<DocumentRelation> descendants) {
    this.lines = lines;
    this.descendants = descendants;
  }
  
  @Override
  public Document save() {
    return save(lines, descendants);
  }
 
  public Document save(List<DocumentLine> lines, List<DocumentRelation> descendants) {
    /*
     * Save documents and assign ancestor and descendant,
     * then save relations.
     */
    if(descendants != null) {
      DocumentWrapper dw;
      for(DocumentRelation dr : descendants) {
        if(!dr.isEmpty()) {
          if(!dr.getDescendant().isEmpty()) {
            dr.getDescendant().setForcedView(entity.getForcedView());
            if(dr.getDescendant().getId() == 0) {
              if(entity.isDraft()) {
                dr.getDescendant().setDraft(true);
              }
            } else {
              if(!entity.isDraft()) {
                dr.getDescendant().setDraft(false);
              }
            }
            for(DocumentLine l : dr.getDescendant().getDocumentLines()) {
              if(!l.isEmpty() && l.getStore() != null) {
                UsualUsages.addUsage(entity.getUsualPath() + "df:" + l.getDocument().getDocumentDefinition().getId());
                UsualUsages.addUsage(dr.getDescendant().getUsualPath() + "st:" + l.getStore().getId());
                UsualUsages.addUsage(dr.getDescendant().getUsualPath() + "cp:" + l.getConcept());
              }
            }
            dw = new DocumentWrapper(dr.getDescendant());
            dr.setDescendant(dw.save(null, null));
          }
        }
      }
    }
    if(lines != null) {
      entity.assignTableOrder(lines);
      for(DocumentLine l : lines) {
        if(!l.isEmpty() && l.getStore() != null) {
          UsualUsages.addUsage(entity.getUsualPath() + "st:" + l.getStore().getId());
          UsualUsages.addUsage(entity.getUsualPath() + "cp:" + l.getConcept());
        }
      }
    }
    entity.clearEmpties();
    if(entity.isDraft()) {
      entity.getRegisters().clear();
    } else {
      getRegisters();
      Iterator<Register> ir = entity.getRegisters().iterator();
      while(ir.hasNext()) {
        Register r = ir.next();
        r.removeEmpties();
        r.setRegisterDate(entity.getReceiptDate());
        if(r.isEmpty()) {
          r.setDocument(null);
          ir.remove();
        } else if(entity.getForcedView() != null) {
          if(r.getView().getId() != entity.getForcedView().getId()) {
            r.setView(entity.getForcedView());
            r.setIdRegister(0);
          }
        }
      }
    }
 
    /* Save relations for future saving */
    ArrayList<DocumentRelation> relancestors = new ArrayList<DocumentRelation>(entity.getAncestors());
    ArrayList<DocumentRelation> reldescendants = new ArrayList<DocumentRelation>(entity.getDescendants());
    entity.getAncestors().clear();
    entity.getDescendants().clear();
 
    entity.prepareSave();
    super.save();
 
    /* Save relations from editing entity */
    if(lines != null) {
      for(DocumentRelation dr : relancestors) {
        if(!dr.isEmpty()) {
          dr.setDescendant(entity);
          getDao().saveObject(dr);
        }
      }
      for(DocumentRelation dr : reldescendants) {
        if(!dr.isEmpty() && dr.getDescendant().getId() != 0) {
          dr.setAncestor(entity);
          getDao().saveObject(dr);
        }
      }
    }
    return entity;
  }
 
  @Override
  public boolean delete() {
    // Check if possible and notify
    getDao().executeUpdate(
            "delete from DocumentRelation " +
            "where ancestor = ? " +
            "or descendant = ?",
            new Object[] { entity, entity });
    entity.getAncestors().clear();
    entity.getDescendants().clear();
    return super.delete();
  }
 
  @Override
  protected boolean shouldLog() {
    return true;
  }
  
  public List<Document> getRelatedAncestor(DocumentDefinition documentDefinition) {
    return getDao().getResultList(
            "select doc from Document as doc " +
            "where doc.descendants is empty " +
            "and doc.documentDefinition = ? " +
            "and doc.contract = ?",
            new Object[] { documentDefinition, entity.getContract() });
  }
 
  public List<Document> getRelatedDescendant(DocumentDefinition documentDefinition) {
    return getDao().getResultList(
            "select doc from Document as doc " +
            "where doc.ancestors is empty " +
            "and doc.documentDefinition = ? " +
            "and doc.contract = ?",
            new Object[] { documentDefinition, entity.getContract() });
  }
  
  public double getAdvance() {
    if(entity.getContract() != null) {
      return new ContractWrapper(entity.getContract()).getAdvance(entity.getForcedView());
    }
    return 0.0d;
  }
 
}