☰

Derpibooru - Download All

Adds a download button that lets you download all images from the current search query

  1. // ==UserScript==
  2. // @name        Derpibooru - Download All
  3. // @namespace   Selbi
  4. // @include     http*://*derpibooru.org/*
  5. // @version     2.0.1
  6. // @description Adds a download button that lets you download all images from the current search query
  7. // ==/UserScript==
  8.  
  9. const states = {
  10.   ready: 0,
  11.   fetching: 1,
  12.   readyForDownload: 2,
  13.   downloading: 3
  14. };
  15. var state = states.ready;
  16.  
  17. var imageLinks = [];
  18.  
  19. let randomImageButton = document.querySelector("section.block__header > div:last-child > a:first-child");
  20. let downloadButton = randomImageButton.cloneNode(true);
  21.  
  22. let dlButtonText = downloadButton.querySelector("span");
  23. let dlButtonClasses = downloadButton.querySelector("i");
  24. downloadButton.onclick = function() {
  25.   if (state === states.ready) {
  26.     let params = new URLSearchParams(window.location.search);
  27.     let page = params.get("page");
  28.     if (page != null && parseInt(page) > 1) {
  29.       alert("Warning: You are not on page 1 of the gallery! Only images starting at this page will be downloaded.")
  30.     }
  31.     
  32.     state = states.fetching;
  33.     dlButtonClasses.classList = "fa fa-spinner fa-spin fa-pulse";
  34.     
  35.     fetchLinksForPage(document);
  36.     
  37.     state = states.readyForDownload;
  38.     dlButtonClasses.classList = "fa fa-check";
  39.     dlButtonText.innerHTML = "Download Ready! (" + imageLinks.length + ")";
  40.   } else if (state === states.fetching) {
  41.     alert("Please wait until all links have been fetched!");
  42.   } else if (state === states.readyForDownload) {
  43.     downloadImages(imageLinks);
  44.   } else if (state === downloading) {
  45.     alert("Download is currently in progress, please wait for its completion!") 
  46.   }
  47. };
  48.  
  49. downloadButton.href = "#";
  50. downloadButton.title = "Download All Images";
  51. dlButtonClasses.classList = "fa fa-download";
  52. dlButtonText.innerHTML = "Download All";
  53.  
  54. randomImageButton.before(downloadButton);
  55.  
  56. var parser = new DOMParser();
  57.  
  58. function fetchLinksForPage(page) {
  59.   
  60.   // Get images of current page
  61.   let images = page.querySelectorAll(".image-container");
  62.   for (img of images) {    
  63.     let dataUris = img.getAttribute("data-uris");
  64.     let full = JSON.parse(dataUris)["full"];
  65.     let download = full.replace("/view/", "/download/"); // god I hope this never breaks
  66.     imageLinks.push(download);
  67.     dlButtonText.innerHTML = "Fetching Links... (" + imageLinks.length + ")";
  68.   }
  69.   
  70.   // Go to next page, if there is any
  71.   let nextButton = page.querySelector(".js-next");
  72.   if (nextButton != null) {
  73.     let nextPageHref = nextButton.href;
  74.  
  75.     let xhr = new XMLHttpRequest();
  76.     xhr.open("GET", nextPageHref, false);
  77.     xhr.send();
  78.     let responseText = xhr.responseText;
  79.     let responseDom = parser.parseFromString(responseText, 'text/html');
  80.     
  81.     fetchLinksForPage(responseDom);
  82.   }
  83. }
  84.  
  85. function downloadImages(links) {
  86.   for (link of links) {
  87. 	  window.open(link, '_blank');
  88.   }
  89.   state = states.readyForDownload;
  90. }