☰

Kemono助手

提供更好的Kemono使用体验

As of 15/03/2024. See the latest version.

  1. // ==UserScript==
  2. // @name         Kemono助手
  3. // @version      2.0.18
  4. // @description  提供更好的Kemono使用体验
  5. // @author       ZIDOUZI
  6. // @match        https://*.kemono.party/*
  7. // @match        https://*.kemono.su/*
  8. // @icon         https://kemono.su/static/favicon.ico
  9. // @require      https://cdn.jsdelivr.net/npm/sweetalert2@11
  10. // @require      https://kit.fontawesome.com/101092ae56.js
  11. // @grant        GM_xmlhttpRequest
  12. // @grant        GM_getValue
  13. // @grant        GM_setValue
  14. // @grant        GM_registerMenuCommand
  15. // @grant        GM_unregisterMenuCommand
  16. // @license      GPL-3.0
  17. // @namespace https://greasyfork.org/users/448292
  18. // ==/UserScript==
  19.  
  20. (async function () {
  21.     'use strict';
  22.     await main().catch(function (error) {
  23.         console.log(error)
  24.         alert(`Kemono助手发生错误: ${error.status} ${error.statusText}`)}
  25.     );
  26. })();
  27.  
  28. async function main() {
  29.  
  30.     const language = navigator.language || navigator.userLanguage;
  31.  
  32.     const domain = window.location.href.match(/https:\/\/([^/]+)/)[1];
  33.  
  34.     const menuId = GM_registerMenuCommand('Kemono助手设置', showDialog);
  35.  
  36.     const data = await (async () => {
  37.         let _vimMode = GM_getValue('vimMode', false)
  38.         let _rpc = GM_getValue('rpc', 'http://localhost:6800/jsonrpc')
  39.         let _token = GM_getValue('token', '')
  40.         let _dir = GM_getValue('dir', '')
  41.         let _first = GM_getValue('first', true)
  42.         if (_dir === '' && !_first) {
  43.             try {
  44.                 _dir = await fetchDownloadDir(_rpc, _token) + "/kemono/{service}-{artist_id}/{post}-{post_id}/{index}_{auto_named}.{extension}";
  45.             } catch (e) {
  46.                 // ignore
  47.             }
  48.             GM_setValue('dir', _dir);
  49.         }
  50.         return {
  51.             get vimMode() {
  52.                 return _vimMode;
  53.             }, set vimMode(value) {
  54.                 _vimMode = value;
  55.                 GM_setValue('vimMode', value);
  56.             }, get rpc() {
  57.                 return _rpc;
  58.             }, set rpc(value) {
  59.                 _rpc = value;
  60.                 GM_setValue('rpc', value);
  61.             }, get token() {
  62.                 return _token;
  63.             }, set token(value) {
  64.                 _token = value;
  65.                 GM_setValue('token', value);
  66.             }, get dir() {
  67.                 return _dir;
  68.             }, set dir(value) {
  69.                 // if (/[:*?"<>|]/g.test(value)) {
  70.                 //     alert('下载目录中不能包含以下字符： : * ? " < > |\n您可以选择换为全角字符')
  71.                 //     return;
  72.                 // }
  73.                 _dir = value;
  74.                 GM_setValue('dir', value);
  75.             }, format(date) {
  76.                 return date.toISOString().split('.')[0].replace('T', ' ').replaceAll(':', '-')
  77.             }, formatDir(post, index, name, extension, padStart = 3) {
  78.                 const indexString = index.toString().padStart(padStart, '0');
  79.                 const shouldReplace = /^([0-9a-fA-F]{4}-?){8}$/.test(name)
  80.                     || (/^[a-zA-Z0-9]+$/.test(name) && post.service === "fanbox") || name.length + extension.length > 255;
  81.                 if (name.length + extension.length > 255) name = name.slice(0, 255 - extension.length);
  82.                 return _dir
  83.                     .replaceAll(/\{js:(.*)#}/g, function (_, code) {
  84.                         return eval(code);
  85.                     })
  86.                     .replaceAll(/[\/\\]/g, '≸∱').replaceAll(':', '∱≸')
  87.                     .replaceAll('{service}', post.service)
  88.                     .replaceAll('{artist_id}', post.user)
  89.                     .replaceAll('{date}', this.format(new Date(Date.parse(post.published))))
  90.                     .replaceAll('{post}', post.title)
  91.                     .replaceAll('{post_id}', post.id)
  92.                     .replaceAll('{time}', this.format(new Date()))
  93.                     .replaceAll('{index0}', indexString)
  94.                     .replaceAll('{index}', index === 0 ? '' : indexString)
  95.                     .replaceAll('{auto_named}', shouldReplace ? indexString : name)
  96.                     .replaceAll('{name}', name)
  97.                     .replaceAll('{extension}', extension)
  98.                     // avoid illegal characters in windows folder and file name
  99.                     .replaceAll(':', '：')
  100.                     .replaceAll('*', '＊')
  101.                     .replaceAll('?', '？')
  102.                     .replaceAll('"', '“')
  103.                     .replaceAll('<', '《')
  104.                     .replaceAll('>', '》')
  105.                     .replaceAll('|', '｜')
  106.                     .replaceAll('/', '／')
  107.                     .replaceAll('\\', '＼')
  108.                     .replaceAll('≸∱', '\\').replace('∱≸', ':')
  109.                     .replaceAll(/[\s.]+\\/g, '\\'); // remove space and dot
  110.             }
  111.         }
  112.     })()
  113.  
  114.     const postContent = document.querySelector('.post__content')
  115.     if (postContent) {
  116.         replaceAsync(postContent.innerHTML, /(?<!a href="|<a [^>]+">)(https?:\/\/[^\s<]+)/g, async function (match) {
  117.             let [service, id, post] = await getKemonoUrl(match);
  118.             if (service === null) return `<a href="${match}" target="_blank">${match}</a>`;
  119.             id = id || window.location.href.match(/\/user\/(\d+)/)[1];
  120.             const domain = window.location.href.match(/https:\/\/([^/]+)/)[1];
  121.             const url = `${service}/user/${id}${post ? `/post/${post}` : ""}`;
  122.             return `<a href="https://${domain}/${url}" target="_self">[已替换]${match}</a>`;
  123.         }).then(function (result) {
  124.             postContent.innerHTML = result
  125.                 .replace(/<a href="(https:\/\/[^\s<]+)">\1<\/a>\n?(#[^\s<]+)/g, `<a href="$1$2">$1$2</a>`)
  126.                 .replace(/<a href="(https:\/\/[^\s<]+)">(.*?)<\/a>\n?(#[^\s<]+)/g, `<a href="$1$3">$2</a>`)
  127.         })
  128.     }
  129.  
  130.     const prev = document.querySelector(".post__nav-link.prev");
  131.     if (prev) {
  132.         document.addEventListener("keydown", function (e) {
  133.             if (e.key === "Right" || e.key === "ArrowRight" || data.vimMode && (e.key === "h" || e.key === "H")) {
  134.                 prev.click();
  135.             }
  136.         });
  137.     }
  138.  
  139.     const next = document.querySelector(".post__nav-link.next");
  140.     if (next) {
  141.         document.addEventListener("keydown", function (e) {
  142.             if (e.key === "Left" || e.key === "ArrowLeft" || data.vimMode && (e.key === "l" || e.key === "L")) {
  143.                 next.click();
  144.             }
  145.         });
  146.     }
  147.  
  148.     if (language === 'zh-CN') {
  149.         const dms = document.querySelector('.user-header__dms');
  150.  
  151.         if (dms) dms.innerHTML = '私信'
  152.  
  153.         const flagText = document.querySelector('.post__flag')
  154.             ?.querySelector('span:last-child');
  155.  
  156.         if (flagText) {
  157.             flagText.textContent = '标记';
  158.             flagText.title = '标记为需要重新导入的内容'
  159.         }
  160.     }
  161.  
  162.     async function downloadPostContent(post) {
  163.         if (Object.keys(post.file).length !== 0) post.attachments.unshift(post.file)
  164.         const padStart = Math.max(3, post.attachments.length.toString().length);
  165.         for (let [i, { name, path }] of post.attachments.entries()) {
  166.             const extension = path.split('.').pop();
  167.             const filename = name.replace(/\.\w+$/, '');
  168.             await downloadContent(data.rpc, data.token, data.formatDir(post, i, filename, extension, padStart), `https://${domain}/data${path}`);
  169.         }
  170.     }
  171.  
  172.     let mode;
  173.     let header;
  174.     let listener;
  175.     if (window.location.href.match(/\/user\/\w+\/post\/\w+/)) {
  176.         mode = 'post';
  177.         header = document.querySelector('.post__actions');
  178.         listener = async () => await downloadPostContent(await (await fetch(`/api/v1${window.location.pathname}`)).json())
  179.     } else if (window.location.href.match(/\/user\/\w+/)) {
  180.         mode = 'user';
  181.         header = document.querySelector('.user-header__actions');
  182.         listener = async () => {
  183.             for (let post of await getPosts(window.location.pathname)) await downloadPostContent(post)
  184.         }
  185.     } else if (window.location.href.match(/\/favorites/)) {
  186.         mode = 'favor';
  187.         header = document.querySelector('.dropdowns');
  188.         const type = document.querySelector('.dropdowns>select:nth-child(2)>option:nth-child(1)');
  189.         listener = async () => {
  190.             const posts = type.selected !== true
  191.                 ? await (await fetch(`/api/v1/account/favorites?type=post`)).json()
  192.                 : await (async () => {
  193.                     const response = await fetch(`/api/v1/account/favorites?type=artist`)
  194.                     const result = []
  195.                     for (const artist of await response.json()) {
  196.                         result.push(...await getPosts(`${artist.service}/user/${artist.id}`))
  197.                     }
  198.                     return result;
  199.                 })()
  200.             for (let post of posts) await downloadPostContent(post)
  201.         }
  202.     }
  203.  
  204.     if (header) {
  205.         const settings = document.createElement('button');
  206.         settings.classList.add(`${mode}-header__settings`);
  207.         settings.style.backgroundColor = 'transparent';
  208.         settings.style.borderColor = 'transparent';
  209.         settings.style.color = 'white';
  210.         settings.innerHTML = `
  211.             <i class="fa-solid fa-gear ${mode}-header_settings-icon"/>
  212.         `;
  213.         settings.addEventListener('click', showDialog);
  214.  
  215.         const download = document.createElement('button');
  216.         download.classList.add(`${mode}-header__download`);
  217.         download.style.backgroundColor = 'transparent';
  218.         download.style.borderColor = 'transparent';
  219.         download.style.color = 'white';
  220.         download.innerHTML = `
  221.             <i class="fa-solid fa-download ${mode}-header_download-icon"/>
  222.             <span class="${mode}-header__download-text">下载</span>
  223.         `;
  224.  
  225.         download.addEventListener('click', async function () {
  226.             download.innerHTML = `
  227.                 <i class="fa-solid fa-spinner fa-spin-pulse ${mode}-header_download-icon"></i>
  228.                 <span class="${mode}-header__download-text">下载中...</span>
  229.             `;
  230.             await listener();
  231.             download.innerHTML = `
  232.                 <i class="fa-solid fa-check ${mode}-header_download-icon"/>
  233.                 <span class="${mode}-header__download-text">下载/推送完成</span>
  234.             `;
  235.         }, { once: true });
  236.  
  237.         header.appendChild(settings);
  238.         header.appendChild(download);
  239.     } else if (mode !== undefined) {
  240.         alert('未找到插入位置, 请将本页源代码发送给开发者以解决此问题');
  241.     }
  242.  
  243.     function showDialog() {
  244.         swal.fire({
  245.             title: '设置', html: `
  246.                 <div>
  247.                     <label for="rpc">Aria2 RPC地址</label>
  248.                     <input type="text" id="rpc" value="${data.rpc}">
  249.                 </div>
  250.                 <div>
  251.                     <label for="token">Aria2 Token</label>
  252.                     <input type="text" id="token" value="${data.token}">
  253.                 </div>
  254.                 <div>
  255.                     <label for="dir">下载目录</label>
  256.                     <i class="fa-solid fa-info-circle" title="支持的变量:
  257.                      {service}: 服务器, 如fanbox
  258.                      {artist_id}: 作者ID
  259.                      {date}: 发布时间
  260.                      {post}: 作品标题
  261.                      {post_id}: 作品ID
  262.                      {time}: 下载时间
  263.                      {index0}: 从0开始的序号。cover将编号为0
  264.                      {index}: 从1开始的序号。cover的编号为空白
  265.                      {auto_named}: 当文件名为uuid时将使用空命名。否则使用文件名
  266.                      {name}: 文件名 
  267.                      {extension}: 文件扩展名。必须带有此项
  268.                      {js:...#}: js代码. 可用参数请参考源代码85行处formatDir方法"></i>
  269.                     <textarea cols="20" id="dir">${data.dir}</textarea>
  270.                 </div>
  271.                 <div>
  272.                     <label for="vimMode">Vim模式</label>
  273.                     <input type="checkbox" id="vimMode" checked="${data.vimMode}">
  274.                 </div>
  275.             `, showCancelButton: true, confirmButtonText: '保存', cancelButtonText: '取消'
  276.         }).then((result) => {
  277.             if (result.isConfirmed) {
  278.                 data.rpc = document.getElementById('rpc').value;
  279.                 data.token = document.getElementById('token').value;
  280.                 data.dir = document.getElementById('dir').value;
  281.                 data.vimMode = document.getElementById('vimMode').checked;
  282.                 location.reload();
  283.             }
  284.         });
  285.     }
  286.  
  287. }
  288.  
  289. async function replaceAsync(str, regex, asyncFn) {
  290.     const promises = [];
  291.     str.replace(regex, (match, ...args) => {
  292.         const promise = asyncFn(match, ...args);
  293.         promises.push(promise);
  294.     });
  295.     const data = await Promise.all(promises);
  296.     return str.replace(regex, () => data.shift());
  297. }
  298.  
  299. async function getKemonoUrl(url) {
  300.  
  301.     function getFanbox(creatorId) {
  302.         // 同步执行promise
  303.         return new Promise((resolve, reject) => {
  304.             GM_xmlhttpRequest({
  305.                 method: "GET", url: `https://api.fanbox.cc/creator.get?creatorId=${creatorId}`, headers: {
  306.                     "Content-Type": "application/json",
  307.                     "Accept": "application/json",
  308.                     "Origin": "https://www.fanbox.cc",
  309.                     "Referer": "https://www.fanbox.cc/"
  310.                 }, onload: function (response) {
  311.                     if (response.status === 200) {
  312.                         resolve(JSON.parse(response.responseText))
  313.                     } else {
  314.                         reject({ status: response.status, statusText: response.statusText })
  315.                     }
  316.                 }, onerror: function (response) {
  317.                     reject({ status: response.status, statusText: response.statusText })
  318.                 }
  319.             })
  320.         })
  321.     }
  322.  
  323.     const pixiv_user = /https:\/\/www\.pixiv\.net\/users\/(\d+)/i;
  324.     const fantia_user = /https:\/\/fantia\.jp\/fanclubs\/(\d+)(\/posts(\S+))?/i;
  325.     const fanbox_user1 = /https:\/\/www\.fanbox\.cc\/@([^/]+)(?:\/posts\/(\d+))?/i;
  326.     const fanbox_user2 = /https:\/\/(.+)\.fanbox\.cc(?:\/posts\/(\d+))?/i;
  327.     const dlsite_user = /https:\/\/www.dlsite.com\/.+?\/profile\/=\/maker_id\/(RG\d+).html/i;
  328.     const patreon_user1 = /https:\/\/www.patreon.com\/user\?u=(\d+)/i;
  329.     const patreon_user2 = /https:\/\/www.patreon.com\/(\w+)/i;
  330.     const patreon_post1 = /https:\/\/www.patreon.com\/posts\/(\d+)/i;
  331.     const patreon_post2 = /https:\/\/www.patreon.com\/posts\/video-download-(\d+)/i;
  332.  
  333.     let service;
  334.     let id;
  335.     let post = null;
  336.  
  337.     if (pixiv_user.test(url)) {
  338.         //pixiv artist
  339.         service = "fanbox"
  340.         id = url.match(pixiv_user)[1]
  341.     } else if (fantia_user.test(url)) {
  342.         //fantia
  343.         service = "fantia"
  344.         id = url.match(fantia_user)[1]
  345.     } else if (dlsite_user.test(url)) {
  346.         service = "dlsite"
  347.         id = url.match(dlsite_user)[1]
  348.     } else if (fanbox_user1.test(url) || fanbox_user2.test(url)) {
  349.         //fanbox
  350.         service = "fanbox"
  351.         let matches = fanbox_user1.test(url) ? url.match(fanbox_user1) : url.match(fanbox_user2);
  352.         id = (await getFanbox(matches[1])).body.user.userId.toString()
  353.         post = matches[2]
  354.     } else if (patreon_user1.test(url)) {
  355.         // patreon
  356.         service = "patreon"
  357.         id = url.match(patreon_user1)[1]
  358.     } else if (patreon_post1.test(url)) {
  359.         // patreon post
  360.         service = "patreon"
  361.         post = url.match(patreon_post1)[1]
  362.     } else if (patreon_post2.test(url)) {
  363.         // patreon post
  364.         service = "patreon"
  365.         post = url.match(patreon_post2)[1]
  366.     } else {
  367.         return null;
  368.     }
  369.  
  370.     return [service, id, post]
  371. }
  372.  
  373. async function getPosts(path, order = 0) {
  374.     let posts = [];
  375.     while (true) {
  376.         const response = await fetch(`/api/v1/${path}?o=${order}`)
  377.         // TODO: 429 too many requests, 80 request per minute
  378.         if (response.status === 429) {
  379.             await new Promise(resolve => setTimeout(resolve, 60000))
  380.             continue;
  381.         }
  382.         if (response.status !== 200) throw { status: response.status, statusText: response.statusText }
  383.         const items = await response.json();
  384.         posts.push(...items);
  385.         if (items.length < 50) break;
  386.         order += items.length;
  387.     }
  388.     return posts;
  389. }
  390.  
  391. /**
  392.  * send request to aria2 for download
  393.  * @param {string} rpc
  394.  * @param {string} token
  395.  * @param {string} file
  396.  * @param {string} url
  397.  */
  398. async function downloadContent(rpc, token, file, ...url) {
  399.     const dir = file.replace(/(.+?[\/\\])[^\/\\]+$/, "$1");
  400.     const out = file.slice(dir.length);
  401.     const params = token === undefined
  402.         ? out === ""
  403.             ? [url, { "dir": dir }]
  404.             : [url, { "dir": dir, "out": out }]
  405.         : out === ""
  406.             ? [`token:${token}`, url, { "dir": dir }]
  407.             : [`token:${token}`, url, { "dir": dir, "out": out }]
  408.     return new Promise((resolve) => {
  409.         GM_xmlhttpRequest({
  410.             method: "POST", url: rpc, data: JSON.stringify({
  411.                 jsonrpc: "2.0", id: `kemono-${crypto.randomUUID()}`, method: "aria2.addUri", params: params
  412.             }), onload: function (response) {
  413.                 if (response.status === 200) {
  414.                     resolve(JSON.parse(response.responseText))
  415.                 } else {
  416.                     console.log(`添加下载任务失败: ${response.status} ${response.statusText}`)
  417.                 }
  418.             }, onerror: function (response) {
  419.                 console.log(`添加下载任务失败: ${response.status} ${response.statusText}`)
  420.             }
  421.         })
  422.     })
  423. }
  424.  
  425. async function fetchDownloadDir(rpc, token) {
  426.     return new Promise((resolve, reject) => {
  427.         GM_xmlhttpRequest({
  428.             method: "POST", url: rpc, headers: {
  429.                 "Content-Type": "application/json", "Accept": "application/json",
  430.             }, data: JSON.stringify({
  431.                 jsonrpc: "2.0", id: "Kemono", method: "aria2.getGlobalOption", params: token ? [`token:${token}`] : []
  432.             }), onload: function (response) {
  433.                 if (response.status === 200) {
  434.                     resolve(JSON.parse(response.responseText))
  435.                 } else {
  436.                     reject({ status: response.status, statusText: response.statusText })
  437.                 }
  438.             }, onerror: function (response) {
  439.                 reject({ status: response.status, statusText: response.statusText })
  440.             }
  441.         })
  442.     }).then(function (result) {
  443.         return result.result.dir;
  444.     })
  445. }