/*
 * Copyright (C) 2015 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/>.
 */

#pragma once

#include <string>
#include <vector>

#include <boost/optional.hpp>

#include <Wt/WDateTime.h>
#include <Wt/Dbo/Dbo.h>

#include "TrackArtistLink.hpp"
#include "Types.hpp"

namespace Database
{

class Cluster;
class ClusterType;
class Release;
class Track;
class User;

class Artist : public Wt::Dbo::Dbo<Artist>
{
	public:

		using pointer = Wt::Dbo::ptr<Artist>;

		Artist() {}
		Artist(const std::string& name, const std::string& MBID = "");

		// Accessors
		static pointer			getByMBID(Wt::Dbo::Session& session, const std::string& MBID);
		static pointer			getById(Wt::Dbo::Session& session, IdType id);
		static std::vector<pointer>	getByName(Wt::Dbo::Session& session, const std::string& name);
		static std::vector<pointer> 	getByFilter(Wt::Dbo::Session& session,
								const std::set<IdType>& clusters);		// at least one track that belongs to  these clusters
		static std::vector<pointer> 	getByFilter(Wt::Dbo::Session& session,
								const std::set<IdType>& clusters,		// at least one track that belongs to  these clusters
								const std::vector<std::string>& keywords,	// name must match all of these keywords
								boost::optional<std::size_t> offset,
								boost::optional<std::size_t> size,
								bool& moreExpected);

		static std::vector<pointer>	getAll(Wt::Dbo::Session& session, boost::optional<std::size_t> offset = {}, boost::optional<std::size_t> size = {});
		static std::vector<pointer>	getAllOrphans(Wt::Dbo::Session& session); // No track related
		static std::vector<pointer>	getLastAdded(Wt::Dbo::Session& session, Wt::WDateTime after, boost::optional<std::size_t> size = {});

		// Accessors
		const std::string& getName(void) const { return _name; }
		const std::string& getMBID(void) const { return _MBID; }

		std::vector<Wt::Dbo::ptr<Release>>	getReleases(const std::set<IdType>& clusterIds = std::set<IdType>()) const;
		std::vector<Wt::Dbo::ptr<Track>>	getTracks(boost::optional<TrackArtistLink::Type> linkType = {}) const;
		std::vector<Wt::Dbo::ptr<Track>>	getTracksWithRelease(boost::optional<TrackArtistLink::Type> linkType = {}) const;
		std::vector<Wt::Dbo::ptr<Track>>	getRandomTracks(boost::optional<std::size_t> count) const;

		// Get the cluster of the tracks made by this artist
		// Each clusters are grouped by cluster type, sorted by the number of occurence
		// size is the max number of cluster per cluster type
		std::vector<std::vector<Wt::Dbo::ptr<Cluster>>> getClusterGroups(std::vector<Wt::Dbo::ptr<ClusterType>> clusterTypes, std::size_t size) const;

		void setMBID(const std::string& mbid) { _MBID = mbid; }
		void setSortName(const std::string& sortName);

		// Create
		static pointer create(Wt::Dbo::Session& session, const std::string& name, const std::string& MBID = "");


		template<class Action>
			void persist(Action& a)
			{
				Wt::Dbo::field(a, _name, "name");
				Wt::Dbo::field(a, _name, "sort_name");
				Wt::Dbo::field(a, _MBID, "mbid");

				Wt::Dbo::hasMany(a, _trackArtistLinks, Wt::Dbo::ManyToOne, "artist");
				Wt::Dbo::hasMany(a, _starringUsers, Wt::Dbo::ManyToMany, "user_release_starred", "", Wt::Dbo::OnDeleteCascade);
			}

	private:

		static const std::size_t _maxNameLength = 128;

		std::string _name;
		std::string _sortName;
		std::string _MBID;	// Musicbrainz Identifier

		Wt::Dbo::collection<Wt::Dbo::ptr<TrackArtistLink>> _trackArtistLinks; // Tracks involving this artist
		Wt::Dbo::collection<Wt::Dbo::ptr<User>>		_starringUsers; // Users that starred this artist
};

} // namespace Database