rust-rcon/js/rconService.js

function RconService() {

  var ConnectionStatus = {
    'CONNECTING': 0,
    'OPEN': 1,
    'CLOSING': 2,
    'CLOSED': 3
  };

  var Service = {
    Socket: null,
    Address: null,
    Callbacks: {}
  };

  var LastIndex = 1001;

  Service.Connect = function(addr, pass) {
    this.Socket = new WebSocket("ws://" + addr + "/" + pass);
    this.Address = addr;

    this.Socket.onmessage = function(e) {
      var data = angular.fromJson(e.data);

      //
      // This is a targetted message, it has an identifier
      // So feed it back to the right callback.
      //
      if (data.Identifier > 1000) {
        var cb = Service.Callbacks[data.Identifier];
        if (cb != null) {
          cb.scope.$apply(function() {
            cb.callback(data);
          });
        }
        Service.Callbacks[data.Identifier] = null;

        return;
      }

      //
      // Generic console message, let OnMessage catch it
      //
      if (Service.OnMessage != null) {
        Service.OnMessage(data);
      }
    };

    this.Socket.onopen = this.OnOpen;
    this.Socket.onclose = this.OnClose;
    this.Socket.onerror = this.OnError;
  }

  Service.Disconnect = function() {
    if (this.Socket) {
      this.Socket.close();
      this.Socket = null;
    }

    this.Callbacks = {};
  }

  Service.Command = function(msg, identifier) {
    if (this.Socket === null)
      return;

    if (!this.IsConnected())
      return;

    if (identifier === null)
      identifier = -1;

    var packet = {
      Identifier: identifier,
      Message: msg,
      Name: "WebRcon"
    };

    this.Socket.send(JSON.stringify(packet));
  };

  //
  // Make a request, call this function when it returns
  //
  Service.Request = function(msg, scope, callback) {
    LastIndex++;
    this.Callbacks[LastIndex] = {
      scope: scope,
      callback: callback
    };
    Service.Command(msg, LastIndex);
  }

  //
  // Returns true if websocket is connected
  //
  Service.IsConnected = function() {
    if (this.Socket == null)
      return false;

    return this.Socket.readyState === ConnectionStatus.OPEN;
  }

  //
  // Helper for installing connectivity logic
  //
  // Basically if not connected, call this function when we are
  // And if we are - then call it right now.
  //
  Service.InstallService = function(scope, func) {
    scope.$on("OnConnected", function() {
      func();
    });

    if (this.IsConnected()) {
      func();
    }
  }

  Service.getPlayers = function(scope, success) {
    this.Request("playerlist", scope, function(response) {
      var players = JSON.parse(response.Message);

      if (typeof success === 'function') {
        success.call(scope, players);
      }
    });
  }

  return Service;
}