2 * This file is part of Jenirok.
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.
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.
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/>.
22 #include <QtCore/QObject>
23 #include <QtCore/QString>
24 #include <QtCore/QVector>
25 #include <QtCore/QSet>
26 #include <QtCore/QRegExp>
27 #include <QtCore/QUrl>
28 #include <QtNetwork/QHttp>
30 class Eniro: public QObject
36 enum Site {FI, SE, DK};
37 static const int SITE_COUNT = 3;
39 enum SearchType {YELLOW_PAGES, PERSONS};
41 enum Error {NO_ERROR, CONNECTION_FAILURE, INVALID_LOGIN};
56 SearchDetails(QString const& query = "",
57 QString const& location = "",
58 SearchType type = PERSONS);
67 Eniro(Site site, QObject *parent = 0);
71 void login(QString const& username, QString const& password);
75 void setMaxResults(unsigned int value);
76 void setFindNumber(bool value);
77 bool search(SearchDetails const& details);
80 const QString& errorString() const;
81 static QMap<Site, SiteDetails> getSites();
82 static Site stringToSite(QString const& str);
85 void resultAvailable(Eniro::Result const& result, Eniro::SearchDetails const& details);
86 void requestFinished(QVector <Eniro::Result> const& results, Eniro::SearchDetails const& details, bool error);
87 void loginStatus(bool success);
90 void httpReady(int id, bool error);
94 Q_DISABLE_COPY(Eniro);
104 SearchDetails details;
105 QVector <Result> results;
106 unsigned int foundNumbers;
107 unsigned int numbersTotal;
110 QUrl createUrl(QString const& query, QString const& location);
111 void loadResults(int id, QString const& data);
112 void loadNumber(int id, QString const& data);
113 void getNumberForResult(int id, int index, SearchDetails const& details);
114 void emitRequestFinished(int key, SearchData* data, bool error);
115 QString ucFirst(QString& string);
116 QString& cleanUpNumber(QString& number);
117 QString& stripTags(QString& string);
125 QString errorString_;
126 unsigned int maxResults_;
128 typedef QMap <int, SearchData*> searchMap;
129 typedef QMap <int, NumberData*> numberMap;
131 searchMap pendingSearches_;
132 numberMap pendingNumberRequests_;
133 QSet <int> pendingLoginRequests_;
135 static QRegExp numberCleaner_;
136 static QRegExp tagStripper_;