/*
 * Copyright (C) 2014 Emeric Poupon
 *
 * This file is part of LMS.
 *
 * LMS 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.
 *
 * LMS 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 LMS.  If not, see <http://www.gnu.org/licenses/>.
 */
#include "Playlist.hpp"

#include <cassert>

#include "User.hpp"
#include "Track.hpp"

namespace Database {

Playlist::Playlist()
: _isPublic(false)
{

}

Playlist::Playlist(std::string name, bool isPublic, Wt::Dbo::ptr<User> user)
: _name(name),
 _isPublic(isPublic),
 _user(user)
{

}

Playlist::pointer
Playlist::create(Wt::Dbo::Session& session, std::string name, bool isPublic, Wt::Dbo::ptr<User> user)
{
	return session.add( std::make_unique<Playlist>(name, isPublic, user) );
}

PlaylistEntry::PlaylistEntry()
{
}

PlaylistEntry::pointer
PlaylistEntry::getById(Wt::Dbo::Session& session, PlaylistEntry::id_type id)
{
	return session.find<PlaylistEntry>().where("id = ?").bind(id);
}

Playlist::pointer
Playlist::get(Wt::Dbo::Session& session, std::string name, Wt::Dbo::ptr<User> user)
{
	return session.find<Playlist>().where("name = ? AND user_id = ?").bind(name).bind(user.id());
}

std::vector<Playlist::pointer>
Playlist::getAll(Wt::Dbo::Session& session, Wt::Dbo::ptr<User> user)
{
	Wt::Dbo::collection<Playlist::pointer> res = session.find<Playlist>().where("user_id = ?").bind(user.id()).orderBy("name");

	return std::vector<Playlist::pointer>(res.begin(), res.end());
}

PlaylistEntry::PlaylistEntry(Wt::Dbo::ptr<Track> track, Wt::Dbo::ptr<Playlist> playlist)
: _track(track),
 _playlist(playlist)
{

}

PlaylistEntry::pointer
PlaylistEntry::create(Wt::Dbo::Session& session, Wt::Dbo::ptr<Track> track, Wt::Dbo::ptr<Playlist> playlist)
{
	return session.add( std::make_unique<PlaylistEntry>( track, playlist) );
}


std::vector<Wt::Dbo::ptr<PlaylistEntry>>
Playlist::getEntries(int offset, int size, bool& moreResults) const
{
	assert(session());

	moreResults = false;

	Wt::Dbo::collection<Wt::Dbo::ptr<PlaylistEntry>> entries =
		session()->find<PlaylistEntry>()
		.where("playlist_id = ?").bind(self().id())
		.orderBy("id")
		.limit(size != -1 ? size + 1 : -1)
		.offset(offset);

	std::vector<Wt::Dbo::ptr<PlaylistEntry>> res;

	for (auto entry : entries)
	{
		if (size != -1 && res.size() == static_cast<std::size_t>(size))
		{
			moreResults = true;
			break;
		}

		res.push_back(entry);
	}

	return res;
}

Wt::Dbo::ptr<PlaylistEntry>
Playlist::getEntry(std::size_t pos) const
{
	Wt::Dbo::ptr<PlaylistEntry> res;

	bool moreResults;
	auto entries = getEntries(pos, 1, moreResults);
	if (!entries.empty())
		res = entries.front();

	return res;
}

std::size_t
Playlist::getCount() const
{
	return _entries.size();
}

} // namespace Database