Civitai script

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

Dovrai installare un'estensione come Tampermonkey, Greasemonkey o Violentmonkey per installare questo script.

Dovrai installare un'estensione come Tampermonkey o Violentmonkey per installare questo script.

Dovrai installare un'estensione come Tampermonkey o Violentmonkey per installare questo script.

Dovrai installare un'estensione come Tampermonkey o Userscripts per installare questo script.

Dovrai installare un'estensione come ad esempio Tampermonkey per installare questo script.

Dovrai installare un gestore di script utente per installare questo script.

(Ho già un gestore di script utente, lasciamelo installare!)

Dovrai installare un'estensione come ad esempio Stylus per installare questo stile.

Dovrai installare un'estensione come ad esempio Stylus per installare questo stile.

Dovrai installare un'estensione come ad esempio Stylus per installare questo stile.

Dovrai installare un'estensione per la gestione degli stili utente per installare questo stile.

Dovrai installare un'estensione per la gestione degli stili utente per installare questo stile.

Dovrai installare un'estensione per la gestione degli stili utente per installare questo stile.

(Ho già un gestore di stile utente, lasciamelo installare!)

// ==UserScript==
// @name Civitai script
// @namespace http://tampermonkey-script-exemple
// @version 1.3.1.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/*
// @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');
    if (closeTab) {
        window.close(); // Close the current tab
    }
    else
    {
        return;
    }
    
  }

  const newLinks = [...storedLinks, currentLinks];
  await GM_setValue('storedLinks', JSON.stringify(newLinks));
  if (closeTab) {
    window.close(); // Close the current tab
  }
}

function clickPreview() {
    // get all elements with class 'mantine-Alert-root'
    const alertElements = document.querySelectorAll('.mantine-Alert-root');

    //if there are no alert elements, return
    if (!alertElements.length) {
        return false;
    }
    else {
        // loop through all alert elements
        alertElements.forEach(alertElement => {
        // find the span element within the alert element
        const spanElement = alertElement.querySelector('span');

        // click on the span element, if found
        if (spanElement) {
            //if the texte of the span cntain : 'Notify me when it's available.'
            if(spanElement.textContent == 'Notify me when it\'s available.'){
                spanElement.click();
            }
        }
        });
        //make an alert to show that the element has been added to the waitlist )
        //alert('element added to waitlist');
        return true;
    }
}




window.addEventListener('load', function() {
  if (window.location.href.startsWith('https://civitai.com/models/'))
  {
      if (!clickPreview())
      {
          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
      {
          const closeTab = addCloseTabOption();
            if (closeTab) {
                window.close(); // Close the current tab
            }
      }
  }
  else
  {
     addCloseTabOption();
  }

});