☰

IMDB List Importer

Import list of titles or people in the imdb list

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

  1. // ==UserScript==
  2. // @name            IMDB List Importer
  3. // @namespace       Neinei0k_imdb
  4. // @include         https://www.imdb.com/list/ls*/edit/*
  5. // @version         10.1
  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. var request_data_add_item = {
  11.     "query": "mutation AddConstToList($listId: ID!, $constId: ID!, $includeListItemMetadata: Boolean!, $refTagQueryParam: String, $originalTitleText: Boolean) {\n  addItemToList(input: {listId: $listId, item: {itemElementId: $constId}}) {\n    listId\n    modifiedItem {\n      ...ListItemMetadata\n      listItem @include(if: $includeListItemMetadata) {\n        ... on Title {\n          ...TitleListItemMetadata\n        }\n        ... on Name {\n          ...NameListItemMetadata\n        }\n        ... on Image {\n          ...ImageListItemMetadata\n        }\n        ... on Video {\n          ...VideoListItemMetadata\n        }\n      }\n    }\n  }\n}\n\nfragment ListItemMetadata on ListItemNode {\n  itemId\n  createdDate\n  description {\n    originalText {\n      markdown\n      plaidHtml(showLineBreak: true)\n      plainText\n    }\n  }\n}\n\nfragment TitleListItemMetadata on Title {\n  ...BaseTitleCard\n  plot {\n    plotText {\n      plainText\n    }\n  }\n  latestTrailer {\n    id\n  }\n  series {\n    series {\n      id\n      originalTitleText {\n        text\n      }\n      releaseYear {\n        endYear\n        year\n      }\n      titleText {\n        text\n      }\n    }\n  }\n}\n\nfragment BaseTitleCard on Title {\n  id\n  titleText {\n    text\n  }\n  titleType {\n    id\n    text\n    canHaveEpisodes\n    displayableProperty {\n      value {\n        plainText\n      }\n    }\n  }\n  originalTitleText {\n    text\n  }\n  primaryImage {\n    id\n    width\n    height\n    url\n    caption {\n      plainText\n    }\n  }\n  releaseYear {\n    year\n    endYear\n  }\n  ratingsSummary {\n    aggregateRating\n    voteCount\n  }\n  runtime {\n    seconds\n  }\n  certificate {\n    rating\n  }\n  canRate {\n    isRatable\n  }\n  titleGenres {\n    genres(limit: 3) {\n      genre {\n        text\n      }\n    }\n  }\n  canHaveEpisodes\n}\n\nfragment NameListItemMetadata on Name {\n  id\n  primaryImage {\n    url\n    caption {\n      plainText\n    }\n    width\n    height\n  }\n  nameText {\n    text\n  }\n  primaryProfessions {\n    category {\n      text\n    }\n  }\n  knownFor(first: 1) {\n    edges {\n      node {\n        summary {\n          yearRange {\n            year\n            endYear\n          }\n        }\n        title {\n          id\n          originalTitleText {\n            text\n          }\n          titleText {\n            text\n          }\n          titleType {\n            canHaveEpisodes\n          }\n        }\n      }\n    }\n  }\n  bio {\n    displayableArticle {\n      body {\n        plaidHtml(\n          queryParams: $refTagQueryParam\n          showOriginalTitleText: $originalTitleText\n        )\n      }\n    }\n  }\n}\n\nfragment ImageListItemMetadata on Image {\n  id\n  url\n  height\n  width\n  caption {\n    plainText\n  }\n  names(limit: 4) {\n    id\n    nameText {\n      text\n    }\n  }\n  titles(limit: 1) {\n    id\n    titleText {\n      text\n    }\n    originalTitleText {\n      text\n    }\n    releaseYear {\n      year\n      endYear\n    }\n  }\n}\n\nfragment VideoListItemMetadata on Video {\n  id\n  thumbnail {\n    url\n    width\n    height\n  }\n  name {\n    value\n    language\n  }\n  description {\n    value\n    language\n  }\n  runtime {\n    unit\n    value\n  }\n  primaryTitle {\n    id\n    originalTitleText {\n      text\n    }\n    titleText {\n      text\n    }\n    titleType {\n      canHaveEpisodes\n    }\n    releaseYear {\n      year\n      endYear\n    }\n  }\n}",
  12.     "operationName": "AddConstToList",
  13.     "variables": {
  14.         "listId": "",
  15.         "constId": "",
  16.         "includeListItemMetadata": true,
  17.         "refTagQueryParam": "lsedt_add_items",
  18.         "originalTitleText": false
  19.     }
  20. }
  21.  
  22. var request_data_add_description = {
  23.     "query": "mutation EditListItemDescription($listId: ID!, $itemId: ID!, $itemDescription: String!) {\n  editListItemDescription(\n    input: {listId: $listId, itemId: $itemId, itemDescription: $itemDescription}\n  ) {\n    formattedItemDescription {\n      originalText {\n        markdown\n        plaidHtml(showLineBreak: true)\n        plainText\n      }\n    }\n  }\n}",
  24.     "operationName": "EditListItemDescription",
  25.     "variables": {
  26.         "listId": "",
  27.         "itemId": "",
  28.         "itemDescription": ""
  29.     }
  30. }
  31.  
  32. var request_data_reorder_item = {
  33.     "query": "mutation reorderListItems($input: ReorderListInput!) {\n  reorderList(input: $input) {\n    listId\n  }\n}",
  34.     "operationName": "reorderListItems",
  35.     "variables": {
  36.         "input": {
  37.             "newPositions": [
  38.                 /*{
  39.                     "position": -1,
  40.                     "itemId": ""
  41.                 }*/
  42.             ],
  43.             "listId": ""
  44.         }
  45.     }
  46. }
  47.  
  48. var elements = createHTMLForm();
  49.  
  50. function log(level, message) {
  51. 	console.log("(IMDB List Importer) " + level + ": " + message);
  52. }
  53.  
  54. function setStatus(message) {
  55. 	elements.status.textContent = message;
  56. }
  57.  
  58. function createHTMLForm() {
  59. 	let elements = {};
  60.  
  61.   try {
  62.     let root = createRoot();
  63.     elements.text = createTextField(root);
  64.  
  65.     if (isFileAPISupported()) {
  66.       elements.file = createFileInput(root);
  67.       elements.isFromFile = createFromFileCheckbox(root);
  68.     } else {
  69.       createFileAPINotSupportedMessage(root);
  70.     }
  71.  
  72.     elements.isCSV = createCSVCheckbox(root);
  73.     elements.isUnique = createUniqueCheckbox(root);
  74.     
  75.     elements.isReverse = createReverseCheckbox(root);
  76.     elements.insert = createInsertRadio(root);
  77.     elements.insertOther = createInsertOtherInput(root);
  78.     
  79.     elements.status = createStatusBar(root);
  80.     
  81.     createImportButton(root);
  82.   } catch (message) {
  83.    	log("Error", message);
  84.   }
  85.  
  86. 	return elements;
  87. }
  88.  
  89. function isFileAPISupported() {
  90. 	return window.File && window.FileReader && window.FileList && window.Blob;
  91. }
  92.  
  93. function createRoot() {
  94. 	let container = document.querySelector('section.ipc-page-section--base');
  95. 	if (container === null) {
  96. 		throw "section.section.ipc-page-section--base element not found";
  97. 	}
  98. 	let root = document.createElement('div');
  99. 	root.setAttribute('class', 'search-bar ipc-list-card--base ipc-list-card--border-line');
  100. 	root.style.height = 'initial';
  101.   root.style.marginTop = '30px';
  102.   root.style.marginBottom = '30px';
  103.   root.style.padding = '10px';
  104. 	container.insertBefore(root, container.children[1]);
  105.  
  106. 	return root;
  107. }
  108.  
  109. function createTextField(root) {
  110. 	let text = document.createElement('textarea');
  111. 	text.style = "background-color: white; width: 100%; height: 100px; overflow: initial;";
  112. 	root.appendChild(text);
  113. 	root.appendChild(document.createElement('br'));
  114.  
  115. 	return text;
  116. }
  117.  
  118. function createFileInput(root) {
  119. 	let file = document.createElement('input');
  120. 	file.type = 'file';
  121. 	file.disabled = true;
  122.   file.style.marginBottom = '10px';
  123. 	root.appendChild(file);
  124. 	root.appendChild(document.createElement('br'));
  125.  
  126. 	return file;
  127. }
  128.  
  129. function createFromFileCheckbox(root) {
  130. 	let isFromFile = createCheckbox("Import from file (otherwise import from text)");
  131. 	root.appendChild(isFromFile.label);
  132. 	root.appendChild(document.createElement('br'));
  133.  
  134. 	isFromFile.checkbox.addEventListener('change', fromFileOrTextChangeHandler, false);
  135. 	
  136. 	return isFromFile.checkbox;
  137. }
  138.  
  139. function createCheckbox(textContent) {
  140. 	let checkbox = document.createElement('input');
  141. 	checkbox.type = 'checkbox';
  142. 	checkbox.style = 'width: initial;';
  143.  
  144. 	let text = document.createElement('span');
  145. 	text.style = 'font-weight: normal;';
  146. 	text.textContent = textContent;
  147.  
  148. 	let label = document.createElement('label');
  149. 	label.appendChild(checkbox);
  150. 	label.appendChild(text);
  151.  
  152. 	return {label: label, checkbox: checkbox};
  153. }
  154.  
  155. function createRadio(name, value, textContent) {
  156.   let radio = document.createElement('input');
  157. 	radio.type = 'radio';
  158. 	radio.style = 'width: initial;';
  159.   radio.name = name;
  160.   radio.value = value;
  161.  
  162. 	let text = document.createElement('span');
  163. 	text.style = 'font-weight: normal;';
  164. 	text.textContent = textContent;
  165.  
  166. 	let label = document.createElement('label');
  167. 	label.appendChild(radio);
  168. 	label.appendChild(text);
  169.  
  170. 	return {label: label, radio: radio};
  171. }
  172.  
  173. function fromFileOrTextChangeHandler(event) {
  174. 	let isChecked = event.target.checked;
  175. 	elements.text.disabled = isChecked;
  176. 	elements.file.disabled = !isChecked;
  177. }
  178.  
  179. function createFileAPINotSupportedMessage(root) {
  180. 	let notSupported = document.createElement('div');
  181. 	notSupported.style = 'font-weight: normal;';
  182.   notSupported.style.marginTop = '10px';
  183.   notSupported.style.marginBottom = '10px';
  184. 	notSupported.textContent = "Your browser does not support File API for reading local files.";
  185. 	root.appendChild(notSupported);
  186. }
  187.  
  188. function createCSVCheckbox(root) {
  189. 	let isCSV = createCheckbox("Data from .csv file (otherwise extract ids from text)");
  190. 	isCSV.checkbox.checked = true;
  191. 	root.appendChild(isCSV.label);
  192. 	root.appendChild(document.createElement('br'));
  193.  
  194. 	return isCSV.checkbox;
  195. }
  196.  
  197. function createUniqueCheckbox(root) {
  198. 	let isUnique = createCheckbox("Add only unique elements");
  199. 	root.appendChild(isUnique.label);
  200. 	root.appendChild(document.createElement('br'));
  201.  
  202. 	return isUnique.checkbox;
  203. }
  204.  
  205. function createReverseCheckbox(root) {
  206.   let isReverse = createCheckbox("Reverse Items on Insertion");
  207.   root.appendChild(document.createElement('br'));
  208. 	root.appendChild(isReverse.label);
  209. 	root.appendChild(document.createElement('br'));
  210.  
  211. 	return isReverse.checkbox;
  212. }
  213.  
  214. function createInsertRadio(root) {
  215.   let insertBegin = createRadio("imdb_list_importer_insert", "1", "Insert in the Beginning");
  216.   let insertEnd = createRadio("imdb_list_importer_insert", "-1", "Insert in the End");
  217.   let insertOther = createRadio("imdb_list_importer_insert", "0", "Insert in Other Position");
  218.   
  219.   insertEnd.radio.checked = true;
  220.   
  221.   root.appendChild(insertBegin.label);
  222.   root.appendChild(document.createElement('br'));
  223.   root.appendChild(insertEnd.label);
  224.   root.appendChild(document.createElement('br'));
  225.   root.appendChild(insertOther.label);
  226.   root.appendChild(document.createElement('br'));
  227.   
  228.   insertBegin.radio.addEventListener('change', isOtherHandler, false);
  229.   insertEnd.radio.addEventListener('change', isOtherHandler, false);
  230.   insertOther.radio.addEventListener('change', isOtherHandler, false);
  231.   
  232.   return {'begin': insertBegin.radio, 'end': insertEnd.radio, 'other': insertOther.radio};
  233. }
  234.  
  235. function createInsertOtherInput(root) {
  236.   let insertOtherInput = document.createElement('input');
  237.   insertOtherInput.type = 'text';
  238.   insertOtherInput.disabled = true;
  239.   root.appendChild(insertOtherInput);
  240.   root.appendChild(document.createElement('br'));
  241.   root.appendChild(document.createElement('br'));
  242.   
  243.   return insertOtherInput;
  244. }
  245.  
  246. function isOtherHandler(event) {
  247.   let isDisable = event.target.value != "0";
  248.   elements.insertOther.disabled = isDisable;
  249. }
  250.  
  251.  
  252. function createStatusBar(root) {
  253. 	let status = document.createElement('div');
  254. 	status.textContent = "Set-up parameters. Insert text or choose file. Press 'Import List' button.";
  255.   status.style.marginTop = '10px';
  256.   status.style.marginBottom = '10px';
  257. 	root.appendChild(status);
  258.  
  259. 	return status;
  260. }
  261.  
  262. function createImportButton(root) {
  263. 	let importList = document.createElement('button');
  264. 	importList.class = 'btn';
  265. 	importList.textContent = "Import List";
  266. 	root.appendChild(importList);
  267.  
  268. 	importList.addEventListener('click', importListClickHandler, false);
  269. }
  270.  
  271. function importListClickHandler(event) {
  272. 	if (elements.hasOwnProperty('isFromFile') && elements.isFromFile.checked) {
  273. 		readFile();
  274. 	} else {
  275. 		importList(extractItems(elements.text.value));
  276. 	}
  277. }
  278.  
  279. function readFile() {
  280. 	let file = elements.file.files[0];
  281. 	if (file !== undefined) {
  282. 		log("Info", "Reading file " + file.name);
  283. 		setStatus("Reading file " + file.name);
  284. 		let fileReader = new FileReader();
  285. 		fileReader.onload = fileOnloadHandler;
  286. 		fileReader.readAsText(file);
  287. 	} else {
  288. 		setStatus("Error: File is not selected");
  289. 	}
  290. }
  291.  
  292. function fileOnloadHandler(event) {
  293. 	if (event.target.error === null) {
  294. 		importList(extractItems(event.target.result));
  295. 	} else {
  296. 		log("Error", e.target.error);
  297. 		setStatus("Error: " + e.target.error);
  298. 	}
  299. }
  300.  
  301. function extractItems(text) {
  302. 	try {
  303. 		let itemRegExp = getRegExpForItems();
  304.  
  305. 		if (elements.isCSV.checked) {
  306. 			return extractItemsFromCSV(itemRegExp, text);
  307. 		} else {
  308. 			return extractItemsFromText(itemRegExp, text);
  309. 		}
  310. 	} catch (message) {
  311. 		log("Error", message);
  312. 		setStatus("Error: " + message);
  313. 		return [];
  314. 	}
  315. }
  316.  
  317. function getRegExpForItems() {
  318. 	return "[a-z]{2}[0-9]{7,8}";
  319. }
  320.  
  321. function extractItemsFromCSV(re, text) {
  322. 	let table = parseCSV(text);
  323. 	let fields = findFieldNumbers(table);
  324.  
  325.   if (fields.description !== -1) {
  326. 		log("Info", "Found csv file fields Const(" + fields.const + ") and Description(" + fields.description + ")");
  327.   } else {
  328.     log("Info", "Found csv file field Const(" + fields.const + "). Description field is not found.");
  329.   }
  330.  
  331. 	re = new RegExp("^" + re + "$");
  332. 	let items = [];
  333. 	// Add elements to the list
  334. 	for (let i = 1; i < table.length; i++) {
  335. 		let row = table[i];
  336. 		if (re.exec(row[fields.const]) === null) {
  337. 			throw "Invalid 'const' field format on line " + (i+1);
  338. 		}
  339. 		if (elements.isUnique.checked) {
  340. 			let exists = items.findIndex(function(v){
  341. 				return v.const === row[fields.const];
  342. 			});
  343. 			if (exists !== -1) continue;
  344. 		}
  345. 		items.push({const: row[fields.const], description: (fields.description == -1 ? "" : row[fields.description])});
  346. 	}
  347.  
  348. 	return items;
  349. }
  350.  
  351. function parseCSV(text) {
  352. 	let lines = text.split(/\r|\n/);
  353. 	let table = [];
  354. 	for (let i=0; i < lines.length; i++) {
  355. 		if (isEmpty(lines[i])) {
  356. 			continue;
  357. 		}
  358. 		let isInsideString = false;
  359. 		let row = [""];
  360. 		for (let j=0; j < lines[i].length; j++) {
  361. 			if (!isInsideString && lines[i][j] === ',') {
  362. 				row.push("");
  363. 			} else if (lines[i][j] === '"') {
  364. 				isInsideString = !isInsideString;
  365. 			} else {
  366. 				row[row.length-1] += lines[i][j];
  367. 			}
  368. 		}
  369. 		table.push(row);
  370. 		if (isInsideString) {
  371. 			throw "Wrong number of \" on line " + (i+1);
  372. 		}
  373. 		if (row.length != table[0].length) {
  374. 			throw "Wrong number of fields on line " + (i+1) + ". Expected " + table[0].length + " but found " + row.length + ".";
  375. 		}
  376. 	}
  377.  
  378. 	return table;
  379. }
  380.  
  381. function isEmpty(str) {
  382. 	return str.trim().length === 0;
  383. }
  384.  
  385. function findFieldNumbers(table) {
  386. 	let fieldNames = table[0];
  387. 	let fieldNumbers = {'const': -1, 'description': -1};
  388.  
  389. 	for (let i = 0; i < fieldNames.length; i++) {
  390. 		let fieldName = fieldNames[i].toLowerCase().trim();
  391. 		if (fieldName === 'const') {
  392. 			fieldNumbers.const = i;
  393. 		} else if (fieldName === 'description') {
  394. 			fieldNumbers.description = i;
  395. 		}
  396. 	}
  397.  
  398. 	if (fieldNumbers.const === -1) {
  399. 		throw "Field 'const' not found."
  400. 	}
  401. 	return fieldNumbers;
  402. }
  403.  
  404. function extractItemsFromText(re, text) {
  405. 	re = new RegExp(re);
  406. 	let items = [];
  407. 	let e;
  408. 	while ((e = re.exec(text)) !== null) {
  409. 		let flag = '';
  410. 		if (elements.isUnique.checked)
  411. 			flag = 'g';
  412. 		text = text.replace(new RegExp(e[0], flag), '');
  413. 		items.push({const: e[0], description: ""});
  414. 	}
  415. 	return items;
  416. }
  417.  
  418. async function importList(list) {
  419. 	if (list.length === 0)
  420. 		return;
  421.  
  422. 	let msg = "Elements to add: ";
  423. 	for (let i = 0;  i < list.length; i++)
  424. 		msg += list[i].const + ",";
  425. 	log("Info", msg);
  426.  
  427. 	let list_id = /ls[0-9]{1,}/.exec(location.href)[0];
  428.   
  429.   if (elements.isReverse.checked) {
  430.    	list.reverse();
  431.   }
  432.  
  433.   let items = [];
  434.   for (let i = 0; i < list.length; ++i) {
  435.     log("Info", `Adding element ${String(i+1)}: ${list[i].const}...`);
  436.  
  437.     request_data_add_item.variables.listId = list_id;
  438.   	request_data_add_item.variables.constId = list[i].const;
  439.     let response = await sendRequest(request_data_add_item);
  440.     
  441.     let listItemId = response.data.addItemToList.modifiedItem.itemId;
  442.     log("Info", `${list[i].const} added as ${listItemId}`);
  443.     items.push(listItemId);
  444.       
  445. 		if (list[i].description.length !== 0) {
  446.       log("Info", `Adding description to ${listItemId}...`);
  447.       request_data_add_description.variables.listId = list_id;
  448.     	request_data_add_description.variables.itemId = listItemId;
  449.     	request_data_add_description.variables.itemDescription = list[i].description;
  450.       await sendRequest(request_data_add_description);
  451.     }
  452.     
  453.     setStatus(`Ready ${String(i+1)} of ${list.length}.`);  
  454.   }
  455.   
  456.   let insertPosition = -1;
  457.   if (elements.insert.begin.checked) {
  458.    	insertPosition = 1;
  459.   } else if (elements.insert.other.checked) {
  460.    	insertPosition = Number(elements.insertOther.value);
  461.     if (isNaN(insertPosition) || insertPosition < 1) {
  462.      	insertPosition = -1; 
  463.     }
  464.   }
  465.   if (insertPosition != -1) {
  466.     request_data_reorder_item.variables.input.newPositions = [];
  467.     request_data_reorder_item.variables.input.listId = list_id;
  468.  
  469.     for (let i = items.length - 1; i >= 0; i--) {
  470.       request_data_reorder_item.variables.input.newPositions.push({
  471.         "position": insertPosition,
  472.         "itemId": items[i]
  473.       });
  474.     }
  475.  
  476.     log("Info", `Moving items to position ${insertPosition}...`);
  477.     await sendRequest(request_data_reorder_item);
  478.   }
  479.   
  480.   location.reload();
  481. }
  482.  
  483. function sendRequest(data) {
  484.   return fetch("https://api.graphql.imdb.com/", {
  485.     "credentials": "include",
  486.     "headers": {
  487.         "Accept": "application/graphql+json, application/json",
  488.         "content-type": "application/json",
  489.     },
  490.     "referrer": "https://www.imdb.com/",
  491.     "body": JSON.stringify(data),
  492.     "method": "POST",
  493.     "mode": "cors"
  494. 	}).then((response) => {
  495.   	if (!response.ok) {
  496.       throw new Error(`Request failed with status code ${response.status}`);
  497.     }
  498.  
  499.     return response.json();
  500.   });
  501. }