Add support for "specialized" sockets.

Thiese can be use for pointer events and button pressing, and possibly more.
the-louie · Jun 29, 2016 · 2c593b6 · 2c593b6
1 parent 1dfc226
commit 2c593b6
Show file tree

Hide file tree

Showing 2 changed files with 61 additions and 1 deletion.
diff --git a/index.js b/index.js
@@ -11,6 +11,26 @@ var WebSocketClient = require('websocket').client; // for communication with TV
 var EventEmitter = require('events').EventEmitter;
 var util =  require('util');
 
+var SpecializedSocket = function (ws) {
+    this.send = function(type, payload) {
+        payload = payload || {};
+        // the message should be key:value pairs, one per line,
+        // with an extra blank line to terminate.
+        var message =
+            Object.keys(payload)
+                .reduce(function(acc, k) {
+                    return acc.concat([k + ':' + payload[k]]);
+                }, ['type:' + type])
+                .join('\n') + '\n\n';
+
+        ws.send(message);
+    };
+
+    this.close = function() {
+        ws.close();
+    };
+};
+
 var LGTV = function (config) {
     if (!(this instanceof LGTV)) return new LGTV(config);
     var that = this;
@@ -41,6 +61,8 @@ var LGTV = function (config) {
     var isPaired = false;
     var autoReconnect = config.reconnect;
 
+    var specializedSockets = {};
+
     var callbacks = {};
     var cidCount = 0;
     var cidPrefix = ('0000000' + (Math.floor(Math.random() * 0xFFFFFFFF).toString(16))).slice(-8);
@@ -200,6 +222,40 @@ var LGTV = function (config) {
         connection.send(json);
     };
 
+    this.getSocket = function(url, cb) {
+        if (specializedSockets[url]) {
+            cb(null, specializedSockets[url]);
+            return;
+        }
+
+        that.request(url, function(err, data) {
+            if (err) {
+                cb(err);
+                return;
+            }
+
+            var special = new WebSocketClient();
+            special
+                .on('connect', function(conn) {
+                    conn
+                        .on('error', function(error) {
+                            that.emit('error', error);
+                        })
+                        .on('close', function() {
+                            delete specializedSockets[url];
+                        });
+
+                    specializedSockets[url] = new SpecializedSocket(conn);
+                    cb(null, specializedSockets[url]);
+                })
+                .on('connectFailed', function(error) {
+                    that.emit('error', error);
+                })
+
+            special.connect(data.socketPath);
+        });
+    };
+
     /**
      *      Connect to TV using a websocket url (eg "ws://192.168.0.100:3000")
      *
@@ -219,6 +275,10 @@ var LGTV = function (config) {
     this.disconnect = function () {
         connection.close();
         autoReconnect = false;
+
+        Object.keys(specializedSockets).forEach(
+            function (k) { specializedSockets[k].close(); }
+        );
     };
 
     setTimeout(function () {

diff --git a/package.json b/package.json
@@ -1,6 +1,6 @@
 {
   "name": "lgtv2",
-  "version": "1.0.3",
+  "version": "1.1.0",
   "description": "Simple module to remote control LG WebOS smart TVs",
   "main": "index.js",
   "scripts": {