From 4e9d8e06c99fd25e3cf5462eb5209a051fd62645 Mon Sep 17 00:00:00 2001
From: Romain Beaumont <romain.rom1@gmail.com>
Date: Sun, 20 Sep 2015 21:12:56 +0200
Subject: [PATCH] transform client.js into an es6 class

---
 src/client.js | 290 +++++++++++++++++++++++++-------------------------
 1 file changed, 145 insertions(+), 145 deletions(-)

diff --git a/src/client.js b/src/client.js
index 156d3b1..689aea3 100644
--- a/src/client.js
+++ b/src/client.js
@@ -1,161 +1,161 @@
-var EventEmitter = require('events').EventEmitter
-  , util = require('util')
-  , debug = require('./debug')
-  , serializer = require('./transforms/serializer')
-  , compression = require('./transforms/compression')
-  , framing = require('./transforms/framing')
-  , crypto = require('crypto')
-  , states = serializer.states
-  ;
+var EventEmitter = require('events').EventEmitter;
+var debug = require('./debug');
+var serializer = require('./transforms/serializer');
+var compression = require('./transforms/compression');
+var framing = require('./transforms/framing');
+var crypto = require('crypto');
+var states = serializer.states;
 var version = require('./version');
 var packets = require('minecraft-data')(version.majorVersion).protocol.states;
 var readPackets = require("./packets").readPackets;
 var packetIndexes = readPackets(packets, states);
 
-module.exports = Client;
 
-function Client(isServer) {
-  EventEmitter.call(this);
+class Client extends EventEmitter
+{
+  constructor(isServer) {
+    super();
 
-  var socket;
-  this.packetsToParse = {};
+    var socket;
+    this.packetsToParse = {};
 
-  this.serializer = serializer.createSerializer({ isServer });
-  this.compressor = null;
-  this.framer = framing.createFramer();
-  this.cipher = null;
+    this.serializer = serializer.createSerializer({ isServer });
+    this.compressor = null;
+    this.framer = framing.createFramer();
+    this.cipher = null;
 
-  this.decipher = null;
-  this.splitter = framing.createSplitter();
-  this.decompressor = null;
-  this.deserializer = serializer.createDeserializer({ isServer, packetsToParse: this.packetsToParse });
+    this.decipher = null;
+    this.splitter = framing.createSplitter();
+    this.decompressor = null;
+    this.deserializer = serializer.createDeserializer({ isServer, packetsToParse: this.packetsToParse });
 
-  this._state = states.HANDSHAKING;
-  Object.defineProperty(this, "state", {
-    get: function() {
-      return this.serializer.protocolState;
-    },
-    set: function(newProperty) {
-      var oldProperty = this.serializer.protocolState;
-      this.serializer.protocolState = newProperty;
-      this.deserializer.protocolState = newProperty;
-      this.emit('state', newProperty, oldProperty);
+    this._state = states.HANDSHAKING;
+    Object.defineProperty(this, "state", {
+      get: function() {
+        return this.serializer.protocolState;
+      },
+      set: function(newProperty) {
+        var oldProperty = this.serializer.protocolState;
+        this.serializer.protocolState = newProperty;
+        this.deserializer.protocolState = newProperty;
+        this.emit('state', newProperty, oldProperty);
+      }
+    });
+    Object.defineProperty(this, "compressionThreshold", {
+      get: () => this.compressor == null ? -2 : this.compressor.compressionThreshold,
+      set: (threshold) => this.setCompressionThreshold(threshold)
+    });
+
+    this.isServer = !!isServer;
+
+    this.on('newListener', function(event, listener) {
+      var direction = this.isServer ? 'toServer' : 'toClient';
+      if(packetIndexes.packetStates[direction].hasOwnProperty(event) || event === "packet") {
+        if(typeof this.packetsToParse[event] === "undefined") this.packetsToParse[event] = 1;
+        else this.packetsToParse[event] += 1;
+      }
+    });
+    this.on('removeListener', function(event, listener) {
+      var direction = this.isServer ? 'toServer' : 'toClient';
+      if(packetIndexes.packetStates[direction].hasOwnProperty(event) || event === "packet") {
+        this.packetsToParse[event] -= 1;
+      }
+    });
+  }
+
+  setSocket(socket) {
+    var ended = false;
+
+    // TODO : A lot of other things needs to be done.
+    var endSocket = () => {
+      if(ended) return;
+      ended = true;
+      this.socket.removeListener('close', endSocket);
+      this.socket.removeListener('end', endSocket);
+      this.socket.removeListener('timeout', endSocket);
+      this.emit('end', this._endReason);
+    };
+
+    var onFatalError = (err) => {
+      this.emit('error', err);
+      endSocket();
+    };
+
+    var onError = (err) => this.emit('error', err);
+
+    this.socket = socket;
+
+    if(this.socket.setNoDelay)
+      this.socket.setNoDelay(true);
+
+    this.socket.on('connect', () => this.emit('connect'));
+
+    this.socket.on('error', onFatalError);
+    this.socket.on('close', endSocket);
+    this.socket.on('end', endSocket);
+    this.socket.on('timeout', endSocket);
+    this.serializer.on('error', onError);
+    this.deserializer.on('error', onError);
+    this.framer.on('error', onError);
+    this.splitter.on('error', onError);
+
+    this.socket.pipe(this.splitter).pipe(this.deserializer);
+    this.serializer.pipe(this.framer).pipe(this.socket);
+
+    this.deserializer.on('data', (parsed) => {
+      this.emit('packet', parsed.data, parsed.metadata);
+      this.emit(parsed.metadata.name, parsed.data, parsed.metadata);
+      this.emit('raw.' + parsed.metadata.name, parsed.buffer, parsed.metadata);
+      this.emit('raw', parsed.buffer, parsed.metadata);
+    });
+  }
+
+  end(reason) {
+    this._endReason = reason;
+    this.socket.end();
+  }
+
+  setEncryption(sharedSecret) {
+    if (this.cipher != null)
+      throw new Error("Set encryption twice !");
+    this.cipher = crypto.createCipheriv('aes-128-cfb8', sharedSecret, sharedSecret);
+    this.cipher.on('error', (err) => this.emit('error', err));
+    this.framer.unpipe(this.socket);
+    this.framer.pipe(this.cipher).pipe(this.socket);
+    this.decipher = crypto.createDecipheriv('aes-128-cfb8', sharedSecret, sharedSecret);
+    this.decipher.on('error', (err) => this.emit('error', err));
+    this.socket.unpipe(this.splitter);
+    this.socket.pipe(this.decipher).pipe(this.splitter);
+  }
+
+  setCompressionThreshold(threshold) {
+    if (this.compressor == null) {
+      this.compressor = compression.createCompressor(threshold);
+      this.compressor.on('error', (err) => this.emit('error', err));
+      this.serializer.unpipe(this.framer);
+      this.serializer.pipe(this.compressor).pipe(this.framer);
+      this.decompressor = compression.createDecompressor(threshold);
+      this.decompressor.on('error', (err) => this.emit('error', err));
+      this.splitter.unpipe(this.deserializer);
+      this.splitter.pipe(this.decompressor).pipe(this.deserializer);
+    } else {
+      this.decompressor.threshold = threshold;
+      this.compressor.threshold = threshold;
     }
-  });
-  Object.defineProperty(this, "compressionThreshold", {
-    get: () => this.compressor == null ? -2 : this.compressor.compressionThreshold,
-    set: (threshold) => this.setCompressionThreshold(threshold)
-  });
+  }
 
-  this.isServer = !!isServer;
+  write(packetName, params) {
+    debug("writing packet " + this.state + "." + packetName);
+    debug(params);
+    this.serializer.write({ packetName, params });
+  }
 
-  this.on('newListener', function(event, listener) {
-    var direction = this.isServer ? 'toServer' : 'toClient';
-    if(packetIndexes.packetStates[direction].hasOwnProperty(event) || event === "packet") {
-      if(typeof this.packetsToParse[event] === "undefined") this.packetsToParse[event] = 1;
-      else this.packetsToParse[event] += 1;
-    }
-  });
-  this.on('removeListener', function(event, listener) {
-    var direction = this.isServer ? 'toServer' : 'toClient';
-    if(packetIndexes.packetStates[direction].hasOwnProperty(event) || event === "packet") {
-      this.packetsToParse[event] -= 1;
-    }
-  });
-}
-
-util.inherits(Client, EventEmitter);
-
-Client.prototype.setSocket = function(socket) {
-  var ended = false;
-
-  // TODO : A lot of other things needs to be done.
-  var endSocket = () => {
-    if(ended) return;
-    ended = true;
-    this.socket.removeListener('close', endSocket);
-    this.socket.removeListener('end', endSocket);
-    this.socket.removeListener('timeout', endSocket);
-    this.emit('end', this._endReason);
-  };
-
-  var onFatalError = (err) => {
-    this.emit('error', err);
-    endSocket();
-  };
-
-  var onError = (err) => this.emit('error', err);
-
-  this.socket = socket;
-
-  if(this.socket.setNoDelay)
-    this.socket.setNoDelay(true);
-
-  this.socket.on('connect', () => this.emit('connect'));
-
-  this.socket.on('error', onFatalError);
-  this.socket.on('close', endSocket);
-  this.socket.on('end', endSocket);
-  this.socket.on('timeout', endSocket);
-  this.serializer.on('error', onError);
-  this.deserializer.on('error', onError);
-  this.framer.on('error', onError);
-  this.splitter.on('error', onError);
-
-  this.socket.pipe(this.splitter).pipe(this.deserializer);
-  this.serializer.pipe(this.framer).pipe(this.socket);
-
-  this.deserializer.on('data', (parsed) => {
-    this.emit('packet', parsed.data, parsed.metadata);
-    this.emit(parsed.metadata.name, parsed.data, parsed.metadata);
-    this.emit('raw.' + parsed.metadata.name, parsed.buffer, parsed.metadata);
-    this.emit('raw', parsed.buffer, parsed.metadata);
-  });
-};
-
-Client.prototype.end = function(reason) {
-  this._endReason = reason;
-  this.socket.end();
-};
-
-Client.prototype.setEncryption = function(sharedSecret) {
-  if (this.cipher != null)
-    throw new Error("Set encryption twice !");
-  this.cipher = crypto.createCipheriv('aes-128-cfb8', sharedSecret, sharedSecret);
-  this.cipher.on('error', (err) => this.emit('error', err));
-  this.framer.unpipe(this.socket);
-  this.framer.pipe(this.cipher).pipe(this.socket);
-  this.decipher = crypto.createDecipheriv('aes-128-cfb8', sharedSecret, sharedSecret);
-  this.decipher.on('error', (err) => this.emit('error', err));
-  this.socket.unpipe(this.splitter);
-  this.socket.pipe(this.decipher).pipe(this.splitter);
-}
-
-Client.prototype.setCompressionThreshold = function(threshold) {
-  if (this.compressor == null) {
-    this.compressor = compression.createCompressor(threshold);
-    this.compressor.on('error', (err) => this.emit('error', err));
-    this.serializer.unpipe(this.framer);
-    this.serializer.pipe(this.compressor).pipe(this.framer);
-    this.decompressor = compression.createDecompressor(threshold);
-    this.decompressor.on('error', (err) => this.emit('error', err));
-    this.splitter.unpipe(this.deserializer);
-    this.splitter.pipe(this.decompressor).pipe(this.deserializer);
-  } else {
-    this.decompressor.threshold = threshold;
-    this.compressor.threshold = threshold;
+  writeRaw(buffer) {
+    if (this.compressor === null)
+      this.framer.write(buffer);
+    else
+      this.compressor.write(buffer);
   }
 }
 
-Client.prototype.write = function(packetName, params) {
-  debug("writing packet " + this.state + "." + packetName);
-  debug(params);
-  this.serializer.write({ packetName, params });
-};
-
-Client.prototype.writeRaw = function(buffer) {
-  if (this.compressor === null)
-    this.framer.write(buffer);
-  else
-    this.compressor.write(buffer);
-};
+module.exports = Client;