☰

Pixiv Image Searches and Stuff

Searches Danbooru for pixiv IDs, adds IQDB image search links

  1. // ==UserScript==
  2. // @name         Pixiv Image Searches and Stuff
  3. // @namespace    https://gf.qytechs.cn/scripts/4296
  4. // @description  Searches Danbooru for pixiv IDs, adds IQDB image search links
  5. // @include      *://www.pixiv.net/*
  6. // @grant        GM_deleteValue
  7. // @grant        GM_xmlhttpRequest
  8. // @version      2018.02.01
  9. // ==/UserScript==
  10.  
  11. var iqdbURL = "";//Replace with "http://danbooru.iqdb.org/?url=" (Danbooru) or "http://iqdb.org/?url=" (multi-service) to add IQDB search links (replaces bookmark counts)
  12. var addSourceSearch = true;//Danbooru post search (looks for matching pixiv IDs); **Requires GM_xmlhttpRequest**
  13. var danbooruBase = "http://danbooru.donmai.us";//Base URL to use when linking to or querying Danbooru
  14.  
  15. //Source search options
  16. var danbooruLogin = "";//username
  17. var danbooruApiKey = "";//api key as listed on your profile
  18. var styleSourceFound = "color:green; font-weight: bold;";
  19. var styleSourceMissing = "color:red;";
  20. var sourceTimeout = 20;//seconds to wait before retrying query
  21. var maxAttempts = 10;//# of times to try a query before completely giving up on source searches
  22.  
  23. //////////////////////////////////////////////////////////////////////////////////////
  24.  
  25. var anyBookmarks = false, favList = [], jsonData = [];
  26.  
  27. if( typeof(custom) != "undefined" )
  28. 	custom();
  29.  
  30. if( iqdbURL )
  31.     try{ jsonData = JSON.parse( document.querySelector("[id*='js-'][data-items]").getAttribute("data-items") ); }catch(e) {}
  32.  
  33. //Source search requires GM_xmlhttpRequest()
  34. if( addSourceSearch && typeof(GM_xmlhttpRequest) == "undefined" )
  35. 	addSourceSearch = false;
  36.  
  37. //Manga images have to be handled specially
  38. if( location.search.indexOf("mode=manga") >= 0 )
  39. {
  40. 	let searchID = addSourceSearch && location.search.match(/illust_id=(\d+)/);
  41. 	if( searchID )
  42. 	{
  43. 		function linkManga(stuff)
  44. 		{
  45. 			let thumbList = [], images = stuff.querySelectorAll(".item-container img");
  46. 			for( let i = 0; i < images.length; i++ )
  47. 				thumbList.push({ link: images[i].parentNode.appendChild( document.createElement("div") ).appendChild( document.createElement("a") ), pixiv_id: searchID[1], page: i });
  48. 			sourceSearch( thumbList );
  49. 			return thumbList.length;
  50. 		}
  51. 		
  52. 		if( !linkManga( document ) )
  53. 		{
  54. 			//Normal manga images not found; set up a mutation observer in case a certain other script executes after this one and adds them.
  55. 			new MutationObserver( function(mutationSet)
  56. 			{
  57. 				mutationSet.forEach( function(mutation)
  58. 				{
  59. 					for( let x = 0; x < mutation.addedNodes.length; x++ )
  60. 						linkManga( mutation.addedNodes[x] );
  61. 				} );
  62. 			}).observe( document.body, { childList:true, subtree:true } );
  63. 		}
  64. 	}
  65. }
  66. else if( window == window.top )//Don't run if inside an iframe
  67. {
  68.     if( typeof(GM_deleteValue) != "undefined" )
  69.         GM_deleteValue("minFavs");
  70. 	
  71. 	//Prevent added links sometimes being hidden for thumbnails with long titles
  72. 	let style = document.createElement('style');
  73. 	style.type = 'text/css';
  74. 	style.innerHTML = 'li.image-item{ height:auto !important; overflow:visible !important; padding:5px 0px !important } ';
  75. 	document.getElementsByTagName('head')[0].appendChild(style);
  76.  
  77. 	processThumbs( [document] );
  78.  
  79. 	//Monitor for changes caused by other scripts
  80. 	new MutationObserver( function(mutationSet)
  81. 	{
  82. 		mutationSet.forEach( function(mutation){ processThumbs( mutation.addedNodes ); } );
  83. 	}).observe( document.body, { childList:true, subtree:true } );
  84. }
  85.  
  86. //====================================== Functions ======================================
  87.  
  88. function processThumbs(target)
  89. {
  90.     //Take care not to match on profile images, like those shown in the "Following" box on user profiles...
  91.     var xPattern = [ "descendant-or-self::li/a[contains(@href,'mode=medium')]//img[not(@pisas)]",//member_illust.php, response.php, bookmark.php, new_illust.php, member.php...
  92.                      "descendant-or-self::div/figure/div/a/div[contains(@class,'lazy') and not(span) and not(@pisas)]",//search.php, bookmark_new_illust.php -- "contains(@class,'lazy')" prevents double matching on ugoira
  93.                      "descendant-or-self::div/a[contains(@href,'mode=medium')]//img[not(@pisas)]",//response.php (image being responded to), front page, ranking.php
  94.                      "descendant-or-self::div[@class='works_display']//a[contains(@href,'mode=ugoira_view') and not(@pisas)]", //member_illust.php (ugoira image)
  95.                      "descendant-or-self::div[@class='works_display']/a//img[not(@pisas)]"//member_illust.php (single image)
  96.  
  97.                     //"descendant-or-self::div[@class='works_display']/div/img[not(@pisas)]"
  98.                     //"descendant-or-self::li[@class='image-item']/a//img[not(@pisas)]",//bad hits on front page
  99.                     //"descendant-or-self::section/a[contains(@href,'mode=medium')]//img[not(@pisas)]"
  100.                    ];
  101.  
  102. 	let thumbSearch = [], thumbList = [];
  103. 	
  104. 	//Combine the results over all targets to minimize queries by source search
  105. 	for( let i = 0; i < target.length; i++ )
  106. 	{
  107.         for( let j = 0; j < xPattern.length; j++ )
  108.         {
  109.             let xSearch = document.evaluate( xPattern[j], target[i], null, XPathResult.ORDERED_NODE_SNAPSHOT_TYPE, null);
  110.             if( xSearch.snapshotLength > 0 )
  111.             {
  112.                 for( let k = 0; k < xSearch.snapshotLength; k++ )
  113.                 {
  114.                     (thumbSearch[thumbSearch.length] = xSearch.snapshotItem(k)).setAttribute("pisas","done");
  115.                     //console.debug( pixivIllustID( xSearch.snapshotItem(k).src ) );
  116.                 }
  117.                 //console.debug( "PISAS - "+xSearch.snapshotLength+" hit(s) for xPattern["+j+"]: "+xPattern[j] );
  118.             }
  119.         }
  120. 	}
  121. 	
  122. 	for( let i = 0; i < thumbSearch.length; i++ )
  123. 	{
  124. 		let thumbCont, thumbPage = null, thumbImg = thumbSearch[i];
  125. 		
  126. 		for( thumbCont = thumbImg.parentNode; !thumbCont.classList.contains("works_display"); thumbCont = thumbCont.parentNode )
  127. 			if( thumbCont.tagName == "A" )
  128. 			{
  129. 				thumbPage = thumbCont;
  130. 				thumbCont = thumbPage.parentNode;
  131.                 if( thumbCont.parentNode.tagName == "FIGURE" )
  132.                     thumbCont = thumbCont.parentNode.parentNode;//search.php
  133. 				break;
  134. 			}
  135. 		
  136. 		let bookmarkCount = 0, bookmarkLink = thumbCont.querySelector("a[href*='bookmark_detail.php']");
  137. 		let sourceContainer = thumbCont;
  138.  
  139. 		if( thumbImg.tagName == "IMG" )
  140. 		{
  141. 			//Disable lazy loading
  142. 			if( thumbImg.getAttribute("data-src") )
  143. 				thumbImg.src = thumbImg.getAttribute("data-src");
  144. 			
  145. 			//Skip generic restricted thumbs
  146. 			if( thumbImg.src.indexOf("//source.pixiv.net/") >= 0 )
  147. 				continue;
  148. 			
  149. 			//Skip special thumbs except on image pages (daily rankings on main page, ...)
  150. 			if( location.search.indexOf("mode=") < 0 && thumbPage && ( thumbImg.src.indexOf("_100.") > 0 || thumbPage.href.indexOf("_ranking") > 0 ) )
  151. 				continue;
  152. 		}
  153.  
  154. 		if( bookmarkLink )
  155. 		{
  156. 			//Thumb has bookmark info
  157. 			bookmarkCount = parseInt( bookmarkLink.getAttribute("data-tooltip","x").replace(/([^\d]+)/g,'') ) || 1;
  158. 			sourceContainer = bookmarkLink.parentNode;
  159. 		}
  160. 		else if( iqdbURL )
  161. 		{
  162. 			//Thumb doesn't have bookmark info.  Add a fake bookmark link to link with the IQDB.
  163. 			bookmarkLink = document.createElement("a");
  164. 			bookmarkLink.className = "bookmark-count";
  165. 			if( anyBookmarks )
  166. 			{
  167. 				bookmarkLink.className += " ui-tooltip";
  168. 				bookmarkLink.setAttribute("data-tooltip", "0 bookmarks");
  169. 			}
  170. 			
  171. 			//Dummy div to force new line when needed
  172. 			thumbCont.appendChild( document.createElement("div") );
  173. 			thumbCont.appendChild( bookmarkLink );
  174. 		}
  175. 		else
  176. 		{
  177. 			//Dummy div to force new line when needed
  178. 			thumbCont.appendChild( document.createElement("div") );
  179. 		}
  180.  
  181. 		if( anyBookmarks )
  182. 		{
  183. 			favList.push({ thumb: thumbCont, favcount: bookmarkCount });
  184. 		}
  185.  
  186.         if( iqdbURL )
  187.         {
  188.             //For search.php: Check the JSON data to get the thumb URLs.
  189.             let directURL = thumbImg.src || thumbImg.style.backgroundImage.replace(/.*(http[^"]+).*/,'$1');
  190.             if( !directURL )
  191.             {
  192.                 let pid = pixivIllustID( thumbPage.href );
  193.                 for( let j = 0; j < jsonData.length; j++ )
  194.                     if( pid == jsonData[j].illustId )
  195.                     {
  196.                         directURL = jsonData[j].url;
  197.                         break;
  198.                     }
  199.             }
  200.             if( directURL )
  201.             {
  202.                 bookmarkLink.href = iqdbURL+directURL+(thumbPage ? "&fullimage="+thumbPage.href : "");
  203.                 bookmarkLink.innerHTML = "(IQDB)";
  204.             }
  205.         }
  206.  
  207. 		if( addSourceSearch && ( !thumbImg.src || thumbImg.src.indexOf("/novel/") < 0 ) && pixivIllustID( thumbImg.src || thumbImg.href || thumbPage.href ) )
  208. 		{
  209. 			sourceContainer.appendChild( document.createTextNode(" ") );
  210. 			thumbList.push({ link: sourceContainer.appendChild( document.createElement("a") ), pixiv_id: pixivIllustID( thumbImg.src || thumbImg.href || thumbPage.href ), page: -1 });
  211. 		}
  212. 	}
  213. 	
  214. 	sourceSearch( thumbList );
  215. }
  216.  
  217. function pixivIllustID(url)
  218. {
  219.     try{
  220.         return ( url.match(/\/(\d+)(_|\.)[^\/]+$/) || url.match(/illust_id=(\d+)/) )[1];
  221.     }catch(e){
  222.         console.warn("PISAS - pixivIllustID(): Unable to parse ID from URL - " + url);
  223.         return 0;
  224.     }
  225. }
  226.  
  227. function pixivPageNumber(url)
  228. {
  229.     try{
  230.         return url.match(/_p(\d+)(_master\d+)?\./)[1];
  231.     }catch(e){
  232.         return "x";
  233.     }
  234. }
  235.  
  236. function sourceSearch( thumbList, attempt, page )
  237. {
  238. 	//thumbList[index] = { link, id, page? }
  239. 	
  240. 	if( page === undefined )
  241. 	{
  242. 		//First call.  Finish initialization
  243. 		attempt = page = 1;
  244. 		
  245. 		for( let i = 0; i < thumbList.length; i++ )
  246. 		{
  247. 			if( !thumbList[i].status )
  248. 				thumbList[i].status = thumbList[i].link.parentNode.appendChild( document.createElement("span") );
  249. 			thumbList[i].link.textContent = "Searching...";
  250. 			thumbList[i].posts = [];
  251. 		}
  252. 	}
  253. 	
  254. 	if( attempt >= maxAttempts )
  255. 	{
  256. 		//Too many failures (or Downbooru); give up. :(
  257. 		for( let i = 0; i < thumbList.length; i++ )
  258. 		{
  259. 			thumbList[i].status.style.display = "none";
  260. 			if( thumbList[i].link.textContent[0] != '(' )
  261. 				thumbList[i].link.textContent = "(error)";
  262. 			thumbList[i].link.setAttribute("style","color:blue; font-weight: bold;");
  263. 		}
  264. 		return;
  265. 	}
  266. 	
  267. 	//Is there actually anything to process?
  268. 	if( thumbList.length == 0 )
  269. 		return;
  270. 	
  271. 	//Retry this call if timeout
  272. 	var retry = (function(a,b,c){ return function(){ setTimeout( function(){ sourceSearch(a,b,c); }, maxAttempts == 0 ? 0 : 1000 ); }; })( thumbList, attempt + 1, page );
  273. 	var sourceTimer = setTimeout( retry, sourceTimeout*1000 );
  274. 	
  275. 	var idList = [];
  276. 	for( let i = 0; i < thumbList.length; i++ )
  277. 	{
  278. 		thumbList[i].status.textContent = " ["+attempt+"]";
  279. 		if( idList.indexOf( thumbList[i].pixiv_id ) < 0 )
  280. 			idList.push( thumbList[i].pixiv_id );
  281. 	}
  282. 	
  283. 	GM_xmlhttpRequest(
  284. 	{
  285. 		method: "GET",
  286. 		url: danbooruBase+'/posts.json?limit=100&tags=status:any+pixiv:'+idList.join()+'&login='+danbooruLogin+'&api_key='+danbooruApiKey+'&page='+page,
  287. 		onload: function(responseDetails)
  288. 		{
  289. 			clearTimeout(sourceTimer);
  290. 			
  291. 			//Check server response for errors
  292. 			let result = false, status = null;
  293. 			
  294. 			if( /^ *$/.test(responseDetails.responseText) )
  295. 				status = "(error)";//No content
  296. 			else if( responseDetails.responseText.indexOf("<title>Downbooru</title>") > 0 )
  297. 			{
  298. 				addSourceSearch = maxAttempts = 0;//Give up
  299. 				status = "(Downbooru)";
  300. 			}
  301. 			else if( responseDetails.responseText.indexOf("<title>Failbooru</title>") > 0 )
  302. 				status = "(Failbooru)";
  303. 			else try
  304. 			{
  305. 				result = JSON.parse(responseDetails.responseText);
  306. 				if( result.success !== false )
  307. 					status = "Searching...";
  308. 				else
  309. 				{
  310. 					status = "(" + ( result.message || "error" ) + ")";
  311. 					addSourceSearch = maxAttempts = 0;//Give up
  312. 					result = false;
  313. 				}
  314. 			}
  315. 			catch(err) {
  316. 				result = false;
  317. 				status = "(parse error)";
  318. 			}
  319. 			
  320. 			//Update thumbnail messages
  321. 			for( let i = 0; i < thumbList.length; i++ )
  322. 				thumbList[i].link.textContent = status;
  323. 			
  324. 			if( result === false )
  325. 				return retry();//Hit an error; try again?
  326. 			
  327. 			for( let i = 0; i < thumbList.length; i++ )
  328. 			{
  329. 				//Collect the IDs of every post with the same pixiv_id/page as the pixiv image
  330. 				for( let j = 0; j < result.length; j++ )
  331. 					if( thumbList[i].pixiv_id == result[j].pixiv_id && thumbList[i].posts.indexOf( result[j].id ) < 0 && ( thumbList[i].page < 0 || thumbList[i].page == pixivPageNumber( result[j].source ) ) )
  332. 					{
  333. 						thumbList[i].link.title = result[j].tag_string+" user:"+result[j].uploader_name+" rating:"+result[j].rating+" score:"+result[j].score;
  334. 						thumbList[i].posts.push( result[j].id );
  335. 					}
  336. 				
  337. 				if( thumbList[i].posts.length == 1 )
  338. 				{
  339. 					//Found one post; link directly to it
  340. 					thumbList[i].link.textContent = "post #"+thumbList[i].posts[0];
  341. 					thumbList[i].link.href = danbooruBase+"/posts/"+thumbList[i].posts[0];
  342. 					thumbList[i].link.setAttribute("style",styleSourceFound);
  343. 				}
  344. 				else if( thumbList[i].posts.length > 1 )
  345. 				{
  346. 					//Found multiple posts; link to tag search
  347. 					thumbList[i].link.textContent = "("+thumbList[i].posts.length+" sources)";
  348. 					
  349. 					if( location.href.indexOf("mode=manga") > 0 )
  350. 						thumbList[i].link.href = danbooruBase+"/posts?tags=status:any+id:"+thumbList[i].posts;
  351. 					else
  352. 						thumbList[i].link.href = danbooruBase+"/posts?tags=status:any+pixiv:"+thumbList[i].pixiv_id;
  353. 					
  354. 					thumbList[i].link.setAttribute("style",styleSourceFound);
  355. 					thumbList[i].link.removeAttribute("title");
  356. 				}
  357. 			}
  358. 			
  359. 			if( result.length == 100 )
  360. 				sourceSearch( thumbList, attempt + 1, page + 1 );//Max results returned, so fetch the next page
  361. 			else for( let i = 0; i < thumbList.length; i++ )
  362. 			{
  363. 				//No more results will be forthcoming; hide the status counter and set the links for the images without any posts
  364. 				thumbList[i].status.style.display = "none";
  365. 				if( thumbList[i].posts.length == 0 )
  366. 				{
  367. 					thumbList[i].link.textContent = "(no sources)";
  368. 					thumbList[i].link.setAttribute("style",styleSourceMissing);
  369. 				}
  370. 			}
  371. 		},
  372. 		onerror: retry,
  373. 		onabort: retry
  374. 	});
  375. }

欢迎访问Greasy Fork镜像

土豆服务器，请按需使用

镜像地址随时可能被墙，建议加群获取最新地址

加入Q群

QingJ © 2025

镜像随时可能失效，请加Q群300939539或关注我们的公众号极客氢云获取最新地址