☰

ChatGPT Utils

Modifies the behavior of the chat interface on the OpenAI website

当前为 2022-12-26 提交的版本，查看 最新版本。

  1. // ==UserScript==
  2. // @name         ChatGPT Utils
  3. // @description  Modifies the behavior of the chat interface on the OpenAI website
  4. // @namespace    ChatGPTUtils
  5. // @version      1.7.5
  6. // @author       CriDos
  7. // @match        https://chat.openai.com/*
  8. // @icon         https://www.google.com/s2/favicons?sz=64&domain=chat.openai.com
  9. // @require      https://cdnjs.cloudflare.com/ajax/libs/he/1.2.0/he.min.js
  10. // @require      https://cdnjs.cloudflare.com/ajax/libs/jquery/3.6.1/jquery.min.js
  11. // @grant        GM_xmlhttpRequest
  12. // @run-at       document-end
  13. // @license      MIT
  14. // ==/UserScript==
  15.  
  16. 'use strict';
  17.  
  18. console.log(`ChatGPT Utils initializing...`);
  19.  
  20. let debug = true;
  21.  
  22. setInterval(() => {
  23.     try {
  24.         addAutoTranslate();
  25.         //addTranslateButtons();
  26.     } catch (error) {
  27.         console.error(error);
  28.     }
  29.  
  30.     try {
  31.         findAndHookTextareaElement();
  32.     } catch (error) {
  33.         console.error(error);
  34.     }
  35. }, 100);
  36.  
  37. function addAutoTranslate() {
  38.     var messages = document.querySelectorAll(".markdown.prose");
  39.  
  40.     for (var i = 0; i < messages.length; i++) {
  41.         const msgMarkdownNode = messages[i];
  42.         const parentMsgMarkdown = msgMarkdownNode.parentElement;
  43.  
  44.         if (parentMsgMarkdown.isAutoTranslate) {
  45.             continue;
  46.         }
  47.         parentMsgMarkdown.isAutoTranslate = true;
  48.  
  49.         setInterval(async () => {
  50.             await translateNode(msgMarkdownNode);
  51.         }, 500);
  52.     }
  53. }
  54.  
  55. function addTranslateButtons() {
  56.     var messages = document.querySelectorAll(".markdown.prose");
  57.  
  58.     for (var i = 0; i < messages.length; i++) {
  59.         const msgMarkdownNode = messages[i];
  60.         const msgIcon = msgMarkdownNode.parentElement.parentElement.parentElement.previousElementSibling;
  61.  
  62.         if (!msgIcon.querySelector(".translate-button")) {
  63.             var btn = document.createElement("button");
  64.             btn.textContent = "Tr";
  65.             btn.classList.add("translate-button");
  66.             btn.style.cssText = "width: 30px; height: 30px;";
  67.             msgIcon.insertBefore(btn, msgIcon.firstChild);
  68.  
  69.             btn.addEventListener("click", async () => {
  70.                 await translateNode(msgMarkdownNode);
  71.             });
  72.         }
  73.     }
  74. }
  75.  
  76. function findAndHookTextareaElement() {
  77.     const targetElement = document.querySelector("textarea");
  78.     if (targetElement === null) {
  79.         return;
  80.     }
  81.  
  82.     if (targetElement.isAddHookKeydownEvent === true) {
  83.         return;
  84.     }
  85.  
  86.     targetElement.isAddHookKeydownEvent = true;
  87.  
  88.     console.log(`Textarea element found. Adding keydown event listener.`);
  89.     targetElement.addEventListener("keydown", async event => await handleSubmit(event, targetElement), true);
  90. }
  91.  
  92. async function handleSubmit(event, targetElement) {
  93.     console.log(`Keydown event detected: type - ${event.type}, key - ${event.key}`);
  94.  
  95.     if (event.shiftKey && event.key === "Enter") {
  96.         return;
  97.     }
  98.  
  99.     if (window.isActiveOnSubmit === true) {
  100.         return;
  101.     }
  102.  
  103.     if (event.key === "Enter") {
  104.         window.isActiveOnSubmit = true;
  105.         event.stopImmediatePropagation();
  106.  
  107.         const request = targetElement.value;
  108.         targetElement.value = "";
  109.  
  110.         const translatedText = await translate(request, "ru", "en", "text");
  111.  
  112.         targetElement.focus();
  113.         targetElement.value = translatedText;
  114.         const enterEvent = new KeyboardEvent("keydown", {
  115.             bubbles: true,
  116.             cancelable: true,
  117.             key: "Enter",
  118.             code: "Enter"
  119.         });
  120.         targetElement.dispatchEvent(enterEvent);
  121.  
  122.         window.isActiveOnSubmit = false;
  123.     }
  124. }
  125.  
  126. async function translateNode(node) {
  127.     const translateClassName = "translate-markdown";
  128.     const parentNode = node.parentElement;
  129.     const nodeContent = $(node).html();
  130.  
  131.     if (node.storeContent == nodeContent) {
  132.         return;
  133.     }
  134.     node.storeContent = nodeContent;
  135.  
  136.     var translateNode = parentNode.querySelector(`.${translateClassName}`);
  137.     if (translateNode == null) {
  138.         translateNode = node.cloneNode(true);
  139.         translateNode.classList.add(translateClassName);
  140.         parentNode.insertBefore(translateNode, parentNode.firstChild);
  141.     }
  142.  
  143.     var translatedContent = await translateHTML(nodeContent, "en", navigator.language)
  144.     translatedContent = translatedContent.replace(/<\/div>$/, '');
  145.  
  146.     $(translateNode).html(translatedContent + `<p>.......... конец_перевода ..........</p></div>`);
  147. }
  148.  
  149. async function translateHTML(html, sLang, tLang) {
  150.     const excludeTagRegex = /<(pre|code)[^>]*>([\s\S]*?)<\/(pre|code)>/g;
  151.     const excludeTags = [];
  152.     const excludePlaceholder = 'e0x1c';
  153.  
  154.     let htmlContent = html;
  155.  
  156.     let excludeTagsMatch;
  157.     while (excludeTagsMatch = excludeTagRegex.exec(html)) {
  158.         excludeTags.push(excludeTagsMatch[0]);
  159.         htmlContent = htmlContent.replace(excludeTagsMatch[0], `<${excludePlaceholder}${excludeTags.length - 1}>`);
  160.     }
  161.  
  162.     if (debug) {
  163.         console.log(`preTranslateHTML: ${html}`);
  164.     }
  165.  
  166.     htmlContent = await translate(htmlContent, sLang, tLang, "html", true);
  167.  
  168.     for (let i = 0; i < excludeTags.length; i++) {
  169.         htmlContent = htmlContent.replace(`<${excludePlaceholder}${i}>`, excludeTags[i]);
  170.     }
  171.  
  172.     if (debug) {
  173.         console.log(`postTranslateHTML: ${htmlContent}`);
  174.     }
  175.  
  176.     return htmlContent;
  177. }
  178.  
  179. async function translate(text, sLang, tLang, format, encode) {
  180.     try {
  181.         if (debug) {
  182.             console.log(`preTranslate: ${text}`);
  183.         }
  184.  
  185.         if (encode === true) {
  186.             text = he.encode(text);
  187.         }
  188.  
  189.         if (format == null) {
  190.             format = "html";
  191.         }
  192.  
  193.         const result = await new Promise((resolve, reject) => {
  194.             GM_xmlhttpRequest({
  195.                 method: "POST",
  196.                 url: "https://translate.google.com/translate_a/single",
  197.                 data: `format=${format}&client=gtx&sl=${sLang}&tl=${tLang}&dt=t&q=${encodeURIComponent(text)}`,
  198.                 headers: {
  199.                     "Content-Type": "application/x-www-form-urlencoded"
  200.                 },
  201.                 onload: response => {
  202.                     const translatedText = JSON.parse(response.responseText)[0]
  203.                         .map(text => text[0])
  204.                         .join(" ");
  205.  
  206.                     resolve(translatedText);
  207.                 },
  208.                 onerror: response => {
  209.                     reject(response.statusText);
  210.                 }
  211.             });
  212.         });
  213.  
  214.         if (debug) {
  215.             console.log(`postTranslate: ${result}`);
  216.         }
  217.  
  218.         return result;
  219.     } catch (error) {
  220.         console.error(error);
  221.     }
  222. }