6af26b2f7e0f8911d8fdadd5ad23435d95ecec4f
[jenirok] / src / common / source.h
1 /*
2  * This file is part of Jenirok.
3  *
4  * Jenirok is free software: you can redistribute it and/or modify
5  * it under the terms of the GNU General Public License as published by
6  * the Free Software Foundation, either version 3 of the License, or
7  * (at your option) any later version.
8  *
9  * Jenirok is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with Jenirok.  If not, see <http://www.gnu.org/licenses/>.
16  *
17  */
18
19 #ifndef SOURCE_H
20 #define SOURCE_H
21
22 #include <QtCore/QObject>
23 #include <QtCore/QString>
24 #include <QtCore/QTimerEvent>
25 #include <QtCore/QUrl>
26 #include <QtCore/QByteArray>
27 #include <QtNetwork/QHttp>
28
29 class Source : public QObject
30 {
31
32     Q_OBJECT
33
34 public:
35
36     struct Result
37     {
38         QString name;
39         QString street;
40         QString city;
41         QString number;
42     };
43
44     enum SearchType {YELLOW_PAGES, PERSONS, BOTH};
45
46     struct SearchDetails
47     {
48         QString query;
49         QString location;
50         SearchType type;
51         SearchDetails(QString const& query = "",
52                       QString const& location = "",
53                       SearchType type = PERSONS);
54     };
55
56
57     enum Error {NO_ERROR, CONNECTION_FAILURE, INVALID_LOGIN, TIMEOUT};
58     enum SourceId {ENIRO, MOBIL1881, DASTELEFONBUCH};
59     static int const SOURCE_COUNT = 3;
60
61     struct SourceDetails
62     {
63         SourceId type;
64         QString name;
65         QString id;
66     };
67
68     static unsigned int const DEFAULT_MAX_RESULTS = 30;
69
70     static Source* getSource(SourceId id, QObject* parent = 0);
71     static void getSources(QList<SourceDetails>& list);
72     static SourceId stringToId(QString const& str);
73     Source(QObject* parent = 0);
74     virtual ~Source();
75     static Source* getSource();
76     virtual void abort();
77     virtual void search(SearchDetails const& details) = 0;
78     virtual void getSearchTypes(QList<SearchType>& types) const;
79     void setMaxResults(unsigned int results);
80     unsigned int getMaxResults() const;
81     void setTimeout(unsigned int ms);
82     unsigned int getTimeout() const;
83     void resetTimeout();
84     void setFindNumber(bool value);
85     bool getFindNumber() const;
86     Error error() const;
87     const QString& errorString() const;
88
89 signals:
90     void resultAvailable(Source::Result const& result, Source::SearchDetails const& details);
91     void requestFinished(QVector <Source::Result> const& results, Source::SearchDetails const& details, bool error);
92
93 protected:
94     void setError(Error error, QString const& errorString = "");
95     virtual void timerEvent(QTimerEvent *te);
96     static QString ucFirst(QString& string);
97     static QString& cleanUpNumber(QString& number);
98     static QString& stripTags(QString& string);
99     static QString& htmlEntityDecode(QString& string);
100     static bool isPhoneNumber(QString const& string);
101     void fixUrl(QUrl& url);
102     QHttp http_;
103
104 private slots:
105     void httpReady(int id, bool error);
106
107 private:
108     Q_DISABLE_COPY(Source);
109     virtual void handleHttpData(int id, QByteArray const& data) = 0;
110     virtual void handleHttpError(int id) = 0;
111     unsigned int maxResults_;
112     unsigned int timeout_;
113     int timerId_;
114     bool findNumber_;
115     Error error_;
116     QString errorString_;
117     QString username_;
118     QString password_;
119     bool loggedIn_;
120
121     static QRegExp numberCleaner_;
122     static QRegExp tagStripper_;
123
124 };
125
126 #endif