☰

洛谷验证码自动识别并填写

使用卷积神经网络训练的验证码识别模型，自动识别填写洛谷提交时的验证码

// ==UserScript==
// @name         洛谷验证码自动识别并填写
// @namespace    https://github.com/mayoi-Akira
// @version      0.1.0
// @description  使用卷积神经网络训练的验证码识别模型，自动识别填写洛谷提交时的验证码
// @match        *://www.luogu.com.cn/*
// @icon         https://www.luogu.com.cn/favicon.ico
// @grant        GM_xmlhttpRequest
// @author       Akira
// @license      MIT
// ==/UserScript==


(() => {
  const server = 'http://8.130.187.204:3636';

  const canvas = document.createElement('canvas');
  const ctx = canvas.getContext('2d');

  function recognize(img, callback) {
    canvas.width = img.naturalWidth;
    canvas.height = img.naturalHeight;
    ctx.drawImage(img, 0, 0);
    const data = canvas.toDataURL('image/jpeg').split(',')[1];

    GM_xmlhttpRequest({
      method: 'POST',
      url: server,
      headers: { 'Content-Type': 'application/json' },
      data: JSON.stringify({ image: data }),
      onload: resp => {
        try {
          const { prediction } = JSON.parse(resp.responseText);
          callback(prediction);
        } catch (e) {
          console.error('OCR 获取失败', e);
          callback('');
        }
      },
      onerror: err => {
        console.error('OCR 请求失败', err);
        callback('');
      },
      timeout: 5000
    });
  }

  const observer = new MutationObserver(mutations => {
    for (const m of mutations) {
      let img = null;
      if (m.type === 'childList') {
        img = [...m.addedNodes].find(n => n.nodeName === 'IMG' && n.src.includes('captcha'));
      } else if (m.type === 'attributes' && m.target.nodeName === 'IMG' && m.target.src.includes('captcha')) {
        img = m.target;
      }
      if (!img) continue;

      const input = document.querySelector('input[placeholder*="验证码"]');
      if (!input) {
        console.warn('未找到输入框');
        continue;
      }

      const run = () => {
        recognize(img, text => {
          input.value = text;
          input.dispatchEvent(new Event('input', { bubbles: true }));
          input.dispatchEvent(new Event('change', { bubbles: true }));
        });
      };

      if (img.complete) {
        run();
      } else {
        img.onload = run;
      }
    }
  });
  const root = document.querySelector('#captcha-container') || document.body;
  observer.observe(root, { childList: true, subtree: true, attributes: true });
})();