package org.gcube.dataharvest.dao;

import java.sql.Array;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.Calendar;
import java.util.List;

import org.gcube.dataharvest.datamodel.Harvest;
import org.gcube.dataharvest.utils.Utils;
import org.postgresql.util.PSQLException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class Dao {
	private Connection conn = null;
	private static Logger logger = LoggerFactory.getLogger(Dao.class);

	/**
	 * Class constructor. This method must be called before any other class
	 * method.
	 * 
	 * @throws DaoException
	 */
	public void init() throws DaoException {
		try {
			Class.forName("org.postgresql.Driver");
		} catch (ClassNotFoundException ex) {
			logger.error(ex.getLocalizedMessage());
			throw new DaoException(ex.getLocalizedMessage(), ex.getCause());
		}
	}

	/**
	 * Connect a database
	 * 
	 * @param url
	 * @param user
	 * @param password
	 * @throws DaoException
	 */
	public void connect(String url, String user, String password) throws DaoException {
		try {
			conn = DriverManager.getConnection(url, user, password);
			logger.debug("Connected to: " + url);
		} catch (SQLException ex) {
			logger.error(ex.getLocalizedMessage());
			throw new DaoException(ex.getLocalizedMessage(), ex.getCause());
		}
	}

	/**
	 * Release a database connection
	 *
	 * @throws DaoException
	 */
	public void disconnect() throws DaoException {
		try {
			if (conn != null) {
				conn.close();
				conn = null;
				logger.debug("Disconnecting from database");
			}
		} catch (SQLException ex) {
			logger.error(ex.getLocalizedMessage());
			throw new DaoException(ex.getLocalizedMessage(), ex.getCause());
		}
	}

	/**
	 * Check the connection with the database.
	 * 
	 * @return true|false
	 */
	public boolean isConnected() {
		return (!(conn == null));
	}

	/**
	 * Getter for the database connection
	 * 
	 * @return Connection class
	 */
	public Connection getConnection() {
		return conn;
	}

	/**
	 * Read on the database all active VRES
	 * 
	 * @return
	 */
	public String[] getActiveVres() {
		String query = "select dname from context where dname is not null and dismissed is null";
		logger.debug(query);
		Statement s = null;
		ResultSet rs = null;
		ArrayList<String> list = new ArrayList<>();
		try {
			s = conn.createStatement();
			rs = s.executeQuery(query);
			while (rs.next()) {
				list.add(rs.getString("dname"));
			}

			return list.toArray(new String[list.size()]);
		} catch (Exception ex) {
			logger.error(ex.getLocalizedMessage());
			return null;
		} finally {
			if (rs != null) {
				try {
					rs.close();
				} catch (SQLException ex) {
					// do nothing
				}
			}
			if (s != null) {
				try {
					s.close();
				} catch (SQLException ex) {
					// do nothing
				}
			}
		}
	}

	/**
	 * This method insert/update data in the monthly_measure database table.
	 * 
	 * @param data
	 * @param from
	 * @param to
	 * @throws DaoException
	 */
	public void insertMonthlyMeasure(List<Harvest> data, java.util.Date from, java.util.Date to, boolean doUpdate)
			throws DaoException {
		// first of all: check if data of the same type are already in the
		// database.
		// In this case data will be updated.
		Calendar cFrom = Utils.dateToCalendar(from);
		Calendar cTo = Utils.dateToCalendar(to);
		int monthFrom = cFrom.get(Calendar.MONTH);
		int yearFrom = cFrom.get(Calendar.YEAR);
		int monthTo = cTo.get(Calendar.MONTH);
		int yearTo = cTo.get(Calendar.YEAR);

		if ((monthFrom != monthTo) || (yearFrom != yearTo)) {
			String err = "Invalid time period. The time period MUST refer one month.";
			logger.error(err);
			throw new DaoException(err, null);
		}

		if (data == null || data.size() <= 0) {
			String err = "No data passed in input. Aborting operation.";
			logger.error(err);
			throw new DaoException(err, null);
		}
		monthFrom++; // because january = 0...

		try {
			for (Harvest harvest : data) {
				String query = "select id from monthly_measure where measure_type_id=" + harvest.getDataType()
						+ " and context_id=(select id from context where dname='" + harvest.getContext()
						+ "') and month=" + monthFrom + " and year=" + yearFrom;
				logger.debug(query);
				Statement sel = conn.createStatement();
				ResultSet rs = sel.executeQuery(query);
				if (rs.next()) {
					if (doUpdate) {
						// record found: update it
						Statement s = conn.createStatement();
						int id = rs.getInt("id");
						String update = "update monthly_measure set measure=" + harvest.getMeasure() + " where id="
								+ id;
						logger.debug(update);
						s.execute(update);
						s.close();
					} else {
						logger.warn("Skipped " + harvest.getContext());
					}
				} else {
					// record not found: insert new record
					Statement s = conn.createStatement();
					String insert = "insert into monthly_measure (year, month, measure, measure_type_id, context_id, day) values (";
					insert += yearFrom + "," + monthFrom + "," + harvest.getMeasure() + "," + harvest.getDataType()
							+ ",";
					insert += "(select id from context where dname='" + harvest.getContext() + "'),";
					insert += "'" + yearFrom + "-" + monthFrom + "-01'";
					insert += ")";
					logger.debug(insert);
					s.execute(insert);
					s.close();
				}

				rs.close();
				sel.close();

			}
		} catch (PSQLException x) {
			// handled exception: try to iterate...
			logger.error(x.getLocalizedMessage());
		} catch (Exception x) {
			// not handled exception: stop
			logger.error(x.getLocalizedMessage());
			throw new DaoException(x.getClass().getName() + "::" + x.getLocalizedMessage(), x);
		}
	}

	public ArrayList<Integer> getSubTree(Integer rootId) throws DaoException {
		String queryBase = "select id from context where context_id in (%%)";
		ArrayList<Integer> subTree = new ArrayList<>();
		ArrayList<Integer> temp = new ArrayList<>();
		Statement s = null;
		ResultSet rs = null;
		temp.add(rootId);
		subTree.add(rootId);
		boolean again = true;
		for (int i = 0; (i < 10) && (again); i++) {
			try {
				String listId = "";
				for (Integer id : temp) {
					listId += "," + id;
				}
				listId = listId.substring(1);

				String query = queryBase.replace("%%", listId);
				s = conn.createStatement();
				rs = s.executeQuery(query);
				if (rs.next()) {
					temp = new ArrayList<>();
					Integer dbId = rs.getInt("id"); 
					subTree.add(dbId);
					temp.add(dbId);
					while (rs.next()) {
						dbId = rs.getInt("id"); 
						subTree.add(dbId);
						temp.add(dbId);
					}
				}
				else {
					again = false;
				} 
				rs.close();
				s.close();
			} catch (Exception x) {
				logger.error(x.getLocalizedMessage());
				throw new DaoException(x.getClass().getName() + "::" + x.getLocalizedMessage(), x);
			}
		}
		
		return subTree;
	}

	public void createSocialReport(int contextId, int orderBy) throws DaoException {
		Statement rep = null;
		ResultSet rs = null;
		try {
			String report = "insert into report (context_id, name, orderby, chart_type, x_text, y_text) values ";
			report += "(" + contextId + ", \'VRE Social\', " + orderBy
					+ ", \'column\', \'Period\', \'Social interaction\')";
			logger.debug(report);
			rep = conn.createStatement();
			rep.execute(report);
			String query = "select id from report where name=\'VRE Social\' order by id desc";
			rs = rep.executeQuery(query);
			if (rs.next()) {
				int id = rs.getInt("id");
				String reportItem = "insert into report_item (report_id, type_id_1, type_id_1_name) ";
				reportItem += "values (" + id + ", 8, \'mt\'), (" + id + ", 9, \'mt\'), (" + id + ", 10, \'mt\')";
				rep.execute(reportItem);
			} else
				throw new DaoException("No report id.", null);

		} catch (SQLException ex) {
			logger.error(ex.getLocalizedMessage());
			throw new DaoException(ex.getLocalizedMessage(), ex.getCause());
		} finally {
			try {
				if (rs != null) {
					rs.close();
				}
				if (rep != null) {
					rep.close();
				}
			} catch (SQLException ex) {
				logger.error(ex.getLocalizedMessage());
			}
		}
	}
	

	/**
	 * Dummy tester
	 * 
	 * @Deprecated
	 * @param originator
	 * @throws DaoException
	 * 
	 */
	public void dummyTest(String originator) throws DaoException {
		String insert = "insert into dummy (now, originator) values(current_timestamp, '" + originator + "')";
		logger.debug(insert);
		Statement s = null;
		try {
			s = conn.createStatement();
			s.execute(insert);
		} catch (SQLException ex) {
			logger.error(ex.getLocalizedMessage());
			throw new DaoException(ex.getLocalizedMessage(), ex.getCause());
		} finally {
			if (s != null)
				try {
					s.close();
				} catch (SQLException ex) {
					logger.error(ex.getLocalizedMessage());
				}
		}
	}

}