#include "i2-icinga.h"

using namespace icinga;

IcingaApplication::Ptr SubscriptionComponent::GetIcingaApplication(void) const
{
	return static_pointer_cast<IcingaApplication>(GetApplication());
}

string SubscriptionComponent::GetName(void) const
{
	return "subscriptioncomponent";
}

void SubscriptionComponent::Start(void)
{
	m_SubscriptionEndpoint = make_shared<VirtualEndpoint>();
	m_SubscriptionEndpoint->RegisterMethodHandler("message::Subscribe", bind_weak(&SubscriptionComponent::SubscribeMessageHandler, shared_from_this()));
	m_SubscriptionEndpoint->RegisterMethodHandler("message::Provide", bind_weak(&SubscriptionComponent::ProvideMessageHandler, shared_from_this()));
	m_SubscriptionEndpoint->RegisterMethodSource("message::Subscribe");
	m_SubscriptionEndpoint->RegisterMethodSource("message::Provide");

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

void SubscriptionComponent::Stop(void)
{
	IcingaApplication::Ptr app = GetIcingaApplication();

	if (app) {
		EndpointManager::Ptr mgr = app->GetEndpointManager();
		mgr->UnregisterEndpoint(m_SubscriptionEndpoint);
	}
}

int SubscriptionComponent::SyncSubscription(Endpoint::Ptr target, string type, const NewMethodEventArgs& nmea)
{
	JsonRpcRequest request;
	request.SetVersion("2.0");
	request.SetMethod(type);

	SubscriptionMessage subscriptionMessage;
	subscriptionMessage.SetMethod(nmea.Method);
	request.SetParams(subscriptionMessage);

	EndpointManager::Ptr endpointManager = GetIcingaApplication()->GetEndpointManager();
	endpointManager->SendUnicastRequest(m_SubscriptionEndpoint, target, request);

	return 0;
}

int SubscriptionComponent::SyncSubscriptions(Endpoint::Ptr target, const NewEndpointEventArgs& neea)
{
	Endpoint::Ptr source = neea.Endpoint;

	if (!source->IsLocal())
		return 0;

	source->ForeachMethodSink(bind(&SubscriptionComponent::SyncSubscription, this, target, "message::Subscribe", _1));
	source->ForeachMethodSource(bind(&SubscriptionComponent::SyncSubscription, this,  target, "message::Provide", _1));

	// TODO: bind to endpoint's events
	//endpoint->OnNewMethodSink...

	return 0;
}

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

	EndpointManager::Ptr mgr = GetIcingaApplication()->GetEndpointManager();
	mgr->ForeachEndpoint(bind(&SubscriptionComponent::SyncSubscriptions, this, neea.Endpoint, _1));

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

	return 0;
}

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

	SubscriptionMessage subscriptionMessage = params;

	string method;
	if (!subscriptionMessage.GetMethod(&method))
		return 0;

	nrea.Sender->RegisterMethodSink(method);
	return 0;
}

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

	SubscriptionMessage subscriptionMessage = params;

	string method;
	if (!subscriptionMessage.GetMethod(&method))
		return 0;

	nrea.Sender->RegisterMethodSource(method);
	return 0;
}