D-Net

package eu.dnetlib.goldoa.service;

import eu.dnetlib.goldoa.domain.Budget;

import eu.dnetlib.goldoa.domain.BudgetFile;

import eu.dnetlib.goldoa.domain.BudgetInfo;

import eu.dnetlib.goldoa.domain.OrganizationManagerException;

import eu.dnetlib.goldoa.domain.Person;

import eu.dnetlib.goldoa.domain.PersonManagerException;

import eu.dnetlib.goldoa.service.dao.BudgetDAO;

import eu.dnetlib.goldoa.service.utils.EmailUtils;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.transaction.annotation.Transactional;

import java.io.InputStream;

import java.util.ArrayList;

import java.util.Date;

import java.util.List;

import java.util.UUID;

import java.util.concurrent.ExecutorService;

/**

 * Created by antleb on 4/2/15.

 */

@Transactional

public class BudgetManagerImpl implements BudgetManager {

	@Autowired

	private BudgetDAO budgetDAO;

	@Autowired

	private PersonManager personManager;

	@Autowired

	private OrganizationManager organizationManager;

	@Autowired

	ExecutorService executorService;

	@Autowired

	private EmailUtils emailUtils;

	@Override

	public Budget saveBudget(final Budget budget) {

		if (budget.getId() == null) {

			budget.setId(UUID.randomUUID().toString());

			budget.setDate(new Date());

			budget.setRemaining(budget.getAmountGranted());

		}

		budgetDAO.saveBudget(budget);

		budgetDAO.deleteBudgetRelations(budget);

		budgetDAO.insertBudgetRelations(budget);

		if (Budget.Status.SUBMITTED.equals(getStatus(budget.getStatusCode()))) {

			executorService.submit(new Runnable() {

				@Override

				public void run() {

					try {

						BudgetInfo budgetInfo = getBudgetInfo(budget);

						List<Person> moderators = personManager.getModerators();

						emailUtils.sendUserNewBudgetEmail(budgetInfo);

						for (Person moderator : moderators)

							emailUtils.sendModeratorNewBudgetEmail(moderator, budgetInfo);

					} catch (Exception e) {

						e.printStackTrace();

					}

				}

			});

		}

		return budget;

	}

	@Override

	public BudgetInfo getBudget(String budgetId) {

		return getBudgetInfo(budgetDAO.getBudget(budgetId));

	}

	@Override

	public List<BudgetInfo> getBudgets() {

		List<BudgetInfo> res = new ArrayList<BudgetInfo>();

		for (Budget budget : budgetDAO.getBudgets())

			res.add(getBudgetInfo(budget));

		return res;

	}

	@Override

	public List<BudgetInfo> getBudgetsForUser(String userId) {

		List<BudgetInfo> res = new ArrayList<BudgetInfo>();

		for (Budget budget : budgetDAO.getBudgetsForUser(userId))

			res.add(getBudgetInfo(budget));

		return res;

	}

	@Override

	public List<BudgetInfo> getBudgetsForOrganization(List<String> organizationIds) {

		List<BudgetInfo> res = new ArrayList<BudgetInfo>();

		for (Budget budget : budgetDAO.getBudgetsForOrganization(organizationIds))

			res.add(getBudgetInfo(budget));

		return res;

	}

	@Override

	public void approveBudget(final String budgetId, String comment, float amountGranted) {

		budgetDAO.approveBudget(budgetId, amountGranted);

		budgetDAO.updateBudgetComment(budgetId, comment);

		executorService.submit(new Runnable() {

			@Override

			public void run() {

				try {

					BudgetInfo budgetInfo = getBudget(budgetId);

					List<Person> moderators = personManager.getModerators();

					emailUtils.sendUserAcceptedBudgetEmail(budgetInfo);

					for (Person moderator : moderators)

						emailUtils.sendModeratorAcceptedBudgetEmail(moderator, budgetInfo);

				} catch (Exception e) {

					e.printStackTrace();

				}

			}

		});

	}

	@Override

	public void rejectBudget(final String budgetId, String comment) {

		budgetDAO.rejectBudget(budgetId);

		budgetDAO.updateBudgetComment(budgetId, comment);

		executorService.submit(new Runnable() {

			@Override

			public void run() {

				try {

					BudgetInfo budgetInfo = getBudget(budgetId);

					List<Person> moderators = personManager.getModerators();

					emailUtils.sendUserRejectedBudgetEmail(budgetInfo);

					for (Person moderator : moderators)

						emailUtils.sendModeratorRejectedBudgetEmail(moderator, budgetInfo);

				} catch (Exception e) {

					e.printStackTrace();

				}

			}

		});

	}

	@Override

	public void uploadTerms(String id, String contentType, InputStream inputStream) {

		budgetDAO.uploadTerms(id, contentType, inputStream);

	}

	@Override

	public BudgetFile downloadBudgetTerms(String budgetId) {

		return budgetDAO.downloadTerms(budgetId);

	}

	private BudgetInfo getBudgetInfo(Budget budget) {

		BudgetInfo budgetInfo = new BudgetInfo();

		budgetInfo.setId(budget.getId());

		budgetInfo.setDate(budget.getDate());

		budgetInfo.setStartDate(budget.getStartDate());

		budgetInfo.setEndDate(budget.getEndDate());

		budgetInfo.setAmountGranted(budget.getAmountGranted());

		budgetInfo.setAmountRequested(budget.getAmountRequested());

		budgetInfo.setRemaining(budget.getRemaining());

		budgetInfo.setCurrency(budget.getCurrency());

		budgetInfo.setStatusCode(budget.getStatusCode());

		budgetInfo.setStatus(getStatus(budget.getStatusCode()));

		budgetInfo.setComment(budgetDAO.getBudgetComment(budget.getId()));

		budgetInfo.setBankAccount(budget.getBankAccount());

		if (budget.getUser() != null) {

			try {

				budgetInfo.setUser(personManager.getById(budget.getUser()));

			} catch (PersonManagerException e) {

				e.printStackTrace();

			}

		}

		if (budget.getOrganisation() != null) {

			try {

				budgetInfo.setOrganisation(organizationManager.getById(budget.getOrganisation()));

			} catch (OrganizationManagerException e) {

				e.printStackTrace();

			}

		}

		return budgetInfo;

	}

	private Budget.Status getStatus(int statusCode) {

		for (Budget.Status s : Budget.Status.values())

			if ((statusCode & s.getCode()) == s.getCode())

				return s;

		return null;

	}

}