traininfoservice/banedk/RejseplanenFetcher.java

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<DepartureEntry> lookupDepartures(int stationcode, String type, boolean arrival) throws Exception {
                
                List<DepartureEntry> departureList = new ArrayList<DepartureEntry>();
                
            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") );
                                }
                                
                                
                                
                                DepartureEntry departure = new DepartureEntry();
                                
                                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;
        }
}
1	package dk.thoerup.traininfoservice.banedk;
2
3	/* Work in progress
4	*
5	*/
6
7	import java.text.SimpleDateFormat;
8	import java.util.ArrayList;
9	import java.util.Date;
10	import java.util.List;
11	import java.util.logging.Logger;
12
13	import com.gargoylesoftware.htmlunit.BrowserVersion;
14	import com.gargoylesoftware.htmlunit.WebClient;
15	import com.gargoylesoftware.htmlunit.html.DomNodeList;
16	import com.gargoylesoftware.htmlunit.html.HtmlElement;
17	import com.gargoylesoftware.htmlunit.html.HtmlPage;
18
19	public class RejseplanenFetcher {
20
21	Logger logger = Logger.getLogger(RejseplanenFetcher.class.getName());
22
23	SimpleDateFormat format = new SimpleDateFormat("HH:mm");
24
25	public String getNow() {
26	return format.format( new Date() );
27	}
28
29	public List<DepartureEntry> lookupDepartures(int stationcode, String type, boolean arrival) throws Exception {
30
31	List<DepartureEntry> departureList = new ArrayList<DepartureEntry>();
32
33	final WebClient webClient = new WebClient( BrowserVersion.FIREFOX_3 );
34	webClient.getCookieManager().setCookiesEnabled( false );
35	webClient.setTimeout(2500);
36	webClient.setJavaScriptEnabled(false);
37
38
39
40	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";
41
42	HtmlPage page = webClient.getPage(uri);
43
44
45	HtmlElement table = page.getElementById("hafasSqResults");
46
47	if (table != null) {
48	DomNodeList<HtmlElement> tableRows = table.getElementsByTagName("tr");
49
50	for (HtmlElement currentRow : tableRows) {
51	String id = currentRow.getId();
52	logger.info("tr.id=" + id);
53	String rowClass = currentRow.getAttribute("class");
54	if (rowClass != null && rowClass.indexOf("sqToggleDetails") != -1 ) {
55	DomNodeList<HtmlElement> fields = currentRow.getElementsByTagName("td");
56
57	logger.severe("td count=" + fields.size());
58	if (fields.size() == 0)
59	continue;
60
61	for (int f=0; f<fields.size();f++) {
62	logger.severe("" + f + "=" + fields.get(f).getAttribute("class") );
63	}
64
65
66
67	DepartureEntry departure = new DepartureEntry();
68
69	String time = fields.get(0).asText();
70	departure.setTime(time);
71
72	int updated = 0; //updated not supported by rejseplanen
73	departure.setUpdated(updated);
74
75	String trainNumber = fields.get(1).asText();
76	departure.setTrainNumber(trainNumber);
77
78
79	//field 2 is the track number
80
81
82	String destination = fields.get(3).
83	getElementsByTagName("div").
84	get(0).
85	getElementsByAttribute("span", "class", "bold").
86	get(0).
87	asText();
88	departure.setDestination(destination);
89
90	//origin may take a bit more work
91	//String origin = fields.get(4).asText();
92	//departure.setOrigin(origin);
93
94	/*
95	//rejseplanen doesn't support location
96	String location = ""; //fields.get(5).asText();
97	departure.setLocation(location);
98
99	String status = fields.get(6).asText();
100	departure.setStatus(status);
101
102	String note = extractNote( fields.get(7) );
103	departure.setNote(note);
104
105	departure.setType(type);
106	*/
107	departureList.add(departure);
108
109	}
110	}
111	} else {
112	logger.warning("No departures found for station=" + stationcode + ", type=" + type);
113	}
114	webClient.closeAllWindows();
115
116	return departureList;
117	}
118	}