package dk.thoerup.traininfoservice.banedk;

 /* Work in progress 
  * 
  */

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.logging.Logger;

import com.gargoylesoftware.htmlunit.BrowserVersion;
import com.gargoylesoftware.htmlunit.WebClient;
import com.gargoylesoftware.htmlunit.html.DomNodeList;
import com.gargoylesoftware.htmlunit.html.HtmlElement;
import com.gargoylesoftware.htmlunit.html.HtmlPage;

public class RejseplanenFetcher {
	
	Logger logger = Logger.getLogger(RejseplanenFetcher.class.getName());
	
	SimpleDateFormat format = new SimpleDateFormat("HH:mm");
	
	public String getNow() {
		return format.format( new Date() ); 
	}
	
	public List<DepartureBean> lookupDepartures(int stationcode, String type, boolean arrival) throws Exception {
		
		List<DepartureBean> departureList = new ArrayList<DepartureBean>();
		
	    final WebClient webClient = new WebClient( BrowserVersion.FIREFOX_3 );
	    webClient.getCookieManager().setCookiesEnabled( false );
	    webClient.setTimeout(2500);
	    webClient.setJavaScriptEnabled(false);

	    	    	   
	    
	    String uri = "http://www.rejseplanen.dk/bin/stboard.exe/mn?ml=m&input=" + stationcode + "&boardType=dep&time=" + getNow() + "&productsFilter=11111000001&selectDate=today&maxJourneys=200&start=yes";
	    
	    HtmlPage page = webClient.getPage(uri);
	    
	    
	    HtmlElement table = page.getElementById("hafasSqResults");
	    
	    if (table != null) {
		    DomNodeList<HtmlElement> tableRows =  table.getElementsByTagName("tr");

		    for (HtmlElement currentRow : tableRows) {
		    	String id = currentRow.getId();
		    	logger.info("tr.id=" + id);
		    	String rowClass = currentRow.getAttribute("class");
		    	if (rowClass != null  && rowClass.indexOf("sqToggleDetails") != -1	) {
		    		DomNodeList<HtmlElement> fields = currentRow.getElementsByTagName("td");
	
		    		logger.severe("td count=" + fields.size());
		    		if (fields.size() == 0)
		    			continue;

		    		for (int f=0; f<fields.size();f++) {
		    			logger.severe("" + f + "=" + fields.get(f).getAttribute("class") );
		    		}
		    		
		    		
		    		
		    		DepartureBean departure = new DepartureBean();
		    		
		    		String time = fields.get(0).asText();	
		    		departure.setTime(time);
		    		
		    		int updated = 0; //updated not supported by rejseplanen
		    		departure.setUpdated(updated);
		    		
		    		String trainNumber = fields.get(1).asText();
		    		departure.setTrainNumber(trainNumber);
		    		
		    		
		    		//field 2 is the track number
		    		
		    		
		    		String destination = fields.get(3).
		    			getElementsByTagName("div").
		    			get(0).
		    			getElementsByAttribute("span", "class", "bold").
		    			get(0).
		    			asText();
		    		departure.setDestination(destination);
		    		
		    		//origin may take a bit more work
		    		//String origin = fields.get(4).asText();
		    		//departure.setOrigin(origin);
		    		
		    		/*
		    		//rejseplanen doesn't support location
		    		String location = ""; //fields.get(5).asText();
		    		departure.setLocation(location);
		    		
		    		String status = fields.get(6).asText();
		    		departure.setStatus(status);
		    		
		    		String note = extractNote( fields.get(7) );
		    		departure.setNote(note);
		    		
		    		departure.setType(type);
		    		*/
		    		departureList.add(departure);

		    	}
		    }
	    } else {
	    	logger.warning("No departures found for station=" + stationcode + ", type=" + type);
	    }
	    webClient.closeAllWindows();
	    
	    return departureList;
	}
}