koishi-plugin-booth-get 0.0.1

package/lib/index.d.ts

@@ -0,0 +1,14 @@
+import { Context, Schema } from 'koishi';
+declare module 'koishi' {
+    interface Events {
+        'screenshot/validate'(url: string): string;
+    }
+}
+export declare const name = "booth-get";
+export declare const inject: string[];
+export interface Config {
+    loadTimeout?: number;
+    idleTimeout?: number;
+}
+export declare const Config: Schema<Config>;
+export declare function apply(ctx: Context, config: Config): void;

package/lib/index.js

@@ -0,0 +1,152 @@
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+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 __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var src_exports = {};
+__export(src_exports, {
+  Config: () => Config,
+  apply: () => apply,
+  inject: () => inject,
+  name: () => name
+});
+module.exports = __toCommonJS(src_exports);
+var import_koishi = require("koishi");
+var import_pngjs = require("pngjs");
+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)
+}).description("booth-get");
+function apply(ctx, config) {
+  const logger = ctx.logger("screenshot");
+  const { defaultViewport } = ctx.puppeteer.config;
+  const { loadTimeout, idleTimeout } = config;
+  const booth_url = "https://booth.pm/zh-cn/items/";
+  const maxSize = 1024 * 1024;
+  ctx.command("摊位 <id>").action(async ({ session }, id) => {
+    if (!id)
+      return "请输入ID";
+    let url = booth_url + id;
+    let selector = "div.u-pt-600";
+    const result = ctx.bail("screenshot/validate", url);
+    if (typeof result === "string")
+      return result;
+    let loaded = false;
+    const page = await ctx.puppeteer.page();
+    page.on("load", () => loaded = true);
+    try {
+      await new Promise((resolve, reject) => {
+        logger.debug(`navigating to ${url}`);
+        const _resolve = /* @__PURE__ */ __name(() => {
+          clearTimeout(timer);
+          resolve();
+        }, "_resolve");
+        page.goto(url, {
+          waitUntil: "networkidle0",
+          timeout: idleTimeout
+        }).then(_resolve, () => {
+          return loaded ? _resolve() : reject(new Error("navigation timeout"));
+        });
+        const timer = setTimeout(() => {
+          return loaded ? session.send("正在加载中，请稍等片刻~") : reject(new Error("navigation timeout"));
+        }, loadTimeout);
+      });
+    } catch (error) {
+      page.close();
+      logger.debug(error);
+      return "无法打开页面。";
+    }
+    const shooter = selector ? await page.$(selector) : page;
+    if (!shooter)
+      return "找不到满足该选择器的元素。";
+    return shooter.screenshot().then(async (buffer) => {
+      if (buffer.byteLength > maxSize) {
+        await new Promise((resolve, reject) => {
+          const png = new import_pngjs.PNG();
+          png.parse(buffer, (error, data) => {
+            return error ? reject(error) : resolve(data);
+          });
+        }).then((data) => {
+          const width = data.width;
+          const height = data.height * maxSize / buffer.byteLength;
+          const png = new import_pngjs.PNG({ width, height });
+          data.bitblt(png, 0, 0, width, height, 0, 0);
+          buffer = import_pngjs.PNG.sync.write(png);
+        }).catch(import_koishi.noop);
+      }
+      return import_koishi.h.image(buffer, "image/png");
+    }, (error) => {
+      logger.debug(error);
+      return "截图失败。";
+    }).finally(() => page.close());
+  });
+ctx.command("摊位名称 <query>").action(async ({ session }, query) => {
+  if (!query)
+    return "请输入搜索关键词";
+
+  const tags = [
+    '3Dモデル',
+    'Vrchat'
+  ];
+
+  const tagsParams = tags.map(tag => `tags[]=${encodeURIComponent(tag)}`).join('&');
+
+  let searchUrl = `https://booth.pm/zh-cn/search/${encodeURIComponent(query)}?${tagsParams}&min_price=4500`;
+
+  const page = await ctx.puppeteer.page();
+  try {
+    await page.goto(searchUrl, { waitUntil: 'networkidle0' });
+    const content = await page.content();
+    const regex = /item-card__wrap" id="item_(\d+)"/;
+    const match = content.match(regex);
+
+    if (!match) {
+      return "没有找到任何商品。";
+    }
+
+    const itemId = match[1];
+    const itemUrl = `https://booth.pm/zh-cn/items/${itemId}`;
+
+    await page.goto(itemUrl, { waitUntil: 'networkidle0' });
+
+    const shooter = await page.$('div.u-pt-600');
+    if (!shooter) {
+      return "找不到满足该选择器的元素。";
+    }
+    const buffer = await shooter.screenshot();
+    if (buffer.byteLength > maxSize) {
+    }
+    return import_koishi.h.image(buffer, "image/png");
+  } catch (error) {
+    logger.debug(error);
+    return "操作失败。";
+  } finally {
+    await page.close();
+  }
+});
+}
+__name(apply, "apply");
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  Config,
+  apply,
+  inject,
+  name
+});

package/package.json

@@ -0,0 +1,30 @@
+{
+  "name": "koishi-plugin-booth-get",
+  "description": "通过url检查摊位并反馈用户搜索的图片",
+  "version": "0.0.1",
+  "main": "lib/index.js",
+  "typings": "lib/index.d.ts",
+  "files": [
+    "lib",
+    "dist"
+  ],
+  "license": "MIT",
+"scripts": {
+  "test": "echo \"Error: no test specified\" && exit 1",
+  "build": "echo \"No build script specified\""
+},
+  "keywords": [
+    "chatbot",
+    "koishi",
+    "plugin"
+  ],
+  "peerDependencies": {
+    "koishi": "4.17.9"
+  },
+  "dependencies": {
+    "atsc": "^2.1.0",
+    "koishi-plugin-puppeteer": "^3.8.4",
+    "pngjs": "^7.0.0",
+    "puppeteer-core": "^22.12.1"
+  }
+}