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

#include <menes/standard.hh>

#include <menes/dbi/resultset.hpp>

#include "Library.hpp"

Library::Library(const _R<dbi::Connection>& connection) : connection(connection)
{
	connection->Execute(_B("UPDATE files SET live = FALSE"));
}

_L<MediaFolder> Library::GetFolders() const
{
	_R<dbi::ResultSet> roots(connection->Execute(_B("SELECT DISTINCT root FROM files WHERE live = TRUE ORDER By root")));
	_L<MediaFolder> folders;

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

	return folders;
}

_L<MediaFile> Library::GetFiles(const _L<By>& bys, const _L<cse::String>& items) 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(" = ? ");

	_rforeach (const _L<By>, by, bys)
		statement_ << _B(by + 1 == _set.End() ? "ORDER BY " : ", ") << *by;

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

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

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

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

	return files;
}

_L<cse::String> Library::GetItems(_L<By> bys, _L<cse::String> items) 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;

	_rforeach (const _L<By>, by_, bys)
		statement_ << _B(", ") << *by_;

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

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

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

	while (items_->MoveNext())
		items.InsertLast(items_->GetString(by));

	return items;
}