webui-aria2/js/services/rpc/sockcall.js

angular
  .module("webui.services.rpc.sockcall", [
    "webui.services.deps",
    "webui.services.utils",
    "webui.services.base64",
    "webui.services.alerts"
  ])
  .factory("$sockcall", [
    "$_",
    "$json",
    "$name",
    "$utils",
    "$alerts",
    function(_, JSON, name, utils, alerts) {
      var sockRPC = {
        // true when sockrpc is ready to be used,
        // false when either initializing
        // or no support for web sockets
        initialized: false,

        // ongoing connection handles containing connection id and callbacks
        handles: [],

        // websocket connection socket used for all connections
        sock: null,

        // connection configuration
        conf: null,

        // socket connection scheme, default to unencrypted connection
        scheme: "ws",

        // called when a connection error occurs
        onerror: function(ev) {
          _.each(sockRPC.handles, function(h) {
            h.error();
          });
          sockRPC.handles = [];
          sockRPC.initialized = false;
          if (sockRPC.onready) {
            sockRPC.onready();
            sockRPC.onready = null;
          }
        },
        onclose: function(ev) {
          if (sockRPC.handles && sockRPC.handles.length)
            sockRPC.onerror("Connection reset while calling aria2");
          sockRPC.initialized = false;
          if (sockRPC.onready) {
            sockRPC.onready();
            sockRPC.onready = null;
          }
        },

        // when connection opens
        onopen: function() {
          console.log("websocket initialized!!!");
          sockRPC.initialized = true;
          if (sockRPC.onready) {
            sockRPC.onready();
            sockRPC.onready = null;
          }
        },

        // when message is recieved
        onmessage: function(message) {
          var data = JSON.parse(message.data);

          // reverse loop because we are deleting elements
          // while looping over the old items
          for (var i = sockRPC.handles.length - 1; i >= 0; i--) {
            if (sockRPC.handles[i].id === data.id) {
              sockRPC.handles[i].success(data);
              sockRPC.handles.splice(i, 1);
              return;
            }
          }
        },

        // call to use the rpc
        invoke: function(opts) {
          var data = {
            jsonrpc: 2.0,
            id: utils.uuid(),
            method: opts.name,
            params: opts.params && opts.params.length ? opts.params : undefined
          };

          if (data.params && !data.params.length) data.params = undefined;

          sockRPC.handles.push({
            success: opts.success || angular.noop,
            error: opts.error || angular.noop,
            id: data.id
          });
          sockRPC.sock.send(JSON.stringify(data));
        },

        // should be called initially to start using the sock rpc
        // onready is called when initial connection is resolved
        init: function(conf, onready) {
          sockRPC.initialized = false;
          if (sockRPC.onready) {
            // make previous call is resolved
            sockRPC.onready();
            sockRPC.onready = null;
          }

          if (typeof WebSocket == "undefined") {
            alerts.addAlert("Web sockets are not supported! Falling back to JSONP.", "info");
            onready();
            return;
          }
          sockRPC.conf = conf || sockRPC.conf;

          sockRPC.scheme = sockRPC.conf.encrypt ? "wss" : "ws";

          if (sockRPC.sock) {
            sockRPC.sock.onopen = sockRPC.sock.onmessage = sockRPC.sock.onerror = sockRPC.sock.onclose = null;
            sockRPC.onerror({ message: "Changing the websocket aria2 server details" });
          }

          try {
            var authUrl =
              sockRPC.scheme + "://" + conf.host + ":" + conf.port + (conf.path || "/jsonrpc");
            if (sockRPC.conf.auth && sockRPC.conf.auth.user && sockRPC.conf.auth.pass) {
              authUrl =
                sockRPC.scheme +
                "://" +
                sockRPC.conf.auth.user +
                ":" +
                sockRPC.conf.auth.pass +
                "@" +
                sockRPC.conf.host +
                ":" +
                sockRPC.conf.port +
                (conf.path || "/jsonrpc");
            }

            sockRPC.sock = new WebSocket(authUrl);
            sockRPC.sock.onopen = sockRPC.onopen;
            sockRPC.sock.onclose = sockRPC.onclose;
            sockRPC.sock.onerror = sockRPC.onerror;
            sockRPC.sock.onmessage = sockRPC.onmessage;
            sockRPC.onready = onready;
          } catch (ex) {
            // ignoring IE security exception on local ip addresses
            console.log("not using websocket for aria2 rpc due to: ", ex);
            alerts.addAlert("Web sockets not working due to " + ex.message, "info");
            onready();
          }
        }
      };

      return sockRPC;
    }
  ]);