D-Net

package eu.dnetlib.goldoa.service.dao;

import eu.dnetlib.goldoa.domain.*;

import org.apache.commons.codec.digest.DigestUtils;

import org.apache.commons.io.IOUtils;

import org.hibernate.criterion.Order;

import org.hibernate.criterion.Restrictions;

import org.springframework.stereotype.Repository;

import java.io.ByteArrayOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.math.BigInteger;

import java.util.ArrayList;

import java.util.Date;

import java.util.List;

import java.util.UUID;

//import eu.dnetlib.goldoa.domain.BankAccount;

/*

 * Created by antleb on 4/2/15.

*/

@Repository

public class BudgetDAO extends AbstractDao<String, Budget>{

	private static final String UPDATE_BUDGET_TERMS = "update budget set termsmimetype=?, termsfile=? where id=?";

	private static final String INSERT_BUDGET_TERMS = "insert into budget (termsmimetype, termsfile, id) values (?,?,?)";

//	private static final String GET_BUDGET_FILE = "select termsmimetype, termsfile from budget where id=?";

	public Budget saveBudget(Budget budget) {

		persist(budget);

		return budget;

	}

	public Budget getBudget(String budgetId) {

		return getByKey(budgetId);

	}

	@SuppressWarnings("unchecked")

	public List<Budget> getBudgets() {

		return createEntityCriteria().addOrder(Order.asc("date")).list();

	}

	@SuppressWarnings("unchecked")

	public List<Budget> getBudgetsForUser(String email) {

		return createEntityCriteria().add(Restrictions.eq("user.email",email))

				.addOrder(Order.asc("date")).list();

	}

	@SuppressWarnings("unchecked")

	public List<Budget> getBudgetsForOrganization(final List<String> organizationIds) {

		List<Budget> rs = new ArrayList<>();

		for(String orgID : organizationIds){

			rs.addAll( createEntityCriteria().createAlias("organizations","o")

					.add(Restrictions.eq("o.id",orgID))

					.addOrder(Order.asc("date")).list());

		}

		return rs;

	}

	@SuppressWarnings("unchecked")

	public List<Budget> getBudgetsForPublisher(String publisherEmail) {

		return createEntityCriteria().add(Restrictions.eq("publisher", publisherEmail))

				.addOrder(Order.asc("date")).list();

	}

	@SuppressWarnings("unchecked")

	public List<Budget> getBudgetsForAccounting() {

		return createEntityCriteria().addOrder(Order.asc("date")).list();

	}

	public void initiallyApproveBudget(String budgetId) {

        Budget b = getBudget(budgetId);

        b.setStatus(Budget.Status.INITIALLY_APPROVED);

        saveBudget(b);

	}

	public void approveBudget(String budgetId, float amountGranted) {

        Budget b = getBudget(budgetId);

        b.setStatus(Budget.Status.APPROVED);

        saveBudget(b);

	}

	public void rejectBudget(String budgetId) {

        Budget b = getBudget(budgetId);

        b.setStatus(Budget.Status.REJECTED);

        saveBudget(b);

	}

	public void processingBudget(String budgetId) {

        Budget b = getBudget(budgetId);

        b.setStatus(Budget.Status.ACCOUNTING_PROCESSING);

        saveBudget(b);

	}

	public void accountingDeniedBudget(String budgetId) {

        Budget b = getBudget(budgetId);

        b.setStatus(Budget.Status.ACCOUNTING_DENIED);

        saveBudget(b);

	}

	public void accountingPaidBudget(String budgetId) {

        Budget b = getBudget(budgetId);

        b.setStatus(Budget.Status.ACCOUNTING_PAID);

        saveBudget(b);

	}

	public void accountingOnHoldBudget(String budgetId) {

        Budget b = getBudget(budgetId);

        b.setStatus(Budget.Status.ACCOUNTING_ONHOLD);

        saveBudget(b);

	}

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

		/*try {

			final ByteArrayOutputStream baos = new ByteArrayOutputStream();

			IOUtils.copy(inputStream, baos);

			IOUtils.closeQuietly(baos);

			final PreparedStatementSetter pss = new PreparedStatementSetter() {

				@Override

				public void setValues(PreparedStatement ps) throws SQLException {

					ps.setString(1, contentType);

					ps.setBytes(2, baos.toByteArray());

					ps.setString(3, id);

				}

			};

			if (jdbcTemplate.update(UPDATE_BUDGET_TERMS, pss) == 0) {

				jdbcTemplate.update(INSERT_BUDGET_TERMS, pss);

			}

		} catch (Exception e) {

			e.printStackTrace();

		}*/

	}

	public void saveComment(String budgetId, User user, String comment, String template) {

        String commentId = "portal::" + DigestUtils.md5Hex(comment + new Date());

        Comment c = new Comment();

        c.setId(commentId);

        c.setPerson(user);

        c.setComment(comment);

        CommentTemplate ct = new CommentTemplate();

		//TODO fix comment template

        c.setTemplate(null);

        Budget b = getBudget(budgetId);

        b.getComments().add(c);

        persist(b);

    }

	public List<Comment> getBudgetComments(String budgetId) {

		return getBudget(budgetId).getComments();

	}

	public void uploadBankReceipt(final String budgetId, final String contentType, InputStream inputStream) {

        Budget b;

        ByteArrayOutputStream baos = null;

        try {

            baos = new ByteArrayOutputStream();

            IOUtils.copy(inputStream, baos);

            IOUtils.closeQuietly(baos);

        } catch (IOException e) {

            e.printStackTrace();

        }

        b = getBudget(budgetId);

        b.setBankTransferReceipt(new BankTransferReceipt(contentType,baos.toByteArray()));

        saveBudget(b);

	}

	public void uploadInitialContract(final String budgetId, final String contentType, InputStream inputStream) {

        final String id = "portal::" + UUID.randomUUID().toString();

        ByteArrayOutputStream baos = null;

        try {

            baos = new ByteArrayOutputStream();

            IOUtils.copy(inputStream, baos);

            IOUtils.closeQuietly(baos);

        } catch (IOException e) {

            e.printStackTrace();

        }

		Budgetfile bf = new Budgetfile();

		bf.setId(id);

		bf.setContenttype(contentType);

		bf.setContent(baos.toByteArray());

        Budget b = getBudget(budgetId);

        b.setInitialcontractfile(bf);

        saveBudget(b);

	}

	public void uploadBeneficiaryContract(final String budgetId, final String contentType, InputStream inputStream) {

		try {

			final ByteArrayOutputStream baos = new ByteArrayOutputStream();

			final String id = "portal::" + UUID.randomUUID().toString();

			IOUtils.copy(inputStream, baos);

			IOUtils.closeQuietly(baos);

			BudgetContract bc = new BudgetContract();

			bc.setId(id);

			bc.setContent(baos.toByteArray());

			bc.setContentType(contentType);

			Budget b = getBudget(budgetId);

			b.setBeneficiarysignedcontract(bc);

			saveBudget(b);

		} catch (Exception e) {

			e.printStackTrace();

		}

	}

	public void uploadSignedContract(final String budgetId, final String contentType, InputStream inputStream) {

		try {

			final ByteArrayOutputStream baos = new ByteArrayOutputStream();

			final String id = "portal::" + UUID.randomUUID().toString();

			IOUtils.copy(inputStream, baos);

			IOUtils.closeQuietly(baos);

			BudgetContract bc = new BudgetContract();

			bc.setId(id);

			bc.setContent(baos.toByteArray());

			bc.setContentType(contentType);

			Budget b = getBudget(budgetId);

			b.setSignedcontract(bc);

			saveBudget(b);

		} catch (Exception e) {

			e.printStackTrace();

		}

	}

	public BigInteger getRequestId() {

		return (BigInteger) getSession().createSQLQuery("select nextval('budget_id_seq') as id").list().get(0);

	}

    public Organization getOrganizationForBudget(String budgetID) {

		return (Organization) createEntityCriteria().add(Restrictions.eq("budget.id", budgetID)).list().get(0);

    }

}