☰

tehConnection Snatch Script

Mark snatched torrents for tc

  1. // ==UserScript==
  2. // @name           tehConnection Snatch Script
  3. // @namespace      http://tehconnection.eu
  4. // @description    Mark snatched torrents for tc
  5. // @author         jonls on what.cd  (adapted for TC by Dafreak)
  6. // @include        http://tehconnection.eu/*
  7. // @include        https://tehconnection.eu/*
  8. // @require        http://ajax.googleapis.com/ajax/libs/jquery/1.3/jquery.min.js
  9. // @grant           GM_getValue
  10. // @grant           GM_setValue
  11. // @grant           GM_deleteValue
  12. // @grant           GM_xmlhttpRequest
  13. // @grant           GM_registerMenuCommand
  14. // @grant           GM_addStyle
  15. // @grant           GM_log
  16. // @version        0.2
  17. // @date           2010-10-27
  18. // ==/UserScript==
  19.  
  20. (function() {
  21. 	var DEFAULT_STYLE =
  22. 		'.group_snatched { color: #E5B244; } '+
  23. 		'.snatched { color: #35BF00; } '+
  24. 		'.uploaded { color: red; } '+
  25. 		'.leeching { color: #2B75A1; }'+
  26. 		'.seeding { border-bottom: 1px dotted red; }';
  27. 	
  28. 	var AUTO_UPDATE_INTERVAL = 10*60; /* seconds */
  29.  
  30. 	/* Throttled proxy */
  31. 	function Proxy(url_base, delay) {
  32. 		var last_req = new Date(0);
  33. 		var queue = [];
  34. 		var processing = false;
  35.  
  36. 		return {
  37. 			get: function(req) {
  38. 				var now = new Date();
  39. 				queue.push(req);
  40. 				if (!processing) {
  41. 					/* Race condition: atomic test and set would be appropriate here, to ensure thread safety (is it a problem?) */
  42. 					processing = true;
  43. 					var diff = last_req.getTime() + delay - now.getTime();
  44. 					if (diff > 0) {
  45. 						var that = this;
  46. 						window.setTimeout(function() { that.process_queue(); }, diff);
  47. 					} else {
  48. 						this.process_queue();
  49. 					}
  50. 				}
  51. 			},
  52.  
  53. 			process_queue: function() {
  54. 				var req = queue.shift();
  55. 				this.do_request(req);
  56. 				processing = (queue.length > 0);
  57. 				if (processing) {
  58. 					var that = this;
  59. 					window.setTimeout(function() { that.process_queue(); }, delay);
  60. 				}
  61. 			},
  62.  
  63. 			do_request: function(req) {
  64. 				last_req = new Date();
  65. 				var timer;
  66. 				var req_timed_out = false; /* We cannot abort a request, so we need keep track of whether it timed out */
  67.  
  68. 				/* Create timeout handler */
  69. 				timer = window.setTimeout(function() {
  70. 					/* Race condition: what if the request returns successfully now? */
  71. 					req_timed_out = true;
  72. 					if (req.error) req.error(null, 'Network timeout');
  73. 				}, req.timeout || 10000);
  74.  
  75. 				/* Do the actual request */
  76. 				GM_xmlhttpRequest({
  77. 					method: req.method || 'GET',
  78. 					url: url_base+req.url,
  79. 					headers: { 'User-Agent': navigator.userAgent, 'Accept': req.accept || 'text/xml' },
  80. 					onload: function(response) {
  81. 						window.clearTimeout(timer);
  82. 						if (!req_timed_out) req.callback(response);
  83. 					},
  84. 					onerror: function(response) {
  85. 						window.clearTimeout(timer);
  86. 						if (!req_timed_out && req.error) req.error(response, 'GM_xmlhttpRequest error');
  87. 					}
  88. 				});
  89. 			}
  90. 		};
  91. 	}
  92.  
  93. 	
  94. 	/* Simple rounding (extracted from jQuery Corner)*/
  95. 	$.fn.round = function(radius) {
  96. 		radius = radius || "10px";
  97. 		return this.each(function(i) {
  98. 			if ($.browser.mozilla && /gecko/i.test(navigator.userAgent)) {
  99. 				$(this).css('-moz-border-radius', radius);
  100. 			} else if ($.browser.safari && $.browser.version >= 3) {
  101. 				$(this).css('-webkit-border-radius', radius);
  102. 			}
  103. 		});
  104. 	};
  105. 	
  106. 	/* Global status area - feel free to reuse in your own scripts :)
  107. 	   Requires jQuery and the round extension above. */
  108. 	function StatusBox(title) {
  109. 		/* Setup status area */
  110. 		var status_area = $('#greasemonkey_status_area').eq(0);
  111. 		if (status_area.length == 0) {
  112. 			status_area = $('<div id="greasemonkey_status_area"></div>').css({
  113. 				'position': 'fixed',
  114. 				'top': '0', 'right': '0',
  115. 				'margin': '20px',
  116. 				'width': '20%',
  117. 				'z-index': 500
  118. 			});
  119. 			$('body').append(status_area);
  120. 		}
  121.  
  122. 		/* Create box */
  123. 		var box = $('<div></div>').hide();
  124. 		box.css({
  125. 			'color': 'white',
  126. 			'background-color': 'black',
  127. 			'opacity': 0.5,
  128. 			'margin': '0 0 10px 0',
  129. 			'padding': '10px 10px 20px 10px'}).round();
  130. 		box.append($('<div>'+title+'</div>').css('font-weight', 'bold'));
  131.  
  132. 		/* Create contents area */
  133. 		var contents = $('<div></div>');
  134. 		box.append(contents);
  135.  
  136. 		var timer = null;
  137. 		var timeout = 0;
  138. 		var inhibit_fade = false;
  139.  
  140. 		function set_visible(visible) {
  141. 			if (visible && box.is(':hidden')) box.fadeIn(500);				
  142. 			else if (!visible && box.is(':visible')) box.fadeOut(500);
  143. 		}
  144.  
  145. 		function clear_timer() {
  146. 			if (timer) {
  147. 				window.clearTimeout(timer);
  148. 				timer = null;
  149. 			}
  150. 		}
  151.  
  152. 		function set_timer() {
  153. 			if (!timer && timeout > 0) {
  154. 				timer = window.setTimeout(function() { clear_timer(); set_visible(false); }, timeout);
  155. 			}			
  156. 		}
  157.  
  158. 		function update_timer(t) {
  159. 			clear_timer();
  160. 			timeout = t;
  161. 			if (!inhibit_fade) set_timer();
  162. 		}
  163.  
  164. 		function set_inhibit_fade(inhibit) {
  165. 			inhibit_fade = inhibit;
  166. 			if (!inhibit_fade) { set_timer(); }
  167. 			else clear_timer();
  168. 		}
  169.  
  170. 		/* Register event handlers */
  171. 		box.mouseenter(function(event) {
  172. 			set_inhibit_fade(true);
  173. 			$(this).fadeTo(500, 0.8);
  174. 		});
  175.  
  176. 		box.mouseleave(function(event) {
  177. 			set_inhibit_fade(false);
  178. 			$(this).fadeTo(500, 0.5);
  179. 		});
  180.  
  181. 		box.click(function(event) {
  182. 			clear_timer();
  183. 			set_visible(false);
  184. 		});
  185.  
  186. 		/* Append to global status area */
  187. 		status_area.append(box);
  188.  
  189. 		return {
  190. 			contents: function() {
  191. 				return contents;
  192. 			},
  193.  
  194. 			show: function(t) {
  195. 				t = t || 0;
  196. 				update_timer(t);
  197. 				set_visible(true);
  198. 			},
  199.  
  200. 			hide: function() {
  201. 				clear_timer();
  202. 				set_visible(false);
  203. 			}
  204. 		};
  205. 	}
  206.  
  207. 	/* Cache */
  208. 	function Cache(name, def_value) {
  209. 		var cache;
  210. 		return {
  211. 			serialize: function() {
  212. 				GM_setValue(name, uneval(cache));
  213. 			},
  214. 			unserialize: function() {
  215. 				cache = eval(GM_getValue(name, 'false'));
  216. 				if (!cache) cache = $.extend({}, def_value); /* clone */
  217. 				return cache;
  218. 			},
  219. 			clear: function() {
  220. 				cache = $.extend({}, def_value); /* clone */
  221. 				this.serialize();
  222. 			}
  223. 		};
  224. 	}
  225.  
  226. 	
  227. 	/* Get what.CD base URL */
  228. 	var whatcd_url_base = document.URL.match(/^(https:\/\/tehconnection\.eu|http:\/\/tehconnection\.eu)/)[1];
  229.  
  230. 	/* Create proxy */
  231. 	var whatcd_proxy = Proxy(whatcd_url_base, 1000);
  232.  
  233. 	/* Get user id of this user */
  234. 	var whatcd_id = (function() {
  235. 		var m = $('#header .username').eq(0).attr('href').match(/user\.php\?id=(\d+)/);
  236. 		if (m) return m[1];
  237. 		return null;
  238. 	})();
  239.  
  240. 	if (!whatcd_id) return; /* Exceptional condition: User ID not found */
  241.  
  242. 	/* Create status box */
  243. 	var status = StatusBox('TehConnection.Eu Snatched');
  244.  
  245. 	/* Cache of snatched torrents */
  246. 	var snatch_cache = Cache('snatch_cache', { groups: {}, torrents: {} });
  247.  
  248. 	/* Scan torrent table in doc and mark links as type in cache */
  249. 	function scan_torrent_page(doc, type) {
  250. 		var torrent_table = doc.find('.torrent_table').eq(0);
  251. 		if (torrent_table.length == 0){ 	
  252. 			return 0;
  253. 		}
  254.  
  255. 		var found = 0;
  256.  
  257. 		var d = snatch_cache.unserialize();
  258. 		torrent_table.find('tr').not('.colhead').each(function(i) {
  259. 			/* Find group and torrent ID */
  260. 			var group_id;
  261. 			var torrent_id;
  262. 			var link = $(this).children('td').eq(1).children('a:last').eq(0);
  263. 			var m = link.attr('href').match(/torrents\.php\?id=(\d+)&torrentid=(\d+)/);
  264. 			if (m) {
  265. 				group_id = m[1];
  266. 				torrent_id = m[2];
  267. 			} else {
  268. 				m = link.attr('href').match(/torrents\.php\?id=(\d+)/);
  269. 				if (m) {
  270. 					group_id = m[1];
  271. 					link = $(this).children('td').eq(1).children('span').eq(0).children('a').eq(0);
  272. 					m = link.attr('href').match(/torrents\.php\?action=download&id=(\d+)/);
  273. 					if (m) torrent_id = m[1];
  274. 				}
  275.  
  276. 				if (!m) {
  277. 					status.contents().append('<div><span style="color: red;">Failed:</span> '+$(this).children('td').eq(1).text()+'</div>');
  278. 					status.show();
  279. 				}
  280. 			}
  281.  
  282. 			/* Save in cache */
  283. 			if (group_id && torrent_id) {
  284. 				if (!d.torrents[torrent_id] ||
  285. 						(type != 'seeding' && d.torrents[torrent_id].type != type) ||
  286. 						(type == 'seeding' && !d.torrents[torrent_id].seeding)) {
  287. 					var name = $.trim($(this).children('td').eq(1).clone().children('span, div').remove().end().text().match(/\s+([^[]+)(\s+\[|$)/)[1]);
  288. 					d.groups[group_id] = { name: name };
  289. 					if (type == 'seeding') { /* Special case seeding */
  290. 						if (d.torrents[torrent_id]) d.torrents[torrent_id].seeding = true;
  291. 						else d.torrents[torrent_id] = { type: 'unknown', seeding: true };
  292. 					} else {
  293. 						if (d.torrents[torrent_id]) d.torrents[torrent_id].type = type;
  294. 						else d.torrents[torrent_id] = { type: type, seeding: false };
  295. 					}
  296. 					found += 1;
  297. 				}
  298. 			}
  299. 		});
  300.  
  301. 		if (found == 0) return 0;
  302.  
  303. 		snatch_cache.serialize();
  304. 		return found;
  305. 	}
  306.  
  307. 	/* Fetch and scan all pages of type, call callback when done */
  308. 	function scan_all_torrent_pages(type, page_cb, finish_cb) {
  309. 		var page = 1;
  310. 		var total = 0;
  311.  
  312. 		function request_url() {
  313. 			return '/torrents.php?type='+type+'&userid='+whatcd_id+'&page='+page;
  314. 		}
  315.  
  316. 		function error_handler(response, reason) {
  317. 			alert("Error in loading snatches");
  318. 			status.contents().append('<div><span style="color: red;">Error:</span> Unable to fetch '+type+' page '+page+' ('+reason+')</div>');
  319. 			status.show();
  320. 			finish_cb(total);
  321. 		}
  322.  
  323. 		function page_handler(response) {
  324. 			if (response.status == 200) {
  325. 				
  326. 				var doc = $(new DOMParser().parseFromString(response.responseText, 'text/xml'));
  327. 				
  328. 				page_cb(type, page);
  329. 				
  330. 				var found = scan_torrent_page(doc, type);
  331.  
  332. 				total += found;
  333. 				if (found == 0) { finish_cb(type, total); return; } /* End of asynchronous chain */
  334.  
  335. 				page += 1;
  336. 				whatcd_proxy.get({ url: request_url(), callback: page_handler, error: error_handler });
  337. 			} else {
  338. 				error_handler(response, 'HTTP '+response.status);
  339. 			}
  340. 		}
  341.  
  342. 		whatcd_proxy.get({ url: request_url(), callback: page_handler, error: error_handler });
  343. 	}
  344.  
  345. 	/* Reset command */
  346. 	GM_registerMenuCommand('TehConnection.Eu Snatched: Reset', function() { snatch_cache.clear(); GM_setValue('last_update', '0'); location.reload(); });
  347.  
  348. 	/* Register menu command to enter custom style */
  349. 	var custom_style = GM_getValue('custom_style', DEFAULT_STYLE);
  350. 	GM_registerMenuCommand('TehConnection.Eu Snatched: Enter custom style...', function() {
  351. 		var style = window.prompt('Enter CSS style (or blank to use default)\nClasses: .group_snatched, .snatched, .uploaded, .leeching, .seeding', custom_style);
  352. 		if (style) {
  353. 			GM_setValue('custom_style', style);
  354. 			location.reload();
  355. 		} else if (style == '') {
  356. 			GM_deleteValue('custom_style');
  357. 			location.reload();
  358. 		}
  359. 	});
  360.  
  361. 	/* Inject CSS style */
  362. 	GM_addStyle(custom_style);
  363.  
  364. 	/* Mark all links to torrents that are snatched/uploaded/leeching */
  365. 	function mark_snatched_links() {
  366. 		var d = snatch_cache.unserialize();
  367.  
  368. 		/* Go through all links */
  369. 		$('#content').find('a').each(function(i) {
  370. 			var href = $(this).attr('href');
  371. 			if (href) {
  372. 				var group_id;
  373. 				var torrent_id;
  374.  
  375. 				/* Find and mark links to snatched torrents */
  376. 				var m = href.match(/torrents\.php\?id=(\d+)&torrentid=(\d+)/);
  377. 				if (m) {
  378. 					group_id = m[1];
  379. 					torrent_id = m[2];
  380. 				} else {
  381. 					m = href.match(/torrents\.php\?torrentid=(\d+)/);
  382. 					if (m) {
  383. 						torrent_id = m[1];
  384. 					} else {
  385. 						m = href.match(/torrents\.php\?id=(\d+)/);
  386. 						if (m) group_id = m[1];
  387. 					}
  388. 				}
  389.  
  390. 				/* Add classes */
  391. 				if (group_id && d.groups[group_id] &&
  392. 						(!torrent_id || !$(this).parent().parent().is('.group_torrent'))) {
  393. 					$(this).addClass('group_snatched');
  394. 				}
  395. 				if (torrent_id && d.torrents[torrent_id]) {
  396. 					$(this).addClass(d.torrents[torrent_id].type);
  397. 					if (d.torrents[torrent_id].seeding) $(this).addClass('seeding');
  398. 				}
  399.  
  400. 				/* Change text if text is url */
  401. 				if ($(this).text() == $(this).attr('href') && group_id &&
  402. 						d.groups[group_id] && d.groups[group_id].name) {
  403. 					$(this).text(d.groups[group_id].name);
  404. 				}
  405. 			}
  406. 		});
  407.  
  408. 		/* Mark links on album page in torrent table */
  409. 		if (/tehconnection\.eu\/torrents\.php/.test(document.URL)) {
  410. 			/* Parse search */
  411. 			var search = {};
  412. 			var search_list = document.location.search.substring(1).split('&');
  413. 			for (var i = 0; i < search_list.length; i++) {
  414. 				var pair = search_list[i].split('=');
  415. 				search[pair[0]] = pair[1];
  416. 			}
  417.  
  418. 			if (search.id) {
  419. 				/* Album page */
  420. 				$('#content .torrent_table:first tr.group_torrent').each(function(i) {
  421. 					/* Find torrent id */
  422. 					var torrent_id;
  423. 					$(this).children('td').eq(0).children('span').eq(0).children('a').each(function(i) {
  424. 						var href = $(this).attr('href');
  425. 						if (href) {
  426. 							var m = href.match(/torrents\.php\?torrentid=(\d+)/);
  427. 							if (m) {
  428. 								torrent_id = m[1];
  429. 								$(this).removeClass('group_snatched snatched uploaded leeching seeding');
  430. 								return false;
  431. 							}
  432. 						}
  433. 					});
  434.  
  435. 					if (torrent_id && d.torrents[torrent_id]) {
  436. 						var link = $(this).children('td').eq(0).children('a').eq(0);
  437. 						link.addClass(d.torrents[torrent_id].type);
  438. 						if (d.torrents[torrent_id].seeding) link.addClass('seeding');
  439. 					}
  440. 				});
  441. 			}
  442. 		}
  443. 	}
  444.  
  445. 	/* Mark torrent as leeching when download link is clicked */
  446. 	function mark_download_links() {
  447. 		$('#content').find('a').each(function(i) {
  448. 			var href = $(this).attr('href');
  449. 			if (href) {
  450. 				/* Find download links */
  451. 				var m = href.match(/torrents\.php\?action=download&id=(\d+)/);
  452. 				if (m) {
  453. 					var torrent_id = m[1];
  454. 					$(this).click(function(event) {
  455. 						var d = snatch_cache.unserialize();
  456. 						d.torrents[torrent_id] = { type: 'leeching', seeding: false };
  457. 						snatch_cache.serialize();
  458. 						mark_snatched_links();
  459. 					});
  460. 				}
  461. 			}
  462. 		});
  463. 	}
  464. 	
  465.  
  466. 	/* Scan current page */
  467. 	if (/tehconnection\.eu\/torrents\.php/.test(document.URL)) {
  468. 		/* Parse search */
  469. 		var search = {};
  470. 		var search_list = document.location.search.substring(1).split('&');
  471. 		for (var i = 0; i < search_list.length; i++) {
  472. 			var pair = search_list[i].split('=');
  473. 			search[pair[0]] = pair[1];
  474. 		}
  475.  
  476. 		if ((search.type == 'snatched' || search.type == 'uploaded'
  477. 				|| search.type == 'seeding' || search.type == 'leeching') && search.userid == whatcd_id) {
  478. 			var scan_status = $('<div>Scanning current page... <span></span></div>');
  479. 			status.contents().append(scan_status);
  480. 			status.show();
  481.  
  482. 			/* Scan current page */
  483. 			var found = scan_torrent_page($(document), search.type);
  484.  
  485. 			scan_status.children('span').text('Done ('+((found > 0) ? (found+' updates found') : 'no updates found')+')')
  486. 			status.show(5000);
  487. 		}
  488. 	}
  489.  
  490. 	/* Mark links */
  491. 	mark_download_links();
  492. 	mark_snatched_links();
  493.  
  494. 	/* Auto update */
  495. 	var now = new Date();
  496. 	var last_update = parseInt(GM_getValue('last_update', '0'));
  497. 	if (last_update + AUTO_UPDATE_INTERVAL*1000 < now.getTime()) {
  498. 		GM_setValue('last_update', now.getTime().toString());
  499. 		var jobs = 4;
  500. 		var total_found = {};
  501.  
  502. 		/* Show auto update status */
  503. 		if (last_update == 0) {
  504. 			var update_status = {
  505. 				snatched: $('<div>Updating snatched: <span>Initializing...</span></div>'),
  506. 				uploaded: $('<div>Updating uploaded: <span>Initializing...</span></div>'),
  507. 				leeching: $('<div>Updating leeching: <span>Initializing...</span></div>'),
  508. 				seeding: $('<div>Updating seeding: <span>Initializing...</span></div>')
  509. 			};
  510.  
  511. 			for (var type in update_status) status.contents().append(update_status[type]);
  512. 			status.show();
  513. 		}
  514.  
  515. 		function scan_page_handler(type, page) {
  516. 			if (last_update == 0) {
  517. 				update_status[type].children('span').text('Page '+page+'...');
  518. 				status.show();
  519. 			}
  520. 		}
  521.  
  522. 		function scan_finished_handler(type, found) {
  523. 			if (last_update == 0) {
  524. 				update_status[type].children('span').text('Done ('+((found > 0) ? (found+' updates found') : 'no updates found')+')');
  525. 			}
  526.  
  527. 			jobs -= 1;
  528. 			total_found[type] = found;
  529.  
  530. 			if (jobs == 0) {
  531. 				mark_snatched_links();
  532. 				if (last_update == 0) {
  533. 					var total = [];
  534. 					for (var type in total_found) if (total_found[type] > 0) total.push(type+': '+total_found[type]);
  535. 					status.contents().append('<div>Auto update done</div>');
  536. 					status.show(5000);
  537. 				}
  538. 			}
  539. 		}
  540.  
  541. 		/* Rescan all types of torrent lists */
  542. 		scan_all_torrent_pages('snatched', scan_page_handler, scan_finished_handler);
  543. 		scan_all_torrent_pages('uploaded', scan_page_handler, scan_finished_handler);
  544. 		scan_all_torrent_pages('leeching', scan_page_handler, scan_finished_handler);
  545. 		scan_all_torrent_pages('seeding', scan_page_handler, scan_finished_handler);
  546. 	}
  547. })();