// Library
//
// Douglas Thrift
//
// $Id$

#include <cxx/standard.hh>

#include <dbi/resultset.hpp>

#include "Library.hpp"

Library::Library(const _R<Connector>& connector) : connector(connector)
{
	_R<dbi::Connection> connection(connector->Connect());

	connection->Execute(_B("UPDATE files SET live = FALSE"));

	connector->Release(connection);
}

_L<MediaFolder> Library::GetFolders(unsigned page) const
{
	_R<dbi::Connection> connection(connector->Connect());
	_R<dbi::ResultSet> roots(connection->Parse(_B("SELECT DISTINCT root FROM files WHERE live = TRUE ORDER By root LIMIT 50 OFFSET ?"))->Execute(page * 50));
	_L<MediaFolder> folders;

	while (roots->MoveNext())
		folders.InsertLast(GetFolder(roots->GetString(_B("root"))));

	connector->Release(connection);

	return folders;
}

unsigned Library::GetFoldersPages() const
{
	_R<dbi::Connection> connection(connector->Connect());
	_R<dbi::ResultSet> count(connection->Execute(_B("SELECT count(DISTINCT root) FROM files WHERE live = TRUE")));
	unsigned pages(0);

	if (count->MoveNext())
		pages = (count->Get<unsigned>(_B("count")) + 49) / 50;

	connector->Release(connection);

	return pages;
}

_L<MediaFile> Library::GetFiles(const _L<By>& bys, const _L<cse::String>& items, unsigned page) const
{
	if (bys.IsEmpty() || ext::Contains(bys, By::LAST) || bys.GetSize() != items.GetSize())
		return _L<MediaFile>();

	_S<ios::String> statement_;

	statement_ << _B("SELECT path FROM files WHERE live = TRUE ");

	_foreach (const _L<By>, by, bys)
		statement_ << _B("AND ") << *by << _B(" = ? ");

	statement_ << _B("ORDER BY path LIMIT 50 OFFSET ?");

	_R<dbi::Connection> connection(connector->Connect());
	_R<dbi::Statement> statement(connection->Parse(statement_));

	_foreach (const _L<cse::String>, item, items)
		statement->Set(_index, *item);

	statement->Set(items.GetSize(), page * 50);

	_R<dbi::ResultSet> paths(statement->Execute());
	_L<MediaFile> files;

	while (paths->MoveNext())
		files.InsertLast(GetFile(paths->GetString(_B("path"))));

	connector->Release(connection);

	return files;
}

unsigned Library::GetFilesPages(const _L<By>& bys, const _L<cse::String>& items) const
{
	if (bys.IsEmpty() || ext::Contains(bys, By::LAST) || bys.GetSize() != items.GetSize())
		return 0;

	_S<ios::String> statement_;

	statement_ << _B("SELECT count(path) FROM files WHERE live = TRUE ");

	_foreach (const _L<By>, by, bys)
		statement_ << _B("AND ") << *by << _B(" = ?");

	_R<dbi::Connection> connection(connector->Connect());
	_R<dbi::Statement> statement(connection->Parse(statement_));

	_foreach (const _L<cse::String>, item, items)
		statement->Set(_index, *item);

	_R<dbi::ResultSet> count(statement->Execute());
	unsigned pages(0);

	if (count->MoveNext())
		pages = (count->Get<unsigned>(_B("count")) + 49) / 50;

	connector->Release(connection);

	return pages;
}

_L<cse::String> Library::GetItems(_L<By> bys, _L<cse::String> items, unsigned page) const
{
	if (bys.IsEmpty() || ext::Contains(bys, By::LAST) || bys.GetSize() - 1 != items.GetSize())
		return _L<cse::String>();

	_S<ios::String> statement_;
	By by(bys.Last());

	statement_ << _B("SELECT DISTINCT ") << by << _B(" FROM files WHERE live = TRUE ");

	bys.RemoveLast();

	_foreach (const _L<By>, by_, bys)
		statement_ << _B("AND ") << *by_ << _B(" = ? ");

	statement_ << _B("ORDER BY ") << by << _B(" LIMIT 50 OFFSET ?");

	_R<dbi::Connection> connection(connector->Connect());
	_R<dbi::Statement> statement(connection->Parse(statement_));

	_foreach (const _L<cse::String>, item, items)
		statement->Set(_index, *item);

	statement->Set(items.GetSize(), page * 50);

	_R<dbi::ResultSet> items_(statement->Execute());
	
	items.Clear();

	while (items_->MoveNext())
		items.InsertLast(items_->GetString(lexical_cast<cse::String>(by)));

	connector->Release(connection);

	return items;
}

unsigned Library::GetItemsPages(_L<By> bys, const _L<cse::String>& items) const
{
	if (bys.IsEmpty() || ext::Contains(bys, By::LAST) || bys.GetSize() - 1 != items.GetSize())
		return 0;

	_S<ios::String> statement_;
	By by(bys.Last());

	statement_ << _B("SELECT count(DISTINCT ") << by << _B(") FROM files WHERE live = TRUE");

	bys.RemoveLast();

	_foreach (const _L<By>, by_, bys)
		statement_ << _B(" AND ") << *by_ << _B(" = ?");

	_R<dbi::Connection> connection(connector->Connect());
	_R<dbi::Statement> statement(connection->Parse(statement_));

	_foreach (const _L<cse::String>, item, items)
		statement->Set(_index, *item);

	_R<dbi::ResultSet> count(statement->Execute());
	unsigned pages(0);

	if (count->MoveNext())
		pages = (count->Get<unsigned>(_B("count")) + 49) / 50;

	connector->Release(connection);

	return pages;
}