// ==UserScript==
// @name         Wallhaven一键下载
// @version      1.8
// @description  在Wallhaven缩略图上添加下载按钮和复选框，支持单张下载、逐个下载和打包下载。
// @match        *://wallhaven.cc/*
// @require      https://cdnjs.cloudflare.com/ajax/libs/jszip/3.7.1/jszip.min.js
// @require      https://cdnjs.cloudflare.com/ajax/libs/FileSaver.js/2.0.5/FileSaver.min.js
// @grant        GM_addStyle
// @license      MIT
// @author       EPC_SG
// @namespace 这是干啥的，不是很懂
// ==/UserScript==

(function() {
    'use strict';

    // 注入CSS样式
    GM_addStyle(`
        .thumb { position: relative; }
        .thumb .download-btn {
            position: absolute;
            top: 5px;
            left: 6px;
            z-index: 1000;
            background: #4CAF50;
            color: white;
            border: none;
            border-radius: 5px;
            cursor: pointer;
            padding: 5px 10px;
        }
        .thumb .download-checkbox {
            position: absolute;
            top: 11px;
            left: 42px;
            z-index: 1000;
            width: 15px;
            height: 15px;
            visibility: visible;
        }
        .control-btn {
            background: #4CAF50;
            color: white;
            padding: 5px 10px;
            border-radius: 5px;
            cursor: pointer;
            margin: 0 5px 5px 0;
            font-size: 12px;
        }
        #repo-window {
            position: fixed;
            top: 110px;
            right: 10px;
            width: 250px;
            background: rgba(0, 0, 0, 0.8);
            color: white;
            padding: 10px;
            border-radius: 5px;
            z-index: 2000;
        }
        #repo-header {
            cursor: pointer;
            margin: 0 0 10px;
            font-size: 16px;
            display: flex;
            justify-content: space-between;
            align-items: center;
        }
        #repo-header .arrow { transition: transform 0.2s; }
        #repo-header .arrow.expanded { transform: rotate(0deg); }
        #repo-header .arrow:not(.expanded) { transform: rotate(90deg); }
        #repo-content {
            display: none;
            max-height: 350px;
            overflow-y: auto;
        }
        #repo-content.expanded { display: block; }
        .repo-item {
            display: flex;
            justify-content: space-between;
            align-items: center;
            padding: 5px 0;
        }
        .repo-item a {
            color: #1e90ff;
            text-decoration: none;
        }
        .repo-item a:hover { text-decoration: underline; }
        .repo-buttons {
            display: flex;
            gap: 5px;
        }
        .repo-item button {
            border: none;
            color: white;
            padding: 2px 5px;
            cursor: pointer;
            border-radius: 3px;
            font-size: 10px;
            height: 18px;
        }
        .repo-item .delete-btn { background: #ff4444; }
        .repo-item .download-btn { background: #4CAF50; }
        #clear-btn { background: #ff4444; }
        #abort-btn {
            background: #ff4444;
            color: white;
            padding: 2px 5px;
            border: none;
            border-radius: 3px;
            cursor: pointer;
            margin-left: 5px;
            font-size: 12px;
        }
    `);

    // 工具函数
    const getImageUrl = (id, isPng) => {
        const format = isPng ? 'png' : 'jpg';
        return `https://w.wallhaven.cc/full/${id.slice(0, 2)}/wallhaven-${id}.${format}`;
    };

    const fetchWithRetry = async (url, retries = 3) => {
        for (let i = 0; i < retries; i++) {
            try {
                const response = await fetch(url);
                if (!response.ok) throw new Error('Network error');
                return await response.blob();
            } catch (err) {
                if (i === retries - 1) throw err;
                await new Promise(resolve => setTimeout(resolve, 1000 * (i + 1)));
            }
        }
    };

    const fetchWithLimit = async (urls, limit = 5, onProgress = () => {}, abortSignal = null) => {
        const results = [];
        const total = urls.length;
        let completed = 0;

        for (let i = 0; i < urls.length; i += limit) {
            if (abortSignal && abortSignal.aborted) throw new Error('Download aborted');
            const chunk = urls.slice(i, i + limit);
            const promises = chunk.map(url => fetchWithRetry(url));
            const blobs = await Promise.all(promises);
            results.push(...blobs);
            completed += blobs.length;
            onProgress(Math.round((completed / total) * 100), completed, total);
        }
        return results;
    };

    const debounce = (fn, delay) => {
        let timeout;
        return (...args) => {
            clearTimeout(timeout);
            timeout = setTimeout(() => fn(...args), delay);
        };
    };

    // Worker相关
    const createWorker = (fn) => {
        const blob = new Blob([`(${fn.toString()})()`], { type: 'application/javascript' });
        return new Worker(URL.createObjectURL(blob));
    };

    const workerScript = () => {
        self.importScripts('https://cdnjs.cloudflare.com/ajax/libs/jszip/3.7.1/jszip.min.js');
        self.onmessage = async (e) => {
            const { files } = e.data;
            const zip = new JSZip();
            files.forEach(file => zip.file(file.name, file.blob));
            const blob = await zip.generateAsync({
                type: 'blob',
                compression: 'DEFLATE',
                compressionOptions: { level: 1 }
            }, ({ percent }) => {
                self.postMessage({ type: 'progress', percent: Math.round(percent) });
            });
            self.postMessage({ type: 'complete', blob });
        };
    };

    // 仓库管理（改进部分）
    const repo = {
        items: (() => {
            const rawData = JSON.parse(localStorage.getItem('wallhaven_repo') || '[]');
            // 兼容旧数据：如果旧数据是纯 ID 数组，转换为新格式（假设所有图片为 JPG）
            if (Array.isArray(rawData) && rawData.length > 0 && typeof rawData[0] === 'string') {
                return rawData.map(id => ({ id, isPng: false }));
            }
            return rawData;
        })(),
        save: () => localStorage.setItem('wallhaven_repo', JSON.stringify(repo.items)),
        add: (id, isPng) => {
            if (!repo.items.some(item => item.id === id)) {
                repo.items.push({ id, isPng });
                repo.save();
            }
        },
        remove: (id) => {
            repo.items = repo.items.filter(item => item.id !== id);
            repo.save();
        },
        clear: () => {
            repo.items = [];
            repo.save();
        },
        has: (id) => repo.items.some(item => item.id === id),
        getUrls: () => repo.items.map(item => getImageUrl(item.id, item.isPng))
    };

    // DOM 缓存和状态管理
    let checkboxes = [];
    let ratioDisplay, progressDisplay, repoWindow, repoContent, repoArrow;
    let abortController = null; // 用于中止下载
    let currentWorker = null; // 用于中止打包

    const updateCheckboxes = () => {
        checkboxes = Array.from(document.querySelectorAll('.thumb .download-checkbox'));
    };

    const updateRatio = () => {
        const selected = repo.items.length;
        ratioDisplay.textContent = `仓库中 ${selected} 张`;
    };
    const updateRatioDebounced = debounce(updateRatio, 100);

    const showProgress = (text, showAbort = false) => {
        progressDisplay.textContent = text;
        if (showAbort) {
            const abortBtn = document.createElement('button');
            abortBtn.id = 'abort-btn';
            abortBtn.textContent = '中止';
            abortBtn.addEventListener('click', () => {
                if (abortController) abortController.abort();
                if (currentWorker) currentWorker.terminate();
                showProgress('操作已中止');
            });
            progressDisplay.appendChild(abortBtn);
        } else {
            const existingAbortBtn = progressDisplay.querySelector('#abort-btn');
            if (existingAbortBtn) existingAbortBtn.remove();
        }
    };

    // 渲染悬浮窗（改进部分）
    const renderRepoWindow = () => {
        repoContent.innerHTML = '';
        repo.items.forEach(item => {
            const div = document.createElement('div');
            div.className = 'repo-item';
            div.innerHTML = `
                <a href="https://wallhaven.cc/w/${item.id}" target="_blank">${item.id}</a>
                <div class="repo-buttons">
                    <button class="delete-btn">删除</button>
                    <button class="download-btn">下载</button>
                </div>
            `;
            div.querySelector('.delete-btn').addEventListener('click', () => {
                repo.remove(item.id);
                renderRepoWindow();
                updateRatio();
                syncCheckboxes();
            });
            div.querySelector('.download-btn').addEventListener('click', () => {
                const url = getImageUrl(item.id, item.isPng);
                fetchWithRetry(url)
                    .then(blob => saveAs(blob, url.split('/').pop()))
                    .catch(err => showProgress(`下载 ${item.id} 失败，请重试。`));
            });
            repoContent.appendChild(div);
        });
    };

    // 切换展开/收缩状态
    const toggleRepoWindow = () => {
        repoContent.classList.toggle('expanded');
        repoArrow.classList.toggle('expanded');
    };

    // 同步复选框状态
    const syncCheckboxes = () => {
        checkboxes.forEach(cb => {
            const id = cb.closest('.thumb').dataset.wallpaperId;
            cb.checked = repo.has(id);
        });
    };

    // 全选/取消功能
    const toggleAll = (e) => {
        e.preventDefault();
        const allChecked = checkboxes.every(cb => cb.checked);
        checkboxes.forEach(cb => {
            const thumb = cb.closest('.thumb');
            const id = thumb.dataset.wallpaperId;
            const isPng = !!thumb.querySelector('.thumb-info .png');
            if (!allChecked) {
                cb.checked = true;
                repo.add(id, isPng);
            } else {
                cb.checked = false;
                repo.remove(id);
            }
        });
        renderRepoWindow();
        updateRatio();
    };

    // 添加按钮和复选框
    const addButtons = (container) => {
        container.querySelectorAll('.thumb:not([data-processed])').forEach(thumb => {
            const id = thumb.dataset.wallpaperId;
            if (!id) return;

            thumb.dataset.processed = 'true';
            const isPng = !!thumb.querySelector('.thumb-info .png');
            const url = getImageUrl(id, isPng);

            const dlBtn = document.createElement('button');
            dlBtn.className = 'download-btn';
            dlBtn.innerHTML = '<i class="fas fa-download"></i>';
            thumb.appendChild(dlBtn);

            const checkbox = document.createElement('input');
            checkbox.type = 'checkbox';
            checkbox.className = 'download-checkbox';
            checkbox.value = url;
            checkbox.checked = repo.has(id);
            checkbox.addEventListener('change', () => {
                if (checkbox.checked) repo.add(id, isPng);
                else repo.remove(id);
                renderRepoWindow();
                updateRatioDebounced();
            });
            thumb.appendChild(checkbox);
        });
        updateCheckboxes();
        syncCheckboxes();
        updateRatio();
    };

    // 批量逐个下载
    const batchDownload = async () => {
        if (!repo.items.length) return showProgress('仓库为空！');
        const urls = repo.getUrls();
        abortController = new AbortController();
        try {
            const blobs = await fetchWithLimit(urls, 5, (percent, completed, total) => {
                showProgress(`正在下载：${percent}% (${completed}/${total})`, true);
            }, abortController.signal);
            blobs.forEach((blob, i) => saveAs(blob, urls[i].split('/').pop()));
            showProgress('逐个下载完成！');
        } catch (err) {
            if (err.message === 'Download aborted') {
                showProgress('下载已中止');
            } else {
                showProgress('下载失败，请重试。');
            }
        } finally {
            abortController = null;
        }
    };

    // 打包下载
    const packDownload = async () => {
        if (!repo.items.length) return showProgress('仓库为空！');
        const urls = repo.getUrls();
        abortController = new AbortController();
        try {
            const blobs = await fetchWithLimit(urls, 5, (percent, completed, total) => {
                showProgress(`正在下载：${percent}% (${completed}/${total})`, true);
            }, abortController.signal);

            currentWorker = createWorker(workerScript);
            currentWorker.onmessage = (e) => {
                if (e.data.type === 'progress') {
                    showProgress(`正在打包：${e.data.percent}%`, true);
                } else if (e.data.type === 'complete') {
                    saveAs(e.data.blob, 'images.zip');
                    showProgress('打包完成！');
                    currentWorker = null;
                }
            };

            const files = blobs.map((blob, i) => ({
                name: urls[i].split('/').pop(),
                blob
            }));
            currentWorker.postMessage({ files });
        } catch (err) {
            if (err.message === 'Download aborted') {
                if (currentWorker) currentWorker.terminate();
                showProgress('打包已中止');
            } else {
                showProgress('打包失败，请重试。');
            }
        } finally {
            abortController = null;
        }
    };

    // 清空仓库
    const clearRepo = () => {
        repo.clear();
        renderRepoWindow();
        updateRatio();
        syncCheckboxes();
    };

    // 初始化控制面板
    const initControls = () => {
        const toolbar = document.querySelector('.expanded') || document.body;
        [ratioDisplay, progressDisplay, repoWindow] = [
            Object.assign(document.createElement('span'), { style: 'color: white;' }),
            Object.assign(document.createElement('span'), { style: 'color: white;' }),
            document.createElement('div')
        ];
        repoWindow.id = 'repo-window';

        const header = document.createElement('h3');
        header.id = 'repo-header';
        header.innerHTML = '图片仓库 <span class="arrow">▼</span>';
        header.addEventListener('click', toggleRepoWindow);
        repoArrow = header.querySelector('.arrow');
        repoWindow.appendChild(header);

        repoContent = document.createElement('div');
        repoContent.id = 'repo-content';
        repoWindow.appendChild(repoContent);

        const selectAllBtn = document.createElement('button');
        selectAllBtn.textContent = '全选/取消';
        selectAllBtn.className = 'control-btn';
        selectAllBtn.addEventListener('click', toggleAll);
        toolbar.appendChild(selectAllBtn);

        const batchBtn = document.createElement('button');
        batchBtn.textContent = '逐个下载';
        batchBtn.className = 'control-btn';
        batchBtn.addEventListener('click', batchDownload);
        repoWindow.insertBefore(batchBtn, repoContent);

        const packBtn = document.createElement('button');
        packBtn.textContent = '打包下载';
        packBtn.className = 'control-btn';
        packBtn.addEventListener('click', packDownload);
        repoWindow.insertBefore(packBtn, repoContent);

        const clearBtn = document.createElement('button');
        clearBtn.textContent = '清空';
        clearBtn.id = 'clear-btn';
        clearBtn.className = 'control-btn';
        clearBtn.addEventListener('click', clearRepo);
        repoWindow.insertBefore(clearBtn, repoContent);

        [ratioDisplay, progressDisplay].forEach((el, i) => {
            el.style.marginLeft = `${5 + i * 5}px`;
            toolbar.appendChild(el);
        });
        document.body.appendChild(repoWindow);

        // 默认展开
        toggleRepoWindow();
        renderRepoWindow();
    };

    // 主逻辑
    initControls();
    const listingPage = document.querySelector('.thumb-listing-page');
    if (listingPage) addButtons(listingPage);

    const thumbListing = document.querySelector('.thumb-listing');
    if (thumbListing) {
        const observer = new MutationObserver(mutations => {
            mutations.forEach(m => {
                m.addedNodes.forEach(node => {
                    if (node.classList?.contains('thumb-listing-page')) addButtons(node);
                });
            });
        });
        observer.observe(thumbListing, { childList: true, subtree: true });

        thumbListing.addEventListener('click', (e) => {
            const btn = e.target.closest('.download-btn');
            if (btn) {
                const url = btn.nextElementSibling.value;
                fetchWithRetry(url)
                    .then(blob => saveAs(blob, url.split('/').pop()))
                    .catch(err => showProgress('下载失败，请重试。'));
            }
        });
    }
})();