smsdaemon/plugins/TrainInfo.cpp


#include "TrainInfo.h"

#include "string_nocase.h"
#include "Util.h"
#include "Exceptions.h"
#include "Logger.h"

#include "HttpClient.h"

#include <vector>
#include <sstream>
#include <stdexcept>


using namespace std;

string ConvertUpdateInfo(char ch)
{
        string result;
        switch( ch)
        {
                case '1':
                        result = "< 3 min";
                        break;
                case '2':
                        result = "3 - 10 min";
                        break;
                case '3':
                        result = "> 10 min";
                        break;
                case '4':
                        result = "No info";
                        break;
                default:
                        result = "Unknown";
        }
        return result;
}

inline string clean(string& str)
{
        using namespace Util;

        str = str_replace(str, "&nbsp;", " ");

        return str_trim(str_characters(str_striptags(str)));
}

void CleanTrainInfo(TrainInfo& info)
{
        info.time = clean( info.time );
        info.type = clean( info.type );
        info.destination = clean( info.destination );
        info.origin = clean( info.origin );
        info.current = clean( info.current );
        info.status = clean( info.status );
        info.note = clean( info.note );
}

vector<TrainInfo> GetTrainInfo(string stationcode, string stationname)
{
        vector<TrainInfo> result;

        string url = string("http://www.bane.dk/visStation.asp?W=FJRN&S=") + stationcode + "&artikelId=4275&statnavn=" + stationname;

        //string raw_doc = Util::readUrl(url, "/tmp/sms_tog.tmp" );
        
        string raw_doc;
        try
        { 
                raw_doc = HttpClient::GetString(url);
        } 
        catch (httpexception& e)
        {
                Logger::logMessage(e.what());
        }

        if (raw_doc == "")
        {
                throw std::runtime_error("Connection timeout");
        }

        raw_doc = Util::str_replace(raw_doc, "\"", "'");

        string_nocase html = raw_doc.c_str();

        unsigned int pos =0;

        unsigned int ankomstpos = html.find("id='ankomsttabel'");

        while(1)
        {
                //start with time
                pos = html.find("<td class='tid'>", pos);

                if (pos == string::npos)
                        break;

                if (pos > ankomstpos)
                        break;

                TrainInfo info;

                int start = pos+16;
                int stop = html.find("</td>", pos);

                info.time = html.substr(start, stop-start).c_str();

                //update info
                pos = html.find("<td class='opdater'>", pos);
                pos += 54;
                char update = html.at(pos);
                info.update = ConvertUpdateInfo(update);

                //type
                pos = html.find( "<td class='togtype'>", pos);
                start = pos + 20;
                stop = html.find("</td>", pos);
                info.type = html.substr(start, stop-start).c_str();

                //destination
                pos = html.find( "<td class='til'>", pos);
                start = pos + 16;
                stop = html.find( "</td>", pos);
                info.destination = html.substr(start, stop-start).c_str();

                //origin
                pos = html.find( "<td class='visikke'>", pos);
                start = pos + 20;
                stop = html.find( "</td>", pos);
                info.origin = html.substr(start, stop-start).c_str();

                //Current location
        pos++; //since origin and current use the same search pattern, we must increase the position
                      //so we don't get origin twice.
                
                pos = html.find( "<td class='visikke'>", pos);
                start = pos + 20;
                stop = html.find( "</td>", pos);
                info.current = html.substr(start, stop-start).c_str();

                //status
                pos = html.find( "<td class='status'>", pos);
                start = pos + 19;
                stop = html.find( "</td>", pos);
                info.status = html.substr(start, stop-start).c_str();

                //note
                pos = html.find( "<td class='bem'>", pos);
                start = pos + 16;
                stop = html.find( "</td>", pos);
                info.note = html.substr(start, stop-start).c_str();

                CleanTrainInfo(info);

                result.push_back(info);
        }

        return result;
}
1	torben	44
2			#include "TrainInfo.h"
3
4			#include "string_nocase.h"
5	torben	158	#include "Util.h"
6	torben	193	#include "Exceptions.h"
7			#include "Logger.h"
8	torben	44
9	torben	193	#include "HttpClient.h"
10
11	torben	44	#include <vector>
12			#include <sstream>
13	torben	81	#include <stdexcept>
14	torben	44
15	torben	81
16	torben	44	using namespace std;
17
18			string ConvertUpdateInfo(char ch)
19			{
20			string result;
21			switch( ch)
22			{
23			case '1':
24			result = "< 3 min";
25			break;
26			case '2':
27			result = "3 - 10 min";
28			break;
29			case '3':
30			result = "> 10 min";
31			break;
32			case '4':
33			result = "No info";
34			break;
35			default:
36			result = "Unknown";
37			}
38			return result;
39			}
40
41			inline string clean(string& str)
42			{
43			using namespace Util;
44
45			str = str_replace(str, " ", " ");
46
47			return str_trim(str_characters(str_striptags(str)));
48			}
49
50			void CleanTrainInfo(TrainInfo& info)
51			{
52			info.time = clean( info.time );
53			info.type = clean( info.type );
54			info.destination = clean( info.destination );
55			info.origin = clean( info.origin );
56			info.current = clean( info.current );
57			info.status = clean( info.status );
58			info.note = clean( info.note );
59			}
60
61			vector<TrainInfo> GetTrainInfo(string stationcode, string stationname)
62			{
63			vector<TrainInfo> result;
64
65			string url = string("http://www.bane.dk/visStation.asp?W=FJRN&S=") + stationcode + "&artikelId=4275&statnavn=" + stationname;
66
67	torben	193	//string raw_doc = Util::readUrl(url, "/tmp/sms_tog.tmp" );
68
69			string raw_doc;
70			try
71			{
72			raw_doc = HttpClient::GetString(url);
73			}
74			catch (httpexception& e)
75			{
76			Logger::logMessage(e.what());
77			}
78	torben	81
79			if (raw_doc == "")
80			{
81			throw std::runtime_error("Connection timeout");
82			}
83
84	torben	44	raw_doc = Util::str_replace(raw_doc, "\"", "'");
85
86			string_nocase html = raw_doc.c_str();
87
88	torben	64	unsigned int pos =0;
89	torben	44
90	torben	64	unsigned int ankomstpos = html.find("id='ankomsttabel'");
91
92	torben	44	while(1)
93			{
94			//start with time
95			pos = html.find("<td class='tid'>", pos);
96
97			if (pos == string::npos)
98			break;
99
100	torben	64	if (pos > ankomstpos)
101			break;
102
103	torben	44	TrainInfo info;
104
105			int start = pos+16;
106			int stop = html.find("</td>", pos);
107
108			info.time = html.substr(start, stop-start).c_str();
109
110			//update info
111			pos = html.find("<td class='opdater'>", pos);
112			pos += 54;
113			char update = html.at(pos);
114			info.update = ConvertUpdateInfo(update);
115
116			//type
117			pos = html.find( "<td class='togtype'>", pos);
118			start = pos + 20;
119			stop = html.find("</td>", pos);
120			info.type = html.substr(start, stop-start).c_str();
121
122			//destination
123			pos = html.find( "<td class='til'>", pos);
124			start = pos + 16;
125			stop = html.find( "</td>", pos);
126			info.destination = html.substr(start, stop-start).c_str();
127
128			//origin
129			pos = html.find( "<td class='visikke'>", pos);
130			start = pos + 20;
131			stop = html.find( "</td>", pos);
132			info.origin = html.substr(start, stop-start).c_str();
133
134			//Current location
135			pos++; //since origin and current use the same search pattern, we must increase the position
136			//so we don't get origin twice.
137
138			pos = html.find( "<td class='visikke'>", pos);
139			start = pos + 20;
140			stop = html.find( "</td>", pos);
141			info.current = html.substr(start, stop-start).c_str();
142
143			//status
144			pos = html.find( "<td class='status'>", pos);
145			start = pos + 19;
146			stop = html.find( "</td>", pos);
147			info.status = html.substr(start, stop-start).c_str();
148
149			//note
150			pos = html.find( "<td class='bem'>", pos);
151			start = pos + 16;
152			stop = html.find( "</td>", pos);
153			info.note = html.substr(start, stop-start).c_str();
154
155			CleanTrainInfo(info);
156
157			result.push_back(info);
158			}
159
160			return result;
161			}