koishi-plugin-booth-get 0.0.3 → 0.0.5

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,
@@ -141,9 +140,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.3",
+  "version": "0.0.5",
   "contributors": [
     "rixiang <1148147857@qq.com>"
   ],
@@ -29,7 +29,7 @@
     "booth"
   ],
   "peerDependencies": {
-    "koishi": "4.17.9"
+    "koishi": "4.17.12"
   },
   "dependencies": {
     "atsc": "^2.1.0",