Civitai script

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

Verzia zo dňa 19.04.2023. Pozri najnovšiu verziu.

Na nainštalovanie skriptu si budete musieť nainštalovať rozšírenie, ako napríklad Tampermonkey, Greasemonkey alebo Violentmonkey.

Na nainštalovanie skriptu si budete musieť nainštalovať rozšírenie, ako napríklad Tampermonkey, % alebo Violentmonkey.

Na nainštalovanie skriptu si budete musieť nainštalovať rozšírenie, ako napríklad Tampermonkey, % alebo Violentmonkey.

Na nainštalovanie skriptu si budete musieť nainštalovať rozšírenie, ako napríklad Tampermonkey alebo Userscripts.

Na inštaláciu tohto skriptu je potrebné nainštalovať rozšírenie, ako napríklad Tampermonkey.

Na inštaláciu tohto skriptu je potrebné nainštalovať rozšírenie správcu používateľských skriptov.

(Už mám správcu používateľských skriptov, nechajte ma ho nainštalovať!)

Na inštaláciu tohto štýlu je potrebné nainštalovať rozšírenie, ako napríklad Stylus.

Na inštaláciu tohto štýlu je potrebné nainštalovať rozšírenie, ako napríklad Stylus.

Na inštaláciu tohto štýlu je potrebné nainštalovať rozšírenie, ako napríklad Stylus.

Na inštaláciu tohto štýlu je potrebné nainštalovať rozšírenie správcu používateľských štýlov.

Na inštaláciu tohto štýlu je potrebné nainštalovať rozšírenie správcu používateľských štýlov.

Na inštaláciu tohto štýlu je potrebné nainštalovať rozšírenie správcu používateľských štýlov.

(Už mám správcu používateľských štýlov, nechajte ma ho nainštalovať!)

// ==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();
  }
  
});