zeus/tools/gateway/appl/GateWay.cpp

/** @file
 * @author Edouard DUPIN
 * @copyright 2016, Edouard DUPIN, all right reserved
 * @license APACHE v2.0 (see license file)
 */

#include <appl/GateWay.hpp>
#include <appl/debug.hpp>
#include <enet/TcpServer.hpp>
#include <appl/DirectInterface.hpp>
#include <appl/RouterInterface.hpp>


namespace appl {
	class TcpServerInput {
		private:
			enet::TcpServer m_interface;
			std::thread* m_thread;
			bool m_threadRunning;
			appl::GateWay* m_gateway;
		public:
			TcpServerInput(appl::GateWay* _gateway) :
			  m_thread(nullptr),
			  m_threadRunning(false),
			  m_gateway(_gateway) {
				
			}
			virtual ~TcpServerInput() {}
			void start(const std::string& _host, uint16_t _port) {
				m_interface.setHostNane(_host);
				m_interface.setPort(_port);
				m_interface.link();
				m_threadRunning = true;
				APPL_INFO("Start waiting on " << _host << " " << _port);
				m_thread = new std::thread([&](void *){ this->threadCallback();}, nullptr);
				if (m_thread == nullptr) {
					m_threadRunning = false;
					APPL_ERROR("creating callback thread!");
					return;
				}
			}
			void stop() {
				if (m_thread != nullptr) {
					m_threadRunning = false;
				}
				m_interface.unlink();
				if (m_thread != nullptr) {
					m_thread->join();
					delete m_thread;
					m_thread = nullptr;
				}
			}
			void threadCallback() {
				// get datas:
				while (m_threadRunning == true) {
					// READ section data:
					enet::Tcp data = std::move(m_interface.waitNext());
					APPL_VERBOSE("New connection");
					m_gateway->newService(std::move(data));
				}
			}
	};
}

void appl::GateWay::newService(enet::Tcp _connection) {
	APPL_WARNING("New TCP connection (service)");
	ememory::SharedPtr<appl::DirectInterface> tmp = ememory::makeShared<appl::DirectInterface>(std::move(_connection));
	tmp->start(this);
	m_listIODirect.push_back(tmp);
}

appl::GateWay::GateWay() :
  m_idIncrement(10),
  propertyUserName(this, "user", "no-name", "User name of the interface"), // must be set befor start ...
  propertyRouterIp(this, "router-ip", "127.0.0.1", "Ip to listen client", &appl::GateWay::onPropertyChangeClientIp),
  propertyRouterPort(this, "router-port", 1984, "Port to listen client", &appl::GateWay::onPropertyChangeClientPort),
  propertyServiceIp(this, "service-ip", "127.0.0.1", "Ip to listen client", &appl::GateWay::onPropertyChangeServiceIp),
  propertyServicePort(this, "service-port", 1985, "Port to listen client", &appl::GateWay::onPropertyChangeServicePort),
  propertyServiceMax(this, "service-max", 80, "Maximum of client at the same time", &appl::GateWay::onPropertyChangeServiceMax) {
	m_interfaceNewService = ememory::makeShared<appl::TcpServerInput>(this);
}

appl::GateWay::~GateWay() {
	
}

void appl::GateWay::addIO(const ememory::SharedPtr<appl::IOInterface>& _io) {
	m_listIO.push_back(_io);
}

void appl::GateWay::removeIO(const ememory::SharedPtr<appl::IOInterface>& _io) {
	auto it = m_listIO.begin();
	while (it != m_listIO.end()) {
		if (it->get() == _io.get()) {
			it = m_listIO.erase(it);
		} else {
			++it;
		}
	}
}

uint16_t appl::GateWay::getId() {
	return m_idIncrement++;
}

void appl::GateWay::start() {
	m_routerClient = ememory::makeShared<appl::RouterInterface>(*propertyRouterIp, *propertyRouterPort, *propertyUserName, this);
	
	m_interfaceNewService->start(*propertyServiceIp, *propertyServicePort);
}

void appl::GateWay::stop() {
	m_routerClient.reset();
}

bool appl::GateWay::serviceExist(const std::string& _service) {
	for (auto &it : m_listIO) {
		if (it == nullptr) {
			continue;
		}
		for (auto &srvIt : it->getServiceList()) {
			if (srvIt == _service) {
				return true;
			}
		}
	}
	return false;
}

uint16_t appl::GateWay::serviceClientIdGet(const std::string& _service) {
	for (auto &it : m_listIO) {
		if (it == nullptr) {
			continue;
		}
		for (auto &srvIt : it->getServiceList()) {
			if (srvIt == _service) {
				return it->getId();
			}
		}
	}
	return 0;
}

/*
ememory::SharedPtr<appl::ServiceInterface> appl::GateWay::get(const std::string& _serviceName) {
	for (auto &it : m_serviceList) {
		if (it == nullptr) {
			continue;
		}
		if (it->getName() != _serviceName) {
			continue;
		}
		return it;
	}
	return nullptr;
}
*/

std::vector<std::string> appl::GateWay::getAllServiceName() {
	std::vector<std::string> out;
	// TODO : Change this it is old and deprecated ...
	for (auto &it : m_listIO) {
		if (it == nullptr) {
			continue;
		}
		for (auto &srvIt : it->getServiceList()) {
			out.push_back(srvIt);
		}
	}
	return out;
}


bool appl::GateWay::send(ememory::SharedPtr<zeus::Buffer> _data) {
	auto it = m_listIO.begin();
	uint16_t id = _data->getDestinationId();
	while (it != m_listIO.end()) {
		if (*it == nullptr) {
			continue;
		}
		if ((*it)->getId() == id) {
			(*it)->send(_data);
			return true;
		}
		++it;
	}
	return false;
}

void appl::GateWay::cleanIO() {
	APPL_VERBOSE("Check if something need to be clean ...");
	/*
	auto it = m_listIODirect.begin();
	while (it != m_listIODirect.end()) {
		if (*it != nullptr) {
			if ((*it)->isAlive() == false) {
				it = m_listIODirect.erase(it);
				continue;
			}
		} else {
			it = m_listIODirect.erase(it);
			continue;
		}
		++it;
	}
	*/
	if (m_routerClient != nullptr) {
		m_routerClient->clean();
	}
}

void appl::GateWay::onClientConnect(const bool& _value) {
	APPL_TODO("Client connection: " << _value);
}

void appl::GateWay::onServiceConnect(const bool& _value) {
	APPL_TODO("Service connection: " << _value);
}

void appl::GateWay::onPropertyChangeClientIp() {
	
}

void appl::GateWay::onPropertyChangeClientPort() {
	
}

void appl::GateWay::onPropertyChangeClientMax() {
	
}

void appl::GateWay::onPropertyChangeServiceIp() {
	
}

void appl::GateWay::onPropertyChangeServicePort() {
	
}

void appl::GateWay::onPropertyChangeServiceMax() {
	
}