☰

Greasy Fork 还支持 简体中文。

IMDB List Importer

Import list of titles or people in the imdb list

目前為 2019-08-07 提交的版本，檢視 最新版本。

  1. // ==UserScript==
  2. // @name         IMDB List Importer
  3. // @namespace    Neinei0k_imdb
  4. // @include      https://www.imdb.com/list/*
  5. // @version	 7.0
  6. // @grant        GM.xmlHttpRequest
  7. // @description  Import list of titles or people in the imdb list
  8. // ==/UserScript==
  9.  
  10. var o = {
  11.  
  12. init: function(e) {
  13. 	this.etext = e.querySelector('textarea');
  14. 	this.efile = e.querySelector('input[type="file"]');
  15. 	this.eready = e.children[10]; // DOM element with messages for the user.
  16. 	var checkboxes = e.querySelectorAll('input[type="checkbox"]');
  17. 	this.source = checkboxes[0];
  18. 	this.csv = checkboxes[1];
  19. 	this.unique = checkboxes[2];
  20.   
  21.   var hidden_element =  document.querySelector('#main > input'); // Unknown hidden element. Data needs to be send with all requests.
  22.   if (hidden_element == null) {
  23.    	this.log('e','Hidden element not found');
  24.   } else {
  25.   	this.hidden_data = hidden_element.id + "=" + hidden_element.value;
  26.   }
  27. },
  28.  
  29. run: function(event) {
  30. 	this.text = this.etext.value;
  31. 	if (this.source.checked) { // read data from file
  32. 		var file = this.efile.files[0];
  33. 		if (file !== undefined) {
  34. 			this.log("i","Reading file " + file.name);
  35. 			var r = new FileReader();
  36. 			r.onload = this.file_onload.bind(this);
  37. 			r.readAsText(file);
  38. 		} else {
  39. 			this.log("e","File is undefined");
  40. 		}
  41. 	} else { // read data from input element
  42. 		this.add_list(this.create_list());
  43. 	}
  44. },
  45.  
  46. file_onload: function(e) {
  47. 	if (e.target.error === null) {
  48. 		this.text = e.target.result;
  49. 		this.add_list(this.create_list());
  50. 	} else {
  51. 		this.log("e","File reading error: " + e.target.error);
  52. 	}
  53. },
  54.  
  55. log: function(level,msg) {
  56. 	var l = "";
  57.   switch (level) {
  58.     case 'i': l = "Info: "; break;
  59.     case 'w': l = "Warning: "; break;
  60.     case 'e': l = "Error: "; break;
  61.   }
  62. 	if (l.length !== 0)
  63. 		console.log("IMDB List Importer: " + l + msg);
  64. 	if (level == "n" || level == "e")
  65. 		this.eready.innerText = msg;
  66. },
  67.  
  68. create_list: function() {
  69. 	var re;
  70. 	// Find type of the list
  71. 	/*if (document.querySelector('[data-type="Characters"]') !== null) {
  72. 		this.log("i", "List type: characters");
  73. 		re = "ch";
  74. 	} else*/
  75.   if (document.querySelector('[data-type="People"]') !== null) {
  76. 		this.log("i", "List type: people");
  77. 		re = "nm";
  78. 	} else if (document.querySelector('[data-type="Titles"]') !== null) {
  79. 		this.log("i", "List type: titles");
  80. 		re = "tt";
  81. 	} else {
  82. 		this.log("e","Could not determine type of the list");
  83. 		return [];
  84. 	}
  85. 	re += "[0-9]{7}";
  86.  
  87. 	if (this.csv.checked) {
  88. 		return this.read_csv(re);
  89. 	} else {
  90. 		re = new RegExp(re);
  91. 		var list = [];
  92. 		var e;
  93. 		var text = this.text;
  94. 		while ((e = re.exec(text)) !== null) {
  95. 			var flag = '';
  96. 			if (this.unique.checked) flag = 'g';
  97. 			text = text.replace(new RegExp(e[0], flag), '');
  98. 			list.push({const: e[0], description: ""});
  99. 		}
  100. 		return list;
  101. 	}
  102. },
  103.  
  104. read_csv: function(re) {
  105. 	re = new RegExp("^" + re + "$");
  106. 	var list = [];
  107.   
  108.   // Parse csv
  109.   var text = this.text.split('\n'); // Separate by lines
  110.   this.text = null; // Variable may have lots of data which is no longer needed
  111.   var parsed_text = [];
  112.   for (var i in text) { // For each line
  113.     if (text[i].trim().length === 0) { // Ignore empty lines including lines with only white space characters
  114.       continue;
  115.     }
  116.     var state = 0; // 0 - outside of double quotes (comma character is the separator), 1 - inside double quotes (comma character is part of a field)
  117.     var parsed_line = [""];
  118.     for (var j in text[i]) {
  119.       if (state == 0 && text[i][j] == ',') {
  120.         parsed_line.push("");
  121.       } else if (text[i][j] == '"') {
  122.         state = (state + 1) % 2;
  123.       } else {
  124.         parsed_line[parsed_line.length-1] += text[i][j];
  125.       }
  126.     }
  127.     parsed_text.push(parsed_line);
  128.   }
  129.   text = parsed_text;
  130.   
  131.   // console.log(text); // print parsed data
  132.  
  133. 	// Find const and description field numbers.
  134. 	try {
  135.     if (text.length < 2) { // There must be at least 2 rows in the data
  136.     	throw "No data";
  137.     }
  138. 		var fl = text[0];
  139. 		var fll = fl.length;
  140. 		var const_field = fl.indexOf('const');
  141. 		if (const_field === -1) {
  142.       const_field = fl.indexOf('Const');
  143.       if (const_field === -1) {
  144.       	throw "Field 'const' not found.";
  145.       }
  146.     }
  147. 	} catch (err) {
  148. 		this.log("e","Input line 1: " + err);
  149. 		return [];
  150. 	}
  151.   var desc_field = fl.indexOf('description');
  152. 		if (desc_field === -1) {
  153.       desc_field = fl.indexOf('Description');
  154.     }
  155.   
  156. 	this.log("i","Found csv file fields const(" + const_field +
  157. 	             ") and description(" + desc_field + ")");
  158. 	text.shift();
  159.  
  160. 	// Add elements to the list
  161. 	for (var i = 0; i < text.length; i++) {
  162. 		if (text[i].length === 0)
  163. 			continue;
  164. 		try {
  165. 			fl = text[i];
  166. 			if (fll !== fl.length) throw "Invalid number of fields.";
  167. 			if (re.exec(fl[const_field]) === null) throw "Invalid 'const' field.";
  168. 		} catch (err) {
  169. 			this.log("e","Input line " + (i+2) + ": " + err);
  170. 			return [];
  171. 		}
  172. 		if (this.unique.checked) {
  173. 			var exists = list.findIndex(function(v){
  174. 				return v.const === fl[const_field];
  175. 			});
  176. 			if (exists !== -1) continue;
  177. 		}
  178. 		list.push({const: fl[const_field],description: (desc_field == -1 ? "" : fl[desc_field])});
  179. 	}
  180.  
  181.   // console.log(list); // Print final list
  182.   
  183. 	return list;
  184. },
  185. 	
  186. add_list: function(list) {
  187. 	if (list.length === 0)
  188. 		return;
  189.  
  190. 	var msg = "Elements to add: ";
  191. 	for (var i in list)
  192. 		msg += list[i].const + ",";
  193. 	this.log("i",msg);
  194.  
  195. 	var l = {};
  196. 	l.list = list;
  197. 	l.ready = 0;
  198. 	l.list_id = /ls[0-9]{1,}/.exec(location.href)[0];
  199.   
  200. 	this.sendNext(l);
  201. },
  202.  
  203. sendNext: function(l) {
  204. 	this.log("i",'Add element ' + l.ready + ': ' + l.list[l.ready].const);
  205. 	this.send_request(this.check_item, l, 'https://www.imdb.com/list/' + l.list_id + '/' + l.list[l.ready].const + '/add', this.hidden_data);
  206. },
  207.  
  208. send_request: function(f,l,u,d) {
  209.   GM.xmlHttpRequest({
  210.   	method: "POST",
  211.     url: u,
  212.     data: d,
  213.     headers: {
  214.       "Content-Type": "application/x-www-form-urlencoded"
  215.     },
  216.     onreadystatechange: f.bind(this,l)
  217.   });
  218. 	/*var x = new XMLHttpRequest();
  219. 	x.onreadystatechange = f.bind(this,l);
  220. 	x.open('POST', u, true);
  221. 	x.setRequestHeader('Content-Type',
  222. 	  'application/x-www-form-urlencoded');
  223. 	x.send(d);*/
  224. },
  225.  
  226. check_item: function(l, e) {
  227. 	this.log("i","Add element(" + l.list[l.ready].const +
  228. 	             ") request: readyState(" + e.readyState +
  229. 	             "), status(" + e.status + ")");
  230. 	if (e.readyState == 4 && e.status == 200) {
  231. 		if (l.list[l.ready].description.length !== 0) {
  232. 			this.send_request(this.check_item_desc, l, 'https://www.imdb.com/list/' + l.list_id + '/edit/itemdescription',
  233. 			                  'newDescription=' + l.list[l.ready].description +
  234. 			                  '&listItem=' + JSON.parse(e.responseText).list_item_id + '&' + this.hidden_data);
  235. 		} else {
  236. 			this.showReady(l);
  237. 		}
  238. 	}
  239. },
  240.  
  241. check_item_desc: function(l,e) {
  242. 	this.log("i","Add element(" + l.list[l.ready].const +
  243. 	             ") description request: readyState(" + e.readyState +
  244. 	             "), status(" + e.status + ")");
  245. 	if (e.readyState == 4 && e.status == 200) {
  246. 		this.showReady(l);
  247. 	}
  248. },
  249.  
  250. showReady: function(l) {
  251. 	l.ready += 1;
  252. 	this.log("n",'Ready ' + l.ready + ' of ' + l.list.length + '.');
  253. 	if (l.ready == l.list.length) {
  254. 		location.reload();
  255. 	} else {
  256. 		this.sendNext(l);
  257. 	}
  258. },
  259.  
  260. change: function(e) {
  261. 	var s = e.target.checked;
  262. 	this.etext.disabled = s;
  263. 	this.efile.disabled = !s;
  264. },
  265.  
  266. };
  267.  
  268. var c = window.File && window.FileReader && window.FileList && window.Blob; // Check support of File API
  269. var div = document.createElement('div');
  270. div.setAttribute('class', 'search-bar');
  271. div.style.height = "initial"
  272. var s = '<textarea style="background-color: white; width: 100%; height: 100px; overflow: initial"></textarea><br>';
  273. if (c) {
  274. 	s += '<input type="file" disabled><br>';
  275. 	s += '<label>';
  276. 	s += '<input type="checkbox" style="width: initial;">';
  277. 	s += '<span style="font-weight: normal;">';
  278. 	s += 'Import from file (otherwise import from text)';
  279. 	s += '</span>';
  280. 	s += '</label><br>';
  281. } else {
  282. 	s += '<span style="font-weight: normal;">';
  283. 	s += 'Looks like your browser does not support File API for reading local files.';
  284. 	s += '</span><br>';
  285. }
  286. s += '<label>';
  287. s += '<input type="checkbox" checked style="width: initial;">';
  288. s += '<span style="font-weight: normal;">Data from .csv file</span>';
  289. s += '</label><br>';
  290. s += '<label>';
  291. s += '<input type="checkbox" style="width: initial;">';
  292. s += '<span style="font-weight: normal;">Add only unique elements</span>';
  293. s += '</label><br>';
  294. s += '<div>Set-up parameters. Insert text or choose file. Press \'Import List\' button.</div>';
  295. s += '<button class="btn">Import List</button>';
  296. div.innerHTML = s;
  297.  
  298. o.init(div);
  299. div.querySelector('button').addEventListener('click',o.run.bind(o),false);
  300. if (c) {
  301. 	o.source.addEventListener('change',o.change.bind(o),false);
  302. }
  303.  
  304. var list_edit = document.querySelector('.lister-search');
  305. list_edit.appendChild(div);