Gelbooru Tag Copyer

复制 Gelbooru 的 tags,支持自定义选择 tag 类型 / Quickly copy tags from gelbooru's post so you can use thoes tags to generate AI images.

Tendrás que instalar una extensión para tu navegador como Tampermonkey, Greasemonkey o Violentmonkey si quieres utilizar este script.

Necesitarás instalar una extensión como Tampermonkey o Violentmonkey para instalar este script.

Necesitarás instalar una extensión como Tampermonkey o Violentmonkey para instalar este script.

Necesitarás instalar una extensión como Tampermonkey o Userscripts para instalar este script.

Necesitará instalar una extensión como Tampermonkey para instalar este script.

Necesitarás instalar una extensión para administrar scripts de usuario si quieres instalar este script.

(Ya tengo un administrador de scripts de usuario, déjame instalarlo)

Necesitará instalar una extensión como Stylus para instalar este estilo.

Necesitará instalar una extensión como Stylus para instalar este estilo.

Necesitará instalar una extensión como Stylus para instalar este estilo.

Necesitará instalar una extensión del gestor de estilos de usuario para instalar este estilo.

Necesitará instalar una extensión del gestor de estilos de usuario para instalar este estilo.

Necesitará instalar una extensión del gestor de estilos de usuario para instalar este estilo.

(Ya tengo un administrador de estilos de usuario, déjame instalarlo)

// ==UserScript==
// @name         Gelbooru Tag Copyer
// @namespace    https://greasyfork.org/zh-CN/scripts/439308
// @version      1.5
// @description  复制 Gelbooru 的 tags,支持自定义选择 tag 类型 / 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);

        if(checkedTagsType.includes('character')) {
            checkedTagsType.push("attrTag");
        }

        // sort tags by 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");

        navigator.clipboard.writeText(tags).then(() => {
            window.notice && window.notice('Tags copied.')
        }).catch((e) => {
            // alert('failed copy tags:', e)
            prompt("copied tags:", tags);
        })
    });
})();