Clementine-audio-player-Mac-Linux-Windows/src/podcasts/podcastservicemodel.cpp

/* This file is part of Clementine.
   Copyright 2012, David Sansome <me@davidsansome.com>
   
   Clementine is free software: you can redistribute it and/or modify
   it under the terms of the GNU General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.
   
   Clementine is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU General Public License for more details.
   
   You should have received a copy of the GNU General Public License
   along with Clementine.  If not, see <http://www.gnu.org/licenses/>.
*/

#include "podcastservice.h"
#include "podcastservicemodel.h"
#include "playlist/songmimedata.h"

PodcastServiceModel::PodcastServiceModel(QObject* parent)
  : QStandardItemModel(parent)
{
}

QMimeData* PodcastServiceModel::mimeData(const QModelIndexList& indexes) const {
  SongMimeData* data = new SongMimeData;
  QList<QUrl> urls;

  foreach (const QModelIndex& index, indexes) {
    switch (index.data(InternetModel::Role_Type).toInt()) {
    case PodcastService::Type_Episode:
      MimeDataForEpisode(index, data, &urls);
      break;

    case PodcastService::Type_Podcast:
      MimeDataForPodcast(index, data, &urls);
      break;
    }
  }

  data->setUrls(urls);
  return data;
}

void PodcastServiceModel::MimeDataForEpisode(const QModelIndex& index,
                                             SongMimeData* data, QList<QUrl>* urls) const {
  QVariant episode_variant = index.data(PodcastService::Role_Episode);
  if (!episode_variant.isValid())
    return;

  PodcastEpisode episode(episode_variant.value<PodcastEpisode>());

  // Get the podcast from the index's parent
  Podcast podcast;
  QVariant podcast_variant = index.parent().data(PodcastService::Role_Podcast);
  if (podcast_variant.isValid()) {
    podcast = podcast_variant.value<Podcast>();
  }

  Song song = episode.ToSong(podcast);

  data->songs << song;
  *urls << song.url();
}

void PodcastServiceModel::MimeDataForPodcast(const QModelIndex& index,
                                             SongMimeData* data, QList<QUrl>* urls) const {
  // Get the podcast
  Podcast podcast;
  QVariant podcast_variant = index.data(PodcastService::Role_Podcast);
  if (podcast_variant.isValid()) {
    podcast = podcast_variant.value<Podcast>();
  }

  // Add each child episode
  const int children = index.model()->rowCount(index);
  for (int i=0 ; i<children ; ++i) {
    QVariant episode_variant = index.child(i, 0).data(PodcastService::Role_Episode);
    if (!episode_variant.isValid())
      continue;

    PodcastEpisode episode(episode_variant.value<PodcastEpisode>());
    Song song = episode.ToSong(podcast);

    data->songs << song;
    *urls << song.url();
  }
}