com.axelor.apps.cash.management.service.ForecastService.java Source code

Introduction

Here is the source code for com.axelor.apps.cash.management.service.ForecastService.java

Source

/**
 * Axelor Business Solutions
 *
 * Copyright (C) 2015 Axelor (<http://axelor.com>).
 *
 * This program is free software: you can redistribute it and/or  modify
 * it under the terms of the GNU Affero General Public License, version 3,
 * as published by the Free Software Foundation.
 *
 * 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 com.axelor.apps.cash.management.service;

import java.math.BigDecimal;
import java.util.List;

import org.joda.time.LocalDate;

import com.axelor.apps.base.db.BankDetails;
import com.axelor.apps.base.db.Company;
import com.axelor.apps.base.service.administration.GeneralService;
import com.axelor.apps.cash.management.db.Forecast;
import com.axelor.apps.cash.management.db.ForecastGenerator;
import com.axelor.apps.cash.management.db.ForecastReason;
import com.axelor.apps.cash.management.db.repo.ForecastRepository;
import com.google.inject.Inject;

public class ForecastService {

    @Inject
    protected GeneralService generalService;

    public void generate(ForecastGenerator forecastGenerator) {
        LocalDate fromDate = forecastGenerator.getFromDate();
        LocalDate toDate = forecastGenerator.getToDate();
        LocalDate itDate = new LocalDate(fromDate);
        LocalDate todayDate = generalService.getTodayDate();

        if (forecastGenerator.getForecastList() != null && !forecastGenerator.getForecastList().isEmpty()) {
            List<Forecast> forecastList = forecastGenerator.getForecastList();
            for (Forecast forecast : forecastList) {
                if (forecast.getRealizedSelect() == ForecastRepository.REALISED_SELECT_NO) {
                    forecastList.remove(forecast);
                } else if (forecast.getRealizedSelect() == ForecastRepository.REALISED_SELECT_AUTO
                        && forecast.getEstimatedDate().isAfter(todayDate)) {
                    forecastList.remove(forecast);
                }
            }
        }

        while (!itDate.isAfter(toDate)) {
            Forecast forecast = this.createForecast(forecastGenerator.getCompany(),
                    forecastGenerator.getBankDetails(), forecastGenerator.getTypeSelect(),
                    forecastGenerator.getAmount(), itDate, forecastGenerator.getForecastReason(),
                    forecastGenerator.getComments(), forecastGenerator.getRealizedSelect());
            forecastGenerator.addForecastListItem(forecast);
            itDate.plusMonths(forecastGenerator.getPeriodicitySelect());
        }
    }

    public Forecast createForecast(Company company, BankDetails bankDetails, int typeSelect, BigDecimal amount,
            LocalDate estimatedDate, ForecastReason reason, String comments, int realizedSelect) {

        Forecast forecast = new Forecast();
        forecast.setCompany(company);
        forecast.setBankDetails(bankDetails);
        forecast.setTypeSelect(typeSelect);
        forecast.setAmount(amount);
        forecast.setEstimatedDate(estimatedDate);
        forecast.setForecastReason(reason);
        forecast.setComments(comments);
        forecast.setRealizedSelect(realizedSelect);

        return forecast;
    }

}