NSFC_conclusion_downloader

Greasy Fork is available in English.
NSFC_conclusion_downloader

帮助你直接下载国自然结题报告
当前为 2022-02-04 提交的版本，查看最新版本。
换行
// ==UserScript==
// @name         NSFC_conclusion_downloader
// @namespace    https://blog.rhilip.info/
// @version      1.3
// @description  帮助你直接下载国自然结题报告
// @author       Rhilip
// @match        https://kd.nsfc.gov.cn/finalDetails*
// @require      https://unpkg.com/[email protected]/dist/jspdf.umd.min.js
// @require      https://unpkg.com/[email protected]/dist/jquery.js
// ==/UserScript==

/* globals $, jspdf */

// Copy from https://stackoverflow.com/a/61511955
function waitForElm(selector) {
    return new Promise(resolve => {
        if ($(selector).length > 0) {
            return resolve($(selector));
        }

        const observer = new MutationObserver(mutations => {
            if ($(selector).length > 0) {
                resolve($(selector));
                observer.disconnect();
            }
        });

        observer.observe(document.body, {
            childList: true,
            subtree: true
        });
    });
}

(async function() {
    'use strict';

    // 等待页面加载（出现结题报告窗口）
    const conclusionReportTab = await waitForElm('#related');

    // 准备交互按钮，并插入到页面中
    const downloadBtn = $('<button type="button" class="el-button el-button--default el-button--medium is-round">下载全文</button>');
    conclusionReportTab.find('div.verticalBar_T').after(downloadBtn);

    // 点击交互按钮时需要开始下载操作
    downloadBtn.click(async () => {
        downloadBtn.prop('disabled', true).addClass('is-disabled');
        if (!/暂无结题报告全文/.test(conclusionReportTab.text())) {
            // 获得项目信息： 编号（加密后）、批准号、项目名称
            const urlParams = new URLSearchParams(location.search);
            const dependUintID = urlParams.get('id');
            const projectID = $('.basic_info > div.el-row div:contains("项目批号：") + div').text();
            const projectName = $('.basic_info > div.el-row div:contains("项目名称：") + div').text();

            // 准备需要的PDF文件，并删除初始页
            const { jsPDF } = jspdf;
            const doc = new jsPDF();
            doc.deletePage(1);
            doc.setDocumentProperties({
                title: `${projectID} ${projectName}`,
                subject: `${location.origin}${location.pathname}`,
                creator: 'NSFC_conclusion_downloader'
            });

            // 核心下载方法
            const image = new Image();
            for (let i=1;;i++) {
                downloadBtn.text(`正在下载第 ${i} 页`);

                // 获得图片链接
                const { data: requestData } = await $.post('/api/baseQuery/completeProjectReport', {id: dependUintID, index: i});

                // 获得Blob形式的imageData，这样可以防止image.src和jsPDF.addImage会产生两次图片请求，浪费带宽
                // （实际变成了一次请求服务器和两次请求本地blob）
                const imageDataAsBlob = await $.ajax({
                    url: requestData.url,
                    method: 'GET',
                    xhrFields: { responseType: 'blob'}
                });

                // 加载图片并获得图片的 width, height 属性
                image.src = URL.createObjectURL(imageDataAsBlob);
                await image.decode();

                // 将图片添加进PDF中
                doc.addPage([image.width, image.height], image.width < image.height ? 'p' : 'l');
                doc.addImage(image, "PNG", 0, 0, image.width, image.height);

                if (!requestData.hasnext) break;
            }

            // 我们并没法 await 保存过程，所以直接显示下载完成就好，浏览器处理好会自动显示下载文件
            doc.save(`${projectID} ${projectName}.pdf`);
            downloadBtn.text('下载完成');
        }
    });

})();