搜题 - 源代码

Greasy Fork is available in English.
搜题

在线答题搜答案脚本
换行
"use strict";
/* eslint-disable no-underscore-dangle, @typescript-eslint/no-empty-function */
// ==UserScript==
// @name         搜题
// @namespace    search-answer
// @version      1.4
// @description  在线答题搜答案脚本
// @author       HCLonely
// @include      *
// @run-at       document-start
// @grant        GM_setValue
// @grant        GM_getValue
// @grant        GM_addStyle
// @grant        GM_xmlhttpRequest
// @grant        GM_registerMenuCommand
// @homepage     https://github.com/HCLonely/search-answer
// @require      https://cdn.jsdelivr.net/npm/sweetalert2@11
// @require      https://cdn.jsdelivr.net/npm/[email protected]/dist/jquery.slim.min.js
// @require      http://greasyfork.icu/scripts/418102-tm-request/code/TM_request.js?version=902218
// @require      https://cdn.jsdelivr.net/npm/[email protected]/mammoth.browser.min.js
// @require      https://cdn.jsdelivr.net/npm/[email protected]/dist/xlsx.full.min.js
// @require      https://cdn.jsdelivr.net/npm/[email protected]/dist/tinykeys.umd.min.js
// @require      https://cdn.jsdelivr.net/npm/[email protected]/dist/tesseract.min.js
// @require      https://cdn.jsdelivr.net/npm/[email protected]/build/md5.min.js
// @license      Apache-2.0
// @connect      www.baidu.com
// @connect      www.sogou.com
// @connect      cn.bing.com
// @connect      www.google.com
// ==/UserScript==
(() => {
    window.onblur = () => { };
    window.onfocus = () => { };
    document.onfocusin = () => { };
    document.onfocusout = () => { };
    document._addEventListener = document.addEventListener;
    document.addEventListener = (...argv) => {
        if (['visibilitychange', 'mozvisibilitychange', 'webkitvisibilitychange', 'msvisibilitychange'].includes(argv[0])) {
            return;
        }
        document._addEventListener(...argv);
    };
    document._removeEventListener = document.removeEventListener;
    document.removeEventListener = (...argv) => {
        if (['visibilitychange', 'mozvisibilitychange', 'webkitvisibilitychange', 'msvisibilitychange'].includes(argv[0])) {
            return;
        }
        document._removeEventListener(...argv);
    };
    window.onload = () => {
        window.onblur = () => { };
        window.onfocus = () => { };
        document.onfocusin = () => { };
        document.onfocusout = () => { };
    };
    let { highLightAbswer, startShortcutKey, ocrShortcutKey } = GM_getValue('settings') || {};
    const start = async () => {
        let data;
        let imageData;
        let engine = 'baidu';
        const searchFromWebPage = (text, engine) => {
            switch (engine) {
                case 'baidu':
                    window.open(`https://www.baidu.com/s?wd=${text}`, 'SearchResult', 'resize=yes,scrollbars=yes');
                    break;
                case 'sougou':
                    window.open(`https://www.sogou.com/web?query=${text}`, 'SearchResult', 'resize=yes,scrollbars=yes');
                    break;
                case 'bing':
                    window.open(`https://cn.bing.com/search?q=${text}`, 'SearchResult', 'resize=yes,scrollbars=yes');
                    break;
                case 'google':
                    window.open(`https://www.google.com/search?q=111${text}`, 'SearchResult', 'resize=yes,scrollbars=yes');
                    break;
                default:
                    window.open(`https://www.baidu.com/s?wd=${text}`, 'SearchResult', 'resize=yes,scrollbars=yes');
                    break;
            }
            return null;
        };
        const locate = (text, i = 0) => {
            const local = data.indexOf(text, i);
            if (local > -1) {
                return [local, ...locate(text, local + 1)];
            }
            return [];
        };
        const search = async (text) => {
            if (data === 'none') {
                return searchFromWebPage(text, engine);
            }
            const result = [];
            const local = locate(text);
            const regText = new RegExp(text, 'g');
            for (const i of local) {
                const matchResult = data.slice(i - 100, i + 500).replace(regText, `<font style="color:red">${text}</font>`);
                if (highLightAbswer) {
                    const arr = matchResult.split(text);
                    arr[1] = arr[1].replace(/[\w]+/, '<font style="color:red">$&</font>');
                    result.push(arr.join(text));
                    continue;
                }
                result.push(matchResult);
            }
            return result.filter((e) => e.trim()).map((e) => {
                if (!(e.includes('<img') && imageData && Object.keys(imageData).length > 0)) {
                    return e;
                }
                // eslint-disable-next-line
                Object.keys(imageData).map((imageMd5) => e.includes(`$${imageMd5}$`) && (e = e.replace(`$${imageMd5}$`, imageData[imageMd5])));
                return e;
            })
                .join('<br><hr data-content="分隔线">');
        };
        const readData = async () => {
            try {
                const imagesData = {};
                const data = await new Promise((res) => {
                    // eslint-disable-next-line max-len
                    const input = $('<input type="file" id="search-answer-js" style="width:50%;height:50%;color:red;position:fixed;left:25%;top:25%;background-color:red;z-index:99999999" title="点此加载题库" multiple="multiple">');
                    $('body').append(input);
                    input[0].addEventListener('change', async function selectedFileChanged() {
                        if (this.files?.length) {
                            Swal.fire('读取&处理中...', 'Excel格式文件和题目较多时处理较慢，请耐心等待！');
                            Swal.showLoading();
                            await new Promise((resolve) => {
                                setTimeout(() => {
                                    resolve(true);
                                }, 1000);
                            });
                            const text = (await Promise.all([...(this.files || [])].map((file) => new Promise((resolve) => {
                                const reader = new FileReader();
                                const fileName = file.name;
                                reader.onabort = () => resolve('');
                                reader.onerror = () => resolve('');
                                if (/.*?\.docx?$/.test(fileName)) {
                                    reader.onload = async () => {
                                        const arrayBuffer = reader.result;
                                        const options = {
                                            convertImage: mammoth.images.imgElement((image) => image.read('base64').then((imageBuffer) => {
                                                const imageMd5 = md5(imageBuffer);
                                                imagesData[imageMd5] = `data:${image.contentType};base64,${imageBuffer}`;
                                                return {
                                                    src: `$${imageMd5}$`
                                                };
                                            }))
                                        };
                                        const { value: fileData } = await mammoth.convertToHtml({ arrayBuffer }, options);
                                        resolve(fileData);
                                    };
                                    reader.readAsArrayBuffer(file);
                                }
                                else if (/.*?\.xlsx?$/.test(fileName)) {
                                    reader.onload = async () => {
                                        const arrayBuffer = reader.result;
                                        const { Sheets } = XLSX.read(arrayBuffer);
                                        // eslint-disable-next-line max-len
                                        const fileData = Object.values(Sheets).map((sheet) => XLSX.utils.sheet_to_json(sheet, { header: 1 }).map((cell) => cell.map((value) => value?.toString()?.trim()).filter((value) => value)
                                            .join(' | '))
                                            .join('<br/>'))
                                            .join('<br/>');
                                        resolve(fileData);
                                    };
                                    reader.readAsArrayBuffer(file);
                                }
                                else {
                                    reader.onload = () => {
                                        const fileData = reader.result;
                                        if (!fileData) {
                                            return resolve('');
                                        }
                                        resolve(fileData);
                                    };
                                    reader.readAsText(file);
                                }
                            })))).join('<br/>');
                            GM_setValue('data0', text);
                            GM_setValue('data1', imagesData);
                            input.remove();
                            Swal.fire('题库加载完毕！');
                            res(text);
                        }
                    });
                    document.querySelector('#search-answer-js').click();
                });
                return { text: data, image: imagesData };
            }
            catch (error) {
                console.error(error);
                Swal.fire('题库加载失败！', '详情请查看控制台', 'error');
                return {};
            }
        };
        await Swal.fire({
            title: '是否加载题库？',
            html: '加载题库：如果你有题库，请加载你的题库（推荐）<br/>直接运行：如之前加载过题库，并且不需要重新加载题库<br/>无题库模式：弹出网页显示搜索结果',
            confirmButtonText: '加载题库',
            showCancelButton: true,
            cancelButtonText: '直接运行',
            showDenyButton: true,
            denyButtonText: '无题库模式'
        }).then(async ({ isConfirmed, isDenied }) => {
            if (isConfirmed) {
                data = (await readData()).text;
                imageData = (await readData()).image;
            }
            else if (isDenied) {
                data = 'none';
                const { value: selectedEngine } = await Swal.fire({
                    title: '请选择搜索引擎',
                    input: 'radio',
                    inputOptions: {
                        baidu: '百度',
                        sougou: '搜狗',
                        bing: '必应',
                        google: '谷歌'
                    },
                    inputValidator: (value) => {
                        if (!value) {
                            return '请选择一个搜索引擎！';
                        }
                        return '';
                    }
                });
                if (selectedEngine) {
                    engine = selectedEngine;
                }
            }
            else {
                data = GM_getValue('data0');
                imageData = GM_getValue('data1');
            }
        });
        if (!data)
            return Swal.fire('加载题库失败', '', 'error');
        const icon = document.createElement('div');
        icon.innerHTML = '搜';
        icon.setAttribute('style', '' +
            'width:32px!important;' +
            'height:32px!important;' +
            'display:none!important;' +
            'background:#fff!important;' +
            'border-radius:16px!important;' +
            'box-shadow:4px 4px 8px #888!important;' +
            'position:absolute!important;' +
            'z-index:2147483647!important;' +
            'font-size: 24px;text-align-last: center;' +
            'cursor: pointer;' +
            '');
        icon.setAttribute('title', '搜索');
        document.documentElement.appendChild(icon);
        document.addEventListener('mousedown', (e) => {
            if (e.target === icon || (e.target?.parentNode === icon) || (e.target?.parentNode?.parentNode === icon)) {
                e.preventDefault();
            }
        });
        document.addEventListener('selectionchange', () => {
            if (!window.getSelection()?.toString()
                ?.trim()) {
                icon.style.display = 'none';
            }
        });
        document.addEventListener('mouseup', (e) => {
            if (e.target === icon || (e.target?.parentNode === icon) || (e.target?.parentNode?.parentNode === icon)) {
                e.preventDefault();
                return;
            }
            const text = window.getSelection()?.toString()
                ?.trim();
            if (text && icon.style.display === 'none') {
                icon.style.top = `${e.pageY + 12}px`;
                icon.style.left = `${e.pageX - 18}px`;
                icon.innerHTML = '搜';
                icon.setAttribute('title', '搜索');
                icon.style.display = 'block';
            }
            else if (!text) {
                icon.style.display = 'none';
            }
        });
        icon.addEventListener('click', async () => {
            const text = window.getSelection()?.toString()
                ?.trim();
            if (text) {
                icon.style.display = 'none';
                const result = await search(text);
                if (data && data !== 'none' && result !== null) {
                    Swal.fire({
                        html: result
                    });
                }
            }
        });
    };
    const settings = () => {
        Swal.fire({
            title: '设置',
            // eslint-disable-next-line max-len
            html: `<div class="setting"><input id="high-light-answer" type="checkbox"${highLightAbswer ? ' checked="checked"' : ''}/>高亮答案（仅支持题库模式且题目后面要紧跟"ABCD..."格式的答案）<br/>启动快捷键：<input id="start-shortcut-key" type="text" readonly="readonly" value="${startShortcutKey || ''}"/><br/>启动快捷键：<input id="ocr-shortcut-key" type="text" readonly="readonly" value="${ocrShortcutKey || ''}"/></div>`,
            preConfirm: () => ({
                highLightAbswer: $('#high-light-answer').is(':checked'),
                startShortcutKey: $('#start-shortcut-key').val(),
                ocrShortcutKey: $('#ocr-shortcut-key').val()
            })
        }).then(({ value }) => {
            highLightAbswer = value?.highLightAbswer;
            startShortcutKey = value?.startShortcutKey;
            ocrShortcutKey = value?.ocrShortcutKey;
            GM_setValue('settings', {
                highLightAbswer, startShortcutKey, ocrShortcutKey
            });
        });
        $('#start-shortcut-key,#ocr-shortcut-key').on('keydown', function (event) {
            let functionKey = '';
            if (event.metaKey) {
                functionKey += 'Meta+';
            }
            if (event.ctrlKey) {
                functionKey += 'Control+';
            }
            if (event.altKey) {
                functionKey += 'Alt+';
            }
            if (event.shiftKey) {
                functionKey += 'Shift+';
            }
            const keyValue = event.key.toUpperCase();
            $(this).val(functionKey + (['MEAT', 'ALT', 'CONTROL', 'SHIFT'].includes(keyValue) ? '' : keyValue));
        });
    };
    const OCR = async () => {
        const worker = Tesseract.createWorker({
            logger: (message) => console.log(message)
        });
        await worker.load();
        await worker.loadLanguage('eng+chi_sim+chi_tra');
        await worker.initialize('chi_sim');
        Swal.fire('正在进行OCR识别，请耐心等待...');
        Swal.showLoading();
        for (const element of $.makeArray($('img[src]:not(".ocred")'))) {
            try {
                const { data: { text } } = await worker.recognize(element);
                if (text) {
                    $(element).after(`<div>${text}</div>`);
                }
            }
            catch (e) {
                console.error(e);
            }
            $(element).addClass('ocred');
        }
        /*
        for (const element of $.makeArray($('body *:not(".ocred")')).filter((e) => /^url\(.*?\)$/.test($(e).css('backgroundImage')))) {
          const { data: { text } } = await worker.recognize($(element).css('backgroundImage')
            .replace('url("', '')
            .replace('")', ''));
          if (text) {
            $(element).after(`<div>${text}</div>`);
          }
          $(element).addClass('ocred');
        }
        */
        await worker.terminate();
        Swal.hideLoading();
        Swal.fire('OCR识别完成！', '', 'success');
    };
    const tinykeysOptions = {};
    if (startShortcutKey) {
        tinykeysOptions[startShortcutKey] = start;
    }
    if (ocrShortcutKey) {
        tinykeysOptions[ocrShortcutKey] = OCR;
    }
    window.tinykeys.default(window, tinykeysOptions);
    GM_registerMenuCommand('启动', start);
    GM_registerMenuCommand('设置', settings);
    GM_addStyle(`
.swal2-container {
  z-index: 9999999999 !important;
}
.swal2-html-container *{
  left:0;
  padding-left:0 !important;
  margin-left:0;
  border-left:0;
  width:100%;
}
.swal2-html-container hr{
  color: #a2a9b6;
  border: 0;
  font-size: 12px;
  padding: 1em 0;
  position: relative;
}
.swal2-html-container hr::before {
  content: attr(data-content);
  position: absolute;
  padding: 0 1ch;
  line-height: 1px;
  border: solid #d0d0d5;
  border-width: 0 99vw;
  width: fit-content;
  white-space: nowrap;
  left: 50%;
  transform: translateX(-50%);
}
.swal2-html-container hr::after{
  content: attr(data-content);
  position: absolute;
  padding: 4px 1ch;
  top: 50%; left: 50%;
  transform: translate(-50%, -50%);
  color: transparent;
  border: 1px solid #d0d0d5;
}
.swal2-html-container .setting {
  text-align: left;
}
.swal2-html-container input[type="checkbox"]{
  width: 15px;
}
.swal2-html-container input[type="text"]{
  width: 200px;
  border: 2px solid #00a9fd;
  border-radius: 5px;
  font-size: 15px;
}
`);
})();