☰

Civitai script

Save information when a page is loaded in a same place, then we download content, generate json with alls informations

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

// ==UserScript==
// @name Civitai script
// @namespace http://tampermonkey-script-exemple
// @version 1
// @description Save information when a page is loaded in a same place, then we download content, generate json with alls informations
// @match https://civitai.com/models/*
// @grant GM_getValue
// @grant GM_setValue
// @author viatana35
// @license MIT
// ==/UserScript==

async function findLinks() {
  const links = document.querySelectorAll('a');
  const results = [];

  links.forEach(link => {
    if (link.classList.contains('mantine-UnstyledButton-root', 'mantine-Button-root') &&
        link.type === 'button' &&
        link.getAttribute('data-button') === 'true' &&
        link.getAttribute('download') === '') {

      const hrefParts = link.href.split('/');
      const lastPart = hrefParts[hrefParts.length - 1];

      results.push({ href: link.href, id: lastPart });
    }
  });

  return results[0];
}

async function saveLinks() {
  const currentLinks = await findLinks();
  const storedLinksJSON = await GM_getValue('storedLinks', '[]');
  const storedLinks = JSON.parse(storedLinksJSON);

  if (storedLinks.some(link => link.id === currentLinks.id)) {
    console.log('Link already stored');
    return;
  }

  const newLinks = [...storedLinks, currentLinks];
  await GM_setValue('storedLinks', JSON.stringify(newLinks));
}

window.addEventListener('load', function() {
  saveLinks();
});

// Creation of the button to download the stored content
const header = document.querySelector('header');
const downloadButton = document.createElement('button');
downloadButton.className = 'mantine-Button-inner';
downloadButton.textContent = 'Download stored content';
downloadButton.addEventListener('click', async () => {
  const storedLinksJSON = await GM_getValue('storedLinks', '[]');
  const storedLinks = JSON.parse(storedLinksJSON);
  const data = 'data:text/json;charset=utf-8,' + encodeURIComponent(JSON.stringify(storedLinks));
  const link = document.createElement('a');
  link.setAttribute('href', data);
  link.setAttribute('download', 'storedLinks.json');
  document.body.appendChild(link);
  link.click();
  document.body.removeChild(link);
});
header.appendChild(downloadButton);

// Creation of the button to clear the stored content
const clearButton = document.createElement('button');
clearButton.className = 'mantine-Button-inner';
clearButton.textContent = 'Erase Stored Content';
clearButton.addEventListener('click', () => {
  if (confirm('Are you sure you want to erase stored content?')) {
    GM_setValue('storedLinks', '[]');
  }
});
header.appendChild(clearButton);