☰

画世界下载原图

从画世界网页版（PC端UA）下载原图的脚本

目前为 2025-02-21 提交的版本。查看 最新版本。

// ==UserScript==
// @name         画世界下载原图
// @namespace    画世界下载原图
// @homepage     https://gist.github.com/kaixinol/a7521662de470d427db2eb9d45cd843e/
// @version      0.2
// @license      MIT
// @description  从画世界网页版（PC端UA）下载原图的脚本
// @author       Kaesinol
// @match        https://huashijie.art/work/detail/*
// @grant        GM_xmlhttpRequest
// @grant        GM_download
// @connect      app.huashijie.art
// ==/UserScript==

(function() {
    'use strict';

    // 存储拦截到的图片URL和作品信息
    let interceptedImages = [];
    let workInfo = null;

    // 获取作品ID
    function getWorkId() {
        const path = window.location.pathname;
        return path.split('/').pop();
    }

    // 获取文件扩展名
    function getFileExtension(url) {
        const match = url.match(/\.([^.]+)(?:\?|$)/);
        return match ? match[1].toLowerCase() : 'jpg';
    }

    // 从API获取作品信息
    async function fetchWorkInfo(workId) {
        return new Promise((resolve, reject) => {
            GM_xmlhttpRequest({
                method: 'GET',
                url: `https://app.huashijie.art/api/work/detailV2?workId=${workId}`,
                headers: {
                    'Accept': 'application/json'
                },
                onload: function(response) {
                    try {
                        const data = JSON.parse(response.responseText);
                        if (data && data.data) {
                            resolve(data.data);
                        } else {
                            reject('Invalid API response');
                        }
                    } catch (e) {
                        reject(e);
                    }
                },
                onerror: reject
            });
        });
    }

    // 拦截 canvas.drawImage()
    const originalDrawImage = CanvasRenderingContext2D.prototype.drawImage;
    CanvasRenderingContext2D.prototype.drawImage = function(img, ...args) {
        if (img instanceof HTMLImageElement && !interceptedImages.includes(img.src)) {
            interceptedImages.push(img.src);
            console.log('Intercepted image:', img.src);
        }
        return originalDrawImage.apply(this, [img, ...args]);
    };

    // 添加下载按钮
    async function addDownloadButton() {
        const likeButton = document.querySelector('.common-like.click-hover');
        if (!likeButton) {
            setTimeout(addDownloadButton, 1000);
            return;
        }

        // 获取作品信息
        const workId = getWorkId();
        try {
            workInfo = await fetchWorkInfo(workId);
        } catch (error) {
            console.error('Failed to fetch work info:', error);
        }

        const downloadButton = document.createElement('button');
        downloadButton.textContent = '下载图片';
        downloadButton.style.cssText = `
            margin-left: 10px;
            padding: 5px 10px;
            background-color: #4CAF50;
            color: white;
            border: none;
            border-radius: 4px;
            cursor: pointer;
        `;

        downloadButton.addEventListener('click', async () => {
            if (interceptedImages.length === 0) {
                alert('未检测到图片，请先浏览作品');
                return;
            }

            if (!workInfo) {
                alert('未能获取作品信息，将使用默认文件名');
                return;
            }

            // 下载所有图片
            for (let i = 0; i < interceptedImages.length; i++) {
                const url = interceptedImages[i];
                const extension = getFileExtension(url);
                const filename = `${workInfo.user.nick} - ${workInfo.id} - ${workInfo.userId} - ${i + 1}.${extension}`;

                GM_download({
                    url: url,
                    name: filename,
                    saveAs: false,
                    onerror: (error) => {
                        console.error('Download failed:', error);
                        alert(`下载失败: ${error}`);
                    }
                });
            }
            alert(`开始下载 ${interceptedImages.length} 张图片`);
        });

        likeButton.parentNode.insertBefore(downloadButton, likeButton.nextSibling);
    }

    // 页面加载完成后添加按钮
    window.addEventListener('load', addDownloadButton);
})();