☰

Aria2 RPC

Aria2 RPC Library

This script should not be not be installed directly. It is a library for other scripts to include with the meta directive // @require https://update.greatest.deepsurf.us/scripts/5672/933444/Aria2%20RPC.js 

  1. // ==UserScript==
  2. // @name         Aria2 RPC
  3. // @namespace    moe.jixun
  4. // @version      0.1
  5. // @license      MIT
  6. // ==/UserScript==
  7.  
  8. // Public Class Aria2 ( options )
  9.  
  10. var Aria2 = (function (_isGM, _arrFn, _merge, _format, _isFunction) {
  11. 	var jsonrpc_ver = '2.0';
  12.  
  13. 	if (_isGM) {
  14. 		var doRequest = function ( opts ) {
  15. 			console.warn ([
  16. 				'Warning: You are now using an simple implementation of GM_xmlhttpRequest',
  17. 				'Cross-domain request are not avilible unless configured correctly @ target server.',
  18. 				'',
  19. 				'Some of its features are not avilible, such as `username` and `password` field.'
  20. 			].join('\n'));
  21.  
  22. 			var oReq = new XMLHttpRequest ();
  23. 			var cbCommon = function (cb) {
  24. 				return (function () {
  25. 					cb ({
  26. 						readyState: oReq.readyState,
  27. 						responseHeaders: opts.getHeader ? oReq.getAllResponseHeaders() : null,
  28. 						getHeader: oReq.getResponseHeader.bind (oReq),
  29. 						responseText: oReq.responseText,
  30. 						status: oReq.status,
  31. 						statusText: oReq.statusText
  32. 					});
  33. 				}).bind (opts);
  34. 			};
  35.  
  36. 			if (opts.onload)  oReq.onload   = cbCommon (opts.onload);
  37. 			if (opts.onerror) oReq.onerror  = cbCommon (opts.onerror);
  38.  
  39. 			oReq.open(opts.method || 'GET', opts.url, !opts.synchronous);
  40.  
  41. 			if (opts.headers) {
  42. 				Object.keys(opts.headers).forEach (function (key) {
  43. 					oReq.setRequestHeader (key, opts.headers[key]);
  44. 				});
  45. 			}
  46. 			return oReq.send(opts.data || null);
  47. 		};
  48. 	} else {
  49. 		var doRequest = GM_xmlhttpRequest;
  50. 	}
  51.  
  52. 	var AriaBase = function ( options ) {
  53. 		this.options = _merge ({
  54. 			auth: {
  55. 				type: AriaBase.AUTH.noAuth,
  56. 				user: '',
  57. 				pass: ''
  58. 			},
  59. 			host: '127.0.0.1',
  60. 			port: 6800
  61. 		}, options || {});
  62.  
  63. 		this.id = parseInt (options, 10) || (+ new Date());
  64. 	};
  65. 	
  66. 	// 静态常量
  67. 	AriaBase.AUTH = {
  68. 		noAuth: 0,
  69. 		basic:  1,
  70. 		secret: 2
  71. 	};
  72.  
  73. 	// public 函数
  74. 	AriaBase.prototype = {
  75. 		getBasicAuth: function () {
  76. 			return btoa (_format('%s:%s', this.options.auth.user, this.options.auth.pass));
  77. 		},
  78.  
  79. 		send: function ( bIsDataBatch, data, cbSuccess, cbError ) {
  80. 			var srcTaskObj = { jsonrpc: jsonrpc_ver, id: this.id };
  81.  
  82. 			var payload = {
  83.  
  84. 				method: 'POST',
  85. 				url: _format('http://%s:%s/jsonrpc', this.options.host, this.options.port),
  86. 				headers: {
  87. 					'Content-Type': 'application/json; charset=UTF-8'
  88. 				},
  89. 				data: bIsDataBatch
  90. 					? data.map (function (e) { return _merge ({}, srcTaskObj, e); })
  91. 					: _merge ({}, srcTaskObj, data),
  92. 				onload: function (r) {
  93. 					var repData = JSON.parse (r.responseText);
  94. 					if (repData.error) {
  95. 						cbError && cbError (false, repData);
  96. 					} else {
  97. 						cbSuccess && cbSuccess (repData);
  98. 					}
  99. 				},
  100. 				onerror: cbError ? cbError.bind(null, false) : null
  101. 			};
  102.  
  103. 			switch ( parseInt (this.options.auth.type, 10) ) {
  104. 				case AriaBase.AUTH.noAuth:
  105. 					// DO NOTHING
  106. 					break;
  107.  
  108. 				case AriaBase.AUTH.basic:
  109. 					payload.headers.Authorization = 'Basic ' + this.getBasicAuth();
  110. 					break;
  111.  
  112. 				case AriaBase.AUTH.secret:
  113. 					(function (sToken) {
  114. 						if (bIsDataBatch) {
  115. 							for (var i = 0; i < payload.data.length; i++) {
  116. 								payload.data[i].params.splice(0, 0, sToken);
  117. 							}
  118. 						} else {
  119. 							if (!payload.data.params)
  120. 								payload.data.params = [];
  121. 							payload.data.params.splice(0, 0, sToken);
  122. 						}
  123. 					})(_format('token:%s', this.options.auth.pass));
  124. 					break;
  125.  
  126. 				default:
  127. 					throw new Error('Undefined auth type: ' + this.options.auth.type);
  128. 			}
  129.  
  130. 			payload.data = JSON.stringify ( payload.data );
  131.  
  132. 			return doRequest (payload);
  133. 		},
  134.  
  135. 		// batchAddUri ( foo, { uri: 'http://example.com/xxx', options: { ... } } )
  136. 		batchAddUri: function (fCallback) {
  137. 			console.warn (
  138. 				'This function [%s] has deprecated! Consider use %s instead.',
  139. 				'batchAddUri', 'AriaBase.BATCH'
  140. 			);
  141.  
  142. 			// { url, name }
  143. 			var payload = [].slice.call (arguments, 1).map (function (arg) {
  144. 				return {
  145. 					method: 'aria2.addUri',
  146. 					params: [ arg.uri.map ? arg.uri : [ arg.uri ] ].concat (arg.options || [])
  147. 				};
  148. 			});
  149. 			
  150. 			return this.send (true, payload, fCallback, fCallback);
  151. 		}
  152. 	};
  153.  
  154.  
  155. 	// 添加各类函数
  156. 	AriaBase.fn = {};
  157. 	_arrFn.forEach (function (sMethod) {
  158. 		// 函数链表
  159. 		AriaBase.fn[sMethod] = sMethod;
  160.  
  161. 		// arg1, arg2, ... , [cbSuccess, [cbError]]
  162. 		AriaBase.prototype[sMethod] = function ( ) {
  163. 			var args = [].slice.call (arguments);
  164.  
  165. 			var cbSuccess, cbError;
  166. 			if (args.length && _isFunction(args[args.length - 1])) {
  167. 				cbSuccess = args[args.length - 1];
  168. 				args.splice (-1, 1);
  169.  
  170. 				if (args.length && _isFunction(args[args.length - 1])) {
  171. 					cbError = cbSuccess;
  172. 					cbSuccess = args[args.length - 1];
  173. 					args.splice (-1, 1);
  174. 				}
  175. 			}
  176.  
  177. 			return this.send (false, {
  178. 				method: 'aria2.' + sMethod,
  179. 				params: args
  180. 			}, cbSuccess, cbError);
  181. 		};
  182. 	});
  183.  
  184. 	AriaBase.BATCH = function ( parent, cbSuccess, cbFail ) {
  185. 		if (!(parent instanceof AriaBase))
  186. 			throw new Error ('Parent is not AriaBase!');
  187.  
  188. 		this.parent = parent;
  189. 		this.data = [];
  190.  
  191. 		this.onSuccess = cbSuccess;
  192. 		this.onFail = cbFail;
  193. 	};
  194.  
  195. 	AriaBase.BATCH.prototype = {
  196. 		addRaw: function (fn, args) {
  197. 			this.data.push ({
  198. 				method: 'aria2.' + fn,
  199. 				params: args
  200. 			});
  201. 			return this;
  202. 		},
  203.  
  204. 		add: function (fn) {
  205. 			// People can add more without edit source.
  206. 			if (!AriaBase.fn[fn])
  207. 				throw new Error ('Unknown function: ' + fn + ', please check if you had a typo.');
  208.  
  209. 			return this.addRaw (fn, [].slice.call(arguments, 1));
  210. 		},
  211.  
  212. 		send: function () {
  213. 			// bIsDataBatch, data, cbSuccess, cbError
  214. 			var ret = this.parent.send ( true, this.data, this.onSuccess, this.onFail );
  215. 			this.reset ();
  216. 			return ret;
  217. 		},
  218.  
  219. 		getActions: function () {
  220. 			return this.data.slice();
  221. 		},
  222.  
  223. 		setActions: function (actions) {
  224. 			if (!actions || !actions.map) return ;
  225.  
  226. 			this.data = actions;
  227. 		},
  228.  
  229. 		reset: function () {
  230. 			this.onSuccess = this.onFail = null;
  231. 			this.setActions ( [] );
  232. 		}
  233. 	};
  234. 	
  235. 	return AriaBase;
  236. })
  237. // const 变量
  238. ('undefined' == typeof GM_xmlhttpRequest, [
  239. 	"addUri", "addTorrent", "addMetalink", "remove", "forceRemove",
  240. 	"pause", "pauseAll", "forcePause", "forcePauseAll", "unpause",
  241. 	"unpauseAll", "tellStatus", "getUris", "getFiles", "getPeers",
  242. 	"getServers", "tellActive", "tellWaiting", "tellStopped",
  243. 	"changePosition", "changeUri", "getOption", "changeOption",
  244. 	"getGlobalOption", "changeGlobalOption", "getGlobalStat",
  245. 	"purgeDownloadResult", "removeDownloadResult", "getVersion",
  246. 	"getSessionInfo", "shutdown", "forceShutdown", "saveSession"
  247. ], 
  248. // private 函数
  249. (function (base) {
  250. 	var _isObject = function (obj) {
  251. 		return obj instanceof Object;
  252. 	};
  253. 	var _merge = function (base) {
  254. 		var args = arguments,
  255. 			argL = args.length;
  256. 		for ( var i = 1; i < argL; i++ ) {
  257. 			Object.keys (args[i]).forEach (function (key) {
  258. 				if (_isObject(args[i][key]) && _isObject(base[key])) {
  259. 					base[key] = _merge (base[key], args[i][key]);
  260. 				} else {
  261. 					base[key] = args[i][key];
  262. 				}
  263. 			});
  264. 		}
  265. 		return base;
  266. 	};
  267. 	return _merge;
  268. })(), function (src) {
  269. 	var args = arguments,
  270. 		argL = args.length;
  271.  
  272. 	var ret = src.slice ();
  273. 	for ( var i = 1; i < argL; i++ )
  274. 		ret = ret.replace ('%s', args[i]);
  275. 	return ret;
  276. }, function (foo) {
  277. 	return typeof foo === 'function'
  278. });