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