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