#include "usersimpl.h"
#include <Ice/ObjectAdapter.h>
#include <Ice/Communicator.h>
#include <modifycommand.h>
#include <sql/users/authenticate.sql.h>
#include <sql/users/verify.sql.h>
#include <sql/users/create.sql.h>
#include <sql/users/get.sql.h>
#include <sql/users/getNew.sql.h>
#include <sql/users/find.sql.h>
#include <sql/users/safeDelete.sql.h>
#include <sql/users/track.sql.h>
#include <sql/users/untrack.sql.h>
#include <sql/users/tracked.sql.h>
#include <sql/users/mailshotsent.sql.h>
#include <notifications.h>

namespace Gentoo {
namespace Service {
Users::Users(boost::shared_ptr<AdHoc::ResourcePool<DB::Connection>> d) :
	IceTray::AbstractCachingDatabaseClient(d)
{
}

Gentoo::UserPtr
Users::authenticate(const std::string & username, const std::string & password, const Ice::Current &)
{
	return fetchCache<Gentoo::UserPtr>(sql::users::authenticate, 30, username, password);
}

Gentoo::UserPtr
Users::verify(const std::string & username, const std::string & verifyguid, const Ice::Current &)
{
	return fetch<Gentoo::UserPtr>(sql::users::verify, username, verifyguid);
}

Gentoo::UserPtr
Users::get(Ice::Int id, const Ice::Current &)
{
	return fetch<Gentoo::UserPtr>(sql::users::get, id);
}

Gentoo::NewUserPtr
Users::getNew(const std::string & username, const std::string & password, const Ice::Current &)
{
	return fetch<Gentoo::NewUserPtr>(sql::users::getNew, username, password);
}

Gentoo::UserPtr
Users::find(const std::string & username, const Ice::Current &)
{
	return fetch<Gentoo::UserPtr>(sql::users::find, username);
}

Gentoo::NewUserPtr
Users::authOrCreate(const std::string & username, const std::string & password, const std::string & realname, const std::string & email)
{
	auto existing = fetch<IceUtil::Optional<Gentoo::NewUserPtr>>(sql::users::getNew, username, password);
	if (existing && *existing) {
		return *existing;
	}
	return fetch<Gentoo::NewUserPtr>(sql::users::create, username, password, realname, email);
}

Gentoo::NewUserPtr
Users::create(const std::string & username, const std::string & password, const std::string & realname, const std::string & email, const Ice::Current & current)
{
	auto notifications = NotificationsPrx::checkedCast(current.adapter->getCommunicator()->stringToProxy("notifications"));
	BOOST_ASSERT(notifications);
	BOOST_ASSERT((notifications->ice_ping(), true));
	auto mailServer = MailServerPrx::checkedCast(current.adapter->getCommunicator()->stringToProxy("mailserver"));
	BOOST_ASSERT(mailServer);
	BOOST_ASSERT((mailServer->ice_ping(), true));

	auto dbc = db->get();
	DB::TransactionScope tx(dbc.get());

	auto newUser = authOrCreate(username, password, realname, email);
	auto mail = notifications->getSignup(newUser);
	mailServer->sendEmail(mail);

	return newUser;
}

void
Users::mailshotsent(Ice::Int id, const Ice::Current &)
{
	auto dbc = db->get();
	auto upd = sql::users::mailshotsent.modify(dbc.get());
	upd->bindParamI(0, id);
	upd->execute();
}

void
Users::remove(Ice::Int id, const std::string & password, const Ice::Current &)
{
	auto dbc = db->get();
	auto del = sql::users::safeDelete.modify(dbc.get());
	del->bindParamI(0, id);
	del->bindParamS(1, password);
	del->execute();
}

void
Users::track(Ice::Int userId, Ice::Int packageId, const Ice::Current &)
{
	auto dbc = db->get();
	auto track = sql::users::track.modify(dbc.get());
	track->bindParamI(0, userId);
	track->bindParamI(1, packageId);
	track->execute();
}

void
Users::untrack(Ice::Int userId, Ice::Int packageId, const Ice::Current &)
{
	auto dbc = db->get();
	auto untrack = sql::users::untrack.modify(dbc.get());
	untrack->bindParamI(0, userId);
	untrack->bindParamI(1, packageId);
	untrack->execute();
}

Gentoo::PackageIds
Users::tracked(Ice::Int userId, const Ice::Current &)
{
	return fetchCache<Gentoo::PackageIds>(sql::users::tracked, 10, userId);
}
}
}