☰

IMDB List Importer

Import list of titles or people in the imdb list

当前为 2023-05-17 提交的版本，查看 最新版本。

  1. // ==UserScript==
  2. // @name            IMDB List Importer
  3. // @namespace       Neinei0k_imdb
  4. // @include         https://www.imdb.com/list/*
  5. // @version         8.2
  6. // @license         GNU General Public License v3.0 or later
  7. // @description	Import list of titles or people in the imdb list
  8. // ==/UserScript==
  9.  
  10. let elements = createHTMLForm();
  11.  
  12. function log(level, message) {
  13. 	console.log("(IMDB List Importer) " + level + ": " + message);
  14. }
  15.  
  16. function setStatus(message) {
  17. 	elements.status.textContent = message;
  18. }
  19.  
  20. function createHTMLForm() {
  21. 	let elements = {};
  22.  
  23.   try {
  24.     let root = createRoot();
  25.     elements.text = createTextField(root);
  26.  
  27.     if (isFileAPISupported()) {
  28.       elements.file = createFileInput(root);
  29.       elements.isFromFile = createFromFileCheckbox(root);
  30.     } else {
  31.       createFileAPINotSupportedMessage(root);
  32.     }
  33.  
  34.     elements.isCSV = createCSVCheckbox(root);
  35.     elements.isUnique = createUniqueCheckbox(root);
  36.     elements.status = createStatusBar(root);
  37.     
  38.     createImportButton(root);
  39.   } catch (message) {
  40.    	log("Error", message);
  41.   }
  42.  
  43. 	return elements;
  44. }
  45.  
  46. function isFileAPISupported() {
  47. 	return window.File && window.FileReader && window.FileList && window.Blob;
  48. }
  49.  
  50. function createRoot() {
  51. 	let container = document.querySelector('.lister-search');
  52. 	if (container === null) {
  53. 		throw ".lister-search element not found";
  54. 	}
  55. 	let root = document.createElement('div');
  56. 	root.setAttribute('class', 'search-bar');
  57. 	root.style.height = 'initial';
  58.   root.style.marginBottom = '30px';
  59. 	container.appendChild(root);
  60.  
  61. 	return root;
  62. }
  63.  
  64. function createTextField(root) {
  65. 	let text = document.createElement('textarea');
  66. 	text.style = "background-color: white; width: 100%; height: 100px; overflow: initial;";
  67. 	root.appendChild(text);
  68. 	root.appendChild(document.createElement('br'));
  69.  
  70. 	return text;
  71. }
  72.  
  73. function createFileInput(root) {
  74. 	let file = document.createElement('input');
  75. 	file.type = 'file';
  76. 	file.disabled = true;
  77.   file.style.marginBottom = '10px';
  78. 	root.appendChild(file);
  79. 	root.appendChild(document.createElement('br'));
  80.  
  81. 	return file;
  82. }
  83.  
  84. function createFromFileCheckbox(root) {
  85. 	let isFromFile = createCheckbox("Import from file (otherwise import from text)");
  86. 	root.appendChild(isFromFile.label);
  87. 	root.appendChild(document.createElement('br'));
  88.  
  89. 	isFromFile.checkbox.addEventListener('change', fromFileOrTextChangeHandler, false);
  90. 	
  91. 	return isFromFile.checkbox;
  92. }
  93.  
  94. function createCheckbox(textContent) {
  95. 	let checkbox = document.createElement('input');
  96. 	checkbox.type = 'checkbox';
  97. 	checkbox.style = 'width: initial;';
  98.  
  99. 	let text = document.createElement('span');
  100. 	text.style = 'font-weight: normal;';
  101. 	text.textContent = textContent;
  102.  
  103. 	let label = document.createElement('label');
  104. 	label.appendChild(checkbox);
  105. 	label.appendChild(text);
  106.  
  107. 	return {label: label, checkbox: checkbox};
  108. }
  109.  
  110. function fromFileOrTextChangeHandler(event) {
  111. 	let isChecked = event.target.checked;
  112. 	elements.text.disabled = isChecked;
  113. 	elements.file.disabled = !isChecked;
  114. }
  115.  
  116. function createFileAPINotSupportedMessage(root) {
  117. 	let notSupported = document.createElement('div');
  118. 	notSupported.style = 'font-weight: normal;';
  119.   notSupported.style.marginTop = '10px';
  120.   notSupported.style.marginBottom = '10px';
  121. 	notSupported.textContent = "Your browser does not support File API for reading local files.";
  122. 	root.appendChild(notSupported);
  123. }
  124.  
  125. function createCSVCheckbox(root) {
  126. 	let isCSV = createCheckbox("Data from .csv file (otherwise extract ids from text)");
  127. 	isCSV.checkbox.checked = true;
  128. 	root.appendChild(isCSV.label);
  129. 	root.appendChild(document.createElement('br'));
  130.  
  131. 	return isCSV.checkbox;
  132. }
  133.  
  134. function createUniqueCheckbox(root) {
  135. 	let isUnique = createCheckbox("Add only unique elements");
  136. 	root.appendChild(isUnique.label);
  137. 	root.appendChild(document.createElement('br'));
  138.  
  139. 	return isUnique.checkbox;
  140. }
  141.  
  142. function createStatusBar(root) {
  143. 	let status = document.createElement('div');
  144. 	status.textContent = "Set-up parameters. Insert text or choose file. Press 'Import List' button.";
  145.   status.style.marginTop = '10px';
  146.   status.style.marginBottom = '10px';
  147. 	root.appendChild(status);
  148.  
  149. 	return status;
  150. }
  151.  
  152. function createImportButton(root) {
  153. 	let importList = document.createElement('button');
  154. 	importList.class = 'btn';
  155. 	importList.textContent = "Import List";
  156. 	root.appendChild(importList);
  157.  
  158. 	importList.addEventListener('click', importListClickHandler, false);
  159. }
  160.  
  161. function importListClickHandler(event) {
  162. 	if (elements.hasOwnProperty('isFromFile') && elements.isFromFile.checked) {
  163. 		readFile();
  164. 	} else {
  165. 		importList(extractItems(elements.text.value));
  166. 	}
  167. }
  168.  
  169. function readFile() {
  170. 	let file = elements.file.files[0];
  171. 	if (file !== undefined) {
  172. 		log("Info", "Reading file " + file.name);
  173. 		setStatus("Reading file " + file.name);
  174. 		let fileReader = new FileReader();
  175. 		fileReader.onload = fileOnloadHandler;
  176. 		fileReader.readAsText(file);
  177. 	} else {
  178. 		setStatus("Error: File is not selected");
  179. 	}
  180. }
  181.  
  182. function fileOnloadHandler(event) {
  183. 	if (event.target.error === null) {
  184. 		importList(extractItems(event.target.result));
  185. 	} else {
  186. 		log("Error", e.target.error);
  187. 		setStatus("Error: " + e.target.error);
  188. 	}
  189. }
  190.  
  191. function extractItems(text) {
  192. 	try {
  193. 		let itemRegExp = getRegExpForItems();
  194.  
  195. 		if (elements.isCSV.checked) {
  196. 			return extractItemsFromCSV(itemRegExp, text);
  197. 		} else {
  198. 			return extractItemsFromText(itemRegExp, text);
  199. 		}
  200. 	} catch (message) {
  201. 		log("Error", message);
  202. 		setStatus("Error: " + message);
  203. 		return [];
  204. 	}
  205. }
  206.  
  207. function getRegExpForItems() {
  208. 	let listType;
  209. 	if (isPeopleList()) {
  210. 		log("Info", "List type: people");
  211. 		listType = "nm";
  212. 	} else if (isTitlesList()) {
  213. 		log("Info", "List type: titles");
  214. 		listType = "tt";
  215. 	} else {
  216. 		throw "Could not determine list type";
  217. 	}
  218. 	return listType + "[0-9]{7,8}";
  219. }
  220.  
  221. function isPeopleList() {
  222. 	return document.querySelector('[data-type="People"]') !== null;
  223. }
  224.  
  225. function isTitlesList() {
  226. 	return document.querySelector('[data-type="Titles"]') !== null;
  227. }
  228.  
  229. function extractItemsFromCSV(re, text) {
  230. 	let table = parseCSV(text);
  231. 	let fields = findFieldNumbers(table);
  232.  
  233.   if (fields.description !== -1) {
  234. 		log("Info", "Found csv file fields Const(" + fields.const + ") and Description(" + fields.description + ")");
  235.   } else {
  236.     log("Info", "Found csv file field Const(" + fields.const + "). Description field is not found.");
  237.   }
  238.  
  239. 	re = new RegExp("^" + re + "$");
  240. 	let items = [];
  241. 	// Add elements to the list
  242. 	for (let i = 1; i < table.length; i++) {
  243. 		let row = table[i];
  244. 		if (re.exec(row[fields.const]) === null) {
  245. 			throw "Invalid 'const' field format on line " + (i+1);
  246. 		}
  247. 		if (elements.isUnique.checked) {
  248. 			let exists = items.findIndex(function(v){
  249. 				return v.const === row[fields.const];
  250. 			});
  251. 			if (exists !== -1) continue;
  252. 		}
  253. 		items.push({const: row[fields.const], description: (fields.description == -1 ? "" : row[fields.description])});
  254. 	}
  255.  
  256. 	return items;
  257. }
  258.  
  259. function parseCSV(text) {
  260. 	let lines = text.split(/\r|\n/);
  261. 	let table = [];
  262. 	for (let i=0; i < lines.length; i++) {
  263. 		if (isEmpty(lines[i])) {
  264. 			continue;
  265. 		}
  266. 		let isInsideString = false;
  267. 		let row = [""];
  268. 		for (let j=0; j < lines[i].length; j++) {
  269. 			if (!isInsideString && lines[i][j] === ',') {
  270. 				row.push("");
  271. 			} else if (lines[i][j] === '"') {
  272. 				isInsideString = !isInsideString;
  273. 			} else {
  274. 				row[row.length-1] += lines[i][j];
  275. 			}
  276. 		}
  277. 		table.push(row);
  278. 		if (isInsideString) {
  279. 			throw "Wrong number of \" on line " + (i+1);
  280. 		}
  281. 		if (row.length != table[0].length) {
  282. 			throw "Wrong number of fields on line " + (i+1) + ". Expected " + table[0].length + " but found " + row.length + ".";
  283. 		}
  284. 	}
  285.  
  286. 	return table;
  287. }
  288.  
  289. function isEmpty(str) {
  290. 	return str.trim().length === 0;
  291. }
  292.  
  293. function findFieldNumbers(table) {
  294. 	let fieldNames = table[0];
  295. 	let fieldNumbers = {'const': -1, 'description': -1};
  296.  
  297. 	for (let i = 0; i < fieldNames.length; i++) {
  298. 		let fieldName = fieldNames[i].toLowerCase().trim();
  299. 		if (fieldName === 'const') {
  300. 			fieldNumbers.const = i;
  301. 		} else if (fieldName === 'description') {
  302. 			fieldNumbers.description = i;
  303. 		}
  304. 	}
  305.  
  306. 	if (fieldNumbers.const === -1) {
  307. 		throw "Field 'const' not found."
  308. 	}
  309. 	return fieldNumbers;
  310. }
  311.  
  312. function extractItemsFromText(re, text) {
  313. 	re = new RegExp(re);
  314. 	let items = [];
  315. 	let e;
  316. 	while ((e = re.exec(text)) !== null) {
  317. 		let flag = '';
  318. 		if (elements.isUnique.checked)
  319. 			flag = 'g';
  320. 		text = text.replace(new RegExp(e[0], flag), '');
  321. 		items.push({const: e[0], description: ""});
  322. 	}
  323. 	return items;
  324. }
  325.  
  326. function importList(list) {
  327. 	if (list.length === 0)
  328. 		return;
  329.  
  330. 	let msg = "Elements to add: ";
  331. 	for (let i = 0;  i < list.length; i++)
  332. 		msg += list[i].const + ",";
  333. 	log("Info", msg);
  334.  
  335. 	let l = {};
  336. 	l.list = list;
  337. 	l.ready = 0;
  338. 	l.list_id = /ls[0-9]{1,}/.exec(location.href)[0];
  339. 	l.hiddenElementData = getHiddenElementData(); // Data needs to be send with all requests.
  340.  
  341. 	sendItem(l);
  342. }
  343.  
  344. function getHiddenElementData() {
  345. 	let hiddenElement = document.querySelector('#main > input');
  346. 	if (hiddenElement === null) {
  347. 	 	log("Error", "Hidden element not found. It is required to be sent with every request.");
  348.     setStatus("Error: Hidden element not found. It is required to be sent with every request.");
  349. 		return "";
  350. 	}
  351. 	return hiddenElement.id + "=" + hiddenElement.value;
  352. }
  353.  
  354. function sendItem(l) {
  355. 	log("Info", 'Add element ' + l.ready + ': ' + l.list[l.ready].const);
  356. 	let url = 'https://www.imdb.com/list/' + l.list_id + '/' + l.list[l.ready].const + '/add';
  357. 	sendRequest(sendItemHandler, l, url, l.hiddenElementData);
  358. }
  359.  
  360. function sendRequest(handler, l, url, data) {
  361.   var x = new XMLHttpRequest();
  362. 	x.onreadystatechange = function(event) {
  363. 			handler(l, event);
  364.   }
  365. 	x.open('POST', url, true);
  366. 	x.setRequestHeader('Content-Type', 'application/x-www-form-urlencoded');
  367. 	x.send(data);
  368. }
  369.  
  370. function sendItemHandler(l, event) {
  371.   let target = event.target;
  372. 	log("Info", "Add element(" + l.list[l.ready].const + ") request: readyState(" + target.readyState + "), status(" + target.status + ")");
  373. 	if (target.readyState == 4 && target.status == 200) {
  374. 		let description = l.list[l.ready].description;
  375. 		if (description.length !== 0) {
  376. 			let listItemId = JSON.parse(target.responseText).list_item_id;
  377. 			let url = 'https://www.imdb.com/list/' + l.list_id + '/edit/itemdescription';
  378. 			let data = 'newDescription=' + description + '&listItem=' + listItemId + '&' + l.hiddenElementData
  379. 			sendRequest(sendItemDescriptionHandler, l, url, data);
  380. 		} else {
  381. 			showReady(l);
  382. 		}
  383. 	}
  384. }
  385.  
  386. function sendItemDescriptionHandler(l, event) {
  387.     let target = event.target;
  388. 	log("Info", "Add element(" + l.list[l.ready].const + ") description request: readyState(" + target.readyState + "), status(" + target.status + ")");
  389. 	if (target.readyState == 4 && target.status == 200) {
  390. 		showReady(l);
  391. 	}
  392. }
  393.  
  394. function showReady(l) {
  395. 	l.ready += 1;
  396. 	setStatus('Ready ' + l.ready + ' of ' + l.list.length + '.');
  397. 	if (l.ready == l.list.length) {
  398. 		location.reload();
  399. 	} else {
  400. 		sendItem(l);
  401. 	}
  402. }