DocumentGenerator.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2012 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.Date;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Set;
import org.turro.financials.db.FinancialsPU;
import org.turro.financials.entity.*;
import org.turro.financials.model.register.RegisterWrapper;
import org.turro.financials.model.register.ViewWrapper;
import org.turro.jpa.entity.XMLSerializer;
import org.turro.log.SystemLogType;
import org.turro.log.SystemLogger;
import org.turro.usual.UsualUsages;
 
public class DocumentGenerator {
 
  private DocumentDefinition documentDefinition;
  private Contract contract;
  private String notes;
  private Date receiptDate;
  private Date documentDate;
  private String documentNumber;
  private boolean draft, noAncestors = false, noDescendants = false;
 
  private Document document;
 
  public DocumentGenerator(DocumentDefinition documentDefinition, Contract contract, String notes, Date receiptDate, Date documentDate, String documentNumber, boolean draft) {
    this.documentDefinition = documentDefinition;
    this.contract = contract;
    this.notes = notes;
    this.receiptDate = receiptDate;
    this.documentDate = documentDate;
    this.documentNumber = documentNumber;
    this.draft = draft;
    createDocument();
  }
 
  public DocumentGenerator(Document document) {
    this.document = document;
  }
  
  public Document getDocument() {
    return document;
  }
  
  public DocumentLine addLine(Contract store, String concept, 
          double quantity, double price, double tax, double retention, 
          LineType lineType, ContractPreference contractPreference) {
    DocumentLine dl = new DocumentLine();
    dl.setDocument(document);
    dl.setLineOrder(document.getMaxLineOrder() + 1);
    dl.setStore(store);
    dl.setConcept(concept);
    dl.setQuantity(quantity);
    dl.setPrice(price);
    dl.setTax(tax);
    dl.setRetention(retention);
    dl.setLineType(lineType);
    dl.setContractPreference(contractPreference);
    dl = new FinancialsPU().saveObject(dl);
    document.getDocumentLines().add(dl);
    return dl;
  }
  
  public DocumentLine addLine(Contract store, Product product, String concept, 
          double quantity, double price, double tax, double retention, 
          LineType lineType, ContractPreference contractPreference) {
    DocumentLine dl = new DocumentLine();
    dl.setDocument(document);
    dl.setLineOrder(document.getMaxLineOrder() + 1);
    dl.setStore(store);
    dl.setProduct(product);
    dl.setConcept(concept);
    dl.setQuantity(quantity);
    dl.setPrice(price);
    dl.setTax(tax);
    dl.setRetention(retention);
    dl.setLineType(lineType);
    dl.setContractPreference(contractPreference);
    dl = new FinancialsPU().saveObject(dl);
    document.getDocumentLines().add(dl);
    return dl;
  }
  
  public DocumentLine addLine(Contract store, ProductByContractor product, String concept, 
          double quantity, double price, double tax, double retention, 
          LineType lineType, ContractPreference contractPreference) {
    DocumentLine dl = new DocumentLine();
    dl.setDocument(document);
    dl.setLineOrder(document.getMaxLineOrder() + 1);
    dl.setStore(store);
    dl.setProductByContractor(product);
    dl.setProduct(product.getProduct());
    dl.setConcept(concept);
    dl.setQuantity(quantity);
    dl.setPrice(price);
    dl.setTax(tax);
    dl.setRetention(retention);
    dl.setLineType(lineType);
    dl.setContractPreference(contractPreference);
    dl = new FinancialsPU().saveObject(dl);
    document.getDocumentLines().add(dl);
    return dl;
  }
  
  public Document save() {
    document.clearEmpties();
    for(DocumentLine l : document.getDocumentLines()) {
      if(!l.isEmpty() && l.getStore() != null) {
        UsualUsages.addUsage(document.getUsualPath() + "st:" + l.getStore().getId());
        UsualUsages.addUsage(document.getUsualPath() + "cp:" + l.getConcept());
      }
    }
    if(document.isDraft()) {
      document.getRegisters().clear();
    } else {
      getRegisters();
      Iterator<Register> ir = document.getRegisters().iterator();
      while(ir.hasNext()) {
        Register r = ir.next();
        r.removeEmpties();
        r.setRegisterDate(document.getReceiptDate());
        if(r.isEmpty()) {
          r.setDocument(null);
          ir.remove();
        } else if(document.getForcedView() != null) {
          if(r.getView().getId() != document.getForcedView().getId()) {
            r.setView(document.getForcedView());
            r.setIdRegister(0);
          }
        }
      }
    }
 
    document.prepareSave();
    document = new FinancialsPU().saveObject(document);
    SystemLogger.getInstance().doLog(SystemLogType.LOG_INFO, document, "saved", new XMLSerializer(document).data());
 
    return document;
  }
 
  public Set<RegisterWrapper> getRegisters() {
    Set<RegisterWrapper> s = new HashSet<RegisterWrapper>();
    for(RegisterView rv : ViewWrapper.getViews()) {
      boolean done = false;
      for(Register r : document.getRegisters()) {
        if(r.getView().getId() == rv.getId()) {
          r.setRegisterDate(document.getReceiptDate());
          s.add(new RegisterWrapper(r));
          done = true;
          break;
        }
      }
      if(!done) {
        Register r = new Register();
        r.setView(rv);
        r.setRegisterDate(document.getReceiptDate());
        r.setDocument(document);
        document.getRegisters().add(r);
        s.add(new RegisterWrapper(r));
      }
    }
    return s;
  }
 
  private void createDocument() {
    document = new Document();
    document.setDocumentDefinition(documentDefinition);
    document.setContract(contract);
    document.setNotes(notes);
    document.setReceiptDate(receiptDate == null ? documentDate : receiptDate);
    document.setDocumentDate(documentDate);
    document.setDocumentNumber(documentNumber);
    document.setDraft(draft);
    document.setCurrency(contract.getCurrency());
    document = new FinancialsPU().saveObject(document);
  }
 
}