☰

Aria2 RPC Edit (use GM_xmlhttpRequest)

Aria2 RPC Library 修改自 https://greasyfork.org/scripts/5672-aria2-rpc

目前为 2020-05-05 提交的版本。查看 最新版本。

此脚本不应直接安装，它是一个供其他脚本使用的外部库。如果您需要使用该库，请在脚本元属性加入：// @require https://update.cn-greasyfork.org/scripts/402652/801594/Aria2%20RPC%20Edit%20%28use%20GM_xmlhttpRequest%29.js 

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