git.sur5r.net Git - minitube/blob - src/channelmodel.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 "channelmodel.h"
  22 #include "ytchannel.h"
  23
  24 static const int channelOffset = 2;
  25
  26 ChannelModel::ChannelModel(QObject *parent) :
  27     QAbstractListModel(parent),
  28     hoveredRow(-1) { }
  29
  30
  31 int ChannelModel::rowCount(const QModelIndex &) const {
  32     return channels.isEmpty() ? 0 : channelOffset + channels.size();
  33 }
  34
  35 QVariant ChannelModel::data(const QModelIndex &index, int role) const {
  36     switch (role) {
  37
  38     case ChannelModel::ItemTypeRole:
  39         return typeForIndex(index);
  40
  41     case ChannelModel::DataObjectRole:
  42         if (typeForIndex(index) == ChannelModel::ItemChannel)
  43             return QVariant::fromValue(QPointer<YTChannel>(channelForIndex(index)));
  44         break;
  45
  46     case ChannelModel::HoveredItemRole:
  47         return hoveredRow == index.row();
  48
  49     case Qt::StatusTipRole:
  50         if (typeForIndex(index) == ChannelModel::ItemChannel)
  51             return channelForIndex(index)->getDescription();
  52
  53     }
  54
  55     return QVariant();
  56 }
  57
  58 YTChannel* ChannelModel::channelForIndex(const QModelIndex &index) const {
  59     const int row = index.row();
  60     if (row < channelOffset) return 0;
  61     return channels.at(index.row() - channelOffset);
  62 }
  63
  64 ChannelModel::ItemTypes ChannelModel::typeForIndex(const QModelIndex &index) const {
  65     switch (index.row()) {
  66     case 0:
  67         return ChannelModel::ItemAggregate;
  68     case 1:
  69         return ChannelModel::ItemUnwatched;
  70     default:
  71         return ChannelModel::ItemChannel;
  72     }
  73 }
  74
  75 void ChannelModel::setQuery(const QString &query, const QSqlDatabase &db) {
  76     beginResetModel();
  77     channels.clear();
  78     sqlError = QSqlError();
  79
  80     QSqlQuery q(db);
  81     q.prepare(query);
  82     bool success = q.exec();
  83     if (!success) {
  84         qWarning() << q.lastQuery() << q.lastError().text();
  85         sqlError = q.lastError();
  86     }
  87     channels.reserve(q.size());
  88     while (q.next()) {
  89         YTChannel *channel = YTChannel::forId(q.value(0).toString());
  90         connect(channel, SIGNAL(thumbnailLoaded()), SLOT(updateSender()), Qt::UniqueConnection);
  91         connect(channel, SIGNAL(notifyCountChanged()), SLOT(updateSender()), Qt::UniqueConnection);
  92         connect(channel, SIGNAL(destroyed(QObject *)), SLOT(removeChannel(QObject *)), Qt::UniqueConnection);
  93         channels << channel;
  94     }
  95     channels.squeeze();
  96     endResetModel();
  97 }
  98
  99 QSqlError ChannelModel::lastError() const {
 100     return sqlError;
 101 }
 102
 103 void ChannelModel::updateSender() {
 104     YTChannel *channel = static_cast<YTChannel*>(sender());
 105     if (!channel) {
 106         qWarning() << "Cannot get sender" << __PRETTY_FUNCTION__;
 107         return;
 108     }
 109     updateChannel(channel);
 110 }
 111
 112 void ChannelModel::updateChannel(YTChannel *channel) {
 113     int row = channels.indexOf(channel);
 114     if (row == -1) return;
 115     row += channelOffset;
 116     QModelIndex i = createIndex(row, 0);
 117     emit dataChanged(i, i);
 118 }
 119
 120 void ChannelModel::updateUnwatched() {
 121     QModelIndex i = createIndex(1, 0);
 122     emit dataChanged(i, i);
 123 }
 124
 125 void ChannelModel::removeChannel(QObject *obj) {
 126     YTChannel *channel = static_cast<YTChannel*>(obj);
 127     // qWarning() << "channel" << channel << obj << obj->metaObject()->className();
 128     if (!channel) return;
 129
 130     int row = channels.indexOf(channel);
 131     if (row == -1) return;
 132
 133     int position = row + channelOffset;
 134     beginRemoveRows(QModelIndex(), position, position+1);
 135     channels.removeAt(row);
 136     endRemoveRows();
 137 }
 138
 139 void ChannelModel::setHoveredRow(int row) {
 140     int oldRow = hoveredRow;
 141     hoveredRow = row;
 142     emit dataChanged( createIndex( oldRow, 0 ), createIndex( oldRow, 0 ) );
 143     emit dataChanged( createIndex( hoveredRow, 0 ), createIndex( hoveredRow, 0 ) );
 144 }
 145
 146 void ChannelModel::clearHover() {
 147     emit dataChanged( createIndex( hoveredRow, 0 ), createIndex( hoveredRow, 0 ) );
 148     hoveredRow = -1;
 149 }