package dk.thoerup.traininfoservice.banedk;

import java.io.IOException;
import java.util.List;
import java.util.logging.Logger;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * Servlet implementation class TimetableServlet
 */
public class TimetableServlet extends HttpServlet {
	private static final long serialVersionUID = 1L;
       
	
	Logger logger = Logger.getLogger( TimetableServlet.class.getName() );
	
	TimetableFetcher fetcher = new TimetableFetcher();
	
	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		String train = request.getParameter("train").trim();
		
		String sections[] = train.split(" ");
		if (sections.length != 2) {
			response.sendError(400, "Invalid parameter");
			return;
		}
		
		String type = lookupType(sections[0]);
		String trainID = sections[1];
		try
		{
			List<TimetableBean> timetable = fetcher.lookupTimetable(trainID, type);
		
			String responseXML = formatTimetable(timetable);
		
			response.setContentType("text/xml");
			response.getWriter().print(responseXML);
		} catch (Exception e) {
			logger.warning("failure looking up timetable: train=" +train);
			throw new ServletException(e);
		}
		
	}
	
	String lookupType(String trainCode) {
		trainCode = trainCode.toLowerCase();
		
		if (trainCode.equals("a") || trainCode.equals("b") || trainCode.equals("bx") || trainCode.equals("c") || 
				trainCode.equals("e") || trainCode.equals("f") || trainCode.equals("h")) {
			return "S2";
		} else {
			return "FJRN";
		}
	}
	
	String formatTimetable(List<TimetableBean> timetable) {
		StringBuffer sb = new StringBuffer();
		
		sb.append("<timetable>");
		
		for (TimetableBean bean : timetable) {
			sb.append("<entry");
			if (bean.isCurrent())
				sb.append(" current=\"true\"");
			sb.append(">");
			
			sb.append("<station>").append( bean.getStation() ).append("</station>");
			sb.append("<arrival>").append( bean.getArrival() ).append("</arrival>");
			sb.append("<departure>").append( bean.getDeparture() ).append("</departure>");
			
			sb.append("</entry>");
		}
		
		sb.append("</timetable>");
		
		return sb.toString();
	}

}