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 "http.h"
  23 #include "httputils.h"
  24
  25 YTSuggester::YTSuggester(QObject *parent) : Suggester(parent) {
  26
  27 }
  28
  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
  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 }