// ==UserScript==
// @name         一键下载Steam云存档
// @namespace    SteamCloudSave
// @version      0.1.0
// @description  Auto download Steam Cloud Save
// @author       DreamNya
// @match        https://store.steampowered.com/account/remotestorageapp?appid=*
// @match        https://store.steampowered.com/account/remotestorageapp/?appid=*
// @require      https://cdn.jsdelivr.net/npm/jszip@3.9.1/dist/jszip.min.js
// @grant        GM_xmlhttpRequest
// @grant        GM_download
// @grant        GM_addStyle
// @run-at       document-end
// @license      MIT
// @connect      steampowered.com
// @connect      steamusercontent.com
// ==/UserScript==

/* global JSZip */
/* eslint-disable no-constant-condition */

const settings = {
    // 请求过快可能触发临时ip封禁或404下载错误，暂时不考虑并行下载
    globalDelay: 100, // 存档下载间隔 单位：ms
};

const $ = unsafeWindow.jQuery;

const button = $("<button id='AutoDownload'>一键下载</a>")
    .prependTo("#main_content")
    .one("click", async function () {
        button.css({ cursor: "not-allowed" }).text("下载中...");
        await main();
        button.text("下载成功");
    });

GM_addStyle(`
    #AutoDownload{
        display:inline-block;
        padding:10px 20px;
        border-radius:8px;
        background:#2ECC71;
        color:#ffffff;
        font-size:15px;
        border:1px solid #2ECC71;
        transition:background .2s, transform .1s;
    }
    #AutoDownload:hover{
        background:#27AE60;
    }
    #AutoDownload:active{
        transform:scale(0.97);
    }
`);

async function main() {
    const appid = location.href.match(/appid=(\d+)/)[1];
    const tittle = $("h2:last").text().trim();
    let index = 0;
    const filesInfos = [];

    while (true) {
        const url = `https://store.steampowered.com/account/remotestorageapp?appid=${appid}&index=${index}`;
        const dom = await GetPage(url);

        const filesInfo = GetFilesInfo(dom, index);
        filesInfos.push(...filesInfo);

        const nextPage = CheckNextPage(dom, index);
        if (nextPage) {
            index += 50;
            await sleep();
        } else {
            break;
        }
    }

    await DownloadFiles(filesInfos, tittle);
}

function GetPage(url) {
    return new Promise((resolve, reject) => {
        GM_xmlhttpRequest({
            url,
            method: "GET",
            // 使用text+DOMParser代替document提高性能
            responseType: "text",
            onload: (xhr) => {
                console.log(xhr);
                if (xhr.status == 200) {
                    const dom = new DOMParser().parseFromString(xhr.responseText, "text/html");
                    resolve(dom);
                } else {
                    const msg = `下载失败 url:${url} status:${xhr.status}`;
                    button.text(msg);
                    reject(msg);
                }
            },
        });
    });
}

function GetFilesInfo(dom, globalIndex) {
    // jQuery对DOMParser兼容性较差，直接用原生方法
    return [...dom.querySelectorAll("tr")].splice(1).map((node, index) => {
        const childrens = node.childElements();

        const mainFolder = childrens[0].innerText.trim();
        const subFolder = childrens[1].innerText.trim();

        const path = mainFolder ? `${mainFolder}/${subFolder}` : subFolder;
        const size = childrens[2].innerText.trim();
        const date = childrens[3].innerText.trim();
        const href = childrens[4].querySelector("a").href;

        return { index: globalIndex + index, path, size, date, href };
    });
}

function CheckNextPage(dom, index) {
    const nextIndex = dom.querySelector("#main_content > a:last-child")?.href.match(/index=(\d+)/)[1] ?? 0;
    return nextIndex > index;
}

async function DownloadFiles(filesInfos, tittle) {
    // 3.10.1版本JSZip会卡generateAsync，3.9.1无这个问题，// @require 中已锁死版本
    const zip = new JSZip();
    const result = [["序号,文件名,文件大小,写入日期,下载状态"]];
    const total = filesInfos.length;
    for (const fileInfo of filesInfos) {
        const { blob, status } = await DownloadFile(fileInfo.href);
        result.push([...Object.values(fileInfo).slice(0, 4), `${status} ${status == 200 ? "OK" : "Fail"}`].join(","));
        if (status == 200) {
            zip.file(fileInfo.path, blob);
        }
        button.text(`下载中 ${result.length - 1}/${total}`);
        await sleep();
    }
    // "\uFEFF" 为BOM头，带BOM的UTF-8编码可以防止Excel打开乱码
    zip.file("下载结果.csv", "\uFEFF" + result.join("\n"));
    const content = await zip.generateAsync({ type: "blob" });
    const blobUrl = URL.createObjectURL(content);
    GM_download(blobUrl, `${tittle.replace(/[\\/:*?"<>|]/g, "_")} ${Date.now()}.zip`);
    // URL.revokeObjectURL(blobUrl);
}

function DownloadFile(url) {
    return new Promise((resolve) => {
        GM_xmlhttpRequest({
            url,
            method: "GET",
            responseType: "blob",
            onload: (xhr) => {
                console.log(xhr);
                // 此处直接转发，不做任何判断，打包时进行判断
                resolve({ blob: xhr.response, status: xhr.status });
            },
        });
    });
}

function sleep(ms) {
    return new Promise((resolve) => setTimeout(resolve, ms ?? settings.globalDelay));
}