Better formatting, and include extendLog

[rbdr/cologne] / lib / loggers / socket.js

diff --git a/lib/loggers/socket.js b/lib/loggers/socket.js
index d9ae05577ed4a9815dc8eafaa1a9735b8c562673..b057d77726e8b4b25ae238392f4f3c83d5de354b 100644 (file)
--- a/lib/loggers/socket.js
+++ b/lib/loggers/socket.js
@@ -4,6 +4,7 @@
   // Load up dependencies
   if (typeof require === 'function') {
     var Ne = require('neon');
+    var ioClient = require('socket.io-client');
     Module = Ne.Module;
     Class = Ne.Class;
   } else {
@@ -18,8 +19,20 @@
     prototype : {
       serverUrl : '/',
 
-      init : function () {
-        this._socket = io.connect(this.serverUrl);
+      init : function (config) {
+        var logger = this;
+
+        if (config) {
+          for (property in config) {
+            logger[property] = config[property];
+          }
+        }
+
+        if (!logger.socketIo) {
+          logger.socketIo = ioClient;
+        }
+
+        logger._socket = logger.socketIo.connect(logger.serverUrl);
       },
 
       log : function () {
@@ -29,7 +42,9 @@
           messageArray.push(arguments[i]);
         }
 
-        this._socket.emit('log', messageArray);
+        if (this._socket) {
+          this._socket.emit('log', messageArray);
+        }
       }
     }
   });