Implement message logging to create recorded exchanges.

Signed-off-by: Katharine Berry <[email protected]>

Author: Katharine

app/src/pkjs/lib/message_queue.js

@@ -18,6 +18,7 @@ var MAX_BYTES_IN_FLIGHT = 600;
 
 function MessageQueue() {
     this.queue = [];
+    this.log = null;
     this.messagesInFlight = 0;
     this.bytesInFlight = 0;
 }
@@ -44,7 +45,22 @@ function countBytes(message) {
     return bytes;
 }
 
+MessageQueue.prototype.startLogging = function() {
+    this.log = [];
+};
+
+MessageQueue.prototype.stopLogging = function() {
+    this.log = null;
+}
+
+MessageQueue.prototype.getLog = function() {
+    return this.log || [];
+}
+
 MessageQueue.prototype.enqueue = function(message) {
+    if (this.log) {
+        this.log.push(message);
+    }
     this.queue.push(message);
     if (this.messagesInFlight < 10 && this.bytesInFlight < MAX_BYTES_IN_FLIGHT) {
         console.log('sending immediately, messages in flight: ' + this.messagesInFlight + ', bytes in flight: ' + this.bytesInFlight);

app/src/pkjs/session.js

@@ -14,6 +14,8 @@
  * limitations under the License.
  */
 
+var LOGGING_ENABLED = false;
+
 var location = require('./location');
 var config = require('./config');
 var actions = require('./actions');
@@ -35,6 +37,9 @@ function getSettings() {
 }
 
 Session.prototype.run = function() {
+    if (LOGGING_ENABLED) {
+        messageQueue.startLogging();
+    }
     console.log("Opening websocket connection...");
     var url = API_URL + '?prompt=' + encodeURIComponent(this.prompt) + '&token=' + exports.userToken;
     if (location.isReady() && config.isLocationEnabled()) {
@@ -151,6 +156,10 @@ Session.prototype.handleMessage = function(event) {
         this.enqueue({
             CHAT_DONE: true
         });
+        if (LOGGING_ENABLED) {
+            console.log(JSON.stringify(messageQueue.getLog()));
+            messageQueue.stopLogging();
+        }
     } else if (message[0] == 'a') {
         actions.handleAction(this, this.ws, message.substring(1));
     } else if (message[0] == 't') {