#ifndef FEATURE_CONTROLLER_H
#define FEATURE_CONTROLLER_H

#include <memory>

#include "component/controller/helper/ActivationTranslator.h"
#include "component/controller/Controller.h"

#include "utility/messaging/MessageListener.h"
#include "utility/messaging/type/MessageActivateEdge.h"
#include "utility/messaging/type/MessageActivateFile.h"
#include "utility/messaging/type/MessageActivateNodes.h"
#include "utility/messaging/type/MessageActivateTokenIds.h"
#include "utility/messaging/type/MessageActivateTokenLocations.h"
#include "utility/messaging/type/MessageResetZoom.h"
#include "utility/messaging/type/MessageSearch.h"
#include "utility/messaging/type/MessageSwitchColorScheme.h"
#include "utility/messaging/type/MessageZoom.h"

class StorageAccess;

class FeatureController
	: public Controller
	, public MessageListener<MessageActivateEdge>
	, public MessageListener<MessageActivateFile>
	, public MessageListener<MessageActivateNodes>
	, public MessageListener<MessageActivateTokenIds>
	, public MessageListener<MessageActivateTokenLocations>
	, public MessageListener<MessageResetZoom>
	, public MessageListener<MessageSearch>
	, public MessageListener<MessageSwitchColorScheme>
	, public MessageListener<MessageZoom>
{
public:
	FeatureController(StorageAccess* storageAccess);
	~FeatureController();

private:
	virtual void handleMessage(MessageActivateEdge* message);
	virtual void handleMessage(MessageActivateFile* message);
	virtual void handleMessage(MessageActivateNodes* message);
	virtual void handleMessage(MessageSearch* message);
	virtual void handleMessage(MessageActivateTokenIds* message);
	virtual void handleMessage(MessageActivateTokenLocations* message);
	virtual void handleMessage(MessageResetZoom* message);
	virtual void handleMessage(MessageSwitchColorScheme* message);
	virtual void handleMessage(MessageZoom* message);

	StorageAccess* m_storageAccess;
	ActivationTranslator m_activationTranslator;
};

#endif // FEATURE_CONTROLLER_H