X-Git-Url: https://git.llucax.com/z.facultad/75.42/plaqui.git/blobdiff_plain/a0481d50f6da9cac5efd3502c3657b3fc461ec0d..9317ed8f3f8241a30fd09b6800ff96b2ac138747:/Server/src/server.cpp

diff --git a/Server/src/server.cpp b/Server/src/server.cpp
index d4dd52a..fac8419 100644
--- a/Server/src/server.cpp
+++ b/Server/src/server.cpp
@@ -29,8 +29,9 @@
 #include "plaqui/server/connection.h"
 #include "plaqui/server/controlserver.h"
 #include <sigc++/class_slot.h>
-// FIXME - sacar sstream (a menos que se necesite)
+#include <glibmm/timer.h>
 #include <sstream>
+#include <exception>
 #ifdef DEBUG
 #	include "plaqui/server/string.h"
 #	include <iostream>
@@ -44,58 +45,70 @@ namespace Server {
 
 Server::~Server(void) {
 #ifdef DEBUG
-	cerr << __FILE__ <<  ": destructor." << endl;
+	cerr << __FILE__ << "(" << __LINE__ << ")"
+		<<  ": destructor." << endl;
 #endif // DEBUG
+	// Mando a terminar todas las plantas.
+	plants_mutex.lock();
+	for (PlantList::iterator i = plants.begin(); i != plants.end(); i++) {
+		i->second->finish();
+	}
+	PlantList::size_type count = plants.size();
+	plants_mutex.unlock();
+	// Espero que terminen realmente.
+	while (count) {
+		Glib::usleep(10000); // 10 milisegundos
+		plants_mutex.lock();
+		count = plants.size();
+		plants_mutex.unlock();
+	}
 }
 
-Server::Server(int port):
-		TCPServer(port) {
+Server::Server(const string& plant_filename, const Connection::Port& port)
+		throw(sockerr): TCPServer(port) {
 #ifdef DEBUG
-	cerr << __FILE__ <<  ": port = " << port << endl;
+	cerr << __FILE__ << "(" << __LINE__ << ")"
+		<<  ": port = " << port << endl;
 #endif // DEBUG
+	// FIXME - hacer que se puedan cargar mas plantas bien.
+	Glib::Mutex::Lock lock(plants_mutex);
+	plants["default"] = new Plant(plant_filename);
+	plants["default"]->signal_finished().connect(SigC::bind(
+			SigC::slot_class(*this, &Server::on_plant_finished),
+			"default"));
+	plants["default"]->run();
 }
 
-/// \todo Implementar.
-bool Server::start_transmission(string host, int port) {
-#ifdef DEBUG
-	cerr << __FILE__ <<  ": start_transmission(host = " << host
-		<< " | port = " << port << ")" << endl;
-#endif // DEBUG
-	// TODO
-	return false;
-}
-			
-Connection* Server::new_connection(
-		const sockbuf::sockdesc& sd) {
+Connection* Server::new_connection(const sockbuf::sockdesc& sd) {
 #ifdef DEBUG
-	cerr << __FILE__ <<  ": new_connection(sd = " << sd.sock << ")"
+	cerr << __FILE__ << "(" << __LINE__ << ")"
+		<<  ": new_connection(sd = " << sd.sock << ")"
 		<< endl;
 #endif // DEBUG
 	ControlServer* connection = new ControlServer(sd);
 	// TODO verificar si el new se hace bien? no creo.
-	connection->signal_command_received().connect(
-			SigC::bind<ControlServer*>(
-				SigC::slot_class(*this, &Server::on_control_command_received),
-				connection));
+	connection->signal_command_received().connect(SigC::bind(
+			SigC::slot_class(*this, &Server::on_control_command_received),
+			connection));
 	// TODO: 
 	return connection;
 }
 
-/// \todo Implementar.
-bool Server::stop_transmission(string host, int port) {
+void Server::on_plant_finished(const char* plant) {
 #ifdef DEBUG
-	cerr << __FILE__ <<  ": stop_transmission(host = " << host
-		<< " | port = " << port << ")" << endl;
+	cerr << __FILE__ << "(" << __LINE__ << ")"
+		<<  ": on_plant_finished(plant_name = " << plant << endl;
 #endif // DEBUG
-	// TODO
-	return false;
+	Glib::Mutex::Lock lock(plants_mutex);
+	plants.erase(plant);
 }
 
-/// \todo Implementar.
+/// \todo Terminar de implementar.
 void Server::on_control_command_received(const Command& command,
 		ControlServer* controlserver) {
 #ifdef DEBUG
-	cerr << __FILE__ <<  ": on_control_command_received(target = "
+	cerr << __FILE__ << "(" << __LINE__ << ")"
+		<<  ": on_control_command_received(target = "
 		<< command.get_target() << ", command = " << command.get_command()
 		<< ", args = [" << String::join(command.get_args(), ", ") << "])"
 		<< endl;
@@ -106,12 +119,18 @@ void Server::on_control_command_received(const Command& command,
 		if (command.get_command() == "status") {
 			response = cmd_server_status();
 		} else if (command.get_command() == "stop") {
-			finish();
 			response = new HTTPResponse(HTTPMessage::OK,
-					"El server se apagará en instantes...");
+					"<response desc=\"El server se apagará en instantes...\" />");
+			// XXX - Sin mandar la respuesta enseguida podría ser que el server
+			// cierre la conexión antes de mandar la respuesta.
+			//response->headers["Content-Type"] = "text/xml; charset=iso-8859-1";
+			//controlserver->send(*response);
+			//delete response;
+			finish();
+			//return;
 		} else {
 			response = new HTTPResponse(HTTPMessage::NOT_FOUND,
-					"Invalid command for 'server' taget!");
+					"<response desc=\"Invalid command for 'server' taget!\" />");
 		}
 	} else if (command.get_target() == "connection") {
 		if (command.get_command() == "list") {
@@ -120,19 +139,42 @@ void Server::on_control_command_received(const Command& command,
 			response = cmd_connection_stop(command);
 		} else {
 			response = new HTTPResponse(HTTPMessage::NOT_FOUND,
-					"Invalid command for 'connection' taget!");
+					"<response desc=\"Invalid command for 'connection' taget!\" />");
 		}
 	} else if (command.get_target() == "transmission") {
-		response = new HTTPResponse(HTTPMessage::NOT_FOUND,
-				"Invalid command for 'transmission' taget!");
+		if (command.get_command() == "list") {
+			response = cmd_transmission_list();
+		} else if (command.get_command() == "start") {
+			response = cmd_transmission_start(command);
+		} else if (command.get_command() == "stop") {
+			response = cmd_transmission_stop(command);
+		} else {
+			response = new HTTPResponse(HTTPMessage::NOT_FOUND,
+					"<response desc=\"Invalid command for 'transmission' taget!\" />");
+		}
 	} else if (command.get_target() == "plant") {
-		response = new HTTPResponse(HTTPMessage::NOT_FOUND,
-				"Invalid command for 'plant' taget!");
+		if (command.get_command() == "list") {
+			response = cmd_plant_list();
+		} else if (command.get_command() == "get") {
+			response = cmd_plant_get(command);
+		} else if (command.get_command() == "set") {
+			response = cmd_plant_set(command);
+		} else if (command.get_command() == "set_frequency") {
+			response = cmd_plant_set_frequency(command);
+		} else if (command.get_command() == "start") {
+			response = cmd_plant_start(command);
+		} else if (command.get_command() == "stop") {
+			response = cmd_plant_stop(command);
+		} else {
+			response = new HTTPResponse(HTTPMessage::NOT_FOUND,
+					"<response desc=\"Invalid command for 'plant' taget!\" />");
+		}
 	} else {
-		response = new HTTPResponse(HTTPMessage::NOT_FOUND, "Invalid taget!");
+		response = new HTTPResponse(HTTPMessage::NOT_FOUND,
+				"<response desc=\"Invalid taget!\" />");
 	}
 	// FIXME
-	response->headers["Content-Type"] = "text/html; charset=iso-8859-1";
+	response->headers["Content-Type"] = "text/xml; charset=iso-8859-1";
 	//response->headers["Connection"] = "close";
 	controlserver->send(*response);
 	delete response;
@@ -145,69 +187,32 @@ void Server::on_control_command_received(const Command& command,
 
 HTTPResponse* Server::cmd_server_status(void) const {
 	// FIXME
-	stringstream response_xml;
-	response_xml << "<html>" << endl;
-	response_xml << "    <head>" << endl;
-	response_xml << "        <title>PlaQui v0.7</title>" << endl;
-	response_xml << "    </head>" << endl;
-	response_xml << "    <body>" << endl;
-	response_xml << "        <h1>PlaQui</h1>" << endl;
-	response_xml << "        <p>versión 0.7</p>" << endl;
-/*	response_xml << "        <h2>Comando</h2>" << endl;
-	response_xml << "        <ul>" << endl;
-	response_xml << "           <li><b>Target:</b> " << command.get_target() << endl;
-	response_xml << "           <li><b>Command:</b> " << command.get_command() << endl;
-	response_xml << "           <li><b>Argumentos:</b>" << endl;
-	response_xml << "               <ol>" << endl;
-	for (Command::Arguments::const_iterator i = command.get_args().begin();
-			i != command.get_args().end(); i++) {
-		response_xml << "                   <li>" << *i << "</li>" << endl;
-	}
-	response_xml << "               </ol>" << endl;
-	response_xml << "        </ul>" << endl;
-*/	response_xml << "        <h2>Desarrollado por</h2>" << endl;
-	response_xml << "        <ul>" << endl;
-	response_xml << "            <li>Nicolás Dimov.</li>" << endl;
-	response_xml << "            <li>Leandro Lucarella.</li>" << endl;
-	response_xml << "            <li>Ricardo Markiewicz.</li>" << endl;
-	response_xml << "        </ul>" << endl;
-	response_xml << "        <address>" << endl;
-	response_xml << "             Copyleft 2003 - bajo los " << endl;
-	response_xml << "             términos de la licencia GPL" << endl;
-	response_xml << "        </address>" << endl;
-	response_xml << "    </body>" << endl;
-	response_xml << "</html>" << endl;
-	return new HTTPResponse(HTTPMessage::OK, response_xml.str());
+	stringstream xml;
+	xml << "<serverstatus>" << endl;
+	xml << "\t<version>" VERSION "</version>" << endl;
+	xml << "\t<authors>" << endl;
+	xml << "\t\t<author>Nicolás Dimov</author>" << endl;
+	xml << "\t\t<author>Leandro Lucarella</author>" << endl;
+	xml << "\t\t<author>Ricardo Markiewicz</author>" << endl;
+	xml << "\t</authors>" << endl;
+	xml << "</serverstatus>" << endl;
+	return new HTTPResponse(HTTPMessage::OK, xml.str());
 }
 
 HTTPResponse* Server::cmd_connection_list(void) {
 	// FIXME
 	TCPServer::ConnectionInfoList cil = get_connected();
-	stringstream response_xml;
-	response_xml << "<html>" << endl;
-	response_xml << "    <head>" << endl;
-	response_xml << "        <title>PlaQui v0.7</title>" << endl;
-	response_xml << "    </head>" << endl;
-	response_xml << "    <body>" << endl;
-	response_xml << "        <h1>PlaQui</h1>" << endl;
-	response_xml << "        <p>versión 0.7</p>" << endl;
-	response_xml << "        <h2>Lista de conexiones:</h2>" << endl;
-	response_xml << "        <ul>" << endl;
+	stringstream xml;
+	xml << "<list type=\"connection\">" << endl;
 	for (TCPServer::ConnectionInfoList::const_iterator i = cil.begin();
 			i != cil.end(); i++) {
-		response_xml << "       <li>" << i->host
-			<< ":" << i->port << " [<a href=\"/connection/stop/"
-			<< i->host << "/" << i->port << "\">deconectar</a>]</li>"
-			<< endl;
+		xml << "\t<row>" << endl;
+		xml << "\t\t<cell>" << i->host << "</cell>" << endl;
+		xml << "\t\t<cell>" << i->port << "</cell>" << endl;
+		xml << "\t</row>" << endl;
 	}
-	response_xml << "        </ul>" << endl;
-	response_xml << "        <address>" << endl;
-	response_xml << "             Copyleft 2003 - bajo los " << endl;
-	response_xml << "             términos de la licencia GPL" << endl;
-	response_xml << "        </address>" << endl;
-	response_xml << "    </body>" << endl;
-	response_xml << "</html>" << endl;
-	return new HTTPResponse(HTTPMessage::OK, response_xml.str());
+	xml << "</list>" << endl;
+	return new HTTPResponse(HTTPMessage::OK, xml.str());
 }
 
 HTTPResponse* Server::cmd_connection_stop(const Command& command) {
@@ -215,16 +220,222 @@ HTTPResponse* Server::cmd_connection_stop(const Command& command) {
 	Connection::Port port;
 	if (args.size() < 2) {
 		return new HTTPResponse(HTTPMessage::CONFLICT,
-				"Faltan argumentos.");
-	} else if (disconnect(args[0], String(args[1]).to(port))) {
+				"<response desc=\"Faltan argumentos.\" />");
+	} else if (disconnect(args[0], to(args[1], port))) {
 		return new HTTPResponse(HTTPMessage::OK,
-				string("La conexión a ") + args[0] + ":" + args[1]
-				+ " se cerrará en instantes...");
+				string("<response desc=\"La conexión a ") + args[0] + ":" + args[1]
+				+ " se cerrará en instantes...\" />");
 	} else {
 		return new HTTPResponse(HTTPMessage::NOT_FOUND,
-				string("No existe una conexión a ") + args[0]
-				+ ":" + args[1]);
+				string("<response desc=\"No existe una conexión a ") + args[0]
+				+ ":" + args[1] + "\" />");
+	}
+}
+
+HTTPResponse* Server::cmd_transmission_list(void) {
+	// FIXME
+	stringstream xml;
+	xml << "<list type=\"transmission\">" << endl;
+/*TODO	plants_mutex.lock();
+	for (PlantList::const_iterator i = plants.begin();
+			i != plants.end(); i++) {
+		trans
+		xml << "       <li>" << (*i)->get_host() << ":"
+			<< (*i)->get_port() << " [<a href=\"/transmission/stop/"
+			<< (*i)->get_host() << "/" << (*i)->get_port()
+			<< "\">desconectar</a>]</li>" << endl;
+	}
+	transmissions_mutex.unlock();*/
+	xml << "</list>" << endl;
+	return new HTTPResponse(HTTPMessage::OK, xml.str());
+}
+
+HTTPResponse* Server::cmd_transmission_start(const Command& command) {
+	const Command::Arguments& args = command.get_args();
+	if (args.size() < 3) {
+		return new HTTPResponse(HTTPMessage::CONFLICT,
+				"<response desc=\"Faltan argumentos.\" />");
+	} else {
+		string plant = args[0];
+		string host = args[1];
+		Connection::Port port = to(args[2], port);
+		Glib::Mutex::Lock lock(plants_mutex);
+		PlantList::iterator p = plants.find(plant);
+		if (p == plants.end()) {
+			return new HTTPResponse(HTTPMessage::NOT_FOUND,
+					string("<response desc=\"No existe la planta '") + plant + "'.\" />");
+		// TODO - agregar chequeo de que la transmision a ese host:port no
+		//        exista para otra planta?
+		} else if (plants[plant]->transmission_start(host, port)) {
+			return new HTTPResponse(HTTPMessage::OK,
+					string("<response desc=\"Se empieza a transmitir la planta '") + plant
+					+ "' a " + host + ":" + String().from(port) + ".\" />");
+		} else {
+			return new HTTPResponse(HTTPMessage::INTERNAL_SERVER_ERROR,
+					string("<response desc=\"Error al crear la transmisión a de la planta '")
+					+ plant + "' a " + host + ":" + args[2] + ".\" />");
+		}
+	}
+}
+
+HTTPResponse* Server::cmd_transmission_stop(const Command& command) {
+	const Command::Arguments& args = command.get_args();
+	if (args.size() < 2) {
+		return new HTTPResponse(HTTPMessage::CONFLICT,
+				"<response desc=\"Faltan argumentos.\" />");
+	} else {
+		const string& host = args[0];
+		Connection::Port port = to(args[1], port);
+		for (PlantList::iterator i = plants.begin(); i != plants.end(); i++) {
+			// TODO - agregar chequeo para saber si existe una conexion (para
+			// tirar error de que no hay conexion o de que no se pudo
+			// desconectar.
+			if (i->second->transmission_stop(host, port)) {
+				return new HTTPResponse(HTTPMessage::OK,
+						string("<response desc=\"Se finaliza la transmisión de la planta '")
+						+ i->first + "' a " + host + ":" + args[1] + ".\" />");
+			}
+		}
+		return new HTTPResponse(HTTPMessage::NOT_FOUND,
+				string("<response desc=\"No se puede finalizar la transmisión a ")
+				+ host + ":" + args[1] + ".\" />");
+	}
+}
+
+HTTPResponse* Server::cmd_plant_list(void) {
+	// FIXME
+	stringstream xml;
+	xml << "<list type=\"plant\">" << endl;
+	plants_mutex.lock();
+	for (PlantList::const_iterator i = plants.begin();
+			i != plants.end(); i++) {
+		xml << "\t<row>" << endl;
+		xml << "\t\t<cell>" << i->first << "</cell>" << endl;
+		xml << "\t</row>" << endl;
+	}
+	plants_mutex.unlock();
+	xml << "</list>" << endl;
+	return new HTTPResponse(HTTPMessage::OK, xml.str());
+}
+
+HTTPResponse* Server::cmd_plant_get(const Command& command) {
+	if (!command.get_args().size()) {
+		return new HTTPResponse(HTTPMessage::CONFLICT,
+				"<response desc=\"Faltan argumentos.\" />");
+	}
+	Glib::Mutex::Lock lock(plants_mutex);
+	string plant = command.get_args()[0];
+	if (plants.find(plant) == plants.end()) {
+		return new HTTPResponse(HTTPMessage::NOT_FOUND,
+				string("<response desc=\"No existe la planta ") + plant + "\" />");
+	}
+	// TODO try/catch?
+	string xml = plants[plant]->get_xml();
+	if (xml.length()) {
+		return new HTTPResponse(HTTPMessage::OK, xml);
+	} else {
+		return new HTTPResponse(HTTPMessage::INTERNAL_SERVER_ERROR,
+				("<response desc=\"No se pudo obtener el XML de la planta ") + plant + "\" />");
+	}
+}
+
+HTTPResponse* Server::cmd_plant_set(const Command& command) {
+	const Command::Arguments& args = command.get_args();
+	if (args.size() < 4) {
+		return new HTTPResponse(HTTPMessage::CONFLICT,
+				"<response desc=\"Faltan argumentos.\" />");
+	}
+	string plant = args[0];
+	string element = args[1];
+	string key = args[2];
+	if (key != "open") {
+		return new HTTPResponse(HTTPMessage::NOT_FOUND,
+				string("<response desc=\"La clave '") + key + "' es inválida.\" />");
+	}
+	string value = args[3];
+	Glib::Mutex::Lock lock(plants_mutex);
+	PlantList::iterator p = plants.find(plant);
+	if (p == plants.end()) {
+		return new HTTPResponse(HTTPMessage::NOT_FOUND,
+				string("<response desc=\"No existe la planta '") + plant + "'.\" />");
+	}
+	bool open = true;
+	if ((value == "false") || (value == "0") || (value == "off")
+			|| (value == "no")) {
+		open = false;
+	}
+	if (!plants[plant]->set_open(element, open)) {
+		return new HTTPResponse(HTTPMessage::CONFLICT,
+				string("<response desc=\"No se pudo cambiar el estado del elemento '") + element + "'.\" />");
+	}
+	return new HTTPResponse(HTTPMessage::OK,
+			string("<response desc=\"Se cambió el estado del elemento '") + element + "'.\" />");
+}
+
+HTTPResponse* Server::cmd_plant_set_frequency(const Command& command) {
+	if (command.get_args().size() < 2) {
+		return new HTTPResponse(HTTPMessage::CONFLICT,
+				"<response desc=\"Faltan argumentos.\" />");
+	}
+	Glib::Mutex::Lock lock(plants_mutex);
+	const string name = command.get_args()[0];
+	if (plants.find(name) == plants.end()) {
+		return new HTTPResponse(HTTPMessage::NOT_FOUND,
+				string("<response desc=\"No existe la planta ") + name + "\" />");
+	}
+	unsigned hz;
+	to(command.get_args()[1], hz);
+	plants[name]->set_frequency(hz);
+	return new HTTPResponse(HTTPMessage::OK,
+			string("<response desc=\"La planta '") + name + "' fue pausada.\" />");
+}
+
+HTTPResponse* Server::cmd_plant_start(const Command& command) {
+	if (!command.get_args().size()) {
+		return new HTTPResponse(HTTPMessage::CONFLICT,
+				"<response desc=\"Faltan argumentos.\" />");
+	}
+	Glib::Mutex::Lock lock(plants_mutex);
+	const string name = command.get_args()[0];
+	if (plants.find(name) == plants.end()) {
+		return new HTTPResponse(HTTPMessage::NOT_FOUND,
+				string("<response desc=\"No existe la planta ") + name + "\" />");
+	}
+	plants[name]->set_paused(false);
+	return new HTTPResponse(HTTPMessage::OK,
+			string("<response desc=\"La planta '") + name + "' fue reanudada.\" />");
+}
+
+HTTPResponse* Server::cmd_plant_stop(const Command& command) {
+	if (!command.get_args().size()) {
+		return new HTTPResponse(HTTPMessage::CONFLICT,
+				"<response desc=\"Faltan argumentos.\" />");
+	}
+	Glib::Mutex::Lock lock(plants_mutex);
+	const string name = command.get_args()[0];
+	if (plants.find(name) == plants.end()) {
+		return new HTTPResponse(HTTPMessage::NOT_FOUND,
+				string("<response desc=\"No existe la planta ") + name + "\" />");
+	}
+	plants[name]->set_paused(true);
+	return new HTTPResponse(HTTPMessage::OK,
+			string("<response desc=\"La planta '") + name + "' fue pausada.\" />");
+}
+
+HTTPResponse* Server::cmd_plant_remove(const Command& command) {
+	if (!command.get_args().size()) {
+		return new HTTPResponse(HTTPMessage::CONFLICT,
+				"<response desc=\"Faltan argumentos.\" />");
+	}
+	Glib::Mutex::Lock lock(plants_mutex);
+	const string name = command.get_args()[0];
+	if (plants.find(name) == plants.end()) {
+		return new HTTPResponse(HTTPMessage::NOT_FOUND,
+				string("<response desc=\"No existe la planta ") + name + "\" />");
 	}
+	plants[name]->finish();
+	return new HTTPResponse(HTTPMessage::OK,
+			string("<response desc=\"La planta '") + name + "' se cerrará en instantes...\" />");
 }
 
 } // namespace Server