Basic work on subscriptions

wi/index.js

@@ -23,8 +23,8 @@ const uuid = require('node-uuid');
 const jwt = require('jsonwebtoken');
 var mysql = require('mysql').createPool(settings.db);
 
-var wsi = new websocket_iterface.wsi(server, key);
 var api = new servertalk.api();
+var wsi = new websocket_iterface.wsi(server, key, api);
 api.Init(settings.servertalk.addr, settings.servertalk.port, false, settings.servertalk.key);
 
 app.use(bodyParser.json());

wi/network/servertalk_api.js

@@ -7,6 +7,7 @@ class ServertalkAPI
 		this.client = new servertalk.client();
 		this.client.Init(addr, port, ipv6, 'WebInterface', credentials);
 		this.pending_calls = {};
+		this.subscriptions = {};
 		var self = this;
 		
 		this.client.on('connecting', function() {
@@ -25,28 +26,51 @@ class ServertalkAPI
 		});
 
 		this.client.on('message', function(opcode, packet) {
-			var response = Buffer.from(packet).toString('utf8');
-			try {
-				var res = JSON.parse(response);
-				
-				if(res.id) {
-					if(self.pending_calls.hasOwnProperty(res.id)) {
-						var entry = self.pending_calls[res.id];
-						
-						if(res.error) {
-							var reject = entry[1];
-							reject(res.error);
-						} else {
-							var resolve = entry[0];
-							resolve(res.response);
+			if(opcode == 47) {
+				var response = Buffer.from(packet).toString('utf8');
+				try {
+					var res = JSON.parse(response);
+					
+					if(res.id) {
+						if(self.pending_calls.hasOwnProperty(res.id)) {
+							var entry = self.pending_calls[res.id];
+							
+							if(res.error) {
+								var reject = entry[1];
+								reject(res.error);
+							} else {
+								var resolve = entry[0];
+								resolve(res.response);
+							}
+							
+							delete self.pending_calls[res.id];
 						}
-						
-						delete self.pending_calls[res.id];
 					}
+				} catch(ex) {
+					console.log('Error processing response from server:\n', ex);
+				}
+			} else if(opcode == 104) {
+				var message = Buffer.from(packet).toString('utf8');
+				try {
+					var msg = JSON.parse(message);
+					
+					if(msg.event) {
+						if(self.subscriptions.hasOwnProperty(msg.event)) {
+							var subs = self.subscriptions[msg.event];
+						
+							for(var idx in subs) {
+								try {
+									var sub = subs[idx];
+									sub.emit('subscriptionMessage', msg);
+								} catch(ex) {
+									console.log('Error dispatching subscription message', ex);
+								}
+							}
+						}
+					}
+				} catch(ex) {
+					console.log('Error processing response from server:\n', ex);
 				}
-				
-			} catch(ex) {
-				console.log('Error processing response from server:\n', ex);
 			}
 		});
 	}
@@ -83,6 +107,34 @@ class ServertalkAPI
 		var c = { method: method, params: args };
 		client.Send(47, Buffer.from(JSON.stringify(c)));
 	}
+	
+	Subscribe(event_id, who) {
+		this.Unsubscribe(event_id, who);
+		
+		var subs = this.subscriptions[event_id];
+		if(subs) {
+			console.log('Subscribe', who.uuid, 'to', event_id);
+			subs[who.uuid] = who;
+		} else {
+			console.log('Subscribe', who.uuid, 'to', event_id);
+			this.subscriptions[event_id] = { };
+			this.subscriptions[event_id][who.uuid] = who;
+		}
+	}
+	
+	Unsubscribe(event_id, who) {
+		var subs = this.subscriptions[event_id];
+		if(subs) {
+			console.log('Unsubscribe', who.uuid, 'from', event_id);
+			delete subs[who.uuid];
+		}
+	}
+	
+	UnsubscribeAll(who) {
+		for(var sub_idx in this.subscriptions) {
+			this.Unsubscribe(sub_idx, who);
+		}
+	}
 }
 
 module.exports = {

wi/network/servertalk_client.js

@@ -214,8 +214,8 @@ class ServertalkClient extends EventEmitter
 	
 	ProcessMessage(p) {
 		try {
-			var length = this.m_buffer.readUInt32LE(0);
-			var opcode = this.m_buffer.readUInt16LE(4);
+			var length = p.readUInt32LE(0);
+			var opcode = p.readUInt16LE(4);
 			if(length > 0) {
 				var data = p.slice(6);
 				

wi/test.js

@@ -2,7 +2,7 @@ const WebSocket = require('ws');
 const ws = new WebSocket('ws://localhost:9080');
 
 ws.on('open', function open() {
-	ws.send(JSON.stringify({authorization: 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VybmFtZSI6IktTcHJpdGUxIiwiZXhwIjoxNDg0NzIzNDQxLCJpYXQiOjE0ODQxMTg2NDF9.Lmwm572yMWIu1DUrfer8JVvl1DGEkdnMsMFp5WDzp_A', id: '1', method: 'EQW::IsLocked', params: []}));
+	ws.send(JSON.stringify({authorization: 'eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJ1c2VybmFtZSI6IktTcHJpdGUxIiwiZXhwIjoxNDg0NzIzNDQxLCJpYXQiOjE0ODQxMTg2NDF9.Lmwm572yMWIu1DUrfer8JVvl1DGEkdnMsMFp5WDzp_A', id: '1', method: 'EQW::Subscribe::WorldZoneUpdate', params: []}));
 });
 
 ws.on('message', function(data, flags) {

wi/ws/eqw.js

@@ -1,20 +1,10 @@
-function Register(name, wsi, api) {
-	wsi.Register(name, 
-		function(request) {
-			api.Call(name, request.params)
-				.then(function(value) {
-					wsi.Send(request, value);
-				})
-				.catch(function(reason) {
-					wsi.SendError(request, reason);
-				});
-		}, true);
-}
+const common = require('./wi_common.js');
 
 var RegisterEQW = function(wsi, api) {
-	Register('EQW::IsLocked', wsi, api);
-	Register('EQW::Lock', wsi, api);
-	Register('EQW::Unlock', wsi, api);
+	common.Register('EQW::IsLocked', wsi, api);
+	common.Register('EQW::Lock', wsi, api);
+	common.Register('EQW::Unlock', wsi, api);
+	common.RegisterSubscription('EQW', 'WorldZoneUpdate', wsi, api);
 };
 
 module.exports = {

wi/ws/wi_common.js

@@ -0,0 +1,27 @@
+function Register(name, wsi, api) {
+	wsi.Register(name, 
+		function(request) {
+			api.Call(name, request.params)
+				.then(function(value) {
+					wsi.Send(request, value);
+				})
+				.catch(function(reason) {
+					wsi.SendError(request, reason);
+				});
+		}, true);
+}
+
+function RegisterSubscription(namespace, event, wsi, api) {
+	wsi.Register(namespace + '::Subscribe::' + event, function(request) {
+		api.Subscribe(event, request.ws);
+	});
+	
+	wsi.Register(namespace + '::Unsubscribe::' + event, function(request) {
+		api.Unsubscribe(event, request.ws);
+	});
+}
+
+module.exports = {
+	'Register': Register,
+	'RegisterSubscription': RegisterSubscription
+}

wi/ws/ws_interface.js

@@ -1,18 +1,21 @@
 const WebSocketServer = require('ws').Server;
 const jwt = require('jsonwebtoken');
+const uuid = require('node-uuid');
 
 class WebSocketInterface
 {
-	constructor(server, key) {
+	constructor(server, key, api) {
 		this.wss = new WebSocketServer({ server: server });
 		this.methods = {};
 		var self = this;
 		
 		this.wss.on('connection', function connection(ws) {
 			self.ws = ws;
+			ws.uuid = uuid.v4();
 			ws.on('message', function incoming(message) {
 				try {
 					var request = JSON.parse(message);
+					request.ws = ws;
 					
 					if(request.method) {
 						var method = self.methods[request.method];
@@ -51,6 +54,14 @@ class WebSocketInterface
 					self.SendError(null, 'No method supplied');
 				}
 			});
+			
+			ws.on('close', function() {
+				api.UnsubscribeAll(ws);
+			});
+			
+			ws.on('subscriptionMessage', function(msg) {
+				self.SendRaw(msg);
+			});
 		});
 	}
 	
@@ -92,6 +103,14 @@ class WebSocketInterface
 			console.log(ex);
 		}
 	}
+	
+	SendRaw(obj) {
+		try {
+			this.ws.send(JSON.stringify(obj));
+		} catch(ex) {
+			console.log(ex);
+		}
+	}
 }
 
 module.exports = {