have node retrieve js files via ws instead of http

lib/flowerbox/reporter/console_base.rb

@@ -1,3 +1,5 @@
+require 'flowerbox/reporter/base'
+
 module Flowerbox::Reporter
   class ConsoleBase < Base
     include FileDisplay

lib/flowerbox/reporter/verbose.rb

@@ -1,3 +1,5 @@
+require 'flowerbox/reporter/console_base'
+
 module Flowerbox::Reporter
   class Verbose < ConsoleBase
     def initialize

lib/flowerbox/runner/base.rb

@@ -32,6 +32,10 @@ module Flowerbox
         Flowerbox.reporters
       end
 
+      def load(file)
+        sprockets.find_asset(file.first, :bundle => false).body
+      end
+
       def ensure_alive
         while @count < MAX_COUNT && !finished?
           @count += 1 if @timer_running

lib/flowerbox/runner/node.rb

@@ -66,27 +66,16 @@ jsdom.env(
 
   var gotFlowerbox = false;
 
-  var files = #{sprockets.files.to_json};
+  var socket = new ws('ws://localhost:#{server.port + 1}/');
-  var fileRunner = function() {
+  socket.onopen = function() {
-    if (files.length > 0) {
+    var files = #{sprockets.files.to_json};
-      var file = files.shift();
 
-      var options = {
+    var fileLoader = function() {
-        host: "localhost",
+      if (files.length > 0) {
-        port: #{server.port},
+        var file = files.shift();
-        path: "/__F__/" + file,
-        method: "GET"
-      };
 
-      var request = http.request(options, function(response) {
+        socket.onmessage = function(data) {
-        var data = '';
+          vm.runInNewContext(data.data, context, file);
-
-        response.on('data', function(chunk) {
-          data += chunk;
-        });
-
-        response.on('end', function() {
-          vm.runInNewContext(data, context, file);
 
           for (thing in window) {
             if (!context[thing]) { context[thing] = window[thing] }
@@ -95,18 +84,18 @@ jsdom.env(
           if (!gotFlowerbox && context.Flowerbox) {
             context.Flowerbox.environment = 'node';
             context.Flowerbox.UNKNOWN = '#{Flowerbox::Result::FileInfo::UNKNOWN}';
+            context.Flowerbox.socket = socket;
 
             gotFlowerbox = true;
           }
 
-          fileRunner();
+          fileLoader();
-        });
+        };
-      });
+
+        socket.send(JSON.stringify(['load', file]));
+      } else {
+        socket.onmessage = null;
 
-      request.end();
-    } else {
-      context.Flowerbox.socket = new ws('ws://localhost:#{server.port + 1}/');
-      context.Flowerbox.socket.onopen = function() {
         #{env}
 
         var waitForFinish;
@@ -118,10 +107,11 @@ jsdom.env(
           }
         };
         waitForFinish();
-      };
+      }
-    }
+    };
+
+    fileLoader();
   };
-  fileRunner();
 });
 JS
       end

lib/flowerbox/server.rb

@@ -43,7 +43,9 @@ module Flowerbox
       ws.onmessage { |message|
         command, data = JSON.parse(message)
 
-        runner.send(command, data.flatten)
+        output = runner.send(command, [ data ].flatten)
+
+        ws.send(output) if command == 'load'
       }
     end