#include "i2-icinga.h"

using namespace icinga;

string AuthenticationComponent::GetName(void) const
{
	return "authenticationcomponent";
}

void AuthenticationComponent::Start(void)
{
	m_AuthenticationEndpoint = make_shared<VirtualEndpoint>();
	m_AuthenticationEndpoint->RegisterMethodHandler("auth::SetIdentity", bind_weak(&AuthenticationComponent::IdentityMessageHandler, shared_from_this()));
	m_AuthenticationEndpoint->RegisterMethodSource("auth::Welcome");

	EndpointManager::Ptr mgr = GetEndpointManager();
	mgr->OnNewEndpoint += bind_weak(&AuthenticationComponent::NewEndpointHandler, shared_from_this());
	mgr->ForeachEndpoint(bind(&AuthenticationComponent::NewEndpointHandler, this, _1));
	mgr->RegisterEndpoint(m_AuthenticationEndpoint);
}

void AuthenticationComponent::Stop(void)
{
	EndpointManager::Ptr mgr = GetEndpointManager();

	if (mgr)
		mgr->UnregisterEndpoint(m_AuthenticationEndpoint);
}

int AuthenticationComponent::NewEndpointHandler(const NewEndpointEventArgs& neea)
{
	if (neea.Endpoint->IsLocal() || neea.Endpoint->HasIdentity())
		return 0;

	neea.Endpoint->AddAllowedMethodSinkPrefix("auth::");
	neea.Endpoint->AddAllowedMethodSourcePrefix("auth::");

	neea.Endpoint->RegisterMethodSink("auth::SetIdentity");

	JsonRpcRequest request;
	request.SetMethod("auth::SetIdentity");

	IdentityMessage params;
	params.SetIdentity("keks");
	request.SetParams(params);

	GetEndpointManager()->SendUnicastRequest(m_AuthenticationEndpoint, neea.Endpoint, request);

	return 0;
}

int AuthenticationComponent::IdentityMessageHandler(const NewRequestEventArgs& nrea)
{
	Message params;
	if (!nrea.Request.GetParams(&params))
		return 0;

	IdentityMessage identityMessage = params;

	string identity;
	if (!identityMessage.GetIdentity(&identity))
		return 0;

	nrea.Sender->SetIdentity(identity);

	/* there's no authentication for now, just tell them it's ok to send messages */
	JsonRpcRequest request;
	request.SetMethod("auth::Welcome");

	GetEndpointManager()->SendUnicastRequest(m_AuthenticationEndpoint, nrea.Sender, request);

	return 0;
}