package compbio.cassandra;

import java.util.List;

import org.apache.log4j.Logger;

import com.datastax.driver.core.Row;
import com.datastax.driver.core.Session;
import com.datastax.driver.core.ResultSet;

import compbio.engine.JpredJob;
import compbio.engine.ProteoCachePropertyHelperManager;
import compbio.util.PropertyHelper;

public class CassandraWriter {
	private Session session;
	private static final PropertyHelper ph = ProteoCachePropertyHelperManager.getPropertyHelper();
	private static Logger log = Logger.getLogger(CassandraNativeConnector.class);

	CassandraWriter() {
		Session inis = CassandraNativeConnector.getSession();
		setSession(inis);
	}

	public void setSession(Session s) {
		assert s != null;
		session = s;
	}

	public boolean JobisNotInsterted(String jobid) {
		ResultSet results1 = session.execute("SELECT * FROM ProteinLog WHERE JobID = '" + jobid + "';");
		if (results1.isExhausted()) {
			return true;
		}
		return false;
	}

	public boolean JobisNotArchived(String jobid) {
		ResultSet results1 = session.execute("SELECT * FROM JpredArchive WHERE JobID = '" + jobid + "';");
		if (results1.isExhausted()) {
			return true;
		}
		return false;
	}

	/*
	 * inserting data into the tables for queries
	 */
	public int FormQueryTables(JpredJob job) {
		if (JobisNotInsterted(job.getJobID())) {
			String id = job.getJobID();
			String ip = job.getIP();
			String protein = job.getProtein();
			String finalstatus = job.getFinalStatus();
			String execstatus = job.getExecutionStatus();
			String com1 = "INSERT INTO ProteinLog " + "(JobID, IP, DataBegin, DataEnd, FinalStatus, ExecutionStatus, Protein)"
					+ " VALUES ('" + id + "','" + ip + "','" + job.getStartingTimeStr() + "','" + job.getEndTimeStr() + "','" + finalstatus
					+ "','" + execstatus + "','" + protein + "');";
			session.execute(com1);

			String com2 = "INSERT INTO ProteinData " + "(jobtime, JobID, ExecTime, Protein)" + " VALUES (" + job.getStartingDate() + ",'" + id
					+ "'," + job.getExecutionTime() + ",'" + protein + "');";
			session.execute(com2);

			String allpredictions = "";
			List<FastaSequence> pr = job.getPredictions();
			for (FastaSequence pred : pr) {
				String predictionname = pred.getId();
				String prediction = pred.getSequence().replaceAll("\n", "");
				allpredictions += "'" + predictionname + "':'" + prediction + "',";
			}
			String final_prediction = "";
			if (!allpredictions.equals("")) {
				final_prediction = allpredictions.substring(0, allpredictions.length() - 1);
			}

			String check2 = "SELECT * FROM ProteinRow WHERE JobID = '" + job.getJobID() + "';";
			ResultSet results2 = session.execute(check2);
			if (results2.isExhausted()) {
				String com3 = "INSERT INTO ProteinRow " + "(Protein, JobID, Predictions)" + " VALUES ('" + protein + "','" + id + "',{"
						+ final_prediction + "});";
				session.execute(com3);
			}

			// update some internal query tables
			String check3 = "SELECT * FROM MainParameters WHERE Name = 'EarliestJobDate';";
			ResultSet results3 = session.execute(check3);
			boolean updateparameter = true;
			if (!results3.isExhausted()) {
				Row r = results3.one();
				if (job.getStartingDate() >= Long.parseLong(r.getString("Value")))
					updateparameter = false;
			}
			if (updateparameter) {
				String com = "INSERT INTO MainParameters " + "(Name, Value)" + " VALUES ('EarliestJobDate','" + job.getStartingDateStr()
						+ "');";
				session.execute(com);
			}
			String check4 = "SELECT * FROM JobDateInfo WHERE jobday = " + job.getStartingDate() + ";";
			ResultSet results4 = session.execute(check4);
			updateparameter = true;
			int njobs = 1;
			if (!results4.isExhausted()) {
				Row r = results4.one();
				njobs += r.getLong("Total");
			}
			String com = "INSERT INTO JobDateInfo " + "(jobday, Total)" + " VALUES (" + job.getStartingDate() + "," + njobs + ");";
			session.execute(com);

			return 1;
		}
		return 0;
	}

	/*
	 * insert data from a real Jpred job: timing+IP, Execution Status, Final
	 * status, protein sequence, predictions, alignment, LOG and tar.gz files
	 */
	public int ArchiveData(JpredJob job, String archivepath) {
		if (JobisNotArchived(job.getJobID())) {
			String id = job.getJobID();
			String log = job.getLog().replaceAll("'", "");
			String com = "INSERT INTO JpredArchive (JobID, Protein, IP, StartTime, ExecTime, FinalStatus, LOG, ArchiveLink) VALUES ('" + id + "','"
					+ job.getProtein() + "','" + job.getIP() + "'," + job.getStartingTime() + "," + job.getExecutionTime() + ",'" + job.getFinalStatus() + "','" + log
					+ "','" + archivepath + "');";
			session.execute(com);

			List<FastaSequence> predictions = job.getPredictions();
			for (FastaSequence p : predictions) {
				session.execute("UPDATE JpredArchive SET predictions = predictions + {'" + p.getId() + "':'"
						+ p.getSequence().replaceAll("\n", "") + "'} WHERE JobID = '" + id + "';");
			}

			List<FastaSequence> seqs = job.getAlignment();
			for (FastaSequence s : seqs) {
				session.execute("UPDATE JpredArchive SET alignment = alignment + {'" + s.getId() + "':'"
						+ s.getSequence().replaceAll("\n", "") + "'} WHERE JobID = '" + id + "';");
			}
			return 1;
		}
		return 0;
	}

}