From 27eadc5587d50de7e22db2cdcd5229c99483ac98 Mon Sep 17 00:00:00 2001
From: wltsmrz <wltsmrz@gmail.com>
Date: Tue, 15 Jul 2014 00:37:45 -0700
Subject: [PATCH] Add addStream to request

---
 src/js/ripple/request.js | 42 +++++++++++++++++++++++++++++++++++-----
 1 file changed, 37 insertions(+), 5 deletions(-)

diff --git a/src/js/ripple/request.js b/src/js/ripple/request.js
index 7e20ec64..334caca5 100644
--- a/src/js/ripple/request.js
+++ b/src/js/ripple/request.js
@@ -11,12 +11,19 @@ var Server       = require('./server').Server;
 //  'remoteError'
 //  'remoteUnexpected'
 //  'remoteDisconnected'
+
+/**
+ * Request
+ *
+ * @param {Remote} remote
+ * @param {String} command
+ */
+
 function Request(remote, command) {
   EventEmitter.call(this);
 
   this.remote = remote;
   this.requested = false;
-
   this.message = {
     command: command,
     id: void(0)
@@ -31,15 +38,15 @@ Request.prototype.broadcast = function() {
 };
 
 // Send the request to a remote.
-Request.prototype.request = function(remote) {
+Request.prototype.request = function(callback) {
   if (this.requested) {
-    return;
+    return this;
   }
 
   this.requested = true;
 
   this.on('error', function(){});
-  this.emit('request', remote);
+  this.emit('request', this.remote);
 
   if (this._broadcast) {
     this.remote._servers.forEach(function(server) {
@@ -50,13 +57,15 @@ Request.prototype.request = function(remote) {
     this.remote.request(this);
   }
 
+  this.callback(callback);
+
   return this;
 };
 
 Request.prototype.callback = function(callback, successEvent, errorEvent) {
   var self = this;
 
-  if (this.requested || typeof callback !== 'function') {
+  if (typeof callback !== 'function') {
     return this;
   }
 
@@ -359,6 +368,29 @@ Request.prototype.addBook = function(book, snapshot) {
   }
 
   this.message.books.push(json);
+
+  return this;
+};
+
+Request.prototype.addStream = function(stream) {
+  var self = this;
+
+  if (arguments.length > 1) {
+    for (arg in arguments) {
+      this.addStream(arguments[arg]);
+    }
+    return;
+  }
+
+  switch (stream) {
+    case 'ledger':
+    case 'server':
+    case 'transactions':
+    case 'transactions_proposed':
+      this.message.streams = (this.message.streams || []).concat(stream);
+  }
+
+  return this;
 };
 
 exports.Request = Request;