Gelbooru Tag Copyer

Quickly copy tags from gelbooru's post so you can use thoes tags to generate AI images.
Version au 14/06/2024. Voir la dernière version.
Poser une question, écrire un avis, ou signaler le script.
Enveloppe les lignes
// ==UserScript==
// @name         Gelbooru Tag Copyer
// @namespace    https://greasyfork.org/zh-CN/scripts/439308
// @version      1.3
// @description  Quickly copy tags from gelbooru's post so you can use thoes tags to generate AI images.
// @author       3989364
// @include      *://gelbooru.com/index.php*
// @icon         https://gelbooru.com/favicon.ico
// @grant        none
// @license MIT

// ==/UserScript==

(function () {
  "use strict";
  const tagListEle = document.querySelector("#tag-list");

  const DEFAULT_CHECKED_TAGS = ["general", "copyright", "character"];
  const DEFAULT_QUALITY_TAG = ''; // `((masterpiece)), (((best quality))), ((ultra-detailed)), ((illustration)), (an extremely delicate and beautiful), `;

  const attriTags = ["hair", "eyes", "dress", "sleeves", "bow"];

  const EXCLUDED_TAGS = ["censor", "out-of-frame", "speech bubble"];

  function initUI(tagListEle, defultCheckedTags) {
    /**
     * @param tagListEle
     * @returns [String, ...]
     */
    function parseTags(tagListEle) {
      const tags = Object.create(null);

      tagListEle
        .querySelectorAll('li[class^="tag-type"]')
        .forEach((tagItem) => {
          const tagEle = tagItem.querySelector(".sm-hidden + a");
          const tag = tagEle.textContent;

          const tagCount = parseInt(tagEle.nextElementSibling.textContent) || 0;

          const tagType = tagItem.className.replace("tag-type-", "");
          if (!tags[tagType]) {
            tags[tagType] = [];
          }

          tags[tagType].push({ tag, tagCount });
        });

      // sort general tags by count
      if ("general" in tags) {
        tags["general"].sort((a, b) => b.tagCount - a.tagCount);
      }

      for (const key in tags) {
        tags[key] = tags[key].map((item) => item.tag);
      }

      // add attr tag
      tags["attrTag"] = [];
      return tags;
    }

    function createTagCheckbox(tagType, checked = false) {
      const el = document.createElement("div");

      el.innerHTML = `
        <input
        type="checkbox"
        name="${tagType}"
        ${checked ? "checked" : ""}>${tagType}
    `;

      el.style.marginBottom = "0.15rem";
      el.style.display = "flex";
      el.style.alignItems = "center";

      return [el, el.querySelector("input")];
    }

    const tagCheckboxList = [];
    const tagsObj = parseTags(tagListEle);

    // emphasis character
    if ("character" in tagsObj) {
      tagsObj["character"] = tagsObj["character"].map(
        (character) => `${character}`
      );
    }

    // exclude special tags & resort tag
    if ("general" in tagsObj) {
      tagsObj["attrTag"] = tagsObj["general"].filter((tag) =>
        attriTags.some((attrTag) => tag.includes(attrTag))
      );

      tagsObj["general"] = tagsObj["general"]
        .filter(
          (name) => !EXCLUDED_TAGS.some((exTags) => name.includes(exTags))
        )
        .filter((tag) => !attriTags.some((attrTag) => tag.includes(attrTag)));
    }

    const tagCheckboxContainer = document.createElement("li");

    for (const tagType in tagsObj) {
      if (tagType === "attrTag") {
        continue;
      }

      tagsObj[tagType] = tagsObj[tagType].map(item => item.replaceAll('(', '\\(').replaceAll(')', '\\)'))
      // console.log(tagsObj)
      const [wrapper, ckbox] = createTagCheckbox(
        tagType,
        defultCheckedTags.includes(tagType)
      );

      tagCheckboxContainer.appendChild(wrapper);
      tagCheckboxList.push(ckbox);
    }

    const copyBtn = document.createElement("button");

    copyBtn.innerText = "Copy";
    tagCheckboxContainer.appendChild(copyBtn);
    tagListEle.insertBefore(tagCheckboxContainer, tagListEle.firstChild);

    return {
      copyBtn,
      tagCheckboxList,
      tagsObj,
    };
  }

  const ui = initUI(tagListEle, DEFAULT_CHECKED_TAGS);

  const tagTypeOrderMap = {
    character: 1,
    copyright: 1,
    artist: 1,
    attrTag: 1,
    general: 2,
  };

  ui.copyBtn.addEventListener("click", () => {
    const checkedTagsType = ui.tagCheckboxList
      .filter((item) => item.checked)
      .map((item) => item.name);

    checkedTagsType.push("attrTag");

    // sort tags type
    checkedTagsType.sort((a, b) => {
      return tagTypeOrderMap[a] - tagTypeOrderMap[b];
    });

    const tags =
      DEFAULT_QUALITY_TAG +
      "\n" +
      checkedTagsType
        .map((type) => {
          return ui.tagsObj[type].join(", ");
        })
        .join(",\n");

    prompt("copied tags:", tags);
  });
})();