koishi-plugin-starfx-bot 0.14.3 → 0.15.0

package/lib/index.d.ts

@@ -23,58 +23,11 @@ export interface Config {
     openRepeat: boolean;
     minRepeatTimes: number;
     repeatPossibility: number;
+    originImg: boolean;
+    originImgRSSUrl: string;
+    proxyUrl: string;
     featureControl: string;
 }
-export declare const Config: Schema<Schemastery.ObjectS<{
-    openLock: Schema<boolean, boolean>;
-    openSold: Schema<boolean, boolean>;
-    bangdreamBorder: Schema<boolean, boolean>;
-}> | Schemastery.ObjectS<{
-    record: Schema<boolean, boolean>;
-    tagWeight: Schema<number, number>;
-    saveArchive: Schema<boolean, boolean>;
-}> | Schemastery.ObjectS<{
-    roll: Schema<boolean, boolean>;
-    undo: Schema<boolean, boolean>;
-    echo: Schema<boolean, boolean>;
-}> | Schemastery.ObjectS<{
-    atNotSay: Schema<boolean, boolean>;
-    atNotSayProperty: Schema<number, number>;
-    atNotSayOther: Schema<boolean, boolean>;
-    atNotSayOtherProperty: Schema<number, number>;
-    iLoveYou: Schema<boolean, boolean>;
-    replyBot: Schema<"关闭" | "无需at" | "必须at", "关闭" | "无需at" | "必须at">;
-}> | Schemastery.ObjectS<{
-    openRepeat: Schema<boolean, boolean>;
-    minRepeatTimes: Schema<number, number>;
-    repeatPossibility: Schema<number, number>;
-}> | Schemastery.ObjectS<{
-    featureControl: Schema<string, string>;
-}>, {
-    openLock: boolean;
-    openSold: boolean;
-    bangdreamBorder: boolean;
-} & import("cosmokit").Dict & {
-    record: boolean;
-    tagWeight: number;
-    saveArchive: boolean;
-} & {
-    roll: boolean;
-    undo: boolean;
-    echo: boolean;
-} & {
-    atNotSay: boolean;
-    atNotSayProperty: number;
-    atNotSayOther: boolean;
-    atNotSayOtherProperty: number;
-    iLoveYou: boolean;
-    replyBot: "关闭" | "无需at" | "必须at";
-} & {
-    openRepeat: boolean;
-    minRepeatTimes: number;
-    repeatPossibility: number;
-} & {
-    featureControl: string;
-}>;
+export declare const Config: Schema<Schemastery.ObjectS<{}>, {} & import("cosmokit").Dict>;
 export declare const usage = "<h2>StarFreedomX\u7684\u81EA\u7528\u63D2\u4EF6 \u653E\u4E86\u4E00\u4E9B\u5C0F\u529F\u80FD</h2>\n  ";
 export declare function apply(ctx: Context, cfg: Config): void;

package/lib/index.js

@@ -59,6 +59,11 @@ var import_fs = __toESM(require("fs"));
 var import_node_path = __toESM(require("node:path"));
 var import_sharp = __toESM(require("sharp"));
 var import_jimp = require("jimp");
+var import_rss_parser = __toESM(require("rss-parser"));
+var cheerio = __toESM(require("cheerio"));
+var import_http_proxy_agent = require("http-proxy-agent");
+var import_https_proxy_agent = require("https-proxy-agent");
+var import_axios = __toESM(require("axios"));
 async function addRecord(ctx, gid, avatarUrl) {
   const recordDir = `${assetsDir}/record/${gid}`;
   const avatarBuffer = await ctx.http.get(avatarUrl, { responseType: "arraybuffer" });
@@ -456,6 +461,75 @@ async function undo(cfg, session) {
   }
 }
 __name(undo, "undo");
+async function getXUrl(urls) {
+  const regex = /https:\/\/x\.com\/([^\/]+)\/status\/(\d+)/g;
+  let match;
+  const results = [];
+  while ((match = regex.exec(urls)) !== null) {
+    const [fullUrl] = match;
+    results.push(fullUrl);
+  }
+  return results;
+}
+__name(getXUrl, "getXUrl");
+var parser = new import_rss_parser.default({
+  customFields: {
+    item: ["description", "link"]
+  }
+});
+async function getXNum(session) {
+  const params = session.content.trim().split(" ").slice(1).filter((item) => !isNaN(+item) && item).map((str) => Number(str) - 1);
+  return params;
+}
+__name(getXNum, "getXNum");
+async function getXImage(rssUrl, xUrls) {
+  const xUrlsArray = Array.isArray(xUrls) ? xUrls : [xUrls];
+  const feed = await parser.parseURL(rssUrl);
+  const allImageUrls = [];
+  for (const xUrl of xUrlsArray) {
+    const item = feed.items.find((i) => i.link === xUrl);
+    if (item) {
+      const $ = cheerio.load(item.description);
+      $("img").each((_, el) => {
+        const src = $(el).attr("src");
+        if (src) allImageUrls.push(src);
+      });
+    }
+  }
+  return allImageUrls;
+}
+__name(getXImage, "getXImage");
+function chunk(arr, size) {
+  const res = [];
+  for (let i = 0; i < arr.length; i += size) {
+    res.push(arr.slice(i, i + size));
+  }
+  return res;
+}
+__name(chunk, "chunk");
+async function sendImages(session, cfg, imageUrls) {
+  const chunks = chunk(imageUrls, 10);
+  for (const group of chunks) {
+    const messages = await Promise.all(
+      group.map(async (url) => import_koishi.h.image(await getXImageBase64(url, cfg)))
+    );
+    const message = messages.join("");
+    await session.send(message);
+  }
+}
+__name(sendImages, "sendImages");
+async function getXImageBase64(url, cfg) {
+  const httpAgent = new import_http_proxy_agent.HttpProxyAgent(cfg.proxyUrl);
+  const httpsAgent = new import_https_proxy_agent.HttpsProxyAgent(cfg.proxyUrl);
+  import_axios.default.defaults.httpAgent = httpAgent;
+  import_axios.default.defaults.httpsAgent = httpsAgent;
+  const res = await import_axios.default.get(url, { responseType: "arraybuffer" });
+  const base64 = Buffer.from(res.data, "binary").toString("base64");
+  const dataUrl = `data:image/png;base64,${base64}`;
+  console.log("success");
+  return dataUrl;
+}
+__name(getXImageBase64, "getXImageBase64");
 
 // src/index.ts
 var name = "starfx-bot";
@@ -492,6 +566,17 @@ var Config2 = import_koishi2.Schema.intersect([
     minRepeatTimes: import_koishi2.Schema.number().default(2).description("最少重复次数"),
     repeatPossibility: import_koishi2.Schema.number().role("slider").min(0).max(1).step(0.01).default(0.3).description("复读发生概率")
   }).description("复读功能"),
+  import_koishi2.Schema.object({
+    originImg: import_koishi2.Schema.boolean().default(false).description("根据链接获取原图开关")
+  }).description("自用功能"),
+  import_koishi2.Schema.union([
+    import_koishi2.Schema.object({
+      originImg: import_koishi2.Schema.const(true).required(),
+      originImgRSSUrl: import_koishi2.Schema.string().required().description("推特列表rss地址"),
+      proxyUrl: import_koishi2.Schema.string().default("http://127.0.0.1:7890").description("代理地址")
+    }),
+    import_koishi2.Schema.object({})
+  ]),
   import_koishi2.Schema.object({
     featureControl: import_koishi2.Schema.string().role("textarea", { rows: [15] }).default("{\n\n}").description(`黑/白名单配置，语法为JSON格式(可以不缩进)，<br>
 可配置功能键及语法详见 [项目地址](https://github.com/StarFreedomX/starfx-bot)或[npm发布页](https://www.npmjs.com/package/koishi-plugin-bangdream-ccg)`)
@@ -513,6 +598,7 @@ function apply(ctx, cfg) {
   }
   if (cfg.openSold) {
     ctx.command("卖掉了 [param]").action(async ({ session }, param) => {
+      console.log("ssssss");
       if (detectControl(controlJson, session.guildId, "sold"))
         await session.send(await drawSold(ctx, await getImageSrc(session, param)));
     });
@@ -597,6 +683,24 @@ function apply(ctx, cfg) {
         await undo(cfg, session);
     });
   }
+  if (cfg.originImg) {
+    ctx.command("获取原图 <urls>").action(async ({ session }, urls) => {
+      if (detectControl(controlJson, session.guildId, "originImg")) {
+        let [xUrls, xIndex] = await Promise.all([
+          getXUrl(session.quote.content),
+          getXNum(session)
+        ]);
+        xIndex = xIndex.length ? xIndex : xUrls.map((_, i) => i);
+        console.log(`xIndex:${xIndex}`);
+        console.log(`xUrls:${xUrls}`);
+        const filteredUrls = xIndex.filter((i) => i >= 0 && i < xUrls.length).map((i) => xUrls[i]);
+        console.log(filteredUrls);
+        const imageUrls = await getXImage(cfg.originImgRSSUrl, filteredUrls);
+        console.log(imageUrls);
+        await sendImages(session, cfg, imageUrls);
+      }
+    });
+  }
   ctx.middleware(async (session, next) => {
     const elements = session.elements;
     if (cfg.openRepeat && detectControl(controlJson, session.guildId, "repeat")) {
@@ -621,7 +725,6 @@ function apply(ctx, cfg) {
   });
   if (process.env.NODE_ENV === "development") {
     ctx.command("test").action(async ({ session }) => {
-      await session.send(import_koishi2.h.video("https://video.twimg.com/amplify_video/1920672748596043776/vid/avc1/1080x1920/c3BNP3qg4-sT82fR.mp4?tag=21"));
     });
   }
   function initAssets() {

package/lib/utils.d.ts

@@ -137,4 +137,10 @@ export declare function handleRoll(session: Session): string;
  * @param session session
  */
 export declare function undo(cfg: Config, session: Session): Promise<void>;
+export declare function getXUrl(urls: string): Promise<any[]>;
+export declare function getXNum(session: Session): Promise<number[]>;
+export declare function getXImage(rssUrl: string, xUrls: string | string[]): Promise<string[]>;
+export declare function chunk<T>(arr: T[], size: number): T[][];
+export declare function sendImages(session: Session, cfg: Config, imageUrls: string[]): Promise<void>;
+export declare function test(url: string): Promise<void>;
 export {};

package/package.json

@@ -4,7 +4,7 @@
   "contributors": [
     "StarFreedomX <starfreedomx@outlook.com>"
   ],
-  "version": "0.14.3",
+  "version": "0.15.0",
   "main": "lib/index.js",
   "typings": "lib/index.d.ts",
   "files": [
@@ -36,7 +36,12 @@
     "koishi": "^4.18.7"
   },
   "dependencies": {
+    "axios": "^1.9.0",
+    "cheerio": "^1.0.0",
+    "http-proxy-agent": "^7.0.2",
+    "https-proxy-agent": "^7.0.6",
     "jimp": "^1.6.0",
+    "rss-parser": "^3.13.0",
     "sharp": "^0.34.1"
   }
 }

package/readme.md

@@ -44,19 +44,20 @@ StarFreedomX机器人的小功能，自用
 ### 📌 可配置功能键一览
 
 
-| 键名         | 功能说明                                  |
-|------------|---------------------------------------|
-| `lock`     | 明日方舟封印功能（对应`openLock`）                |
-| `sold`     | 闲鱼“卖掉了”功能（对应`openSold`）               |
-| `repeat`   | 群复读功能（对应`openRepeat`）                 |
-| `record`   | 群语录功能（对应`投稿、语录`）                      |
-| `atNotSay` | “艾特我/他又不说话”系列功能                       |
-| `replyBot` | “我才不是机器人！”系列功能                        |
-| `iLoveYou` | “我也喜欢你”系列功能                           |
-| `bdbd`     | BanG Dream! 边框功能（对应`bangdreamBorder`） |
-| `roll`     | 随机数功能                                 |
-| `undo`     | 撤回功能                                  |
-| `echo`     | echo功能                                |
+| 键名          | 功能说明                                  |
+|-------------|---------------------------------------|
+| `lock`      | 明日方舟封印功能（对应`openLock`）                |
+| `sold`      | 闲鱼“卖掉了”功能（对应`openSold`）               |
+| `repeat`    | 群复读功能（对应`openRepeat`）                 |
+| `record`    | 群语录功能（对应`投稿、语录`）                      |
+| `atNotSay`  | “艾特我/他又不说话”系列功能                       |
+| `replyBot`  | “我才不是机器人！”系列功能                        |
+| `iLoveYou`  | “我也喜欢你”系列功能                           |
+| `bdbd`      | BanG Dream! 边框功能（对应`bangdreamBorder`） |
+| `roll`      | 随机数功能                                 |
+| `undo`      | 撤回功能                                  |
+| `echo`      | echo功能                                |
+| `originImg` | rss获取推特原图                             |
 
 ---
 
@@ -107,3 +108,4 @@ StarFreedomX机器人的小功能，自用
 | `0.13.7` | 修复bdbd无param时出现的异常               |
 | `0.14.0` | echo增加对引用消息的支持，且支持有at触发          |
 | `0.14.3` | echo多at情况支持                      |
+| `0.15.0` | 自用rss获取推特原图                      |