RegisterWrapper.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.register;
 
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.AccountsForDocumentLine;
import org.turro.financials.model.business.CompanyWrapper;
import org.turro.jpa.Dao;
import org.turro.jpa.entity.XMLSerializer;
import org.turro.log.SystemLogType;
import org.turro.log.SystemLogger;
 
public class RegisterWrapper {
 
  protected Register register;
  private Set<Long> bookDefs;
  private boolean generate;
 
  public RegisterWrapper(Register register) {
    this(register, true);
  }
 
  public RegisterWrapper(Register register, boolean generate) {
    this.register = register;
    this.generate = generate;
    bookDefs = new HashSet<Long>();
    generateEntries();
  }
 
  public Register getRegister() {
    return register;
  }
 
  public EntrySet getEntries() {
    return generateEntries();
  }
 
  private EntrySet generateEntries() {
    EntrySet es = new EntrySet();
    if(generate && register.getDocument() != null) {
      register.setRegularizeVAT(register.getDocument().isRegularizeVAT());
      register.setRegularizeIRPF(register.getDocument().isRegularizeIRPF());
      RegisterView view = register.getView();
      if(register.getDocument().getForcedView() != null) {
        view = null;
        if(register.getDocument().getForcedView().getId() != register.getView().getId()) {
          register.reuseLines(es.getRegisterEntries());
          return es;
        }
      }
      bookDefs.clear();
      for(DocumentLine dl : register.getDocument().getDocumentLines()) {
        if(dl.getSubtotal() != 0) {
          for(RegisterEntry re : new AccountsForDocumentLine(dl).getEntries(view, bookDefs)) {
            es.addEntry(re);
          }
        }
      }
      es.roundEntries(register.getDocument().getCurrency().getDefaultFractionDigits());
      new RegisterAssuredBalance(es, register.getDocument()).normalize();
      es.calculateBalance();
      register.reuseLines(es.getRegisterEntries());
      register.killUnusedBooks(bookDefs);      
    } else {
      bookDefs.clear();
      for(RegisterEntry re : register.getRegisterEntries()) {
        if(re.getDebit() != 0 || re.getCredit() != 0) {
          MajorAccount ma = re.getAccount().getParent();
          if(ma != null) {
            BookDefinition bf = ma.getBookDefinition();
            if(bf != null) {
              bookDefs.add(bf.getId());
            }
          }
          es.addEntry(re);
        }
      }
      es.calculateBalance();
    }
    getBookString();
    return es;
  }
 
  public String getBookString() {
    Iterator<BookRegister> itbr = register.getBookRegisters().iterator();
    // Clear temporary assigned books
    while(itbr.hasNext()) {
      BookRegister br = itbr.next();
      boolean found = false;
      for(Long idBd : bookDefs) {
        if(br.getBookDefinition().getId() == idBd) {
          found = true;
          break;
        }
      }
      if(!found && br.getId() == 0) {
        itbr.remove();
      }
    }
    // Generate new books
    StringBuilder sb = new StringBuilder();
    for(Long idBd : bookDefs) {
      boolean found = false;
      for(BookRegister br : register.getBookRegisters()) {
        if(br.getBookDefinition().getId() == idBd) {
          if(sb.length() > 0) {
            sb.append(" | ");
          }
          sb.append(br.getBookDefinition().getDescription());
          sb.append(" [" + (br.getBookOrder() == 0 ? "<auto>" : br.getBookOrder()) +"]");
          found = true;
          break;
        }
      }
      if(!found) {
        BookDefinition bd = new FinancialsPU().find(BookDefinition.class, idBd);
        if(bd != null) {
          BookRegister br = new BookRegister();
          br.setRegister(register);
          br.setBookDefinition(bd);
          register.getBookRegisters().add(br);
          if(sb.length() > 0) {
            sb.append(" | ");
          }
          sb.append(bd.getDescription());
          sb.append(" [<auto>]");
        }
      }
    }
    return sb.toString();
  }
 
  public Register save() {
    register.prepareSave();
    register = new FinancialsPU().saveObject(register);
    SystemLogger.getInstance().doLog(SystemLogType.LOG_INFO, register, "saved", new XMLSerializer(register).data());
    return register;
  }
 
  public boolean delete() {
    if(register.getDocument() == null) {
      SystemLogger.getInstance().doLog(SystemLogType.LOG_INFO, register, "deleted", new XMLSerializer(register).data());
      new FinancialsPU().deleteObject(register);
      return true;
    }
    return false;
  }
 
  public static Register prepareBookRegister(Register register) {
    Dao dao = new FinancialsPU();
    Set<Long> bookDefs = new HashSet<Long>();
    for(RegisterEntry re : register.getRegisterEntries()) {
      if(re.getDebit() != 0 || re.getCredit() != 0) {
        MajorAccount ma = re.getAccount().getParent();
        if(ma != null) {
          BookDefinition bf = ma.getBookDefinition();
          if(bf != null) {
            bookDefs.add(bf.getId());
          }
        }
      }
    }
    for(Long bookDef : bookDefs) {
      BookRegister br = new BookRegister();
      br.setRegister(register);
      br.setBookDefinition(dao.find(BookDefinition.class, bookDef));
      register.getBookRegisters().add(br);
    }
    register.prepareSave();
    return register;
  }
 
  public boolean checkSave() {
    return CompanyWrapper.isValidDate(register.getRegisterDate(), register.isClosing() && register.isExclude());
  }
 
}