server.WebSocketClient.js

var ChatClientBase = require('./server.ChatClientBase.js');
var util = require("util");
var ws = require("ws");

function WebSocketClient(socket) {
    ChatClientBase.call(this);
    this.socket = socket;
}

util.inherits(WebSocketClient, ChatClientBase);

WebSocketClient.prototype.getId = function () {
    return this.socket;
};

WebSocketClient.prototype.getIpAddress = function () {
    var ip = this.socket.upgradeReq.connection.remoteAddress;
    if (ip == "127.0.0.1" || ip == "::1") // only use forward ip when seeing local ips
        ip = this.socket.upgradeReq.headers['x-forwarded-for'] || ip; // fallback to local ip if not set
    return ip;
};

WebSocketClient.prototype.close = function () {
    this.socket.close();
};

WebSocketClient.prototype.send = function (causingClient, data) {
    data.time = Date.now(); // Add timestamp to command
    try {
        if (this.socket.readyState == ws.OPEN) {
            this.socket.send(JSON.stringify(data));
        }
    } catch (e) {
        console.error(e);
    }
};

module.exports = WebSocketClient;