Civitai script

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

À data de 19/04/2023. Ver a versão mais recente.

Terá de instalar uma extensão como Tampermonkey, Greasemonkey ou Violentmonkey para instalar este script.

You will need to install an extension such as Tampermonkey or Violentmonkey to install this script.

Terá de instalar uma extensão como Tampermonkey ou Violentmonkey para instalar este script.

Terá de instalar uma extensão como Tampermonkey ou Userscripts para instalar este script.

Terá de instalar uma extensão como Tampermonkey para instalar este script.

Terá de instalar uma extensão de gestão de scripts de utilizador para instalar este script.

(Já tenho um gestor de scripts de utilizador, deixe-me instalá-lo!)

Terá de instalar uma extensão como Stylus para instalar este estilo.

Terá de instalar uma extensão como Stylus para instalar este estilo.

Terá de instalar uma extensão como Stylus para instalar este estilo.

Terá de instalar uma extensão de gestão de estilos de utilizador para instalar este estilo.

Terá de instalar uma extensão de gestão de estilos de utilizador para instalar este estilo.

Terá de instalar uma extensão de gestão de estilos de utilizador para instalar este estilo.

(Já tenho um gestor de estilos de utilizador, deixe-me instalá-lo!)

// ==UserScript==
// @name Civitai script
// @namespace http://tampermonkey-script-exemple
// @version 1.3
// @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/*
// @match https://civitai.com*
// @grant GM_getValue
// @grant GM_setValue
// @grant GM_registerMenuCommand
// @author viatana35
// @license MIT
// ==/UserScript==

function addCloseTabOption() {
    let closeTab = GM_getValue('closeTab', false);

    GM_registerMenuCommand('Close Tab', () => {
    closeTab = !closeTab;
    GM_setValue('closeTab', closeTab);

    const statusMessage = closeTab ? 'enabled' : 'disabled';
    console.log(`Close tab option is now ${statusMessage}.`);
  });
    console.log("closetab");
  return closeTab;
}


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 closeTab = addCloseTabOption();
  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));
  if (closeTab) {
    window.close(); // Close the current tab
  }
}

window.addEventListener('load', function() {
  if (window.location.href.startsWith('https://civitai.com/models/'))
  {
      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);

      // Creation of the button to delete this page
      const deletePageButton = document.createElement('button');
      deletePageButton.className = 'mantine-Button-inner';
      deletePageButton.textContent = 'Delete this page';
      deletePageButton.addEventListener('click', async () => {
          const currentLinks = await findLinks();
          const storedLinksJSON = await GM_getValue('storedLinks', '[]');
          const storedLinks = JSON.parse(storedLinksJSON);

          const index = storedLinks.findIndex(link => link.id === currentLinks.id);
          if (index > -1) {
              if (confirm('Are you sure you want to delete the stored content of this page?')) {
                  storedLinks.splice(index, 1);
                  await GM_setValue('storedLinks', JSON.stringify(storedLinks));
              }
          } else {
              alert('There is no stored content for this page.');
          }
      });
      header.appendChild(deletePageButton);
  }
  else
  {
     addCloseTabOption();
  }
  
});