portal/src/com/biomed/server/handlers/ScheduleHandler.java

package com.biomed.server.handlers;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.Date;

import javax.inject.Inject;

import com.biomed.server.CalendarApi;
import com.biomed.server.dispatch.ActionHandler;
import com.biomed.server.dispatch.ExecutionContext;
import com.biomed.shared.api.EmptyResult;
import com.biomed.shared.api.ScheduleAction;
import com.google.inject.Provider;

public class ScheduleHandler implements
		ActionHandler<ScheduleAction, EmptyResult> {

	private static final String QUERY = "INSERT INTO workorder "
			+ "(job_date, client_id, tech, job_start, job_end, reason, remarks, job_status_id, job_scheduled_date, caller)"
			+ " VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)";

	private static final String FETCH_DATA_QUERY = "SELECT u.email, c.client_name, c.client_identification, c.address, c.address_2, c.city, c.state, c.zip, c.phone FROM user AS u, client AS c WHERE u.id = ? AND c.id = ?";

	private static final String UPDATE_EVENT_ID_QUERY = "UPDATE workorder SET google_event_id = ? WHERE id = ?";

	@Override
	public Class<ScheduleAction> getType() {
		return ScheduleAction.class;
	}

	@Inject
	Provider<Connection> connectionProvider;

	@Inject
	CalendarApi calendarApi;

	@Override
	public EmptyResult execute(ScheduleAction action, ExecutionContext context)
			throws Exception {
		Connection connection = connectionProvider.get();

		int workorderId = insertWorkorder(connection, action);
		String eventId = updateGoogleCalendar(connection, action);
		updateGoogleEventId(connection, workorderId, eventId);

		connection.close();

		return new EmptyResult();
	}

	private static int insertWorkorder(Connection connection,
			ScheduleAction action) throws SQLException {

		PreparedStatement query = connection.prepareStatement(QUERY,
				Statement.RETURN_GENERATED_KEYS);
		query.setString(1, SqlUtil.formatDate(action.getJobDate()));
		query.setInt(2, action.getClientId());
		query.setInt(3, action.getTechId());
		query.setInt(4, action.getStartTime().toInteger());
		query.setInt(5, action.getEndTime().toInteger());
		query.setInt(6, action.getReason().getId());
		query.setString(7, action.getRemarks());
		query.setInt(8, action.getStatus().getId());
		query.setString(9, SqlUtil.formatDate(new Date()));
		query.setString(10, action.getRequestedBy());
		query.execute();

		ResultSet keys = query.getGeneratedKeys();
		if (keys.next()) {
			return keys.getInt(1);
		} else {
			return -1;
		}
	}

	private String updateGoogleCalendar(Connection connection,
			ScheduleAction action) throws SQLException {
		Date startDate = action.getStartTime().toDate(action.getJobDate());
		Date endDate = action.getEndTime().toDate(action.getJobDate());

		PreparedStatement query2 = connection.prepareStatement(FETCH_DATA_QUERY);
		query2.setInt(1, action.getTechId());
		query2.setInt(2, action.getClientId());
		query2.execute();

		ResultSet results = query2.getResultSet();
		results.first();

		String email = results.getString("email");
		String summary = results.getString("client_name");
		
		String location = formatLocation(results);
		String description = formatDescription(results, action);
		
		return calendarApi.scheduleEvent(email, startDate, endDate, summary,
				description, location);
	}

	private String formatLocation(ResultSet results) throws SQLException {
		String address = results.getString("address");
		String address2 = results.getString("address_2");
		String city = results.getString("city");
		String state = results.getString("state");
		String zip = results.getString("zip");

		if (address2 != null) {
			return String.format("%s %s, %s, %s. %s", address, address2, city, state, zip);
		} else {
			return String.format("%s, %s, %s. %s", address, city, state, zip);
		}		
	}
	
	private String formatDescription(ResultSet results, ScheduleAction action) throws SQLException {
		
		StringBuilder builder = new StringBuilder();

		String address2 = results.getString("address_2");
		
		builder
			.append("Customer:\n")
			.append("\t").append(results.getString("client_identification")).append("\n\n")
			.append("Phone:\n")
			.append("\t").append(results.getString("phone")).append("\n\n")
			.append("Address:\n")
			.append("\t").append(results.getString("address")).append("\n");

		if (address2 != null) {
			builder.append("\t").append(address2).append("\n");
		}
		
		builder
				.append("\t").append(results.getString("city"))
				.append(", ").append(results.getString("state"))
				.append(". ").append(results.getString("zip"))
				.append("\n\n");
		
		if (action.getRequestedBy() != null) {
			builder.append("Requested By:\n")
					.append("\t").append(action.getRequestedBy()).append("\n\n");
		}

		builder
				.append("Reason:\n")
				.append("\t").append(action.getReason().getLabel()).append("\n\n")
				.append("Status:\n")
				.append("\t").append(action.getStatus().getLabel()).append("\n\n");
		
		if (action.getRemarks() != null) {
			builder.append("Remarks:\n")
					.append("\t").append(action.getRemarks()).append("\n");
		}

		return builder.toString();
	}
	
	private static void updateGoogleEventId(Connection connection,
			int workorderId, String eventId) throws SQLException {

		PreparedStatement query = connection
				.prepareStatement(UPDATE_EVENT_ID_QUERY);
		query.setString(1, eventId);
		query.setInt(2, workorderId);

		query.execute();
	}

}