1 /* $BEGIN_LICENSE
2 
3 This file is part of Minitube.
4 Copyright 2009, Flavio Tordini <flavio.tordini@gmail.com>
5 
6 Minitube is free software: you can redistribute it and/or modify
7 it under the terms of the GNU General Public License as published by
8 the Free Software Foundation, either version 3 of the License, or
9 (at your option) any later version.
10 
11 Minitube is distributed in the hope that it will be useful,
12 but WITHOUT ANY WARRANTY; without even the implied warranty of
13 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14 GNU General Public License for more details.
15 
16 You should have received a copy of the GNU General Public License
17 along with Minitube.  If not, see <http://www.gnu.org/licenses/>.
18 
19 $END_LICENSE */
20 
21 #include "ytsuggester.h"
22 #include "http.h"
23 #include "httputils.h"
24 
YTSuggester(QObject * parent)25 YTSuggester::YTSuggester(QObject *parent) : Suggester(parent) {
26 
27 }
28 
suggest(const QString & query)29 void YTSuggester::suggest(const QString &query) {
30     if (query.startsWith(QLatin1String("http"))) return;
31 
32     QString locale = QLocale::system().uiLanguages().at(0);
33 
34     // case for system locales such as "C"
35     if (locale.length() < 2) {
36         locale = "en-US";
37     }
38 
39     QString url =
40             QStringLiteral("https://suggestqueries.google.com/complete/search?ds=yt&output=toolbar&hl=%1&q=%2")
41             .arg(locale, query);
42 
43     QObject *reply = HttpUtils::yt().get(url);
44     connect(reply, SIGNAL(data(QByteArray)), SLOT(handleNetworkData(QByteArray)));
45 }
46 
handleNetworkData(QByteArray response)47 void YTSuggester::handleNetworkData(QByteArray response) {
48     QVector<Suggestion*> suggestions;
49     suggestions.reserve(10);
50     QXmlStreamReader xml(response);
51     while (!xml.atEnd()) {
52         xml.readNext();
53         if (xml.tokenType() == QXmlStreamReader::StartElement) {
54             if (xml.name() == QLatin1String("suggestion")) {
55                 QStringRef str = xml.attributes().value(QLatin1String("data"));
56                 QString value = str.toString();
57                 suggestions << new Suggestion(value);
58             }
59         }
60     }
61     emit ready(suggestions);
62 }
63