koishi-plugin-booth-get 0.0.4 → 2.0.0

package/lib/index.js

@@ -17,7 +17,6 @@ var __copyProps = (to, from, except, desc) => {
 };
 var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
 
-// src/index.ts
 var src_exports = {};
 __export(src_exports, {
   Config: () => Config,
@@ -32,14 +31,21 @@ var name = "booth-get";
 var inject = ["puppeteer"];
 var Config = import_koishi.Schema.object({
   loadTimeout: import_koishi.Schema.natural().role("ms").description("加载页面的最长时间。当一个页面等待时间超过这个值时，如果此页面主体已经加载完成，则会发送一条提示消息“正在加载中，请稍等片刻”并继续等待加载；否则会直接提示“无法打开页面”并终止加载。").default(import_koishi.Time.second * 5),
-  idleTimeout: import_koishi.Schema.natural().role("ms").description("等待页面空闲的最长时间。当一个页面等待时间超过这个值时，将停止进一步的加载并立即发送截图。").default(import_koishi.Time.second * 30)
+  idleTimeout: import_koishi.Schema.natural().role("ms").description("等待页面空闲的最长时间。当一个页面等待时间超过这个值时，将停止进一步的加载并立即发送截图。").default(import_koishi.Time.second * 30),
+  proxyServer: import_koishi.Schema.string().description("自定义代理服务器地址。例如：http://127.0.0.1:1080").default(""),
 }).description("booth-get");
 function apply(ctx, config) {
   const logger = ctx.logger("screenshot");
   const { defaultViewport } = ctx.puppeteer.config;
-  const { loadTimeout, idleTimeout } = config;
+  const { loadTimeout, idleTimeout,proxyServer } = config;
   const booth_url = "https://booth.pm/zh-cn/items/";
   const maxSize = 1024 * 1024;
+  const launchOptions = { ...ctx.puppeteer.config.launchOptions };
+  
+  if (proxyServer) {
+    launchOptions.args = launchOptions.args || [];
+    launchOptions.args.push(`--proxy-server=${proxyServer}`);
+  }
   ctx.command("摊位 <id>").action(async ({ session }, id) => {
     if (!id)
       return "请输入ID";
@@ -49,7 +55,7 @@ function apply(ctx, config) {
     if (typeof result === "string")
       return result;
     let loaded = false;
-    const page = await ctx.puppeteer.page();
+    const page = await ctx.puppeteer.page(launchOptions);
     page.on("load", () => loaded = true);
     try {
       await new Promise((resolve, reject) => {
@@ -141,9 +147,40 @@ ctx.command("摊位名称 <query>").action(async ({ session }, query) => {
     await page.close();
   }
 });
+async function captureBoothPage(url) {
+  const page = await ctx.puppeteer.page();
+  try {
+    await page.goto(url, { waitUntil: 'networkidle0' });
+    const shooter = await page.$('div.u-pt-600');
+    if (!shooter) {
+      throw new Error("找不到满足该选择器的元素。");
+    }
+    const buffer = await shooter.screenshot();
+    return import_koishi.h.image(buffer, "image/png");
+  } catch (error) {
+    throw error;
+  } finally {
+    await page.close();
+  }
+}
+ctx.middleware((session, next) => {
+  const message = session.content;
+  const boothUrlRegex = /https:\/\/booth.pm\/[\w-]+\/items\/\d+/;
+
+  if (boothUrlRegex.test(message)) {
+    const boothUrl = message.match(boothUrlRegex)[0];
+    return captureBoothPage(boothUrl).then(image => {
+      return session.send(image);
+    }).catch(error => {
+      ctx.logger.debug(error);
+      return session.send("无法获取页面截图。");
+    });
+  }
+
+  return next();
+});
 }
 __name(apply, "apply");
-// Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
   Config,
   apply,

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "koishi-plugin-booth-get",
   "description": "通过url与名称检查摊位物品并反馈用户搜索的图片",
-  "version": "0.0.4",
+  "version": "2.0.0",
   "contributors": [
     "rixiang <1148147857@qq.com>"
   ],
@@ -37,4 +37,4 @@
     "pngjs": "^7.0.0",
     "puppeteer-core": "^22.12.1"
   }
-}
+}