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