D-Net

package eu.dnetlib.goldoa.service;

import eu.dnetlib.goldoa.domain.Invoice;

import eu.dnetlib.goldoa.domain.InvoiceFile;

import eu.dnetlib.goldoa.domain.ManagerException;

import org.apache.commons.io.IOUtils;

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

import org.springframework.dao.EmptyResultDataAccessException;

import org.springframework.jdbc.core.JdbcTemplate;

import org.springframework.jdbc.core.PreparedStatementSetter;

import org.springframework.jdbc.core.RowMapper;

import org.springframework.transaction.annotation.Transactional;

import javax.sql.DataSource;

import java.io.ByteArrayOutputStream;

import java.io.InputStream;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.sql.SQLException;

import java.sql.Types;

import java.util.UUID;

/**

 * Created by antleb on 3/21/15.

 */

@Transactional

public class InvoiceManagerImpl implements InvoiceManager {

	// TODO move to dao if needed

	@Autowired

	private DataSource dataSource;

	private static final String INSERT_INVOICE = "insert into invoice (number, alternativeid, date, source, id) values (?, ?, ?, ?, ?)";

	private static final String UPDATE_INVOICE = "update invoice set number=?, alternativeid=?, date=?, source=? where id = ?";

	private static final String GET_INVOICE = "select id, alternativeid, number, date, source from invoice where id = ?";

	private static final String GET_INVOICE_FILE = "select invoice, mimetype, file from file where invoice = ?";

	private static final String INSERT_INVOICE_FILE = "insert into file (mimetype, file, invoice) values (?, ?, ?)";

	private static final String UPDATE_INVOICE_FILE = "update file set mimetype=?, file=? where invoice=?";

	@Override

	public Invoice saveInvoice(Invoice invoice) {

		JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);

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

			invoice.setSource("portal");

			invoice.setId("portal::" + UUID.randomUUID().toString());

		}

		if (jdbcTemplate.update(UPDATE_INVOICE, new Object[]{invoice.getNumber(), invoice.getAlternativeId(), invoice.getDate(), invoice.getSource(), invoice.getId()},

				new int[]{Types.VARCHAR, Types.VARCHAR, Types.TIMESTAMP, Types.VARCHAR, Types.VARCHAR}) == 0) {

			jdbcTemplate.update(INSERT_INVOICE, new Object[]{invoice.getNumber(), invoice.getAlternativeId(), invoice.getDate(), invoice.getSource(), invoice.getId()},

					new int[]{Types.VARCHAR, Types.VARCHAR, Types.TIMESTAMP, Types.VARCHAR, Types.VARCHAR});

		}

		return invoice;

	}

	@Override

	public Invoice getInvoice(String invoiceId) throws ManagerException {

		try {

			return new JdbcTemplate(dataSource).queryForObject(GET_INVOICE, new String[]{invoiceId}, new int[]{Types.VARCHAR}, new RowMapper<Invoice>() {

				@Override

				public Invoice mapRow(ResultSet rs, int rowNum) throws SQLException {

					Invoice invoice = new Invoice();

					invoice.setId(rs.getString("id"));

					invoice.setAlternativeId(rs.getString("alternativeid"));

					invoice.setNumber(rs.getString("number"));

					invoice.setDate(rs.getTimestamp("date"));

					invoice.setSource(rs.getString("source"));

					return invoice;

				}

			});

		} catch (EmptyResultDataAccessException e) {

			throw new ManagerException(ManagerException.ErrorCause.NOT_EXISTS);

		}

	}

	@Override

	public void uploadInvoice(final String invoiceId, final String mimetype, InputStream invoice) throws ManagerException {

		try {

			final ByteArrayOutputStream baos = new ByteArrayOutputStream();

			final JdbcTemplate jdbcTemplate = new JdbcTemplate(dataSource);

			final PreparedStatementSetter pss = new PreparedStatementSetter() {

				@Override

				public void setValues(PreparedStatement ps) throws SQLException {

					ps.setString(1, mimetype);

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

					ps.setString(3, invoiceId);

				}

			};

			IOUtils.copy(invoice, baos);

			IOUtils.closeQuietly(baos);

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

				jdbcTemplate.update(INSERT_INVOICE_FILE, pss);

			}

		} catch (Exception e) {

			throw new ManagerException(ManagerException.ErrorCause.UNKNOWN);

		}

	}

	@Override

	public InvoiceFile downloadInvoice(String invoiceId) throws ManagerException {

		try {

			return new JdbcTemplate(dataSource).queryForObject(GET_INVOICE_FILE, new String[]{invoiceId}, new int[]{Types.VARCHAR}, new RowMapper<InvoiceFile>() {

				@Override

				public InvoiceFile mapRow(ResultSet rs, int rowNum) throws SQLException {

					return new InvoiceFile(rs.getString("invoice"), rs.getString("mimetype"), rs.getBytes("file"));

				}

			});

		} catch (EmptyResultDataAccessException e) {

			throw new ManagerException(ManagerException.ErrorCause.NOT_EXISTS);

		} catch (Exception e) {

			throw new ManagerException(ManagerException.ErrorCause.UNKNOWN);

		}

	}

	public DataSource getDataSource() {

		return dataSource;

	}

	public void setDataSource(DataSource dataSource) {

		this.dataSource = dataSource;

	}

}