☰

Greasy Fork is available in English.

v2ex AI 总结帖子

自定义 api key 等信息，实现 AI 总结帖子，会保留缓存记录到本地避免大量消耗 token。

// ==UserScript==
// @name         v2ex AI 总结帖子
// @namespace    https://github.com/dlzmoe/scripts
// @version      0.0.5
// @description  自定义 api key 等信息，实现 AI 总结帖子，会保留缓存记录到本地避免大量消耗 token。
// @author       dlzmoe
// @match        *://v2ex.com/*
// @match        *://*.v2ex.com/*
// @match        *://www.v2ex.com/*
// @grant        GM_xmlhttpRequest
// @grant        GM_registerMenuCommand
// @grant        GM_unregisterMenuCommand
// @grant        GM_openInTab
// @grant        GM_getValue
// @grant        GM_setValue
// @grant        GM_notification
// @grant        GM_info
// @icon         data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAACAAAAAgCAYAAABzenr0AAAAOGVYSWZNTQAqAAAACAABh2kABAAAAAEAAAAaAAAAAAACoAIABAAAAAEAAAAgoAMABAAAAAEAAAAgAAAAAI9OQMkAAARKSURBVHicrVe7SzNLFP/t7CObbEg0WliJooUxcAsfYKFW2lh8NhdLwUIsbMTCBG3EIoid4J/xNd9t7ZSAIiJcNaSIYBcbNVlMsq/M3MK762YziVE8EGbYOTnnd55zRsD/xBiT7+/vf+u6vlir1VTLskApFfADRAhhiqIgEokYsVjsNJVK/S0Igg0AAgAUCoW/np+fr15fX5WfUPgZ9fT0WP39/dNjY2P/SowxOZfLXZXLZUVRFIyPjyORSCAUCkEQmh3AGIMgCGCMAYC3D/IFiVIKy7Lw8vKCfD6PcrmsALhijEWFu7u7P4+Pj78URcHMzAyi0WjXCoPnQeLxV6tVXFxcwLIsDA0N/UN0XV8EgGQyCU3TQClFo9FAo9EAY8xb/Xv33OWllIJSCsaYt/L4GWPQNA3JZBIAoOv6olSr1VQASCQS3h/9FgWtC54FPcDbB2UkEgkAQK1WUyXLsgAAiqK0MFNKIQgCKKVcl3ajMChPFEXIsgwAsCwLkltq/jh1K/C7RAhxAQmSX3DQwo2NDZydnbWcu4D9vKFQCNvb21hdXfUU8ZIwaIQHwD3w50Aul/PQflZyjuPg6OgIoVAIKysrbRUGc6wFgH8lhLRV2K4Es9ksotEolpaWuuobTSEAPhIPACSpCV/XtL+/D03TMD8/3ySP5xHiB+BnYIxBFEUQQkAI8faiKHJ/fh7GGPb29nB9fd0kryOAIBAXgCRJkCQJhBBv5QGSJKmJ33EcpNNp5PP5JpnBkHXMgcvLy65cXiqVsLm5iaenp6bvpmkinU7j+PgYw8PD3BJuCgEAr+kEAXXqCQMDAzg5OUFfX19LaN7e3pDJZGDbNtcLXA+4ICzL6nj5uHu3WtwwBIn3rQVAJws/6/WlUgmZTAa6rnvKXHDxeByHh4cghHgXFRcAT8Hy8nLXVy/P0nA4jGw2i8HBQTiO09kDfgD+PvDdWUCWZRwcHGB0dBTuhcczkhsCfyfslvwACSHY3d1FKpXy8oinvAlAEATw7tLPEjB4DgBbW1uYnJyEYRhfb8V+hJ0AtPPA+vo6ZmdnYZpmizy3urruhFNTU17n83dDf9dz6z0cDmNtbQ0LCwswDKNpNON1QK4Hgi7a2dnh/qkdMcZgmmbbCYo7DxBCGKVU4CWh/ybjCQwqNwyD2zWDSdhoNAC8P1gkWZZhmiYsy4KiKCCEeAz1ev1LpfdZqYqiCMaY1xNkWQZRVdUAgHK5/PHxY2bzVv/eb43fa/7RPMhPCPGGUVeXqqqGFI1GTyuVyq9isQhN0xCJRBAKhbqy8KvE2PvDpFgsAgA0TTsVGGPy+fn5m67riizLGBkZQTweh6Io31YUJEopbNtGpVLBw8MDbNtGLBaz5ubmopIgCPbt7e00gCtd15VCofAjSjtRLBazent7pwVBsD0TGWPyzc3N72q1uliv11XHcX70eS7LMlRVNSKRyOnExIT3PP8P91unlxYYZf4AAAAASUVORK5CYII=
// @license      Apache-2.0 license
// ==/UserScript==

(function () {
  'use strict';
  var menu_ALL = [
    ['menu_ManualSummary', '是否开启手动总结 / 自动', '是否开启手动总结 / 自动', true],
  ];
  var menu_ID = [];
  for (let i = 0; i < menu_ALL.length; i++) { // 如果读取到的值为 null 就写入默认值
    if (GM_getValue(menu_ALL[i][0]) == null) {
      GM_setValue(menu_ALL[i][0], menu_ALL[i][3])
    };
  }
  registerMenuCommand();

  // 注册脚本菜单
  function registerMenuCommand() {
    if (menu_ID.length > menu_ALL.length) { // 如果菜单ID数组多于菜单数组，说明不是首次添加菜单，需要卸载所有脚本菜单
      for (let i = 0; i < menu_ID.length; i++) {
        GM_unregisterMenuCommand(menu_ID[i]);
      }
    }
    for (let i = 0; i < menu_ALL.length; i++) { // 循环注册脚本菜单
      menu_ALL[i][3] = GM_getValue(menu_ALL[i][0]);
      menu_ID[i] = GM_registerMenuCommand(`${menu_ALL[i][3]?'✅':'❌'} ${menu_ALL[i][1]}`, function () {
        menu_switch(`${menu_ALL[i][3]}`, `${menu_ALL[i][0]}`, `${menu_ALL[i][2]}`)
      });
    }
    menu_ID[menu_ID.length] = GM_registerMenuCommand('⚙️ 设置API key配置', function () {
      setApiConfig();
    });

    menu_ID[menu_ID.length] = GM_registerMenuCommand('💬 建议与反馈！', function () {
      window.GM_openInTab("https://github.com/dlzmoe/scripts", {
        active: true,
        insert: true,
        setParent: true
      });
    });

  }

  function setApiConfig() {

    $('body').append(`
      <div class="v2exaisummary">
  <input type="text" id="v2exaisummary-apikey" placeholder="sk-xxxxxxx">
  <input type="text" id="v2exaisummary-baseurl" placeholder="https://api.openai.com" value="https://api.openai.com">
  <input type="text" id="v2exaisummary-model" placeholder="gpt-4o-mini" value="gpt-4o-mini">
  <button id="v2exaisummary-save">保存</button>
</div>
      `)

    $('.v2exaisummary').show();

    var v2exaisummaryAPI = JSON.parse(localStorage.getItem('v2exaisummaryAPI'));
    if (!v2exaisummaryAPI) {
      v2exaisummaryAPI = {
        apikey: "",
        baseurl: "",
        model: "",
      }
    } else {
      $('#v2exaisummary-apikey').val(v2exaisummaryAPI.apikey);
      $('#v2exaisummary-baseurl').val(v2exaisummaryAPI.baseurl);
      $('#v2exaisummary-model').val(v2exaisummaryAPI.model);
    }

    // 保存
    $('#v2exaisummary-save').click(function () {
      v2exaisummaryAPI = {
        apikey: $('#v2exaisummary-apikey').val(),
        baseurl: $('#v2exaisummary-baseurl').val(),
        model: $('#v2exaisummary-model').val(),
      }
      localStorage.setItem('v2exaisummaryAPI', JSON.stringify(v2exaisummaryAPI));
      $('.v2exaisummary').remove();
    })
  }

  // 菜单开关
  function menu_switch(menu_status, Name, Tips) {
    if (menu_status == 'true') {
      GM_setValue(`${Name}`, false);
      GM_notification({
        text: `已关闭 [${Tips}] 功能\n（点击刷新网页后生效）`,
        timeout: 3500,
        onclick: function () {
          location.reload();
        }
      });
    } else {
      GM_setValue(`${Name}`, true);
      GM_notification({
        text: `已开启 [${Tips}] 功能\n（点击刷新网页后生效）`,
        timeout: 3500,
        onclick: function () {
          location.reload();
        }
      });
    }
    registerMenuCommand(); // 重新注册脚本菜单
  };

  // 返回菜单值
  function menu_value(menuName) {
    for (let menu of menu_ALL) {
      if (menu[0] == menuName) {
        return menu[3]
      }
    }
  }
  $(function () {

  })
  // 手动总结
  function menu_ManualSummary() {
    isCache();
    if (menu_value('menu_ManualSummary')) {
      // 手动总结
      $('.aisummary').click(function () {
        $('.aisummary').hide();
        getPostContent();
      })
    } else {
      // 自动总结
      $('.aisummary').hide();
      // getPostContent();
    }
  }

  if (window.location.pathname.indexOf('/t/') > -1) {
    menu_ManualSummary();
  }

  // 获取帖子内容
  function getPostContent() {
    var v2exaisummaryAPI = JSON.parse(localStorage.getItem('v2exaisummaryAPI'));
    $('.gpt-summary-wrap').show();
    return new Promise((resolve, reject) => {
      const topic_contentdata = $('h1').html() + $('.topic_content').map(function () {
        return $(this).html();
      }).get().join('');

      const v2exprompt = `根据以下帖子内容进行总结，请使用 text 文本返回回答，字数限制 200 字以内，越精炼越好，语言要求返回简体中文，并且进行中英文混排优化。
帖子内容如下：
${topic_contentdata}`;

      fetch(`${v2exaisummaryAPI.baseurl}/v1/chat/completions`, {
          method: "POST",
          headers: {
            "Content-Type": "application/json",
            Authorization: `Bearer ${v2exaisummaryAPI.apikey}`,
          },
          body: JSON.stringify({
            model: v2exaisummaryAPI.model,
            messages: [{
              role: "user",
              content: v2exprompt,
            }],
            temperature: 0.7,
          }),
        })
        .then(response => {
          if (!response.ok) {
            reject(`HTTP error! status: ${response.status}`);
          }
          return response.json();
        })
        .then(gptData => {
          $(".gpt-summary").html(`AI 总结：${gptData.choices[0].message.content}`);
          $('.airegenerate').show();

          let v2exaisummarydata =
            JSON.parse(localStorage.getItem("v2exaisummarydata")) || [];
          const match = window.location.pathname;
          let existingObject = v2exaisummarydata.find((item) => item.name == match);

          let newObject = {
            name: match,
            value: gptData.choices[0].message.content,
          };
          if (existingObject) {
            existingObject.value = newObject.value;
          } else {
            v2exaisummarydata.push(newObject);
          }
          // 将帖子总结的数据缓存
          localStorage.setItem("v2exaisummarydata", JSON.stringify(v2exaisummarydata));
          resolve();

        })
        .catch(error => {
          $(".gpt-summary").html(`抱歉生成失败，请检查配置或者反馈给开发者！`);
          $('.airegenerate').show();
        });
    });
  }

  // 先判断是否有缓存
  function isCache() {
    $("#Main .box>.header").after(`<button type="button" class="aisummary">AI总结</button>`);
    $("#Main .box>.header").after(
      `<div class="gpt-summary-wrap">
       <div class="gpt-summary">AI 总结：正在使用 AI 总结内容中，请稍后...</div>
       <button type="button" class="airegenerate" style="display:none">重新生成</button>
        </div>`
    );

    let v2exaisummarydata = JSON.parse(localStorage.getItem("v2exaisummarydata")) || [];
    const match = window.location.pathname;
    let existingObject = v2exaisummarydata.find((item) => item.name === match);

    if (existingObject) {
      // 存在缓存，拿旧数据
      $('.gpt-summary-wrap').show();
      $(".gpt-summary").html(`AI 总结：${existingObject.value}`);
      $('.airegenerate').show();
      $('.aisummary').hide();

    } else {
      $('.gpt-summary-wrap').hide();

      if (!menu_value('menu_ManualSummary')) {
        getPostContent();
      }
    }

    $('.airegenerate').click(() => {
      $('.gpt-summary').html(`AI 总结：正在使用 AI 总结内容中，请稍后...`)
      $('.airegenerate').hide();
      getPostContent();
    })
  }

  $('body').append(`<style>.gpt-summary-wrap{background:#fffbd9;border-radius:5px;padding:10px;font-size:14px;color:#303030;margin:0;line-height:1.6;text-align:left}.aisummary{display:flex;outline:0;border:1px solid #eee;background:#ffe27d;color:#626262;padding:4px 10px;cursor:pointer;border-radius:3px}.gpt-summary-wrap .airegenerate{margin-top:6px;outline:0;border:1px solid #eee;background:#ffe27d;color:#626262;padding:4px 10px;cursor:pointer;border-radius:3px}.v2exaisummary{position:fixed;bottom:20px;right:20px;z-index:99999;max-width:400px;padding:20px;border:1px solid #ddd;border-radius:8px;box-shadow:0 2px 10px rgba(0,0,0,.1);background-color:#f9f9f9;display:none}.v2exaisummary input[type=text]{width:100%;padding:10px;margin:10px 0;border:1px solid #ccc;border-radius:4px;font-size:16px;transition:border-color .3s}.v2exaisummary input[type=text]:focus{border-color:#007bff;outline:0}.v2exaisummary button{width:100%;padding:10px;background-color:#007bff;color:#fff;border:none;border-radius:4px;font-size:16px;cursor:pointer;transition:background-color .3s}.v2exaisummary button:hover{background-color:#0056b3}</style>`)
})();