git.sur5r.net Git - minitube/blob - src/ytsuggester.cpp

   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 "networkaccess.h"
  23
  24 namespace The {
  25 NetworkAccess* http();
  26 }
  27
  28 YTSuggester::YTSuggester(QObject *parent) : Suggester(parent) {
  29
  30 }
  31
  32 void YTSuggester::suggest(const QString &query) {
  33     if (query.startsWith("http")) return;
  34
  35 #if QT_VERSION >= 0x040800
  36     QString locale = QLocale::system().uiLanguages().first();
  37 #else
  38     QString locale = QLocale::system().name().replace("_", "-");
  39 #endif
  40
  41     // case for system locales such as "C"
  42     if (locale.length() < 2) {
  43         locale = "en-US";
  44     }
  45
  46     QString url =
  47             QString("https://suggestqueries.google.com/complete/search?ds=yt&output=toolbar&hl=%1&q=%2")
  48             .arg(locale, query);
  49
  50     QObject *reply = The::http()->get(url);
  51     connect(reply, SIGNAL(data(QByteArray)), SLOT(handleNetworkData(QByteArray)));
  52 }
  53
  54 void YTSuggester::handleNetworkData(QByteArray response) {
  55     QList<Suggestion*> suggestions;
  56     QXmlStreamReader xml(response);
  57     while (!xml.atEnd()) {
  58         xml.readNext();
  59         if (xml.tokenType() == QXmlStreamReader::StartElement) {
  60             if (xml.name() == QLatin1String("suggestion")) {
  61                 QStringRef str = xml.attributes().value("data");
  62                 QString value = str.toString();
  63                 suggestions << new Suggestion(value);
  64             }
  65         }
  66     }
  67     emit ready(suggestions);
  68 }