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

116 lines
3.5 KiB
JavaScript

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;
alerts.log('Error while talking to aria2 over websocket');
},
onclose: function(ev) {
if (sockRPC.handles && sockRPC.handles.length)
sockRPC.onerror('Connection reset while pending calls to aria2');
sockRPC.initialized = false;
},
// when connection opens
onopen: function() {
alerts.addAlert('Connected to aria2 successfully over websocket!', 'success');
sockRPC.initialized = true;
},
// 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: name + '_' + utils.randStr(),
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
init: function(conf) {
sockRPC.initialized = false;
if (typeof WebSocket == "undefined") {
alerts.addAlert('Web sockets not supported, falling back to jsonp', 'info');
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 {
sockRPC.sock = new WebSocket(sockRPC.scheme + '://' + conf.host + ':' + conf.port + '/jsonrpc');
sockRPC.sock.onopen = sockRPC.onopen;
sockRPC.sock.onclose = sockRPC.onclose;
sockRPC.sock.onerror = sockRPC.onerror;
sockRPC.sock.onmessage = sockRPC.onmessage;
}
catch (ex) {
// ignoring IE securty 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');
}
},
};
return sockRPC;
}]);