LeetCode Assistant

【使用前先看介绍/有问题可反馈】力扣助手 (LeetCode Assistant):为力扣页面增加辅助功能。

当前为 2021-09-11 提交的版本,查看 最新版本

您需要先安装一个扩展,例如 篡改猴Greasemonkey暴力猴,之后才能安装此脚本。

You will need to install an extension such as Tampermonkey to install this script.

您需要先安装一个扩展,例如 篡改猴暴力猴,之后才能安装此脚本。

您需要先安装一个扩展,例如 篡改猴Userscripts ,之后才能安装此脚本。

您需要先安装一款用户脚本管理器扩展,例如 Tampermonkey,才能安装此脚本。

您需要先安装用户脚本管理器扩展后才能安装此脚本。

(我已经安装了用户脚本管理器,让我安装!)

您需要先安装一款用户样式管理器扩展,比如 Stylus,才能安装此样式。

您需要先安装一款用户样式管理器扩展,比如 Stylus,才能安装此样式。

您需要先安装一款用户样式管理器扩展,比如 Stylus,才能安装此样式。

您需要先安装一款用户样式管理器扩展后才能安装此样式。

您需要先安装一款用户样式管理器扩展后才能安装此样式。

您需要先安装一款用户样式管理器扩展后才能安装此样式。

(我已经安装了用户样式管理器,让我安装!)

// ==UserScript==
// @name         LeetCode Assistant
// @namespace    http://tampermonkey.net/
// @version      1.0.1
// @description  【使用前先看介绍/有问题可反馈】力扣助手 (LeetCode Assistant):为力扣页面增加辅助功能。
// @author       cc
// @require      https://cdn.bootcss.com/jquery/3.4.1/jquery.js
// @require      https://greatest.deepsurf.us/scripts/422854-bubble-message.js
// @match        https://leetcode-cn.com/problems/*
// @match        https://leetcode-cn.com/problemset/all/
// @grant        GM_setValue
// @grant        GM_getValue
// ==/UserScript==

(function() {
  var executing = false;
  const bm = new BubbleMessage();
  const config = {
    recommendVisible: false,
  };

  // 通用
  function updateData(obj) {
    var data = GM_getValue('data');
    if (!obj) {
      // 初始化调用
      if (!data) {
        // 未初始化
        data = {};
        Object.assign(data, config);
        GM_setValue('data', data);
      } else {
        // 已初始化
        Object.assign(config, data);
      }
    } else {
      // 更新调用
      Object.assign(config, obj);
      Object.assign(data, config);
      GM_setValue('data', data);
    }
  }

  function insertTextarea() {
    let textarea = document.createElement('textarea');
    textarea.setAttribute('id', 'leetcode-assistant-textarea');
    textarea.style = 'width: 0; height: 0;';
    document.body.appendChild(textarea);
  }

  function executeCopy(value) {
    let textarea = document.getElementById('leetcode-assistant-textarea');
    textarea.value = value;
    textarea.setAttribute('value', value);
    textarea.select();
    document.execCommand('copy');
    bm.message({
      type: 'success',
      message: '复制成功',
      duration: 1500,
    });
  }

  function switchVisible(nodes, visible) {
    if (visible) {
      nodes.forEach(node => node.style.display = '');
    } else {
      nodes.forEach(node => node.style.display = 'none');
    }
  }

  function observeChildList(node, callback) {
    let observer = new MutationObserver(function(mutations) {
      mutations.forEach((mutation) => {
        if (mutation.type == 'childList' && mutation.addedNodes.length > 0) {
          callback([...mutation.addedNodes]);
        }
      });
    });
    observer.observe(node, { childList: true });
  }

  function listenHistoryState() {
    const _historyWrap = function(type) {
      const orig = history[type];
      const e = new Event(type);
      return function() {
          const rv = orig.apply(this, arguments);
          e.arguments = arguments;
          window.dispatchEvent(e);
          return rv;
        };
      };
    history.pushState = _historyWrap('pushState');
    window.addEventListener('pushState', () => {
      if (!executing) {
        executing = true;
        main();
      }
    });
  }

  // 提交记录页
  function requestCode(qid) {
    let query = `
      query mySubmissionDetail($id: ID!) {
        submissionDetail(submissionId: $id) {
          id
          code
          runtime
          memory
          rawMemory
          statusDisplay
          timestamp
          lang
          passedTestCaseCnt
          totalTestCaseCnt
          sourceUrl
          question {
            titleSlug
            title
            translatedTitle
            questionId
            __typename
          }
          ... on GeneralSubmissionNode {
            outputDetail {
              codeOutput
              expectedOutput
              input
              compileError
              runtimeError
              lastTestcase
              __typename
            }
            __typename
          }
          submissionComment {
            comment
            flagType
            __typename
          }
          __typename
        }
      }`;
    $.ajax({
      url: 'https://leetcode-cn.com/graphql/',
      method: 'POST',
      contentType: 'application/json',
      data: JSON.stringify({
        operationName: 'mySubmissionDetail',
        query: query,
        variables: {
          id: qid
        },
      }),
    }).then(res => {
      executeCopy(res.data.submissionDetail.code);
    });
  }

  function insertCopyFunction() {
    let tbody = document.querySelector('.ant-table-tbody');
    let trs = [...tbody.querySelectorAll('tr')];
    let processTr = (tr) => {
      let qid = tr.dataset.rowKey;
      let cell = tr.querySelector(':nth-child(4)');
      cell.title = '点击复制代码'
      cell.style = 'cursor: pointer; color: #007aff';
      cell.addEventListener('click', function() {
        requestCode(qid);
      });
    }
    trs.forEach(processTr);
    observeChildList(tbody, (nodes) => {
      let node = nodes[0];
      if (node.tagName == 'TR')
        processTr(node);
    });
    executing = false;
  }

  // 首页
  function switchRecommendVisible() {
    var nodes = [];
    var target = document.querySelector('.border-divider-border-2');
    while (target) {
      nodes.push(target);
      target = target.previousElementSibling;
    }
    var sidebar = document.querySelector('.col-span-4:nth-child(2)');
    target = sidebar.querySelector('.space-y-4:nth-child(2)');
    while (target) {
      nodes.push(target);
      target = target.nextElementSibling;
    }
    switchVisible(nodes, config.recommendVisible);
    observeChildList(sidebar, (nodes) => {
      switchVisible(nodes, config.recommendVisible);
    });
  }
  
  function insertVisibleSwitch() {
    var insertDiv = document.querySelector('.relative.space-x-5').nextElementSibling;
    insertDiv.style = 'display: inline-flex; align-items: center;';
    var switchCheckbox = document.createElement('input');
    switchCheckbox.type = 'checkbox';
    switchCheckbox.checked = true;
    switchCheckbox.setAttribute('id', 'leetcode-assistant-recommend-switch');
    switchCheckbox.addEventListener('change', function() {
      updateData({ recommendVisible: !this.checked });
      switchRecommendVisible();
    });
    var switchLabel = document.createElement('label');
    switchLabel.setAttribute('for', 'leetcode-assistant-recommend-switch');
    switchLabel.innerText = '简洁模式';
    switchLabel.style.marginLeft = '5px';
    switchLabel.style = 'margin-left: 5px; cursor: default;';
    insertDiv.appendChild(switchCheckbox);
    insertDiv.appendChild(switchLabel);
  }

  // 主函数
  function main() {
    if (location.href.match(/\/submissions\/$/)) {
      (function r() {
        if (document.querySelector('.ant-table-thead')) {
          insertCopyFunction();
        } else {
          setTimeout(r, 500);
        }
      })();
    } else if (location.href == 'https://leetcode-cn.com/problemset/all/') {
      insertVisibleSwitch();
      switchRecommendVisible();
    } else {
      executing = false;
    }
  }

  window.addEventListener('load', () => {
    updateData();
    insertTextarea();
    listenHistoryState();
    main();
  });
})();