/* This file is part of Clementine.
   Copyright 2010, 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 "smartplaylists/generatorinserter.h"

#include <QtConcurrentRun>

#include "core/closure.h"
#include "core/taskmanager.h"
#include "playlist/playlist.h"
#include "smartplaylists/generator.h"

namespace smart_playlists {

GeneratorInserter::GeneratorInserter(TaskManager* task_manager,
                                     LibraryBackend* library, QObject* parent)
    : QObject(parent),
      task_manager_(task_manager),
      library_(library),
      task_id_(-1),
      is_dynamic_(false) {}

static PlaylistItemList Generate(GeneratorPtr generator, int dynamic_count) {
  if (dynamic_count) {
    return generator->GenerateMore(dynamic_count);
  } else {
    return generator->Generate();
  }
}

void GeneratorInserter::Load(Playlist* destination, int row, bool play_now,
                             bool enqueue, bool enqueue_next,
                             GeneratorPtr generator, int dynamic_count) {
  task_id_ = task_manager_->StartTask(tr("Loading smart playlist"));

  destination_ = destination;
  row_ = row;
  play_now_ = play_now;
  enqueue_ = enqueue;
  enqueue_next_ = enqueue_next;
  is_dynamic_ = generator->is_dynamic();

  connect(generator.get(), SIGNAL(Error(QString)), SIGNAL(Error(QString)));

  QFuture<PlaylistItemList> future =
      QtConcurrent::run(Generate, generator, dynamic_count);
  NewClosure(future, this, SLOT(Finished(QFuture<PlaylistItemList>)), future);
}

void GeneratorInserter::Finished(QFuture<PlaylistItemList> future) {
  PlaylistItemList items = future.result();

  if (items.isEmpty()) {
    if (is_dynamic_) {
      destination_->TurnOffDynamicPlaylist();
    }
  } else {
    destination_->InsertItems(items, row_, play_now_, enqueue_);
  }

  task_manager_->SetTaskFinished(task_id_);

  deleteLater();
}

}  // namespace