zeus/tools/router/appl/Router.cpp

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

#include <appl/Router.hpp>
#include <appl/debug.hpp>
#include <enet/TcpServer.hpp>
#include <etk/os/FSNode.hpp>

static std::string g_pathDBName = "USERDATA:router-database.json";

class UserAvaillable {
	public:
		std::string m_name;
		std::string m_basePath;
		bool m_accessMediaCenter;
		FILE* m_subProcess;
};
std::vector<UserAvaillable> g_listUserAvaillable;
bool g_needToStore = false;

static void store_db() {
	if (g_needToStore == false) {
		return;
	}
	APPL_ERROR("Store database [START]");
	ejson::Document database;
	ejson::Array listUserArray;
	database.add("users", listUserArray);
	for (auto &it : g_listUserAvaillable) {
		ejson::Object propObject;
		listUserArray.add(propObject);
		propObject.add("name", ejson::String(it.m_name));
		propObject.add("path", ejson::String(it.m_basePath));
		propObject.add("access-media-center", ejson::Boolean(it.m_accessMediaCenter));
	}
	bool retGenerate = database.storeSafe(g_pathDBName);
	APPL_ERROR("Store database [STOP] : " << g_pathDBName << " ret = " << retGenerate);
	g_needToStore = false;
}

static void load_db() {
	ejson::Document database;
	bool ret = database.load(g_pathDBName);
	if (ret == false) {
		APPL_WARNING("    ==> LOAD error");
	}
	g_listUserAvaillable.clear();
	ejson::Array listUserArray = database["users"].toArray();
	for (const auto itArray: listUserArray) {
		ejson::Object userElement = itArray.toObject();
		UserAvaillable userProperty;
		userProperty.m_name = userElement["name"].toString().get();
		userProperty.m_basePath = userElement["path"].toString().get();
		userProperty.m_accessMediaCenter = userElement["access-media-center"].toBoolean().get();
		APPL_INFO("find USER: '" << userProperty.m_name << "'");
		g_listUserAvaillable.push_back(userProperty);
	}
	g_needToStore = false;
}


namespace appl {
	class TcpServerInput {
		private:
			enet::TcpServer m_interface;
			std::thread* m_thread;
			bool m_threadRunning;
			appl::Router* m_router;
			bool m_service;
		public:
			TcpServerInput(appl::Router* _router, bool _service) :
			  m_thread(nullptr),
			  m_threadRunning(false),
			  m_router(_router),
			  m_service(_service) {
				
			}
			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;
				m_thread = new std::thread([&](void *){ this->threadCallback();}, nullptr);
				if (m_thread == nullptr) {
					m_threadRunning = false;
					ZEUS_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());
					ZEUS_VERBOSE("New connection");
					if (m_service == true) {
						m_router->newClientGateWay(std::move(data));
					} else {
						m_router->newClient(std::move(data));
					}
				}
			}
	};
}

void appl::Router::newClientGateWay(enet::Tcp _connection) {
	ZEUS_WARNING("New TCP connection (service)");
	ememory::SharedPtr<appl::GateWayInterface> tmp = ememory::makeShared<appl::GateWayInterface>(std::move(_connection), this);
	tmp->start();
	m_GateWayList.push_back(tmp);
}

void appl::Router::newClient(enet::Tcp _connection) {
	ZEUS_WARNING("New TCP connection (client)");
	ememory::SharedPtr<appl::ClientInterface> tmp = ememory::makeShared<appl::ClientInterface>(std::move(_connection), this);
	tmp->start();
	m_clientList.push_back(tmp);
}

appl::Router::Router() :
  m_clientUID(2),
  propertyStdOut(this, "stdout", false, "Set the Log of the output"),
  propertyClientIp(this, "client-ip", "127.0.0.1", "Ip to listen client", &appl::Router::onPropertyChangeClientIp),
  propertyClientPort(this, "client-port", 1983, "Port to listen client", &appl::Router::onPropertyChangeClientPort),
  propertyClientMax(this, "client-max", 8000, "Maximum of client at the same time", &appl::Router::onPropertyChangeClientMax),
  propertyGateWayIp(this, "gw-ip", "127.0.0.1", "Ip to listen Gateway", &appl::Router::onPropertyChangeGateWayIp),
  propertyGateWayPort(this, "gw-port", 1984, "Port to listen Gateway", &appl::Router::onPropertyChangeGateWayPort),
  propertyGateWayMax(this, "gw-max", 8000, "Maximum of Gateway at the same time", &appl::Router::onPropertyChangeGateWayMax),
  propertyDelayToStop(this, "delay-to-stop", 0, "Delay before the client stop the connection in second (default: 0=automatic set by the gateway; -1=never disconnect; other the time )") {
	m_interfaceClientServer = ememory::makeShared<appl::TcpServerInput>(this, false);
	m_interfaceGateWayServer = ememory::makeShared<appl::TcpServerInput>(this, true);
	load_db();
}

appl::Router::~Router() {
	
}

void appl::Router::start() {
	m_interfaceClientServer->start(*propertyClientIp, *propertyClientPort);
	m_interfaceGateWayServer->start(*propertyGateWayIp, *propertyGateWayPort);
}

void appl::Router::stop() {
	// TODO : Stop all server ...
	
}

bool appl::Router::userIsConnected(const std::string& _userName) {
	for (auto &it : m_GateWayList) {
		if (it == nullptr) {
			continue;
		}
		if (it->getName() != _userName) {
			continue;
		}
		return true;
	}
	return false;
}

#include <iomanip>
#include <iostream>
#include <cstdio>
#include <unistd.h>


ememory::SharedPtr<appl::GateWayInterface> appl::Router::get(const std::string& _userName) {
	// TODO : Start USer only when needed, not get it all time started...
	for (auto &it : m_GateWayList) {
		if (it == nullptr) {
			continue;
		}
		if (it->getName() != _userName) {
			continue;
		}
		return it;
	}
	// we not find the user ==> check if it is availlable ...
	for (auto &it : g_listUserAvaillable) {
		if (it.m_name == _userName) {
			#if 0
				// start interface:
				std::string cmd = "~/dev/perso/out/Linux_x86_64/debug/staging/clang/zeus-package-base/zeus-package-base.app/bin/zeus-gateway";
				cmd += " --user=" + it.m_name + " ";
				cmd += " --srv=user";
				cmd += " --srv=picture";
				cmd += " --srv=video";
				cmd += " --base-path=" + it.m_basePath;
				cmd += " --elog-file=\"/tmp/zeus.gateway." + it.m_name + ".log\"";
				cmd += "&";
				APPL_ERROR("Start " << cmd);
				it.m_subProcess = popen(cmd.c_str(), "r");
				if (it.m_subProcess == nullptr) {
					perror("popen");
					return nullptr;
				}
				// just trash IO ...
				//pclose(it.m_subProcess);
			#else
				if (fork()) {
					// We're in the parent here.
					// nothing to do ...
					APPL_ERROR("Parent Execution ...");
				} else {
					// We're in the child here.
					APPL_ERROR("Child Execution ...");
					std::string binary = etk::FSNodeGetApplicationPath() + "/zeus-gateway";
					std::string userConf = "--user=" + it.m_name;
					std::string basePath = "--base-path=" + it.m_basePath;
					std::string logFile;
					if (*propertyStdOut == false) {
						logFile = it.m_basePath + "/log/gateway.log";
						if (    logFile.size() != 0
						     && logFile[0] == '~') {
							logFile = etk::FSNodeGetHomePath() + &logFile[1];
						}
						logFile = "--elog-file=" + logFile;
						//std::string logFile = "--elog-file=/home/heero/.local/share/zeus-DATA/SDFGHTHBSDFGSQDHZSRDFGSDFGSDFGSDFG/log/gateway.log";
						//std::string logFile = " ";
						APPL_INFO("New Child log in = " << logFile);
					}
					std::string delay = "--router-delay=" + etk::to_string(*propertyDelayToStop);
					APPL_INFO("execute: " << binary << " " << userConf << " --srv=all " << delay << " " << basePath << " " << logFile);
					int ret = execlp( binary.c_str(),
					                  binary.c_str(), // must repeate the binary name to have the name as first argument ...
					                  userConf.c_str(),
					                  "--srv=all",
					                  "--service-extern=false",
					                  delay.c_str(),
					                  basePath.c_str(),
					                  logFile.c_str(),
					                  NULL);
					APPL_ERROR("Child Execution ret = " << ret);
					exit (-1);
					APPL_ERROR("Must never appear ... child of fork killed ...");
				}
			#endif
			int32_t nbCheckDelayMax = 24;
			while (nbCheckDelayMax-- > 0) {
				std::this_thread::sleep_for(std::chrono::milliseconds(25));
				for (auto &it : m_GateWayList) {
					if (it == nullptr) {
						continue;
					}
					if (it->getName() != _userName) {
						continue;
					}
					return it;
				}
			}
			APPL_ERROR("must be connected ==> and it is not ...");
			break;
		}
	}
	return nullptr;
}


std::vector<std::string> appl::Router::getAllUserName() {
	std::vector<std::string> out;
	/*
	for (auto &it : m_GateWayList) {
		if (it == nullptr) {
			continue;
		}
		out.push_back(it->getName());
	}
	*/
	return out;
}



void appl::Router::cleanIO() {
	store_db();
	auto it = m_GateWayList.begin();
	while (it != m_GateWayList.end()) {
		if (*it != nullptr) {
			if ((*it)->isAlive() == false) {
				it = m_GateWayList.erase(it);
				continue;
			}
		} else {
			it = m_GateWayList.erase(it);
			continue;
		}
		++it;
	}
	auto it2 = m_clientList.begin();
	while (it2 != m_clientList.end()) {
		if (*it2 != nullptr) {
			if ((*it2)->isAlive() == false) {
				(*it2)->stop();
				it2 = m_clientList.erase(it2);
				APPL_INFO("remove DONE ... ");
				continue;
			}
		} else {
			it2 = m_clientList.erase(it2);
			continue;
		}
		++it2;
	}
}

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

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

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

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

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

void appl::Router::onPropertyChangeGateWayIp() {
	
}

void appl::Router::onPropertyChangeGateWayPort() {
	
}

void appl::Router::onPropertyChangeGateWayMax() {
	
}