SpacebrewYun: added new example

1 files changed, 356 insertions, 0 deletions

diff --git a/libraries/SpacebrewYun/SpacebrewYun.cpp b/libraries/SpacebrewYun/SpacebrewYun.cpp
new file mode 100644
index 0000000..a2d04a5
--- /dev/null
+++ b/libraries/SpacebrewYun/SpacebrewYun.cpp
@@ -0,0 +1,356 @@
+#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);						
+		}
+	}
+}
+
+