koishi-plugin-nitter 0.0.1

package/lib/index.d.ts

@@ -0,0 +1,18 @@
+import { Context, Dict, Schema } from 'koishi';
+export declare const name = "nitter";
+export declare const inject: string[];
+export interface Config {
+    apiKey: string;
+    nitterUrl: string;
+    proxy: string;
+    enableTranslate: "google" | "silicon" | "disable";
+    googleApiKey?: string;
+    siliconApiKey?: string;
+    model?: string;
+    prompt?: string;
+    app: string;
+    sendPic: boolean;
+    enableReTweet: boolean;
+}
+export declare const Config: Schema<Config, Dict>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -0,0 +1,347 @@
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __name = (target, value) => __defProp(target, "name", { value, configurable: true });
+var __export = (target, all) => {
+  for (var name2 in all)
+    __defProp(target, name2, { get: all[name2], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.tsx
+var src_exports = {};
+__export(src_exports, {
+  Config: () => Config,
+  apply: () => apply,
+  inject: () => inject,
+  name: () => name
+});
+module.exports = __toCommonJS(src_exports);
+var import_koishi = require("koishi");
+
+// src/translate.ts
+var import_axios = __toESM(require("axios"));
+var import_https_proxy_agent = require("https-proxy-agent");
+async function retry(retries, fn, delay = 500) {
+  try {
+    await fn();
+    return Promise.resolve();
+  } catch (err) {
+    console.log(`剩余${retries}次尝试`, err);
+    await new Promise((resolve) => setTimeout(resolve, delay));
+    return retries > 1 ? retry(retries - 1, fn, delay * 2) : Promise.reject(err);
+  }
+  ;
+}
+__name(retry, "retry");
+var translate;
+function setGoogleTranslate(key, proxy) {
+  translate = /* @__PURE__ */ __name((texts) => {
+    const url = `https://translation.googleapis.com/language/translate/v2?key=${key}`;
+    return Promise.all(texts.map(async (text) => {
+      let data;
+      await retry(3, async () => {
+        ({ data } = await import_axios.default.post(url, {
+          q: text.split("\n"),
+          target: "zh"
+        }, {
+          httpsAgent: new import_https_proxy_agent.HttpsProxyAgent(proxy),
+          proxy: false
+        }));
+      });
+      return data.data.translations.map((item) => item.translatedText).join("\n");
+    }));
+  }, "translate");
+}
+__name(setGoogleTranslate, "setGoogleTranslate");
+function setSiliconTranslate(key, model, prompt) {
+  translate = /* @__PURE__ */ __name(async (texts) => {
+    const url = "https://api.siliconflow.cn/v1/chat/completions";
+    let data;
+    await retry(3, async () => {
+      const response = await (0, import_axios.default)({
+        method: "POST",
+        url,
+        headers: {
+          "Authorization": `Bearer ${key}`,
+          "Content-Type": "application/json"
+        },
+        data: {
+          model: "deepseek-ai/DeepSeek-V3.2-Exp",
+          messages: [
+            {
+              role: "system",
+              content: prompt
+            },
+            {
+              role: "user",
+              content: `请翻译以下${texts.length}条HTML内容，返回JSON数组：
+
+${JSON.stringify(texts)}`
+            }
+          ],
+          stream: false,
+          temperature: 0.3,
+          // 较低的温度值使翻译更稳定
+          response_format: { type: "json_object" }
+        },
+        timeout: 3e4
+        // 30秒超时
+      });
+      if (response.data && response.data.choices && response.data.choices[0]) {
+        const content = response.data.choices[0].message.content;
+        data = JSON.parse(content);
+        if (!Array.isArray(data)) throw new Error("模型未返回数组");
+      } else {
+        throw new Error("API返回数据格式异常");
+      }
+    });
+    return data;
+  }, "translate");
+}
+__name(setSiliconTranslate, "setSiliconTranslate");
+async function addTranslate(page, className) {
+  const elementsHTML = await page.evaluate((className2) => {
+    const elements = document.querySelectorAll(className2);
+    if (!elements.length) return null;
+    return Array.from(elements).map((element) => element.outerHTML);
+  }, className);
+  if (!elementsHTML || !elementsHTML.length) return;
+  const translatedHTMLs = await translate(elementsHTML);
+  await page.evaluate((className2, originalHTMLs, translatedHTMLs2) => {
+    const elements = document.querySelectorAll(className2);
+    elements.forEach((element, index) => {
+      if (element.outerHTML === originalHTMLs[index]) {
+        const tempDiv = document.createElement("div");
+        tempDiv.innerHTML = translatedHTMLs2[index];
+        const newElement = tempDiv.firstElementChild;
+        if (newElement) {
+          element.insertAdjacentElement("afterend", newElement);
+          element.insertAdjacentHTML("afterend", '<div style="height: 1px;background-color: #ccc; margin-top: 10px; margin-bottom: 10px;"></div>');
+        }
+      }
+    });
+  }, className, elementsHTML, translatedHTMLs);
+}
+__name(addTranslate, "addTranslate");
+
+// src/index.tsx
+var import_rettiwt_api = require("rettiwt-api");
+var import_node_cron = require("node-cron");
+var import_jsx_runtime = require("@satorijs/element/jsx-runtime");
+var name = "nitter";
+var inject = ["puppeteer", "subscription"];
+var Config = import_koishi.Schema.intersect([
+  import_koishi.Schema.object({
+    apiKey: import_koishi.Schema.string().required().description("Twitter API Key"),
+    nitterUrl: import_koishi.Schema.string().required().description("Nitter"),
+    proxy: import_koishi.Schema.string().description("代理设置")
+  }).description("API 配置"),
+  import_koishi.Schema.object({
+    enableTranslate: import_koishi.Schema.union([
+      import_koishi.Schema.const("google").description("google cloud translation"),
+      import_koishi.Schema.const("silicon").description("硅基流动"),
+      import_koishi.Schema.const("disable").description("关闭")
+    ]).default("disable").role("radio")
+  }).description("翻译设置"),
+  import_koishi.Schema.union([
+    import_koishi.Schema.object({
+      enableTranslate: import_koishi.Schema.const("google").required(),
+      googleApiKey: import_koishi.Schema.string().required().description("访问https://cloud.google.com/获取,使用v2")
+    }),
+    import_koishi.Schema.object({
+      enableTranslate: import_koishi.Schema.const("silicon").required(),
+      siliconApiKey: import_koishi.Schema.string().required().description("访问https://www.siliconflow.cn/获取"),
+      model: import_koishi.Schema.string().required().description("模型名称"),
+      prompt: import_koishi.Schema.string().required().default("你是一个专业的HTML翻译助手。请将一个HTML数组翻译为中文，严格遵循以下规则：\n1. 翻译文本内容，包括链接标签（如<a>）内的显示文本\n2.保持所有HTML标签、属性、class、id，以及结构和格式，URL链接完全不变\n3. 返回一个严格的由html文本组成的JSON数组，包含与输入数量完全相同的翻译结果，不要添加任何额外说明").role("textarea").description("提示词")
+    }),
+    import_koishi.Schema.object({})
+  ]),
+  import_koishi.Schema.object({
+    app: import_koishi.Schema.string().description("subscription配置中应用名")
+  }).description("订阅配置"),
+  import_koishi.Schema.object({
+    sendPic: import_koishi.Schema.boolean().default(false).description("是否单独发送推文中的图片"),
+    enableReTweet: import_koishi.Schema.boolean().default(false).description("是否发送转推")
+  }).description("推送设置")
+]);
+function apply(ctx, config) {
+  config.nitterUrl = config.nitterUrl.replace(/\/+$/, "");
+  const twitterClient = new import_rettiwt_api.Rettiwt({
+    apiKey: config.apiKey,
+    proxyUrl: config.proxy ? new URL(config.proxy) : void 0
+  });
+  let latestTweetId;
+  let cronJob;
+  (async () => {
+    if (config.enableTranslate == "google") {
+      setGoogleTranslate(config.googleApiKey, config.proxy);
+    } else if (config.enableTranslate == "silicon") {
+      setSiliconTranslate(config.siliconApiKey, config.model, config.prompt);
+    }
+    const tweetList = await getFollowedFeed();
+    for (const data of tweetList) {
+      latestTweetId ||= "";
+      if (latestTweetId < data.id)
+        latestTweetId = data.id;
+    }
+    cronJob = (0, import_node_cron.schedule)("15 * * * * *", checkForUpdates);
+    ctx.on("dispose", () => {
+      cronJob.stop();
+    });
+    ctx.logger("nitter").info("开始监听推特动态");
+  })();
+  ctx.command("nitter.follow", "按照subscription中的订阅配置，使用登录的推特账号关注所有需要订阅的账号", { authority: 3 }).action(async () => {
+    const whiteList = ctx.subscription.getAvailableAccounts(config.app);
+    const { list: followingList } = await twitterClient.user.following();
+    const followingIdList = followingList.map((user) => user.userName);
+    for (const id of whiteList) {
+      if (!followingIdList.includes(id)) {
+        const user = await twitterClient.user.details(id);
+        await new Promise((resolve) => setTimeout(resolve, 3 * 1e3));
+        await twitterClient.user.follow(user.id);
+        console.log(`关注${id}成功`);
+        await new Promise((resolve) => setTimeout(resolve, 30 * 1e3));
+      }
+    }
+    return "关注完成";
+  });
+  ctx.command("nitter <tweetId:string>", "根据推文ID获取推文").action(async ({ session }, tweetId) => {
+    if (!tweetId) {
+      return "请输入推文ID";
+    }
+    try {
+      const [screenshot, imageUrls] = await renderTweetScreenshot(ctx, tweetId, config);
+      let msg = /* @__PURE__ */ (0, import_jsx_runtime.jsx)("img", { src: "data:image/png;base64," + screenshot.toString("base64") });
+      if (imageUrls.length > 0) {
+        msg += /* @__PURE__ */ (0, import_jsx_runtime.jsx)("message", { forward: true, children: imageUrls.map((url) => /* @__PURE__ */ (0, import_jsx_runtime.jsx)("message", { children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("img", { src: url }) })) });
+      }
+      return msg;
+    } catch (error) {
+      ctx.logger("nitter").error("获取推文失败:", error);
+      return "获取推文失败";
+    }
+  });
+  ctx.command("nitter.broadcast <account:string> <tweetId:string>", "根据账号和推文ID推送推文", { authority: 3 }).action(async ({ session }, account, tweetId) => {
+    if (!tweetId) {
+      return "请输入推文ID";
+    }
+    broadcast(account, tweetId);
+  });
+  async function broadcast(account, tweetId) {
+    try {
+      console.log(1);
+      const [screenshot, imageUrls] = await renderTweetScreenshot(ctx, tweetId, config);
+      const screenshotMsg = /* @__PURE__ */ (0, import_jsx_runtime.jsx)("img", { src: "data:image/png;base64," + screenshot.toString("base64") });
+      ctx.subscription.broadcast(config.app, account, screenshotMsg);
+      if (imageUrls.length > 0) {
+        const picsForward = imageUrls.map((url) => /* @__PURE__ */ (0, import_jsx_runtime.jsx)("message", { children: /* @__PURE__ */ (0, import_jsx_runtime.jsx)("img", { src: url }) }));
+        ctx.subscription.broadcastForward(config.app, account, picsForward);
+      }
+    } catch (error) {
+      ctx.logger("twitter").error("获取推文失败:", error);
+    }
+  }
+  __name(broadcast, "broadcast");
+  async function getFollowedFeed() {
+    let tweetList;
+    await retry(3, async () => {
+      ({ list: tweetList } = await twitterClient.user.followed());
+    });
+    return tweetList.reverse();
+  }
+  __name(getFollowedFeed, "getFollowedFeed");
+  async function checkForUpdates() {
+    const tweetList = await getFollowedFeed();
+    for (const data of tweetList) {
+      if (!config.enableReTweet && data.retweetedTweet)
+        continue;
+      if (data.id <= latestTweetId)
+        continue;
+      if (!ctx.subscription.getAvailableAccounts(config.app).includes(data.tweetBy.userName))
+        continue;
+      latestTweetId = data.id;
+      ctx.logger("nitter").info(`检测到推文id:${data.id}，开始推送`);
+      await broadcast(data.tweetBy.userName, data.id);
+    }
+  }
+  __name(checkForUpdates, "checkForUpdates");
+}
+__name(apply, "apply");
+async function renderTweetScreenshot(ctx, tweetId, config) {
+  const puppeteer = ctx.puppeteer;
+  if (!puppeteer) {
+    throw new Error("Puppeteer 服务未找到，请安装 koishi-plugin-puppeteer");
+  }
+  const page = await puppeteer.page();
+  try {
+    const tweetUrl = `${config.nitterUrl}/i/status/${tweetId}`;
+    await retry(3, async () => {
+      await page.goto(tweetUrl);
+      const element2 = await page.$(".main-thread");
+      if (!element2) throw new Error("Rate Limited");
+    }, 2e3);
+    if (config.enableTranslate) {
+      await addTranslate(page, ".main-thread .tweet-content, .main-thread .quote-text");
+    }
+    await new Promise((resolve) => setTimeout(resolve, 200));
+    await page.evaluate(() => {
+      const nav = document.querySelector("nav");
+      if (nav) {
+        nav.style.visibility = "hidden";
+      }
+    });
+    const element = await page.$(".main-thread");
+    const buffer = await element.screenshot({
+      // 不指定path参数，直接返回buffer
+      type: "png",
+      omitBackground: false
+    });
+    if (config.sendPic) {
+      const originalImages = await page.evaluate((url) => {
+        const stillImageLinks = document.querySelectorAll(".main-tweet a.still-image");
+        const imageUrls = [];
+        stillImageLinks.forEach((link) => {
+          const href = link.getAttribute("href");
+          if (href) {
+            imageUrls.push(`${url}${href}`);
+          }
+        });
+        return imageUrls;
+      }, config.nitterUrl);
+      return [buffer, originalImages];
+    }
+    return [buffer, []];
+  } catch (e) {
+    throw e;
+  } finally {
+    await page.close();
+  }
+}
+__name(renderTweetScreenshot, "renderTweetScreenshot");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Config,
+  apply,
+  inject,
+  name
+});

package/lib/translate.d.ts

@@ -0,0 +1,5 @@
+import { Page } from 'puppeteer-core';
+export declare function retry(retries: number, fn: () => any, delay?: number): any;
+export declare function setGoogleTranslate(key: string, proxy: string): void;
+export declare function setSiliconTranslate(key: string, model: string, prompt: string): void;
+export declare function addTranslate(page: Page, className: string): Promise<void>;

package/package.json

@@ -0,0 +1,26 @@
+{
+  "name": "koishi-plugin-nitter",
+  "description": "使用Rettiwt-API订阅推文，并使用nitter渲染",
+  "version": "0.0.1",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "peerDependencies": {
+    "@satorijs/element": "^3.1.8",
+    "axios": "^1.12.2",
+    "koishi": "^4.18.9",
+    "koishi-plugin-subscription": "^0.0.1",
+    "rettiwt-api": "^6.0.6",
+    "https-proxy-agent": "^7.0.6",
+    "node-cron": "^4.2.1"
+  }
+}

package/readme.md

@@ -0,0 +1,5 @@
+# koishi-plugin-twitter
+
+[![npm](https://img.shields.io/npm/v/koishi-plugin-twitter?style=flat-square)](https://www.npmjs.com/package/koishi-plugin-twitter)
+
+推文订阅