Arduino/hardware/arduino/avr/libraries/SpacebrewYun/SpacebrewYun.cpp

#include "SpacebrewYun.h"


SpacebrewYun::SpacebrewYun(const String& _name, const String& _description) {
	name = _name;
	description = _description;

	server = "sandbox.spacebrew.cc";
	port = 9000;

	_connected = false;
	_verbose = false;
	_error_msg = false;

	sub_name = "";
	sub_msg = "";
	sub_type = "";

	read_name = false;
	read_msg = false;
	sub_name_max = 25;
	sub_msg_max = 50;

	for ( int i = 0; i < pidLength; i++ ) {
		pid [i] = '\0';
	}

	for ( int i = 0; i < sbPidsLen; i++ ) {
		sbPids [i] = '\0';
	}

	Console.buffer(64);

}

// boolean SpacebrewYun::_connected = false;
SpacebrewYun::OnBooleanMessage SpacebrewYun::_onBooleanMessage = NULL;
SpacebrewYun::OnRangeMessage SpacebrewYun::_onRangeMessage = NULL;
SpacebrewYun::OnStringMessage SpacebrewYun::_onStringMessage = NULL;
SpacebrewYun::OnCustomMessage SpacebrewYun::_onCustomMessage = NULL;
SpacebrewYun::OnSBOpen SpacebrewYun::_onOpen = NULL;
SpacebrewYun::OnSBClose SpacebrewYun::_onClose = NULL;
SpacebrewYun::OnSBError SpacebrewYun::_onError = NULL;

void SpacebrewYun::onOpen(OnSBOpen function){
	_onOpen = function;
}
void SpacebrewYun::onClose(OnSBClose function){
	_onClose = function;
}
void SpacebrewYun::onRangeMessage(OnRangeMessage function){
	_onRangeMessage = function;
}
void SpacebrewYun::onStringMessage(OnStringMessage function){
	_onStringMessage = function;
}
void SpacebrewYun::onBooleanMessage(OnBooleanMessage function){
	_onBooleanMessage = function;
}
void SpacebrewYun::onCustomMessage(OnCustomMessage function){
	_onCustomMessage = function;
}
void SpacebrewYun::onError(OnSBError function){
	_onError = function;
}

void SpacebrewYun::addPublish(const String& name, const String& type) {
	Publisher *p = new Publisher();
	p->name = createString(name.length() + 1);
	p->type = createString(type.length() + 1);
	name.toCharArray(p->name, name.length() + 1);
	type.toCharArray(p->type, type.length() + 1);

	if (publishers == NULL){
		publishers = p;
	} else {
		Publisher *curr = publishers;
		while(curr->next != NULL){
			curr = curr->next;
		}
		curr->next = p;
	}
}

void SpacebrewYun::addSubscribe(const String& name, const String& type) {
	Subscriber *p = new Subscriber();
	p->name = createString(name.length() + 1);
	p->type = createString(type.length() + 1);
	name.toCharArray(p->name, name.length() + 1);
	type.toCharArray(p->type, type.length() + 1);

	if (subscribers == NULL){
		subscribers = p;
	} 

	else {
		Subscriber *curr = subscribers;
		while(curr->next != NULL){
			curr = curr->next;
		}
		curr->next = p;
	}
}

void SpacebrewYun::connect(String _server, int _port) {
	server = _server;
	port = _port; 
	
	killPids();

	brew.begin("python"); // Process should launch the "curl" command
	brew.addParameter("/usr/lib/python2.7/spacebrew/spacebrew.py"); // Process should launch the "curl" command
	brew.addParameter("--server");
	brew.addParameter(server);
	brew.addParameter("--port");
	brew.addParameter(String(port));
	brew.addParameter("-n");
	brew.addParameter(name);
	brew.addParameter("-d");
	brew.addParameter(description);

	if (subscribers != NULL) {
		Subscriber *curr = subscribers;
		while(curr != NULL){
			if (_verbose) {
				Serial.print(F("Creating subcribers: "));
				Serial.println(curr->name);
			}

			brew.addParameter("-s"); // Add the URL parameter to "curl"
			brew.addParameter(curr->name); // Add the URL parameter to "curl"
			brew.addParameter(","); // Add the URL parameter to "curl"
			brew.addParameter(curr->type); // Add the URL parameter to "curl"

			if (curr->next == NULL) curr = NULL;
			else curr = curr->next;
		}
	}
	if (publishers != NULL) {
		Publisher *curr = publishers;
		while(curr != NULL){
			if (_verbose) {
				Serial.print(F("Creating publishers: "));
				Serial.println(curr->name);
			}
			brew.addParameter("-p"); // Add the URL parameter to "curl"
			brew.addParameter(curr->name); // Add the URL parameter to "curl"
			brew.addParameter(","); // Add the URL parameter to "curl"
			brew.addParameter(curr->type); // Add the URL parameter to "curl"

			if (curr->next == NULL) curr = NULL;
			else curr = curr->next;
		}
	}

	Console.begin();
	brew.runAsynchronously();
	while (!Console) { ; }
}

void SpacebrewYun::monitor() {
	while (Console.available() > 0) {
		char c = Console.read();

		if (c == char(CONNECTION_START) && !_connected) {
			_connected = true;
			if (_verbose) {
				Serial.print(F("Connected to spacebrew server at: "));
				Serial.println(server);
				Serial.print(F("Application name set to: "));
				Serial.println(name);
			}
			if (_onOpen != NULL){
				_onOpen();
			}
		} 	    
		else if (c == char(CONNECTION_END) && _connected) {
			_connected = false;
			if (_verbose) {
				Serial.print(F("Disconnected from spacebrew server at: "));
				Serial.println(server);
			}
			if (_onClose != NULL){
				_onClose();
			}
		} 	 

		if (_verbose) {
			if (c == char(CONNECTION_ERROR)) {
				_error_msg = true;
					Serial.println(F("ERROR :: with Spacebrew.py Connection ::"));
			}
			else if (_error_msg && c == char(MSG_END)) {
				_error_msg = false;
				Serial.println();
			}
			if (_error_msg) {
			    Serial.print(c);			
			}
		}

		if (_connected) {
			if (c == char(MSG_START)) {
				read_name = true;
			} else if (c == char(MSG_DIV) || sub_name.length() > sub_name_max) {
				read_name = false;
				read_msg = true;
			} else if (c == char(MSG_END) || sub_msg.length() > sub_msg_max) {
				read_msg = false;
				onMessage();
			} else {
				if (read_name == true) {
					sub_name += c;
				} else if (read_msg == true) {
					sub_msg += c;
				} 
				else if (_verbose) {
					Serial.print(c);
				}	    	
			}
		}
	}	
}

boolean SpacebrewYun::connected() {
	return SpacebrewYun::_connected;
}

void SpacebrewYun::verbose(boolean verbose = true) {
	_verbose = verbose;
}

void SpacebrewYun::onMessage() {
	if (subscribers != NULL) {
		Subscriber *curr = subscribers;
		while((curr != NULL) && (sub_type == "")){
			if (sub_name.equals(curr->name) == true) {
				sub_type = curr->type;
			}
			if (curr->next == NULL) curr = NULL;
			else curr = curr->next;
		}
	}

	if ( sub_type.equals("range") ) {
		if (_onRangeMessage != NULL) {
			_onRangeMessage( sub_name, int(sub_msg.toInt()) );
		} else {
			Serial.println(F("ERROR :: Range message received, no callback method is registered"));
		}
	} else if ( sub_type.equals("boolean") ) {
		if (_onBooleanMessage != NULL) {
			_onBooleanMessage( sub_name, ( sub_msg.equals("false") ? false : true ) );
		} else {
			Serial.println(F("ERROR :: Boolean message received, no callback method is registered"));
		}
	} else if ( sub_type.equals("string") ) {
		if (_onStringMessage != NULL) {
			_onStringMessage( sub_name, sub_msg );	
		} else {
			Serial.println(F("ERROR :: String message received, no callback method is registered"));
		}
	} else {
		if (_onCustomMessage != NULL) {
			_onCustomMessage( sub_name, sub_msg, sub_type );
		} else {
			Serial.println(F("ERROR :: Custom message received, no callback method is registered"));
		}
	}

	sub_name = "";
	sub_msg = "";
	sub_type = "";
}


void SpacebrewYun::send(const String& name, const String& value){
	Console.print(char(29));
	Console.print(name);
	Console.print(char(30));
	Console.print(value);
	Console.print(char(31));
	Console.flush();

	true;
}


/**
 * method that gets the pid of all spacebrew.py instances running on the linino.
 */
void SpacebrewYun::getPids() {

	// request the pid of all python processes
	pids.begin("python");
	pids.addParameter("/usr/lib/python2.7/spacebrew/getProcPid.py"); // Process should launch the "curl" command
	pids.run();

	if (_verbose) {
		Serial.println(F("Checking if spacebrew process already running"));
	}

	int sbPidsIndex = 0;
	int pidCharIndex = 0;
	char c = '\0';

	while ( pids.available() > 0 ) {

	    c = pids.read();

		if ( c >= '0' && c <= '9' ) {
			pid[pidCharIndex] = c;
			pidCharIndex = (pidCharIndex + 1) % pidLength;
		} 

		else if ( (c == ' ' || c == '\n') && pidCharIndex > 0) {
			sbPids[sbPidsIndex] = atoi(pid);
			if ( sbPidsIndex < (sbPidsLen - 1) ) sbPidsIndex = (sbPidsIndex + 1);    		

			for( int i = 0; i < pidLength; i++ ){ 
				pid[i] = '\0';
				pidCharIndex = 0;
			}
		}
	}
}

/**
 * method that kills all of the spacebrew.py instances that are running 
 * on the linino.
 */
void SpacebrewYun::killPids() {
	getPids();
	delay(400);

	for (int i = 0; i < sbPidsLen; i ++) {
		if (sbPids[i] > 0) {
			char * newPID = itoa(sbPids[i], pid, 10);

			if (_verbose) {
				Serial.print(F("Stopping existing spacebrew processes with pids: "));
				Serial.println(newPID);
			}

			Process p;
			p.begin("kill");
			p.addParameter("-9");
			p.addParameter(newPID);		// Process should launch the "curl" command
			p.run();            		// Run the process and wait for its termination	

			delay(400);						
		}
	}
}