PayrollWrapper.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.ArrayList;
import org.turro.financials.db.FinancialsPU;
import org.turro.financials.entity.Document;
import org.turro.financials.entity.DocumentLine;
import org.turro.financials.entity.LineType;
import org.turro.jpa.Dao;
 
public class PayrollWrapper {
 
  private static final long
          PAYROLL_TYPE = 8, RETENTION_TYPE = 10,
          INSURANCE_TYPE = 9, SEIZURE_TYPE = 159,
          PROTECTION_TYPE = 167, CONTINGENCIES_TYPE = 168;
 
  private Document payroll;
  private DocumentLine payrollLine, retentionLine, insuranceLine, seizureLine,
          protectionLine, contingenciesLine;
 
  public PayrollWrapper(Document payroll) {
    this.payroll = payroll;
    assignLines();
  }
 
  public void save() {
    //TODO: if id is assigned then remove document
    if(!payroll.isEmpty()) {
      new DocumentWrapper(payroll).save(
              new ArrayList<>(payroll.getDocumentLines()),
              new ArrayList<>(payroll.getDescendants()));
    }
  }
 
  public DocumentLine getInsuranceLine() {
    return insuranceLine;
  }
 
  public Document getPayroll() {
    return payroll;
  }
 
  public DocumentLine getPayrollLine() {
    return payrollLine;
  }
 
  public DocumentLine getRetentionLine() {
    return retentionLine;
  }
 
  public DocumentLine getSeizureLine() {
    return seizureLine;
  }
 
  public DocumentLine getProtectionLine() {
    return protectionLine;
  }
 
  public DocumentLine getContingenciesLine() {
    return contingenciesLine;
  }
 
  private void assignLines() {
    if(payroll != null) {
      Dao dao = new FinancialsPU();
      payrollLine = retentionLine = insuranceLine = seizureLine = protectionLine = null;
      for(DocumentLine dl : payroll.getDocumentLines()) {
        if(dl.getLineType().getId() == PAYROLL_TYPE) {
          payrollLine = dl;
        } else if(dl.getLineType().getId() == RETENTION_TYPE) {
          retentionLine = dl;
        } else if(dl.getLineType().getId() == INSURANCE_TYPE) {
          insuranceLine = dl;
        } else if(dl.getLineType().getId() == SEIZURE_TYPE) {
          seizureLine = dl;
        } else if(dl.getLineType().getId() == PROTECTION_TYPE) {
          protectionLine = dl;
        } else if(dl.getLineType().getId() == CONTINGENCIES_TYPE) {
          contingenciesLine = dl;
        }
      }
      if(payrollLine == null) {
        payrollLine = new DocumentLine();
        payrollLine.setDocument(payroll);
        payrollLine.setLineType(dao.find(LineType.class, PAYROLL_TYPE));
        // avoid equalities
        payrollLine.setLineOrder(payroll.getDocumentLines().size() + 90001);
        payroll.getDocumentLines().add(payrollLine);
      }
      if(retentionLine == null) {
        retentionLine = new DocumentLine();
        retentionLine.setDocument(payroll);
        retentionLine.setLineType(dao.find(LineType.class, RETENTION_TYPE));
        // avoid equalities
        retentionLine.setLineOrder(payroll.getDocumentLines().size() + 90001);
        payroll.getDocumentLines().add(retentionLine);
      }
      if(insuranceLine == null) {
        insuranceLine = new DocumentLine();
        insuranceLine.setDocument(payroll);
        insuranceLine.setLineType(dao.find(LineType.class, INSURANCE_TYPE));
        // avoid equalities
        insuranceLine.setLineOrder(payroll.getDocumentLines().size() + 90001);
        payroll.getDocumentLines().add(insuranceLine);
      }
      if(seizureLine == null) {
        seizureLine = new DocumentLine();
        seizureLine.setDocument(payroll);
        seizureLine.setLineType(dao.find(LineType.class, SEIZURE_TYPE));
        // avoid equalities
        seizureLine.setLineOrder(payroll.getDocumentLines().size() + 90001);
        payroll.getDocumentLines().add(seizureLine);
      }
      if(protectionLine == null) {
        protectionLine = new DocumentLine();
        protectionLine.setDocument(payroll);
        protectionLine.setLineType(dao.find(LineType.class, PROTECTION_TYPE));
        // avoid equalities
        protectionLine.setLineOrder(payroll.getDocumentLines().size() + 90001);
        payroll.getDocumentLines().add(protectionLine);
      }
      if(contingenciesLine == null) {
        contingenciesLine = new DocumentLine();
        contingenciesLine.setDocument(payroll);
        contingenciesLine.setLineType(dao.find(LineType.class, CONTINGENCIES_TYPE));
        // avoid equalities
        contingenciesLine.setLineOrder(payroll.getDocumentLines().size() + 90001);
        payroll.getDocumentLines().add(contingenciesLine);
      }
    }
  }
 
}