From 4ef706afeee1232c7b2db9a0a4150b7498e81212 Mon Sep 17 00:00:00 2001
From: Ken Moore <moorekou@gmail.com>
Date: Tue, 3 May 2016 11:03:53 -0400
Subject: [PATCH] Get the bridge setup with the rpc/identify API call as well.
 Also get it all setup for adding additional API calls.

---
 src/bridge/BridgeConnection.cpp | 39 ++++++++++++++++++++++++++++++++-
 src/bridge/BridgeConnection.h   |  4 +++-
 src/bridge/BridgeServer.h       |  2 +-
 3 files changed, 42 insertions(+), 3 deletions(-)

diff --git a/src/bridge/BridgeConnection.cpp b/src/bridge/BridgeConnection.cpp
index 1a06c87..32b2e40 100644
--- a/src/bridge/BridgeConnection.cpp
+++ b/src/bridge/BridgeConnection.cpp
@@ -78,7 +78,7 @@ QStringList BridgeConnection::JsonArrayToStringList(QJsonArray array){
 void BridgeConnection::InjectMessage(QString msg){
   //See if this message is directed to the bridge itself, or a client
   if(msg.startsWith("{")){
-
+    HandleAPIMessage(msg);
   }else{
     //Need to read the destination off the message first
     int lb = msg.indexOf("\n"); //line break
@@ -94,6 +94,43 @@ void BridgeConnection::InjectMessage(QString msg){
   }
 }
 
+void BridgeConnection::HandleAPIMessage(QString msg){
+  QJsonObject JM = QJsonDocument::fromJson(msg.toLocal8Bit()).object();
+  QJsonObject out;
+  if(JM.isEmpty() || !JM.contains("namespace") || !JM.contains("name") || !JM.contains("args") || !JM.contains("id") ){
+    //invalid inputs - return 
+    out.insert("namespace","error");
+    out.insert("name","error");
+    out.insert("id", JM.contains("id") ? JM.value("id") : "error");
+    out.insert("args", "");
+  }else if( JM.value("name").toString()=="response" ){
+    // - Return messages first (check ID)
+      QString id = JM.value("id").toString();
+      if(id=="sysadm_bridge_request_ident"){
+        serverconn = (JM.value("args").toObject().value("type").toString() == "server");
+      }
+     //no response needed
+  }else{
+    //API Call
+    QString name, namesp, id; 
+    QJsonObject args = JM.value("args").toObject();
+    name = JM.value("name").toString();
+    namesp = JM.value("namespace").toString();
+    out.insert("id", JM.value("id"));
+    out.insert("namespace", namesp);
+    out.insert("name","reponse");
+    QJsonObject outargs;
+    //There is only a short list of API calls the bridge is capable of:
+    if(namesp == "rpc" && name=="identify"){
+      outargs.insert("type","bridge");
+    }else{
+      out.insert("name","error"); //unknown API call
+    }
+    out.insert("args",outargs);
+    SOCKET->sendTextMessage( QJsonDocument(out).toJson(QJsonDocument::Compact) );
+  }
+ 
+}
 // =====================
 //       PRIVATE SLOTS
 // =====================
diff --git a/src/bridge/BridgeConnection.h b/src/bridge/BridgeConnection.h
index ddc4e47..0e7ab70 100644
--- a/src/bridge/BridgeConnection.h
+++ b/src/bridge/BridgeConnection.h
@@ -16,18 +16,20 @@ public:
 
 	QString ID();
 	void forwardMessage(QString msg);
+	bool isServer();
 
 private:
 	QTimer *idletimer;
 	QWebSocket *SOCKET;
 	QString SockID, SockAuthToken, SockPeerIP;
+	bool serverconn;
 
 	//Simplification functions
 	QString JsonValueToString(QJsonValue);
 	QStringList JsonArrayToStringList(QJsonArray);
 
 	void InjectMessage(QString msg);
-
+	void HandleAPIMessage(QString msg);
 private slots:
 	void checkIdle(); //see if the currently-connected client is idle
 	void checkAuth(); //see if the currently-connected client has authed yet
diff --git a/src/bridge/BridgeServer.h b/src/bridge/BridgeServer.h
index fb2ebe4..4961743 100644
--- a/src/bridge/BridgeServer.h
+++ b/src/bridge/BridgeServer.h
@@ -40,7 +40,7 @@ private slots:
 	//Socket Blacklist function
 	void BlackListConnection(QHostAddress addr);
 
-	// (WebSocket-only) Server signals/slots
+	//Server signals/slots
 	void ServerClosed(); 						//closed() signal
 	void ServerError(QWebSocketProtocol::CloseCode);	//serverError() signal
 	void ConnectError(QAbstractSocket::SocketError);