git.sur5r.net Git - minitube/blob - src/channelsuggest.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 "channelsuggest.h"
  22 #include "networkaccess.h"
  23
  24 namespace The {
  25     NetworkAccess* http();
  26 }
  27
  28 ChannelSuggest::ChannelSuggest(QObject *parent) : Suggester(parent) {
  29
  30 }
  31
  32 void ChannelSuggest::suggest(const QString &query) {
  33     QUrl url("https://www.youtube.com/results");
  34 #if QT_VERSION >= 0x050000
  35         {
  36             QUrl &u = url;
  37             QUrlQuery url;
  38 #endif
  39     url.addQueryItem("search_type", "search_users");
  40     url.addQueryItem("search_query", query);
  41 #if QT_VERSION >= 0x050000
  42             u.setQuery(url);
  43         }
  44 #endif
  45     QObject *reply = The::http()->get(url);
  46     connect(reply, SIGNAL(data(QByteArray)), SLOT(handleNetworkData(QByteArray)));
  47 }
  48
  49 void ChannelSuggest::handleNetworkData(QByteArray data) {
  50     QStringList choices;
  51     QList<Suggestion*> suggestions;
  52
  53     QString html = QString::fromUtf8(data);
  54     QRegExp re("/(?:user|channel)/[a-zA-Z0-9]+[^>]+data-ytid=[\"']([^\"']+)[\"'][^>]+>([a-zA-Z0-9 ]+)</a>");
  55
  56     int pos = 0;
  57     while ((pos = re.indexIn(html, pos)) != -1) {
  58         QString choice = re.cap(2);
  59         if (!choices.contains(choice, Qt::CaseInsensitive)) {
  60             qDebug() << re.capturedTexts();
  61             QString channelId = re.cap(1);
  62             suggestions << new Suggestion(choice, "channel", channelId);
  63             choices << choice;
  64             if (choices.size() == 10) break;
  65         }
  66         pos += re.matchedLength();
  67     }
  68
  69     emit ready(suggestions);
  70 }