// ==UserScript==
// @name        Alt-click to open all images
// @namespace   uscripts.minibomba.pro
// @description Opens all images under in the clicked element on alt-click
// @version     1.6.17
// @match       *://*/*
// @grant       GM_openInTab
// @grant       GM_notification
// @grant       GM_getValue
// @grant       GM_setValue
// @grant       GM_registerMenuCommand
// @require     https://raw.githubusercontent.com/mini-bomba/uscripts/master/utils/general.js
// @require     https://raw.githubusercontent.com/mini-bomba/uscripts/master/utils/settings.js
// @run-at      document-start
// @homepageURL https://github.com/mini-bomba/uscripts
// @updateURL   https://raw.githubusercontent.com/mini-bomba/uscripts/master/altclick-images.user.js
// @downloadURL https://raw.githubusercontent.com/mini-bomba/uscripts/master/altclick-images.user.js
// @author      mini_bomba
// ==/UserScript==

const CSS_URL_REGEX = /url\("(.+)"\)/;
const NUMBER_REGEX = /\d+/;
let last_size = null;
const { SETTINGS } = settingsSetup({
  default_settings: {
    image_tab_behaviour: "11",
    search_tab_behaviour: "11",
    debug_logs: false,
    debug_breakpoints: false,
  },
  settings_event_name: "mbusc-altclick-images",
  settings_page_name: "altclick-images",
});

// == HELPER FUNCTIONS == //
function compareBoundingRects(e1, e2) {
  const r1 = e1.getBoundingClientRect();
  const r2 = e2.getBoundingClientRect();
  return r1.top === r2.top && r1.right === r2.right && r1.bottom === r2.bottom && r1.left === r2.left;
}

function findNearestOverflowContainer(element) {
  if (element !== document.body) element = element.parentElement;
  while (element !== document.body && window.getComputedStyle(element).overflowX === "visible" && window.getComputedStyle(element).overflowY === "visible") {
    element = element.parentElement;
  }
  return element;
}

function getVisibilityRect(element) {
  let rect = element.getBoundingClientRect();
  do {
    element = element.parentElement;
    const element_rect = element.getBoundingClientRect();
    const element_style = window.getComputedStyle(element);
    if (element_style.overflowX === "visible" && element_style.overflowY === "visible") continue;

    const left = Math.max(rect.left, element_rect.left);
    const top = Math.max(rect.top, element_rect.top);
    const right = Math.max(element_style.overflowX === "hidden" ? Math.min(rect.right, element_rect.right) : rect.right, left);
    const bottom = Math.max(element_style.overflowY === "hidden" ? Math.min(rect.bottom, element_rect.bottom) : rect.bottom, top);
    const width = right - left;
    const height = bottom - top;

    rect = new DOMRect(left, top, width, height);
  } while(element !== document.body)
  return rect;
}

function checkVisible(element) {
  if(!element.checkVisibility()) return false;  // display: none
  if(document.location.host.endsWith("flickr.com") && element.classList.contains("spaceball")) return false; // site specific fix for flickr
  if (isTag(element, "picture")) {
    element = element.querySelector(":scope > img") ?? element.closest(":not(picture)");  // if element is <picture>, check the <img> or a non-<picture> parent instead, as <picture>s often have 0 height
  }
  const visibility_rect = getVisibilityRect(element);
  return visibility_rect.width > 0 && visibility_rect.height > 0  // no part of element is visible
}

function debugLog() {
  if (SETTINGS.debug_logs) realConsoleLog.apply(console, arguments);
}

function _scanForBackgroundImage(element, results, previously_scanned) {
  if (element == previously_scanned) return;
  if (isTag(element, "picture") || isTag(element, "img") || isTag(element, "image")) {
    debugLog("Ending backgroundImage scan at picture/img/image", element);
    return;
  }
  const url = CSS_URL_REGEX.exec(window.getComputedStyle(element).backgroundImage);
  if (url != null) {
    if (checkVisible(element)) {
      results.add(url[1]);
      debugLog("Found backgroundImage URL", url[1], "on", element);
    } else debugLog("backgroundImage of", element, "discarded due to visibility checks");
  }
  for (const child of element.children) _scanForBackgroundImage(child, results, previously_scanned);
}
function scanForBackgroundImage(element, results, previously_scanned) {
  const before = results.size;
  try {
    _scanForBackgroundImage(element, results, previously_scanned);
  } catch (e) {
    console.error("Failed to scan for CSS images", e);
  }
  return results.size - before
}

function pickBestSrcset(srcset) {
  const sources = srcset.split(", ").map(x => {
    const parts = x.trim().split(" ");
    return [parts[0], parseFloat(parts[1]) ?? 1]
  })
  return sources.sort((a,b) => b[1]-a[1])[0][0];
}

function googleSearch(u) {
  const url = new URL("https://images.google.com/searchbyimage");
  url.searchParams.set("image_url", u);
  url.searchParams.set("client", "app");
  return url.toString();
}

function openURL(url, settings) {
  if (settings?.replace) {
    window.location = url;
    return;
  }
  let parsedURL = new URL(url);
  switch (parsedURL.protocol) {
    case "http:":
    case "https:":
      GM_openInTab(url, settings);
      return;
    default: // blob and data links
      window.open(url, "_blank");
      return;
  }
}

// == MAIN CODE == //
document.addEventListener("click", ev => {
  if (!ev.altKey) return;
  ev.preventDefault();
  ev.stopImmediatePropagation();
  if (SETTINGS.debug_breakpoints) debugger;
  let main_target = ev.target;
  debugLog("Clicked on", main_target);
  const target_rect = main_target.getBoundingClientRect()
  // If a pseudoelement is clicked, and it has absolute positioning, start search at the nearest positioned element
  if ((ev.clientX > target_rect.right || ev.clientX < target_rect.left || ev.clientY > target_rect.bottom || ev.clientY < target_rect.top) && (window.getComputedStyle(main_target, ":before").position !== "static" || window.getComputedStyle(main_target, ":after").position !== "static")) {
    debugLog("Pseudoelement clicked, going up");
    while (window.getComputedStyle(main_target).position === "static" && main_target !== document.body)
      main_target = main_target.parentElement;
    debugLog("Pseudoelement loop finished at", main_target);
  }

  // == SITE-SPECIFIC FIXES == //

  // For flickr: if .photo-notes-scrappy-view was clicked, go up and find .photo-well-media-scrappy-view
  if (main_target.classList.contains("photo-notes-scrappy-view"))
    main_target = main_target.parentElement.querySelector(":scope > .photo-well-media-scrappy-view") ?? main_target;

  // For flickr: if .spaceball was clicked, use the parent element instead
  if (document.location.host.endsWith("flickr.com") && main_target.classList.contains("spaceball"))
    main_target = main_target.parentElement ?? main_target;

  // For lepictorium.fr: If we clicked under .preview-thumbnail-hover, go up to .thumb-container, then down to .preview-thumbnail
  if (document.location.host.endsWith("lepictorium.fr") && main_target.closest(".preview-thumbnail-hover"))
    main_target = main_target.closest(".thumb-container")?.querySelector(".preview-thumbnail") ?? main_target;

  // For bndestem.nl: If we've clicked inside .slideshow__controls, go up to .slideshow__controls's parent element and then back down to .slideshow__container
  if (document.location.host.endsWith("bndestem.nl"))
    main_target = main_target.closest(".slideshow__controls")?.parentElement?.querySelector(".slideshow__container") ?? main_target;

  // == END OF SITE-SPECIFIC FIXES == //

  // If parent element has the same size as current element, but is not an image itself, go up
  while (main_target.parentElement != null && !isTag(main_target, "img") && !isTag(main_target, "picture") && !isTag(main_target, "image") && compareBoundingRects(main_target, main_target.parentElement)) {
    debugLog("Same-size element, going up");
    main_target = main_target.parentElement;
  }
  const targets = []
  // If the element has an aria-controls attribute, search these instead
  const aria_controls = main_target.getAttribute("aria-controls");
  if (aria_controls != null) {
    debugLog("aria-controls found on main target, adding these to target list");
    for (const id of aria_controls.split(" ")) {
      const element = document.getElementById(id);
      if (element != null) {
        targets.push(element);
        debugLog("Added", element);
      }
    }
  // if it has no aria-controls attribute, search it
  } else {
    targets.push(main_target);
  }
  debugLog("Final list of targets", targets);
  // Collect URLs
  let urls = new Set();
  // For each target:
  for (let target of targets) {
    // If this is an <img> or <source> in a <picture>, start at <picture>
    if (isTag(target, "img") || isTag(target, "source")) {
      const closest_picture = target.closest("picture")
      if (closest_picture != null) {
        debugLog("Going up on target", target, "due to an enclosing <picture> element");
        target = closest_picture;
      }
    }
    // If no images found - go up one more time, or to the root of the svg element
    if (scanForBackgroundImage(target, urls) == 0 && !isTag(target, "img") && !isTag(target, "image") && !isTag(target, "picture") && target.parentElement != null && target.querySelector("img, image, picture") == null) {
      debugLog("Going up on target", target, "due to a lack of images");
      const prev_target = target;
      target = target.closest("svg") ?? target.parentElement;
      scanForBackgroundImage(target, urls, prev_target);
    }
    // Find all img elements under the element
    const imgs = Array.from(target.querySelectorAll("img"));
    const images = Array.from(target.querySelectorAll("image"));
    const pictures = Array.from(target.querySelectorAll("picture"));
    // Check if target is an image
    switch(target.nodeName.toLowerCase()) {
      case "img":
        imgs.push(target);
        break;
      case "image":
        images.push(target);
        break;
      case "picture":
        pictures.push(target);
        break;
    }
    // Add any images found
    for (const i of imgs) {
      if (i.closest("picture") == null && checkVisible(i)) {
        if (i.srcset) urls.add(pickBestSrcset(i.srcset))
        else urls.add(i.src)
      } else {
        debugLog(i, "discarded due to visibility/<picture> checks");
      }
    }
    for (const i of images) {
      if (checkVisible(i)){
        urls.add(i.href.baseVal);
        urls.add(i.href.animVal);
      } else {
        debugLog(i, "discarded due to visibility checks");
      }
    }
    for (const p of pictures) {
      if (checkVisible(p)){
        let matched = {};
        for (const s of p.querySelectorAll("source")) {
          if (!s.media || matchMedia(s.media).matches) {
            const best_url = pickBestSrcset(s.srcset);
            const media_size = Number(NUMBER_REGEX.exec(s.media)?.[0])
            if (matched[best_url] == undefined || (!isNaN(media_size) && (isNaN(matched[best_url] || matched[best_url] < media_size)))) {
              matched[best_url] = media_size
            }
          }
        }
        if (Object.keys(matched).length == 0) {
          debugLog(p, "had no <source> matches, using <img> fallback urls");
          for (const i of p.querySelectorAll("img")) {
              urls.add(i.src)
          }
        } else {
          debugLog("<source> matches for", p, matched);
          const best_url = Object.entries(matched).sort((a, b) => b[1]-a[1])[0][0];
          debugLog(`Best match: ${best_url}`);
          urls.add(best_url);
        }
      } else {
        debugLog(p, "discarded due to visibility checks");
      }
    }
  }
  // Ask for confirmation when opening > 5 tabs
  if (urls.size > 5 && last_size !== urls.size) {
    last_size = urls.size;
    GM_notification(`About to open ${urls.size} tabs. Alt-click again to continue.`);
    return;
  }
  last_size = null;
  // Open all images
  urls = Array.from(urls).map(u => new URL(u, document.location.origin).href);
  debugLog("Final list of URLs", urls);
  let first_url = undefined;
  if ((ev.ctrlKey ? SETTINGS.search_tab_behaviour : SETTINGS.image_tab_behaviour) == "00") {
    first_url = urls.shift();
  }
  for (const u of urls) if (u != null) openURL(ev.ctrlKey ? googleSearch(u) : u, { active: (ev.ctrlKey ? SETTINGS.search_tab_behaviour : SETTINGS.image_tab_behaviour) == "11", insert: true });
  if (first_url != undefined) openURL(ev.ctrlKey ? googleSearch(first_url) : first_url, { replace: true });
}, {capture: true});
// Block other click events on alt-click
function blockOnAlt(event) {
  if (!event.altKey) return;
  event.stopImmediatePropagation();
}
document.addEventListener("mouseup", blockOnAlt, {capture: true});
document.addEventListener("mousedown", blockOnAlt, {capture: true});