package uk.ac.ebi.intenz.mapper;

 import java.sql.Connection;

 import java.sql.PreparedStatement;

 import java.sql.SQLException;

 import java.sql.Timestamp;

 import java.sql.Types;

 import uk.ac.ebi.intenz.domain.constants.Event;

 public class HistoryEventMapper {

         private static final String INSERT_EVENT_SQL =

                 "INSERT INTO history_events" +

                 " (event_id, group_id, event_class, before_id, after_id, event_note, event_year)" +

                 " VALUES (s_history_event_id.nextval, s_history_group_id.nextval, ?, ?, ?, ?, ?)";

         //TODO

         private static final String SELECT_EVENT_SQL =

                 "SELECT group_id, event_id, before_id, after_id, event_year," +

                 " event_note, event_class FROM history_events" +

                 " WHERE before_id = ? OR after_id = ?";

         private static final String UPDATE_EVENT_SQL =

                         "UPDATE history_events" +

                         " SET before_id = ?, after_id = ?, event_note = ?" +

                         " WHERE event_id = ? AND group_id = ?";

         private static final String UPDATE_EVENT_NOTE_SQL =

                         "UPDATE history_events SET event_note = ?" +

                         " WHERE event_id = ? AND group_id = ?";

     public void insertEvent(Event event, Long beforeId, Long afterId,

                         String note, Connection con) throws SQLException{

             PreparedStatement stm = null;

             try {

                     stm = con.prepareStatement(INSERT_EVENT_SQL);

                     stm.setString(1, event.getCode());

                         switch (event) {

                         case CREATION:

                                 stm.setNull(2, Types.NUMERIC);

                                 stm.setLong(3, afterId);

                                 break;

                         case TRANSFER:

                                 stm.setLong(2, beforeId);

                                 stm.setLong(3, afterId);

                                 break;

                         case DELETION:

                                 stm.setLong(2, beforeId);

                                 stm.setNull(3, Types.NUMERIC);

                                 break;

                         default:

                                 throw new IllegalArgumentException("Event not supported: " + event);

                         }

                         if (note == null){

                                 stm.setNull(4, Types.VARCHAR);

                         } else {

                                 stm.setString(4, note);

                         }

                         stm.setTimestamp(5, new Timestamp(System.currentTimeMillis()));

                         stm.executeUpdate();

             } finally {

                     if (stm != null) stm.close();

             }

         }

     /**

      * Updates the note for an event in the history.

      * @param eventId the ID of the event.

      * @param groupId the ID of the event group.

      * @param newNote the new note for the event.

      * @param con a database connection.

      * @throws SQLException

      */

     public void updateEventNote(int eventId, int groupId, String newNote,

                     Connection con) throws SQLException{

             PreparedStatement ps = null;

             try {

                     ps = con.prepareStatement(UPDATE_EVENT_NOTE_SQL);

                     int n = 1;

                     ps.setString(n++, newNote);

                     ps.setInt(n++, eventId);

                     ps.setInt(n++, groupId);

                     ps.executeUpdate();

             } finally {

                     if (ps != null) ps.close();

             }

     }

     /**

      * Modifies an event in the history.

      * @param eventId the event ID.

      * @param groupId the event group ID.

      * @param newNote the new history note.

      * @param beforeId the ID of the enzyme before the event.

      * @param afterId the ID of the enzyme after the event.

      * @param con a database connection.

      * @throws SQLException

      */

     public void updateEvent(int eventId, int groupId, String newNote,

                     Integer beforeId, Integer afterId, Connection con)

         throws SQLException{

             PreparedStatement ps = null;

             try {

                     ps = con.prepareStatement(UPDATE_EVENT_SQL);

                     int n = 1;

                     if (beforeId == null){

                             ps.setNull(n++, Types.INTEGER);

                     } else {

                             ps.setInt(n++, beforeId);

                     }

                     if (afterId == null){

                             ps.setNull(n++, Types.INTEGER);

                     } else {

                             ps.setInt(n++, afterId);

                     }

                     ps.setString(n++, newNote);

                     ps.setInt(n++, eventId);

                     ps.setInt(n++, groupId);

                     ps.executeUpdate();

             } finally {

                     if (ps != null) ps.close();

             }

     }

 }